From 40f9581991e12082a2708fa8fdd454c080f67a4b Mon Sep 17 00:00:00 2001
From: IRONM00N <64110067+IRONM00N@users.noreply.github.com>
Date: Sun, 24 Apr 2022 23:04:49 -0400
Subject: add member ship tracking because I want to make graph

---
 src/lib/extensions/discord-akairo/BushClient.ts |  2 ++
 src/lib/index.ts                                |  1 +
 src/lib/models/shared/MemberCount.ts            | 38 +++++++++++++++++++++++++
 3 files changed, 41 insertions(+)
 create mode 100644 src/lib/models/shared/MemberCount.ts

(limited to 'src/lib')

diff --git a/src/lib/extensions/discord-akairo/BushClient.ts b/src/lib/extensions/discord-akairo/BushClient.ts
index 1e6418c..4f0c6b0 100644
--- a/src/lib/extensions/discord-akairo/BushClient.ts
+++ b/src/lib/extensions/discord-akairo/BushClient.ts
@@ -58,6 +58,7 @@ import { ModLog } from '../../models/instance/ModLog.js';
 import { Reminder } from '../../models/instance/Reminder.js';
 import { StickyRole } from '../../models/instance/StickyRole.js';
 import { Global } from '../../models/shared/Global.js';
+import { MemberCount } from '../../models/shared/MemberCount.js';
 import { Shared } from '../../models/shared/Shared.js';
 import { Stat } from '../../models/shared/Stat.js';
 import { AllowedMentions } from '../../utils/AllowedMentions.js';
@@ -444,6 +445,7 @@ export class BushClient<Ready extends boolean = boolean> extends AkairoClient<Re
 			Stat.initModel(this.sharedDB);
 			Global.initModel(this.sharedDB);
 			Shared.initModel(this.sharedDB);
+			MemberCount.initModel(this.sharedDB);
 			await this.sharedDB.sync({
 				// Sync all tables to fix everything if updated
 				// if another instance restarts we don't want to overwrite new changes made in development
diff --git a/src/lib/index.ts b/src/lib/index.ts
index 59d8b53..8a6b75d 100644
--- a/src/lib/index.ts
+++ b/src/lib/index.ts
@@ -76,6 +76,7 @@ export * from './models/instance/ModLog.js';
 export * from './models/instance/Reminder.js';
 export * from './models/instance/StickyRole.js';
 export * from './models/shared/Global.js';
+export * from './models/shared/MemberCount.js';
 export * from './models/shared/Shared.js';
 export * from './models/shared/Stat.js';
 export * from './utils/AllowedMentions.js';
diff --git a/src/lib/models/shared/MemberCount.ts b/src/lib/models/shared/MemberCount.ts
new file mode 100644
index 0000000..409d1ac
--- /dev/null
+++ b/src/lib/models/shared/MemberCount.ts
@@ -0,0 +1,38 @@
+import { type Sequelize } from 'sequelize';
+const { DataTypes, Model } = (await import('sequelize')).default;
+
+export interface MemberCountModel {
+	timestamp: Date;
+	guildId: string;
+	memberCount: number;
+}
+
+export interface MemberCountCreationAttributes {
+	timestamp?: Date;
+	guildId: string;
+	memberCount: number;
+}
+
+/**
+ * Data specific to a certain instance of the bot.
+ */
+export class MemberCount extends Model<MemberCountModel, MemberCountCreationAttributes> implements MemberCountModel {
+	public declare timestamp: Date;
+	public declare guildId: string;
+	public declare memberCount: number;
+
+	/**
+	 * Initializes the model.
+	 * @param sequelize The sequelize instance.
+	 */
+	public static initModel(sequelize: Sequelize): void {
+		MemberCount.init(
+			{
+				timestamp: { type: DataTypes.DATE, allowNull: false, defaultValue: DataTypes.NOW },
+				guildId: { type: DataTypes.STRING, allowNull: false },
+				memberCount: { type: DataTypes.BIGINT, allowNull: false }
+			},
+			{ sequelize, timestamps: false }
+		);
+	}
+}
-- 
cgit