Merge pull request #78 from Contraversum/inviteTracker-fix

added initial fetch

index.ts

@@ -4,13 +4,15 @@ import { sendQuestion } from './commands/test/test-command';
 import { encrypt, decrypt } from './encryptionUtils';
 import { sendSurveyQuestions, Feedbackquestions } from './startSurvey';
 import * as fs from 'fs';
-import path from 'path'
+import path from 'path';
+import { trackInvites } from './inviteTracker';
 import { google } from 'googleapis';
 import { client, db, ClientWithCommands } from './common';
 
 client.on(Events.ClientReady, async (c) => {
     console.log(`Ready! Logged in as ${c.user.tag}`);
     await db.connect();
+    await trackInvites();
 });
 client.login(process.env.TOKEN); // Log in to the bot
 
@@ -210,5 +212,8 @@ client.on(Events.MessageCreate, async (message) => {
     }
 });
 
+client.on('guildMemberAdd', () => {
+    trackInvites();
+});
 
 export { client, db };

inviteTracker.ts

@@ -1,7 +1,8 @@
+import 'dotenv/config'
 import { Guild, GuildMember, Role, Collection } from 'discord.js';
 import { client, db } from './index';
 
-async function trackInvites() {
+export async function trackInvites() {
     const guildId = process.env.GUILD_ID;
     if (!guildId) {
         console.error('GUILD_ID is not defined in .env');
@@ -29,6 +30,7 @@ async function trackInvites() {
 
             // Increment the invite count for the inviter
             inviteData[inviterId] = (inviteData[inviterId] || 0) + inviteCount;
+            console.log(inviteData);
         }
     });
 
@@ -58,18 +60,18 @@ async function trackInvites() {
 
 async function assignRoles(inviteCount: number, userId: string, guild: Guild) {
     const rolesToAssign = [
-        { role: '1151603003279802498', minInviteCount: 1, maxInviteCount: 2 }, // Invite Duke
-        { role: '1151555451910115411', minInviteCount: 3, maxInviteCount: 6 }, // Invite Prince
-        { role: '1151555734652342372', minInviteCount: 7, maxInviteCount: 19 }, // Invite King
-        { role: '1151555885672443906', minInviteCount: 20, maxInviteCount: 49 }, // Invite Emperor
-        { role: '1151555968140841012', minInviteCount: 50, maxInviteCount: Infinity }, // Invite God
+        { role: process.env.INVITE_DUKE, minInviteCount: 1, maxInviteCount: 2 }, // Invite Duke
+        { role: process.env.INVITE_PRINCE, minInviteCount: 3, maxInviteCount: 6 }, // Invite Prince
+        { role: process.env.INVITE_KING, minInviteCount: 7, maxInviteCount: 19 }, // Invite King
+        { role: process.env.INVITE_EMPEROR, minInviteCount: 20, maxInviteCount: 49 }, // Invite Emperor
+        { role: process.env.INVITE_GOD, minInviteCount: 50, maxInviteCount: Infinity }, // Invite God
     ];
     const rolesToRemove: Collection<string, Role> = new Collection();
 
     const member = await guild.members.fetch(userId);
 
     for (const { role, minInviteCount, maxInviteCount } of rolesToAssign) {
-        const targetRole = guild.roles.cache.get(role);
+        const targetRole = guild.roles.cache.get(role!);
 
         if (!targetRole) {
             console.error(`Role not found for user ${userId} `);
@@ -97,6 +99,3 @@ async function removeRoles(rolesToRemove: Collection<string, Role>, member: Guil
         }
     }
 }
-client.on('guildMemberAdd', () => {
-    trackInvites();
-});