From 0aaa6166af6ff17f9b643997fccb44de1b57b1e5 Mon Sep 17 00:00:00 2001
From: IRONM00N <64110067+IRONM00N@users.noreply.github.com>
Date: Sat, 6 Aug 2022 21:07:39 -0400
Subject: track the number of guilds the bot is in so I can make a graph

---
 src/lib/models/shared/GuildCount.ts  | 39 ++++++++++++++++++++++++++++++++++++
 src/lib/models/shared/MemberCount.ts |  2 +-
 2 files changed, 40 insertions(+), 1 deletion(-)
 create mode 100644 src/lib/models/shared/GuildCount.ts

(limited to 'src/lib/models')

diff --git a/src/lib/models/shared/GuildCount.ts b/src/lib/models/shared/GuildCount.ts
new file mode 100644
index 0000000..51e571a
--- /dev/null
+++ b/src/lib/models/shared/GuildCount.ts
@@ -0,0 +1,39 @@
+import { type Sequelize } from 'sequelize';
+import { Environment } from '../../../../config/Config.js';
+const { DataTypes, Model } = (await import('sequelize')).default;
+
+export interface GuildCountModel {
+	timestamp: Date;
+	environment: Environment;
+	guildCount: number;
+}
+
+export interface GuildCountCreationAttributes {
+	timestamp?: Date;
+	environment: Environment;
+	guildCount: number;
+}
+
+/**
+ * The number of guilds that the bot is in for each environment.
+ */
+export class GuildCount extends Model<GuildCountModel, GuildCountCreationAttributes> implements GuildCountModel {
+	public declare timestamp: Date;
+	public declare environment: Environment;
+	public declare guildCount: number;
+
+	/**
+	 * Initializes the model.
+	 * @param sequelize The sequelize instance.
+	 */
+	public static initModel(sequelize: Sequelize): void {
+		GuildCount.init(
+			{
+				timestamp: { type: DataTypes.DATE, allowNull: false, defaultValue: DataTypes.NOW },
+				environment: { type: DataTypes.STRING, allowNull: false },
+				guildCount: { type: DataTypes.BIGINT, allowNull: false }
+			},
+			{ sequelize, timestamps: false }
+		);
+	}
+}
diff --git a/src/lib/models/shared/MemberCount.ts b/src/lib/models/shared/MemberCount.ts
index 409d1ac..ea8795c 100644
--- a/src/lib/models/shared/MemberCount.ts
+++ b/src/lib/models/shared/MemberCount.ts
@@ -14,7 +14,7 @@ export interface MemberCountCreationAttributes {
 }
 
 /**
- * Data specific to a certain instance of the bot.
+ * The member count of each guild that the bot is in that have over 100 members.
  */
 export class MemberCount extends Model<MemberCountModel, MemberCountCreationAttributes> implements MemberCountModel {
 	public declare timestamp: Date;
-- 
cgit