guildMemberUpdate.ts

import "../libs/loadVariables.js";
import {
  ContainerBuilder,
  EmbedBuilder,
  Events,
  GuildMember,
  MessageFlags,
  PermissionFlagsBits,
  TextChannel,
} from "discord.js";
import {
  registerBoost,
  removeBoost,
  scheduleCustomRoleDeletionAfterGrace,
  clearPendingCustomRoleDeletion,
} from "../services/boosterService.js";
import {
  assignLevelRoles,
  removeAllLevelRoles,
} from "../services/roleService.js";
import { logger } from "../libs/logger.js";
import { prisma } from "../libs/database.js";
import { DiscordClient } from "../base/types/discord.js";
import { SystemColors } from "../libs/colors.js";

export default {
  name: Events.GuildMemberUpdate,
  async execute(
    _client: DiscordClient,
    oldMember: GuildMember,
    newMember: GuildMember,
  ) {
    try {
      if (oldMember.partial) oldMember = await oldMember.fetch();
      if (newMember.partial) newMember = await newMember.fetch();
    } catch (error) {
      const username =
        oldMember?.user?.username ?? oldMember?.id ?? "Unknown user";
      logger.error(`Failed to fetch partials for user ${username}`);
      return;
    }

    const wasBoostingBefore = oldMember.premiumSince !== null;
    const isBoostingNow = newMember.premiumSince !== null;

    if (!wasBoostingBefore && isBoostingNow) {
      try {
        await onBoostStart(newMember);
      } catch (error) {
        logger.error(
          `onBoostStart failed for user ${newMember.id} in guild ${newMember.guild.id}:`,
          error,
        );
      }
    } else if (wasBoostingBefore && !isBoostingNow) {
      try {
        await onBoostEnd(newMember);
      } catch (error) {
        logger.error(
          `onBoostEnd failed for user ${newMember.id} in guild ${newMember.guild.id}:`,
          error,
        );
      }
    }
  },
};

async function getGuildSettings(guildId: string) {
  return prisma.guildSetting.findUnique({ where: { gid: guildId } });
}
async function safeSend(
  channel: TextChannel | undefined,
  payload: Parameters<TextChannel["send"]>[0],
): Promise<void> {
  if (!channel) return;

  const me = channel.guild.members.me;
  const perms = me ? channel.permissionsFor(me) : null;

  if (
    !perms?.has([
      PermissionFlagsBits.ViewChannel,
      PermissionFlagsBits.SendMessages,
    ])
  ) {
    logger.warn(
      `Missing permission to send in channel ${channel.id} (guild ${channel.guild.id})`,
    );
    return;
  }

  try {
    await channel.send(payload);
  } catch (error) {
    logger.error(`Failed to send message in channel ${channel.id}:`, error);
  }
}

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(settings.greetChannelId) as
    | TextChannel
    | undefined;

  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: "Member", value: member.user.tag, inline: true },
    )
    .setThumbnail(member.displayAvatarURL({ size: 512 }))
    .setFooter({ text: "Thank youuu!" })
    .setTimestamp();

  await safeSend(greetChannel, { embeds: [embed] });

  const logChannel = guild.channels.cache.get(settings.logChannelId) as
    | TextChannel
    | undefined;

  const logEmbed = new EmbedBuilder()
    .setColor(0x57f287)
    .setTitle("Boost Started")
    .addFields(
      {
        name: "User",
        value: `${member.user.tag} (${member.id})`,
        inline: false,
      },
      {
        name: "Total Boost Count",
        value: String(record.boostCounts ?? 1),
        inline: true,
      },
    )
    .setThumbnail(member.displayAvatarURL({ size: 512 }))
    .setTimestamp();

  await safeSend(logChannel, { embeds: [logEmbed] });

  try {
    const serverImage = guild.iconURL(); 

    const container = new ContainerBuilder().addSectionComponents(comp => {
      comp.addTextDisplayComponents(text =>
        text.setContent(
          `## Thanks for the boost! <a:booster:1527782487349268480>\n` +
          `Thank you for boosting the server **${guild.name}**\n\n`+
          `Your support helps us unlock new server perks and continue making this community even better for everyone.\n\n`+
          `We truly appreciate you being part of our journey. Enjoy your booster rewards! ✨`
        )
      );
      if (serverImage) {
        comp.setThumbnailAccessory(accessory => accessory.setURL(serverImage));
      }
      return comp;
    }).setAccentColor(SystemColors.main);
    await member.send({
      components: [container],
      flags: [MessageFlags.IsComponentsV2]
    });
  } 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(settings.logChannelId) as
    | TextChannel
    | undefined;

  const logEmbed = new EmbedBuilder()
    .setColor(0xed4245)
    .setTitle("Boost Ended")
    .addFields(
      {
        name: "User",
        value: `${member.user.tag} (${member.id})`,
        inline: false,
      },
      {
        name: "Historical Boost Count",
        value: String(result.boostCounts ?? 0),
        inline: true,
      },
    )
    .setThumbnail(member.displayAvatarURL({ size: 512 }))
    .setTimestamp();

  await safeSend(logChannel, { embeds: [logEmbed] });
}