global handling added

Author: breadddevv

src/events/guildMemberUpdate.ts

@@ -11,6 +11,7 @@ import {
   removeAllLevelRoles,
 } from "../services/roleService.js";
 import { logger } from "../libs/logger.js";
+import { prisma } from "../libs/database.js";
 
 export default {
   name: Events.GuildMemberUpdate,
@@ -19,10 +20,9 @@ export default {
       if (oldMember.partial) oldMember = await oldMember.fetch();
       if (newMember.partial) newMember = await newMember.fetch();
     } catch (error) {
-      logger.error(
-        `Failed to fetch partials for user ${oldMember.user.username}`,
-      );
+      logger.error(`Failed to fetch partials for user ${oldMember.user.username}`);
     }
+
     const wasBoostingBefore = oldMember.premiumSince !== null;
     const isBoostingNow = newMember.premiumSince !== null;
 
@@ -34,104 +34,113 @@ export default {
   },
 };
 
+async function getGuildSettings(guildId: string) {
+  return prisma.guildSetting.findUnique({ where: { gid: guildId } });
+}
+
 async function onBoostStart(member: GuildMember): Promise<void> {
   const guild = member.guild;
 
+  const settings = await getGuildSettings(guild.id);
+  if (!settings) {
+    logger.error(`No guild settings found for guild ${guild.id} — run /setup first.`);
+    return;
+  }
+
   const record = await registerBoost(
     member.id,
     guild.id,
     guild.name,
     guild.iconURL(),
   );
-
   if (!record) return;
 
   await clearPendingCustomRoleDeletion(record.id);
   await assignLevelRoles(member, record.boostCounts ?? 1);
 
-  const greetChannel = guild.channels.cache.get(process.env.GREET_CHANNEL_ID) as
-    | TextChannel
-    | undefined;
+  const greetChannel = guild.channels.cache.get(settings.greetChannelId) as TextChannel | undefined;
   if (greetChannel) {
     const embed = new EmbedBuilder()
       .setColor(0xf47fff)
       .setTitle("New Server Boost! 🎉")
       .setDescription(`${member} has boosted the server!`)
       .addFields(
-        {
-          name: "Total Boosts",
-          value: String(record.boostCounts ?? 1),
-          inline: true,
+        { 
+          name: "Total Boosts", 
+          value: String(record.boostCounts ?? 1), 
+          inline: true 
+        },
+        { name: "Member", 
+          value: member.user.tag, 
+          inline: true 
         },
-        { name: "Member", value: member.user.tag, inline: true },
       )
+      .setThumbnail(member.displayAvatarURL({ size: 512 }))
       .setTimestamp();
-
     await greetChannel.send({ embeds: [embed] });
   }
 
-  const logChannel = guild.channels.cache.get(process.env.GREET_CHANNEL_ID) as
-    | TextChannel
-    | undefined;
+  const logChannel = guild.channels.cache.get(settings.logChannelId) as TextChannel | undefined;
   if (logChannel) {
     const logEmbed = new EmbedBuilder()
       .setColor(0x57f287)
       .setTitle("Boost Started")
       .addFields(
-        {
-          name: "User",
-          value: `${member.user.tag} (${member.id})`,
-          inline: false,
+        { 
+          name: "User", 
+          value: `${member.user.tag} (${member.id})`, 
+          inline: false 
         },
-        {
-          name: "Total Boost Count",
-          value: String(record.boostCounts ?? 1),
-          inline: true,
+        { 
+          name: "Total Boost Count", 
+          value: String(record.boostCounts ?? 1), 
+          inline: true 
         },
       )
+      .setThumbnail(member.displayAvatarURL({ size: 512 }))
       .setTimestamp();
-
     await logChannel.send({ embeds: [logEmbed] });
   }
 
   try {
-    await member.send(
-      `Thank you for boosting the server! You now have access to booster perks.`,
-    );
-  } catch {
-  }
+    await member.send(`Thank you for boosting the server! You now have access to booster perks.`);
+  } catch {}
 }
 
 async function onBoostEnd(member: GuildMember): Promise<void> {
   const guild = member.guild;
 
+  const settings = await getGuildSettings(guild.id);
+  if (!settings) {
+    logger.error(`No guild settings found for guild ${guild.id} — run /setup first.`);
+    return;
+  }
+
   const result = await removeBoost(member.id, guild.id);
   if (!result) return;
 
   await removeAllLevelRoles(member);
   await scheduleCustomRoleDeletionAfterGrace(member.id, guild.id);
 
-  const logChannel = guild.channels.cache.get(process.env.LOG_CHANNEL_ID) as
-    | TextChannel
-    | undefined;
+  const logChannel = guild.channels.cache.get(settings.logChannelId) as TextChannel | undefined;
   if (logChannel) {
     const logEmbed = new EmbedBuilder()
       .setColor(0xed4245)
       .setTitle("Boost Ended")
       .addFields(
-        {
-          name: "User",
-          value: `${member.user.tag} (${member.id})`,
-          inline: false,
+        { 
+          name: "User", 
+          value: `${member.user.tag} (${member.id})`, 
+          inline: false 
         },
-        {
-          name: "Historical Boost Count",
-          value: String(result.boostCounts ?? 0),
-          inline: true,
+        { 
+          name: "Historical Boost Count", 
+          value: String(result.boostCounts ?? 0), 
+          inline: true 
         },
       )
+      .setThumbnail(member.displayAvatarURL({ size: 512 }))
       .setTimestamp();
-
     await logChannel.send({ embeds: [logEmbed] });
   }
-}
+}