feat(bot): add GuildDelete handler

GalvinPython · Jul 16, 2024 · 3b06833 · 3b06833
1 parent b1b4efc
commit 3b06833
Show file tree

Hide file tree

Showing 7 changed files with 58 additions and 10 deletions.
diff --git a/api/db/init.ts b/api/db/init.ts
@@ -9,7 +9,8 @@ export async function initTables() {
 			members INT,
 			cooldown INT DEFAULT 30000,
 			updates_enabled BOOLEAN DEFAULT FALSE,
-			updates_channel_id VARCHAR(255) DEFAULT NULL
+			updates_channel_id VARCHAR(255) DEFAULT NULL,
+			is_in_guild BOOLEAN DEFAULT TRUE
 		)
 	`;
   const createUsersTable = `

diff --git a/api/db/queries/guilds.ts b/api/db/queries/guilds.ts
@@ -14,7 +14,7 @@ export interface Guild {
 
 export async function getGuild(guildId: string): Promise<[QueryError, null] | [null, Guild | null]> {
 	return new Promise((resolve, reject) => {
-		pool.query("SELECT * FROM guilds WHERE id = ?", [guildId], (err, results) => {
+		pool.query("SELECT * FROM guilds WHERE id = ? AND is_in_guild = ?", [guildId, true], (err, results) => {
 			if (err) {
 				reject([err, null]);
 			} else {
@@ -28,18 +28,20 @@ export async function updateGuild(guild: Omit<Guild, "cooldown" | "updates_enabl
 	return new Promise((resolve, reject) => {
 		pool.query(
 			`
-			INSERT INTO guilds (id, name, icon, members)
-			VALUES (?, ?, ?, ?)
+			INSERT INTO guilds (id, name, icon, members, is_in_guild)
+			VALUES (?, ?, ?, ?, ?)
 			ON DUPLICATE KEY UPDATE
 				name = VALUES(name),
 				icon = VALUES(icon),
-				members = VALUES(members)
+				members = VALUES(members),
+				is_in_guild = VALUES(is_in_guild)
 			`,
 			[
 				guild.id,
 				guild.name,
 				guild.icon,
 				guild.members,
+				true,
 			],
 			(err, results) => {
 				if (err) {
@@ -52,6 +54,18 @@ export async function updateGuild(guild: Omit<Guild, "cooldown" | "updates_enabl
 	});
 }
 
+export async function removeGuild(guildId: string): Promise<[QueryError, null] | [null, true]> {
+	return new Promise((resolve, reject) => {
+		pool.query("UPDATE guilds SET is_in_guild = ? WHERE id = ?", [false, guildId], (err) => {
+			if (err) {
+				reject([err, null]);
+			} else {
+				resolve([null, true]);
+			}
+		});
+	});
+}
+
 export async function setCooldown(guildId: string, cooldown: number): Promise<[QueryError, null] | [null, Guild]> {
   return new Promise((resolve, reject) => {
     pool.query("UPDATE guilds SET cooldown = ? WHERE id = ?", [cooldown, guildId], (err, results) => {

diff --git a/api/index.ts b/api/index.ts
@@ -1,7 +1,7 @@
 import express, { type NextFunction, type Request, type Response } from "express";
 import cors from "cors";
 import path from "path";
-import { getBotInfo, getGuild, getUser, getUsers, initTables, pool, updateGuild, enableUpdates, disableUpdates, setCooldown, setUpdatesChannel, setXP, setLevel } from "./db";
+import { getBotInfo, getGuild, getUser, getUsers, initTables, pool, updateGuild, enableUpdates, disableUpdates, setCooldown, setUpdatesChannel, setXP, setLevel, removeGuild } from "./db";
 
 const app = express();
 const PORT = 18103;
@@ -43,6 +43,17 @@ app.post("/post/:guild", authMiddleware, async (req, res) => {
 	}
 });
 
+app.post('/post/:guild/remove', authMiddleware, async (req, res) => {
+	const { guild } = req.params;
+	const [err, results] = await removeGuild(guild);
+
+	if (err) {
+		res.status(500).json({ message: "Internal server error" });
+	} else {
+		res.status(200).json(results);
+	}
+})
+
 app.post("/post/:guild/:user", authMiddleware, async (req, res) => {
 	const { guild, user } = req.params;
 	const { name, pfp, xp, nickname } = req.body;
@@ -339,7 +350,7 @@ app.get("/leaderboard/:guild", async (req, res) => {
 	const [usersErr, usersData] = await getUsers(guild);
 
 	if (!guildData) {
-		return res.status(404).render("error", { error: { status: 404, message: "The guild does not exist" } });
+		return res.status(404).render("error", { error: { status: 404, message: "The guild does not exist. If Chatr is no longer in this server, the data for this guild has been locked from public access" } });
 	}
 
 	if (guildErr) {

diff --git a/api/views/error.ejs b/api/views/error.ejs
@@ -22,9 +22,9 @@
 			<div class="user">
 				<div class="userInfo">
 					<h1 class="userName">Message</h1>
-					<h1>
+					<p>
 						<%= error.message %>
-					</h1>
+					</p>
 				</div>
 			</div>
 		</div>

diff --git a/bot/events/guildRemove.ts b/bot/events/guildRemove.ts
@@ -0,0 +1,12 @@
+import { Events } from "discord.js";
+import client from "../index";
+import { removeGuild } from "../utils/requestAPI";
+
+client.on(Events.GuildDelete, async (guild) => {
+	try {
+		await removeGuild(guild.id);
+		console.log(`Left guild ${guild.name} with ${guild.memberCount} members. The database has been locked`);
+	} catch (e) {
+		console.error(e);
+	}
+})
diff --git a/bot/index.ts b/bot/index.ts
@@ -12,7 +12,7 @@ const client = new Client({
 	intents: [
 		GatewayIntentBits.Guilds,
 		GatewayIntentBits.GuildMessages,
-		GatewayIntentBits.MessageContent,
+		GatewayIntentBits.MessageContent
 	]
 });
 

diff --git a/bot/utils/requestAPI.ts b/bot/utils/requestAPI.ts
@@ -54,6 +54,16 @@ export async function updateGuildInfo(guild: string, name: string, icon: string,
 	})
 }
 
+export async function removeGuild(guild: string) {
+	await fetch(`http://localhost:18103/post/${guild}/remove`, {
+		headers: {
+			'Content-Type': 'application/json',
+			'Authorization': process.env.AUTH as string,
+		},
+		method: 'POST',
+	})
+}
+
 export async function setXP(guild: string, user: string, xp: number) {
 	const response = await fetch(`http://localhost:18103/admin/set/${guild}/xp`, {
 		"headers": {