|
| 1 | +import type { QueryError } from "mysql2"; |
| 2 | +import { pool } from ".."; |
| 3 | + |
| 4 | +export interface Guild { |
| 5 | + id: string; |
| 6 | + name: string; |
| 7 | + icon: string; |
| 8 | + members: number; |
| 9 | + cooldown: number; |
| 10 | + updates_enabled: 0 | 1; |
| 11 | + updates_channel_id: string | null; |
| 12 | +} |
| 13 | + |
| 14 | + |
| 15 | +export async function getGuild(guildId: string): Promise<[QueryError, null] | [null, Guild | null]> { |
| 16 | + return new Promise((resolve, reject) => { |
| 17 | + pool.query("SELECT * FROM guilds WHERE id = ? AND is_in_guild = ?", [guildId, true], (err, results) => { |
| 18 | + if (err) { |
| 19 | + reject([err, null]); |
| 20 | + } else { |
| 21 | + resolve([null, (results as Guild[])[0]]); |
| 22 | + } |
| 23 | + }); |
| 24 | + }); |
| 25 | +} |
| 26 | + |
| 27 | +export async function updateGuild(guild: Omit<Guild, "cooldown" | "updates_enabled" | "updates_channel_id">): Promise<[QueryError | null, null] | [null, Guild[]]> { |
| 28 | + return new Promise((resolve, reject) => { |
| 29 | + pool.query( |
| 30 | + ` |
| 31 | + INSERT INTO guilds (id, name, icon, members, is_in_guild) |
| 32 | + VALUES (?, ?, ?, ?, ?) |
| 33 | + ON DUPLICATE KEY UPDATE |
| 34 | + name = VALUES(name), |
| 35 | + icon = VALUES(icon), |
| 36 | + members = VALUES(members), |
| 37 | + is_in_guild = VALUES(is_in_guild) |
| 38 | + `, |
| 39 | + [ |
| 40 | + guild.id, |
| 41 | + guild.name, |
| 42 | + guild.icon, |
| 43 | + guild.members, |
| 44 | + true, |
| 45 | + ], |
| 46 | + (err, results) => { |
| 47 | + if (err) { |
| 48 | + reject([err, null]); |
| 49 | + } else { |
| 50 | + resolve([null, results as Guild[]]); |
| 51 | + } |
| 52 | + }, |
| 53 | + ); |
| 54 | + }); |
| 55 | +} |
| 56 | + |
| 57 | +export async function removeGuild(guildId: string): Promise<[QueryError, null] | [null, true]> { |
| 58 | + return new Promise((resolve, reject) => { |
| 59 | + pool.query("UPDATE guilds SET is_in_guild = ? WHERE id = ?", [false, guildId], (err) => { |
| 60 | + if (err) { |
| 61 | + reject([err, null]); |
| 62 | + } else { |
| 63 | + resolve([null, true]); |
| 64 | + } |
| 65 | + }); |
| 66 | + }); |
| 67 | +} |
| 68 | + |
| 69 | +export async function setCooldown(guildId: string, cooldown: number): Promise<[QueryError, null] | [null, Guild]> { |
| 70 | + return new Promise((resolve, reject) => { |
| 71 | + pool.query("UPDATE guilds SET cooldown = ? WHERE id = ?", [cooldown, guildId], (err, results) => { |
| 72 | + if (err) { |
| 73 | + reject([err, null]); |
| 74 | + } else { |
| 75 | + resolve([null, (results as Guild[])[0]]); |
| 76 | + } |
| 77 | + }); |
| 78 | + }) |
| 79 | +} |
| 80 | + |
| 81 | +interface BotInfo { |
| 82 | + total_guilds: number; |
| 83 | + total_members: number; |
| 84 | + user_count?: number; |
| 85 | +} |
| 86 | + |
| 87 | +export async function getBotInfo(): Promise<[QueryError | null, BotInfo | null]> { |
| 88 | + return new Promise((resolve, reject) => { |
| 89 | + pool.query("SELECT COUNT(*) AS total_guilds, SUM(members) AS total_members FROM guilds", (err, results) => { |
| 90 | + if (err) { |
| 91 | + reject([err, null]); |
| 92 | + } else { |
| 93 | + const botInfo: BotInfo = { |
| 94 | + total_guilds: (results as BotInfo[])[0].total_guilds, |
| 95 | + total_members: (results as BotInfo[])[0].total_members ?? 0, |
| 96 | + }; |
| 97 | + getUsersCount() |
| 98 | + .then(([userCountError, userCount]) => { |
| 99 | + if (userCountError) { |
| 100 | + reject([userCountError, null]); |
| 101 | + } else { |
| 102 | + botInfo.user_count = userCount; |
| 103 | + resolve([null, botInfo]); |
| 104 | + } |
| 105 | + }) |
| 106 | + .catch((error) => { |
| 107 | + reject([error, null]); |
| 108 | + }); |
| 109 | + } |
| 110 | + }); |
| 111 | + }); |
| 112 | +} |
| 113 | + |
| 114 | +export async function getUsersCount(): Promise<[QueryError | null, number]> { |
| 115 | + return new Promise((resolve, reject) => { |
| 116 | + pool.query("SELECT COUNT(*) AS count FROM users", (err, results) => { |
| 117 | + if (err) { |
| 118 | + reject([err, null]); |
| 119 | + } else { |
| 120 | + resolve([null, (results as { count: number }[])[0].count]); |
| 121 | + } |
| 122 | + }); |
| 123 | + }); |
| 124 | +} |
0 commit comments