-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathConfigStore.js
108 lines (94 loc) · 2.39 KB
/
ConfigStore.js
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
const {Collection} = require("discord.js");
class ConfigStore extends Collection {
constructor(bot, db) {
super();
this.db = db;
this.bot = bot;
};
async create(server, data = {}) {
return new Promise(async (res, rej) => {
try {
await this.db.query(`INSERT INTO configs (
server_id,
banlog_channel,
ban_message,
reprole,
delist_channel,
starboard,
blacklist
) VALUES ($1,$2,$3,$4,$5,$6,$7)`,
[server, data.banlog_channel || "", data.ban_message || "",
data.reprole || "", data.delist_channel || "",
data.starboard, data.blacklist || []])
} catch(e) {
console.log(e);
return rej(e.message);
}
res(await this.get(server));
})
}
async index(server, data = {}) {
return new Promise(async (res, rej) => {
try {
await this.db.query(`INSERT INTO configs (
server_id,
banlog_channel,
ban_message,
reprole,
delist_channel,
starboard,
blacklist
) VALUES ($1,$2,$3,$4,$5,$6,$7)`,
[server, data.banlog_channel || "", data.ban_message || "",
data.reprole || "", data.delist_channel || "",
data.starboard, data.blacklist || []])
} catch(e) {
console.log(e);
return rej(e.message);
}
res();
})
}
async get(server, forceUpdate = false) {
return new Promise(async (res, rej) => {
if(!forceUpdate) {
var config = super.get(server);
if(config) return res(config);
}
try {
var data = await this.db.query(`SELECT * FROM configs WHERE server_id = $1`,[server]);
} catch(e) {
console.log(e);
return rej(e.message);
}
if(data.rows && data.rows[0]) {
this.set(server, data.rows[0])
res(data.rows[0])
} else res(undefined);
})
}
async update(server, data = {}) {
return new Promise(async (res, rej) => {
try {
await this.db.query(`UPDATE configs SET ${Object.keys(data).map((k, i) => k+"=$"+(i+2)).join(",")} WHERE server_id = $1`,[server, ...Object.values(data)]);
} catch(e) {
console.log(e);
return rej(e.message);
}
res(await this.get(server, true));
})
}
async delete(server) {
return new Promise(async (res, rej) => {
try {
await this.db.query(`DELETE FROM configs WHERE server_id = $1`, [server]);
} catch(e) {
console.log(e);
return rej(e.message);
}
super.delete(server);
res();
})
}
}
module.exports = (bot, db) => new ConfigStore(bot, db);