feat: timezone roles (#40)

* init

* small things

* tested and fixed bugs

Author: StrangeGirlMurph

src/commands/RoleCommands.ts

@@ -5,7 +5,7 @@ import {
 	SlashCommandSubcommandsOnlyBuilder,
 } from "discord.js";
 import { hasAdminPerms } from "../util/misc/permissions";
-import { colorPrompt, pronounPrompt } from "../util/roleCommand/rolePrompts";
+import { colorPrompt, pronounPrompt, timezonePrompt } from "../util/roleCommand/rolePrompts";
 import { deleteRoles } from "../util/roleCommand/roleUtil";
 
 class RoleCommand extends Command {
@@ -31,6 +31,9 @@ class RoleCommand extends Command {
 		} else if (subcommand === "color-prompt") {
 			await interaction.deferReply();
 			await colorPrompt(interaction);
+		} else if (subcommand === "timezone-prompt") {
+			await interaction.deferReply();
+			await timezonePrompt(interaction);
 		} else if (subcommand === "delete") {
 			await interaction.deferReply({ ephemeral: true });
 			await deleteRoles(interaction);
@@ -59,6 +62,9 @@ class RoleCommand extends Command {
 							.setMaxValue(4)
 					)
 			)
+			.addSubcommand((option) =>
+				option.setName("timezone-prompt").setDescription("Creates the timezone role prompt.")
+			)
 			.addSubcommand((option) =>
 				option
 					.setName("delete")

src/config.ts

@@ -29,6 +29,33 @@ export const config: BotConfig = {
 		["Medium Aquamarine", "#6cd9a9"],
 		["Sea Serpent", "#5db9cf"],
 	],
+	timezoneRoles: [
+		"UTC-12",
+		"UTC-11",
+		"UTC-10",
+		"UTC-9",
+		"UTC-8",
+		"UTC-7",
+		"UTC-6",
+		"UTC-5",
+		"UTC-4",
+		"UTC-3",
+		"UTC-2",
+		"UTC-1",
+		"UTC",
+		"UTC+1",
+		"UTC+2",
+		"UTC+3",
+		"UTC+4",
+		"UTC+5",
+		"UTC+6",
+		"UTC+7",
+		"UTC+8",
+		"UTC+9",
+		"UTC+10",
+		"UTC+11",
+		"UTC+12",
+	],
 	color: "#F0A5AC",
 	serverDescription:
 		"We're a group of young and mostly queer people having a game jam/hackathon server together. We're a very friendly and welcoming community and are happy to have you join us! \nCheck out <#1022874504525008997> for more information!",
@@ -46,6 +73,8 @@ interface BotConfig {
 	pronounRoles: [string, ComponentEmojiResolvable | null][];
 	// Max 25 color roles to pick from. First argument is the name of the role, second argument is the color.
 	colorRoles: [string, ColorResolvable][];
+	// Timezones users can pick from.
+	timezoneRoles: string[];
 	// The main color the bot will use.
 	color: ColorResolvable;
 	// A description of the server that will be displayed in the welcome message.

src/interactions/selectMenus/roleSelectMenu.ts

@@ -0,0 +1,43 @@
+import { SelectMenu } from "../interactionClasses";
+import { inlineCode, SelectMenuInteraction } from "discord.js";
+import { logger } from "../../logger";
+
+class RoleSelectMenu extends SelectMenu {
+	constructor() {
+		super("role");
+	}
+
+	async execute(interaction: SelectMenuInteraction, subcommand: string[]): Promise<void> {
+		if (subcommand[0] == "timezone") {
+			const timezone = interaction.values[0] ?? "None";
+			const guildRoles = await interaction.guild?.roles.fetch();
+			const member = await interaction.guild?.members.fetch(interaction.user.id);
+			const role = guildRoles?.find((r) => r.name === timezone);
+			if (!role || !member) {
+				await interaction.reply("Can't find the role or you (the member)...");
+				logger.warn("Couldn't find a role or member. Probably is an old prompt still open.");
+				logger.error(role);
+				logger.error(member);
+				return;
+			}
+
+			const currentTimezoneRoles = member.roles.cache.filter((r) => r.name.startsWith("UTC"));
+
+			for (const currentRole of currentTimezoneRoles) {
+				await member.roles.remove(currentRole);
+			}
+
+			if (role) {
+				await member.roles.add(role);
+				await interaction.reply({ content: `Gave you the ${inlineCode(timezone)} role.`, ephemeral: true });
+			} else {
+				await interaction.reply({
+					content: `Removed your ${currentTimezoneRoles.map((c) => inlineCode(c.name)).join(" ")} role.`,
+					ephemeral: true,
+				});
+			}
+		}
+	}
+}
+
+export default new RoleSelectMenu();

src/util/roleCommand/rolePrompts.ts

@@ -6,14 +6,15 @@ import {
 	ButtonStyle,
 	ChatInputCommandInteraction,
 	EmbedBuilder,
+	SelectMenuBuilder,
 } from "discord.js";
 import { addDefaultEmbedFooter } from "../misc/embeds";
 import { createCanvas } from "@napi-rs/canvas";
 import { bringIntoButtonGrid, setUpRoles } from "./roleUtil";
 
 export async function pronounPrompt(interaction: ChatInputCommandInteraction): Promise<void> {
 	const prompt: EmbedBuilder = new EmbedBuilder()
-		.setTitle("Pronoun roles 🌈🏳‍⚧️⚧️")
+		.setTitle("Pronoun roles 🏳‍⚧️⚧️")
 		.setDescription(
 			"Select the pronouns you want others to use when referring to you :)\nIf you don't understand why: https://pronouns.org/what-and-why"
 		);
@@ -127,3 +128,35 @@ export async function colorPrompt(interaction: ChatInputCommandInteraction): Pro
 		logger.error("Failed to set up color roles.");
 	}
 }
+
+export async function timezonePrompt(interaction: ChatInputCommandInteraction): Promise<void> {
+	const prompt: EmbedBuilder = new EmbedBuilder()
+		.setTitle("Timezone roles 🌍")
+		.setDescription(
+			"Select the timezone you life in. That way we can schedule events at times that work for most of us :)"
+		);
+
+	const actionRow: ActionRowBuilder<SelectMenuBuilder> =
+		new ActionRowBuilder<SelectMenuBuilder>().setComponents(
+			new SelectMenuBuilder()
+				.setCustomId("role.timezone")
+				.setPlaceholder("Select your timezone!")
+				.setMinValues(1)
+				.setMaxValues(1)
+				.setOptions(...config.timezoneRoles.map((t) => ({ label: t, value: t })))
+		);
+
+	if (
+		await setUpRoles(
+			interaction.guild,
+			config.timezoneRoles.map((r) => [r]),
+			"- StartTimezoneRoles -",
+			"- EndTimezoneRoles -"
+		)
+	) {
+		await interaction.editReply({ embeds: [addDefaultEmbedFooter(prompt)], components: [actionRow] });
+	} else {
+		await interaction.editReply({ content: "Couldn't set up the roles..." });
+		logger.error("Failed to set up timezone roles.");
+	}
+}