Focus mode + Role management

discord-scripts/focus-mode.ts

@@ -0,0 +1,129 @@
+import {
+  ChannelType,
+  Client,
+  ButtonStyle,
+  ButtonBuilder,
+  ActionRowBuilder,
+  TextChannel,
+  Message,
+} from "discord.js"
+import { Robot } from "hubot"
+import manageRole from "./role-management/role-management.ts"
+
+const focusModeState: Map<string, boolean> = new Map()
+
+async function sendFocusModeMessage(
+  focusChannel: TextChannel,
+  guildId: string,
+  robot: Robot,
+  originalMessage?: Message,
+) {
+  const isEnabled = focusModeState.get(guildId) || false
+
+  const actionRow = new ActionRowBuilder<ButtonBuilder>().addComponents(
+    new ButtonBuilder()
+      .setCustomId("enable-focus")
+      .setLabel("Enable Focus")
+      .setStyle(ButtonStyle.Success)
+      .setDisabled(isEnabled),
+    new ButtonBuilder()
+      .setCustomId("disable-focus")
+      .setLabel("Disable Focus")
+      .setStyle(ButtonStyle.Danger)
+      .setDisabled(!isEnabled),
+  )
+
+  const messageContent = {
+    content:
+      "**Focus Mode**\nSelect an option to enable or disable focus mode.",
+    components: [actionRow],
+  }
+
+  if (originalMessage) {
+    await originalMessage.edit(messageContent)
+  } else {
+    await focusChannel.send(messageContent)
+  }
+
+  robot.logger.info(
+    `Focus mode options updated in the focus channel of guild: ${guildId}`,
+  )
+}
+
+export default async function setupFocusChannels(
+  discordClient: Client,
+  robot: Robot,
+) {
+  discordClient.on("messageCreate", async (message) => {
+    if (message.author.bot) return
+
+    if (
+      message.guild &&
+      message.channel.type === ChannelType.GuildText &&
+      message.channel.name.startsWith("focus")
+    ) {
+      const guildId = message.guild.id
+      const focusChannel = message.channel as TextChannel
+
+      if (!focusModeState.has(guildId)) {
+        focusModeState.set(guildId, false)
+        robot.logger.info(
+          `Initializing focus mode state for guild: ${message.guild.name}`,
+        )
+      }
+
+      robot.logger.info(
+        `Message received in focus channel of guild: ${message.guild.name}`,
+      )
+
+      await sendFocusModeMessage(focusChannel, guildId, robot)
+    }
+  })
+
+  discordClient.on("interactionCreate", async (interaction) => {
+    if (!interaction.isButton()) return
+
+    const guildId = interaction.guildId
+    if (!guildId || !interaction.member) return
+
+    let isEnabled = focusModeState.get(guildId) || false
+
+    if (interaction.customId === "enable-focus") {
+      isEnabled = true
+      await interaction.reply({
+        content: "Focus mode enabled.",
+        ephemeral: true,
+      })
+      await manageRole(
+        discordClient,
+        guildId,
+        interaction.member.user.id,
+        "1205116222161813545",
+        "add",
+      )
+    } else if (interaction.customId === "disable-focus") {
+      isEnabled = false
+      await interaction.reply({
+        content: "Focus mode disabled.",
+        ephemeral: true,
+      })
+      await manageRole(
+        discordClient,
+        guildId,
+        interaction.member.user.id,
+        "1205116222161813545",
+        "remove",
+      )
+    }
+
+    focusModeState.set(guildId, isEnabled)
+    if (interaction.message instanceof Message) {
+      sendFocusModeMessage(
+        interaction.channel as TextChannel,
+        guildId,
+        robot,
+        interaction.message,
+      )
+    }
+  })
+}

discord-scripts/role-management/role-management.ts

@@ -0,0 +1,34 @@
+import { Client } from "discord.js"
+
+export default async function manageRole(
+  discordClient: Client,
+  guildId: string,
+  memberId: string,
+  roleId: string,
+  action: "add" | "remove",
+): Promise<void> {
+  if (!discordClient) {
+    throw new Error("Discord client is not initialized.")
+  }
+
+  const guild = await discordClient.guilds.fetch(guildId)
+  if (!guild) throw new Error("Guild not found.")
+
+  const member = await guild.members.fetch(memberId)
+  if (!member) throw new Error("Member not found.")
+
+  const role = await guild.roles.fetch(roleId)
+  if (!role) throw new Error("Role not found.")
+
+  if (action === "add") {
+    if (member.roles.cache.has(roleId)) {
+      return
+    }
+    await member.roles.add(role)
+  } else if (action === "remove") {
+    if (!member.roles.cache.has(roleId)) {
+      return
+    }
+    await member.roles.remove(role)
+  }
+}