aboutsummaryrefslogtreecommitdiff
path: root/src/lib/models/shared/Stat.ts
blob: d13862066b63c143e60d44fda1a57334ac6d4b6a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import { type Sequelize } from 'sequelize';
import { BaseModel } from '../BaseModel.js';
const { DataTypes } = (await import('sequelize')).default;

type Environment = 'production' | 'development' | 'beta';

export interface StatModel {
	environment: Environment;
	commandsUsed: bigint;
}

export interface StatModelCreationAttributes {
	environment: Environment;
	commandsUsed?: bigint;
}

/**
 * Statistics for each instance of the bot.
 */
export class Stat extends BaseModel<StatModel, StatModelCreationAttributes> implements StatModel {
	/**
	 * The bot's environment.
	 */
	public declare environment: Environment;

	/**
	 * The number of commands used
	 */
	public declare commandsUsed: bigint;

	/**
	 * Initializes the model.
	 * @param sequelize The sequelize instance.
	 */
	public static initModel(sequelize: Sequelize): void {
		Stat.init(
			{
				environment: { type: DataTypes.STRING, primaryKey: true },
				commandsUsed: {
					type: DataTypes.TEXT,
					get: function (): bigint {
						return BigInt(this.getDataValue('commandsUsed'));
					},
					set: function (val: bigint) {
						return this.setDataValue('commandsUsed', <any>`${val}`);
					},
					allowNull: false,
					defaultValue: `${0n}`
				}
			},
			{ sequelize }
		);
	}
}