From 39f1bb68c8a761feb4bac03e45b29431ce9d4b93 Mon Sep 17 00:00:00 2001
From: "Ean Milligan (Bastion)" <ean.milligan@gmail.com>
Date: Wed, 26 Apr 2023 07:23:38 -0400
Subject: [PATCH] Add system to change event between whitelisted and public

---
 db/populateDefaults.ts                   |  2 +
 src/buttons/_index.ts                    |  2 +
 src/buttons/live-event/editEvent.ts      | 69 ++++++++++++++++--------
 src/buttons/live-event/toggleWLStatus.ts | 48 +++++++++++++++++
 4 files changed, 99 insertions(+), 22 deletions(-)
 create mode 100644 src/buttons/live-event/toggleWLStatus.ts

diff --git a/db/populateDefaults.ts b/db/populateDefaults.ts
index e746b70..03fe1ce 100644
--- a/db/populateDefaults.ts
+++ b/db/populateDefaults.ts
@@ -29,6 +29,8 @@ const actions = [
 	'btn-eeChangeAct',
 	'btn-eeChangeTime',
 	'btn-eeChangeDesc',
+	'btn-eeMakePublic',
+	'btn-eeMakeWL',
 ];
 for (const action of actions) {
 	await dbClient.execute('INSERT INTO command_cnt(command) values(?)', [action]).catch((e) => {
diff --git a/src/buttons/_index.ts b/src/buttons/_index.ts
index c35d17e..90aff03 100644
--- a/src/buttons/_index.ts
+++ b/src/buttons/_index.ts
@@ -17,6 +17,7 @@ import { editDateTimeButton } from './live-event/editDateTime.ts';
 import { applyDescriptionButton } from './live-event/applyDescription.ts';
 import { applyDateTimeButton } from './live-event/applyDateTime.ts';
 import { updateEventButton } from './live-event/updateEvent.ts';
+import { toggleWLStatusButton } from './live-event/toggleWLStatus.ts';
 
 export const buttons: Array<Button> = [
 	gameSelectionButton,
@@ -37,4 +38,5 @@ export const buttons: Array<Button> = [
 	applyDescriptionButton,
 	applyDateTimeButton,
 	updateEventButton,
+	toggleWLStatusButton,
 ];
diff --git a/src/buttons/live-event/editEvent.ts b/src/buttons/live-event/editEvent.ts
index 3a79116..4f328c6 100644
--- a/src/buttons/live-event/editEvent.ts
+++ b/src/buttons/live-event/editEvent.ts
@@ -1,4 +1,4 @@
-import { ApplicationCommandFlags, Bot, ButtonStyles, Interaction, InteractionResponseTypes, MessageComponentTypes } from '../../../deps.ts';
+import { ActionRow, ApplicationCommandFlags, Bot, ButtonStyles, Interaction, InteractionResponseTypes, MessageComponentTypes } from '../../../deps.ts';
 import { dbClient, generateGuildSettingKey, lfgChannelSettings, queries } from '../../db.ts';
 import { infoColor1, somethingWentWrong, stopThat } from '../../commandUtils.ts';
 import { idSeparator, pathIdxEnder, pathIdxSeparator } from '../eventUtils.ts';
@@ -6,11 +6,15 @@ import { addTokenToMap, selfDestructMessage } from '../tokenCleanup.ts';
 import utils from '../../utils.ts';
 import { customId as editDescriptionCustomId } from './editDescription.ts';
 import { customId as editDateTimeCustomId } from './editDateTime.ts';
+import { customId as toggleWLStatusCustomId } from './toggleWLStatus.ts';
 
 export const customId = 'editEvent';
 
 const execute = async (bot: Bot, interaction: Interaction) => {
-	if (interaction.data?.customId && interaction.member && interaction.channelId && interaction.guildId && interaction.message) {
+	if (
+		interaction.data?.customId && interaction.member && interaction.channelId && interaction.guildId && interaction.message && interaction.message.components &&
+		interaction.message.components[0].components
+	) {
 		// Light Telemetry
 		dbClient.execute(queries.callIncCnt('btn-editEvent')).catch((e) => utils.commonLoggers.dbError('editEvent.ts', 'call sproc INC_CNT on', e));
 
@@ -24,11 +28,50 @@ const execute = async (bot: Bot, interaction: Interaction) => {
 		// Make sure this is being done by the owner or a Group Up Manager
 		if (interaction.member.id === ownerId || (lfgChannelSetting.managed && interaction.member.roles.includes(lfgChannelSetting.managerRoleId))) {
 			const actionByManager = interaction.member.id !== ownerId;
-			const editIdxPath = `${idSeparator}${interaction.channelId}${pathIdxSeparator}${interaction.message.id}${actionByManager ? pathIdxEnder : ''}`;
+			const baseEditIdxPath = `${idSeparator}${interaction.channelId}${pathIdxSeparator}${interaction.message.id}`;
+			const editIdxPath = `${baseEditIdxPath}${actionByManager ? pathIdxEnder : ''}`;
+			const whitelistedEvent = interaction.message.components[0].components.some((component) => component.customId?.includes(idSeparator));
 
 			// Store token for later use
 			addTokenToMap(bot, interaction, interaction.guildId, interaction.channelId, interaction.member.id);
 
+			const editButtons: ActionRow = {
+				type: MessageComponentTypes.ActionRow,
+				components: [{
+					type: MessageComponentTypes.Button,
+					label: 'Change Activity',
+					style: ButtonStyles.Primary,
+					customId: `a${editIdxPath}`, // TODO: add customId
+				}, {
+					type: MessageComponentTypes.Button,
+					label: 'Change Date/Time',
+					style: ButtonStyles.Primary,
+					customId: `${editDateTimeCustomId}${editIdxPath}`,
+				}, {
+					type: MessageComponentTypes.Button,
+					label: 'Edit Description',
+					style: ButtonStyles.Primary,
+					customId: `${editDescriptionCustomId}${editIdxPath}`,
+				}],
+			};
+			if (!actionByManager) {
+				editButtons.components.push(
+					whitelistedEvent
+						? {
+							type: MessageComponentTypes.Button,
+							label: 'Make Event Public',
+							style: ButtonStyles.Primary,
+							customId: `${toggleWLStatusCustomId}${baseEditIdxPath}`, // TODO: add customId
+						}
+						: {
+							type: MessageComponentTypes.Button,
+							label: 'Make Event Whitelisted',
+							style: ButtonStyles.Primary,
+							customId: `${toggleWLStatusCustomId}${baseEditIdxPath}${pathIdxEnder}`, // TODO: add customId
+						},
+				);
+			}
+
 			// Open Edit Options
 			bot.helpers.sendInteractionResponse(
 				interaction.id,
@@ -50,25 +93,7 @@ const execute = async (bot: Bot, interaction: Interaction) => {
 
 ${selfDestructMessage(new Date().getTime())}`,
 						}],
-						components: [{
-							type: MessageComponentTypes.ActionRow,
-							components: [{
-								type: MessageComponentTypes.Button,
-								label: 'Change Activity',
-								style: ButtonStyles.Primary,
-								customId: `a${editIdxPath}`, // TODO: add customId
-							}, {
-								type: MessageComponentTypes.Button,
-								label: 'Change Date/Time',
-								style: ButtonStyles.Primary,
-								customId: `${editDateTimeCustomId}${editIdxPath}`,
-							}, {
-								type: MessageComponentTypes.Button,
-								label: 'Edit Description',
-								style: ButtonStyles.Primary,
-								customId: `${editDescriptionCustomId}${editIdxPath}`,
-							}],
-						}],
+						components: [editButtons],
 					},
 				},
 			).catch((e: Error) => utils.commonLoggers.interactionSendError('editEvent.ts', interaction, e));
diff --git a/src/buttons/live-event/toggleWLStatus.ts b/src/buttons/live-event/toggleWLStatus.ts
new file mode 100644
index 0000000..dca44fa
--- /dev/null
+++ b/src/buttons/live-event/toggleWLStatus.ts
@@ -0,0 +1,48 @@
+import { ApplicationCommandFlags, Bot, Interaction, InteractionResponseTypes, MessageComponentTypes } from '../../../deps.ts';
+import { safelyDismissMsg, somethingWentWrong, successColor } from '../../commandUtils.ts';
+import { idSeparator, pathIdxEnder, pathIdxSeparator } from '../eventUtils.ts';
+import { deleteTokenEarly } from '../tokenCleanup.ts';
+import utils from '../../utils.ts';
+import { dbClient, queries } from '../../db.ts';
+import { generateLFGButtons } from '../event-creation/utils.ts';
+
+export const customId = 'toggleWLStatus';
+
+const execute = async (bot: Bot, interaction: Interaction) => {
+	if (interaction.data?.customId && interaction.member && interaction.channelId && interaction.guildId) {
+		deleteTokenEarly(bot, interaction, interaction.guildId, interaction.channelId, interaction.member.id);
+		const makeWhitelisted = interaction.data.customId.endsWith(pathIdxEnder);
+		dbClient.execute(queries.callIncCnt(makeWhitelisted ? 'btn-eeMakeWL' : 'btn-eeMakePublic')).catch((e) => utils.commonLoggers.dbError('toggleWLStatus.ts', 'call sproc INC_CNT on', e));
+		const [evtChannelId, evtMessageId] = (interaction.data.customId.replaceAll(pathIdxEnder, '').split(idSeparator)[1] || '').split(pathIdxSeparator).map((id) => BigInt(id || '0'));
+
+		// TODO: verify we can DM owner before applying update
+		bot.helpers.editMessage(evtChannelId, evtMessageId, {
+			components: [{
+				type: MessageComponentTypes.ActionRow,
+				components: generateLFGButtons(makeWhitelisted),
+			}],
+		}).then(() =>
+			bot.helpers.sendInteractionResponse(interaction.id, interaction.token, {
+				type: InteractionResponseTypes.ChannelMessageWithSource,
+				data: {
+					flags: ApplicationCommandFlags.Ephemeral,
+					embeds: [{
+						color: successColor,
+						title: 'Update successfully applied.',
+						description: `This event is now ${makeWhitelisted ? 'whitelisted, meaning you will have to approve join requests from all future members' : 'public'}.\n\n${safelyDismissMsg}`,
+					}],
+				},
+			}).catch((e: Error) => utils.commonLoggers.interactionSendError('toggleWLStatus.ts', interaction, e))
+		).catch((e) => {
+			utils.commonLoggers.messageEditError('toggleWLStatus.ts', 'toggleWLStatusFailed', e);
+			somethingWentWrong(bot, interaction, 'editFailedInToggleWLStatusButton');
+		});
+	} else {
+		somethingWentWrong(bot, interaction, 'noDataFromToggleWLStatusButton');
+	}
+};
+
+export const toggleWLStatusButton = {
+	customId,
+	execute,
+};