db: migrate bot_info to new system

Author: GalvinPython

src/events/ready.ts

@@ -1,42 +1,26 @@
-import { ActivityType, Events, PresenceUpdateStatus } from "discord.js";
+import { Events } from "discord.js";
+import { CronJob } from "cron";
 
 import client from "../index";
 import fetchLatestUploads from "../utils/youtube/fetchLatestUploads";
 import { config } from "../config";
-import { checkIfStreamersAreLive } from "../utils/twitch/checkIfStreamerIsLive";
-import { updateBotInfo } from "../utils/database";
-
-// update the bot's presence
-async function updatePresence() {
-    if (!client?.user) return;
-
-    const servers = client.guilds.cache.size;
-    const members = client.guilds.cache.reduce(
-        (acc, guild) => acc + guild.memberCount,
-        0,
-    );
-
-    await updateBotInfo(servers, members);
-    client.user.setPresence({
-        activities: [
-            {
-                name: `Notifying ${servers.toLocaleString()} servers [${members.toLocaleString()} members]`,
-                type: ActivityType.Custom,
-            },
-        ],
-        status: PresenceUpdateStatus.Online,
-    });
-}
+// import { checkIfStreamersAreLive } from "../utils/twitch/checkIfStreamerIsLive";
+import { cronUpdateBotInfo } from "../utils/cronJobs";
 
 // Log into the bot
 client.once(Events.ClientReady, async (bot) => {
     console.log(`Ready! Logged in as ${bot.user?.tag}`);
 
     // Set the bot's presence and update it every minute
-    await updatePresence();
+    await cronUpdateBotInfo();
     fetchLatestUploads();
-    setInterval(updatePresence, 60000);
+
+    // Set the bot's presence and update it every minute
+    new CronJob("0 * * * * *", async () => {
+        await cronUpdateBotInfo();
+    }).start();
     setInterval(fetchLatestUploads, config.updateIntervalYouTube as number);
+
     // One at a time
     // checkIfStreamersAreLive();
     // setInterval(checkIfStreamersAreLive, config.updateIntervalTwitch as number);

src/types/database.d.ts

@@ -21,3 +21,12 @@ export type dbDiscordTable = {
     platform_user_id: string;
     guild_ping_role: null | string;
 };
+
+export interface dbBotInfo {
+    locked_row: boolean;
+    guilds_total: number;
+    channels_tracked: number;
+    total_members: number;
+    updated_at: string;
+    extended_info_updated_at: string;
+}

src/utils/cronJobs.ts

@@ -0,0 +1,26 @@
+import { ActivityType, Guild, PresenceUpdateStatus } from "discord.js";
+
+import client from "..";
+
+import { updateBotInfo } from "./db/botinfo";
+
+export async function cronUpdateBotInfo() {
+    if (!client?.user) return;
+
+    const servers: number = client.guilds.cache.size;
+    const members: number = client.guilds.cache.reduce(
+        (acc: number, guild: Guild): number => acc + guild.memberCount,
+        0,
+    );
+
+    await updateBotInfo(servers, members);
+    client.user.setPresence({
+        activities: [
+            {
+                name: `Notifying ${servers.toLocaleString()} servers [${members.toLocaleString()} members]`,
+                type: ActivityType.Custom,
+            },
+        ],
+        status: PresenceUpdateStatus.Online,
+    });
+}

src/utils/database.ts

@@ -305,42 +305,6 @@ export async function twitchStopGuildTrackingChannel(
 }
 // #endregion
 
-// #region Bot Info
-export async function getBotInfo() {
-    const query = `SELECT * FROM bot_info`;
-
-    try {
-        const statement = db.prepare(query);
-        const result = statement.get();
-
-        return result;
-    } catch (err) {
-        console.error("Error getting bot info:", err);
-        throw err;
-    }
-}
-
-export async function updateBotInfo(
-    total_servers: number,
-    total_members: number,
-) {
-    console.log("Updating bot info:", total_servers, total_members);
-    const query = `UPDATE bot_info SET total_servers = ?, total_members = ?`;
-
-    try {
-        const statement = db.prepare(query);
-
-        statement.run(total_servers, total_members);
-
-        return true;
-    } catch (err) {
-        console.error("Error updating bot info:", err);
-
-        return false;
-    }
-}
-// #endregion
-
 // #region i have no idea what im doing here
 
 export async function getAllTrackedInGuild(

src/utils/db/botinfo.ts

@@ -0,0 +1,48 @@
+import type { PoolClient, QueryResult } from "pg";
+import { pool } from "../database";
+import type { dbBotInfo } from "../../types/database";
+
+export async function updateBotInfo(
+    guilds_total: number = 0,
+    channels_tracked: number = 0,
+    total_members: number = 0,
+): Promise<void> {
+    const query = `
+        UPDATE bot_info
+        SET guilds_total = $1, channels_tracked = $2, total_members = $3, updated_at = NOW()
+        WHERE locked_row = true;
+    `;
+
+    try {
+        const client = await pool.connect();
+
+        await client.query(query, [
+            guilds_total,
+            channels_tracked,
+            total_members,
+        ]);
+
+        client.release();
+
+        console.log("Bot info updated successfully");
+    } catch (err) {
+        console.error("Error updating bot info:", err);
+    }
+}
+
+export async function getBotInfo() {
+    const query = `SELECT * FROM bot_info`;
+
+    try {
+        const client: PoolClient = await pool.connect();
+        const result: QueryResult<any> = await client.query(query);
+
+        client.release();
+
+        return result.rows[0] as dbBotInfo;
+    } catch (err) {
+        console.error("Error getting bot info:", err);
+
+        return null;
+    }
+}

src/utils/db/init.ts

@@ -76,6 +76,7 @@ export default async function initTables(): Promise<boolean> {
 
     const createBotInfoTable = `
         CREATE TABLE IF NOT EXISTS bot_info (
+            locked_row BOOLEAN NOT NULL DEFAULT TRUE,
             guilds_total INTEGER NOT NULL DEFAULT 0,
             channels_tracked INTEGER NOT NULL DEFAULT 0,
             total_members INTEGER NOT NULL DEFAULT 0,
@@ -125,6 +126,10 @@ export default async function initTables(): Promise<boolean> {
         );
     `;
 
+    const seedBotInfoTable = `
+        INSERT INTO bot_info (locked_row, guilds_total, channels_tracked, total_members) VALUES (true, 0, 0, 0)
+    `;
+
     try {
         const client = await pool.connect();
 
@@ -167,6 +172,11 @@ export default async function initTables(): Promise<boolean> {
         await client.query(createAuditLogsTable);
         console.log("Audit Logs table created");
 
+        await client.query(seedBotInfoTable);
+        console.log("Bot Info table seeded");
+
+        client.release();
+
         return true;
     } catch (err) {
         console.error("Error creating tables:", err);