✨ Purge configuration data of removed guilds (#39)

* ✨ Delete configs of removed guilds * Move deletion to configHelpers * Move logging to reset function Co-authored-by: MarcusOtter <[email protected]>
MarcusOtter · Jan 31, 2022 · 32b90b7 · 32b90b7
1 parent 530b4e8
commit 32b90b7
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/src/helpers/configHelpers.ts b/src/helpers/configHelpers.ts
@@ -1,4 +1,4 @@
-import { Guild } from "discord.js";
+import { Client, Guild } from "discord.js";
 import * as defaultConfig from "../config.json";
 import { resolve as pathResolve } from "path";
 import * as fs from "fs";
@@ -76,9 +76,25 @@ export function resetConfigToDefault(guildId: string): boolean {
 	if (!fs.existsSync(path)) return false;
 	fs.rmSync(path);
 	guildConfigsCache.delete(guildId);
+	console.log(`Deleted data for guild ${guildId}`);
 	return true;
 }
 
+export function deleteConfigsFromUnkownServers(client: Client): void {
+	if (!client.guilds.cache.size) {
+		console.warn("No guilds available; skipping config deletion.");
+		return;
+	}
+
+	const configFiles = fs.readdirSync(CONFIGS_PATH);
+	configFiles.forEach(file => {
+		const guildId = file.split(".")[0];
+		if (!client.guilds.cache.has(guildId)) {
+			resetConfigToDefault(guildId);
+		}
+	});
+}
+
 function readConfigFromFile(guildId: string): NeedleConfig | undefined {
 	const path = getGuildConfigPath(guildId);
 	if (!fs.existsSync(path)) return undefined;

diff --git a/src/index.ts b/src/index.ts
@@ -2,7 +2,7 @@ import { Client, Intents } from "discord.js";
 import { getOrLoadAllCommands } from "./handlers/commandHandler";
 import { handleInteractionCreate } from "./handlers/interactionHandler";
 import { handleMessageCreate } from "./handlers/messageHandler";
-import { getApiToken } from "./helpers/configHelpers";
+import { deleteConfigsFromUnkownServers, getApiToken, resetConfigToDefault } from "./helpers/configHelpers";
 
 (async () => {
 	(await import("dotenv")).config();
@@ -11,10 +11,14 @@ import { getApiToken } from "./helpers/configHelpers";
 	await getOrLoadAllCommands(false);
 
 	const CLIENT = new Client({ intents: [Intents.FLAGS.GUILDS, Intents.FLAGS.GUILD_MESSAGES] });
-	CLIENT.once("ready", () => console.log("Ready!"));
+	CLIENT.once("ready", () => {
+		console.log("Ready!");
+		deleteConfigsFromUnkownServers(CLIENT);
+	});
 
 	CLIENT.on("interactionCreate", interaction => handleInteractionCreate(interaction).catch(e => console.log(e)));
 	CLIENT.on("messageCreate", message => handleMessageCreate(message).catch(e => console.log(e)));
+	CLIENT.on("guildDelete", guild => { resetConfigToDefault(guild.id); });
 
 	CLIENT.login(getApiToken() ?? undefined);
 })();