From 84230e0d443caf99beefe28cd4282c2ba5211aaa Mon Sep 17 00:00:00 2001
From: Frostbide <129122542+Frostbide@users.noreply.github.com>
Date: Sun, 25 May 2025 19:28:24 -0700
Subject: [PATCH] Change messages for logging

---
 src/plugins/database.js      |  4 +++-
 src/plugins/invitemanager.js | 20 +++++++++++++++++---
 2 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/src/plugins/database.js b/src/plugins/database.js
index 482e1ba..fd75904 100644
--- a/src/plugins/database.js
+++ b/src/plugins/database.js
@@ -7,6 +7,8 @@ function main(ctx) {
   bot.database = {
     db: null,
     async loadDataBase() {
+      if (!host || !port)
+        return console.error('Database missing host or port in config');
       this.db = new Client({
         host,
         port,
@@ -17,7 +19,7 @@ function main(ctx) {
 
       try {
         await this.db.connect();
-        console.log('Connected to PostgreSQL.');
+        console.log('Connected to PostgreSQL');
 
         const createMessages = `
           CREATE TABLE IF NOT EXISTS messages (
diff --git a/src/plugins/invitemanager.js b/src/plugins/invitemanager.js
index 024bca4..a7f5f1b 100644
--- a/src/plugins/invitemanager.js
+++ b/src/plugins/invitemanager.js
@@ -15,7 +15,10 @@ function main(ctx) {
             new Map(guildInvites.map((inv) => [inv.code, inv.uses]))
           );
         } catch (err) {
-          bot.logger.log(`Could not fetch invites for ${guild.name}:`, err);
+          bot.logger.log(
+            `Error fetching invites for guild ${guild.name}:`,
+            err
+          );
         }
       });
 
@@ -73,7 +76,7 @@ function main(ctx) {
             new Map(newInvites.map((inv) => [inv.code, inv.uses]))
           );
         } catch (err) {
-          bot.logger.log(`Error managing invite for ${member.user.tag}:`, err);
+          bot.logger.log(`Guild join error from ${member.user.tag}:`, err);
         }
       });
 
@@ -120,7 +123,7 @@ function main(ctx) {
       return result.rows?.[0]?.total_uses || 0;
     },
 
-    async getLeaderboard(guildId, amount = 10) {
+    async getUserLeaderboard(guildId, amount = 10) {
       const result = await database.db.query(
         `SELECT userid, SUM(uses) AS total_uses
          FROM inviteCount
@@ -136,6 +139,17 @@ function main(ctx) {
       }));
     },
 
+    async getInviteLeaderboard(guildId, amount = 10) {
+      const result = await database.db.query(
+        `SELECT userid, SUM(uses) AS total_uses`,
+        [guildId, amount]
+      );
+      return result.rows.map((row) => ({
+        userId: row.userid,
+        uses: row.total_uses,
+      }));
+    },
+
     async setLogChannel(guildId, channelId) {
       await database.db.query(
         `INSERT INTO inviteConfig (serverid, channelid, enabled)