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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
import { Snowflake } from 'discord.js';
import { DataTypes, Optional, Sequelize } from 'sequelize';
import { BaseModel, BushClient } from '..';
export interface GuildModel {
id: string;
prefix: string;
autoPublishChannels: string[];
blacklistedChannels: Snowflake[];
welcomeChannel: Snowflake;
muteRole: Snowflake;
punishmentEnding: string;
}
export type GuildModelCreationAttributes = Optional<
GuildModel,
'prefix' | 'autoPublishChannels' | 'blacklistedChannels' | 'welcomeChannel' | 'muteRole' | 'punishmentEnding'
>;
export class Guild extends BaseModel<GuildModel, GuildModelCreationAttributes> implements GuildModel {
id: string;
prefix: string;
autoPublishChannels: string[];
blacklistedChannels: Snowflake[];
welcomeChannel: Snowflake;
muteRole: Snowflake;
punishmentEnding: string;
static initModel(sequelize: Sequelize, client: BushClient): void {
Guild.init(
{
id: {
type: DataTypes.STRING,
primaryKey: true
},
prefix: {
type: DataTypes.STRING,
allowNull: false,
defaultValue: client.config.prefix
},
autoPublishChannels: {
type: DataTypes.STRING,
get: function () {
return JSON.parse(this.getDataValue('autoPublishChannels') as unknown as string);
},
set: function (val: Snowflake[]) {
return this.setDataValue('autoPublishChannels', JSON.stringify(val) as unknown as Snowflake[]);
},
allowNull: true
},
blacklistedChannels: {
type: DataTypes.STRING,
get: function () {
return JSON.parse(this.getDataValue('blacklistedChannels') as unknown as string);
},
set: function (val: Snowflake[]) {
return this.setDataValue('blacklistedChannels', JSON.stringify(val) as unknown as Snowflake[]);
},
allowNull: true
},
welcomeChannel: {
type: DataTypes.STRING,
allowNull: true
},
muteRole: {
type: DataTypes.STRING,
allowNull: true
},
punishmentEnding: {
type: DataTypes.TEXT,
allowNull: true
}
},
{ sequelize: sequelize }
);
}
}
|
