From b14f057dd080bd8ef87d04f100b6e36ca2246ad5 Mon Sep 17 00:00:00 2001
From: Isaac <git@eartharoid.me>
Date: Fri, 10 Mar 2023 23:28:46 +0000
Subject: [PATCH] feat: update command permissions (closes #392)

---
 .../[guild]/categories/[category]/index.js    | 33 +++++++++++++++++
 .../admin/guilds/[guild]/categories/index.js  | 36 ++++++++++++++++++-
 2 files changed, 68 insertions(+), 1 deletion(-)

diff --git a/src/routes/api/admin/guilds/[guild]/categories/[category]/index.js b/src/routes/api/admin/guilds/[guild]/categories/[category]/index.js
index 2aeae9b..9900ea1 100644
--- a/src/routes/api/admin/guilds/[guild]/categories/[category]/index.js
+++ b/src/routes/api/admin/guilds/[guild]/categories/[category]/index.js
@@ -1,6 +1,7 @@
 const { logAdminEvent } = require('../../../../../../../lib/logging');
 const { updateStaffRoles } = require('../../../../../../../lib/users');
 const { randomUUID } = require('crypto');
+const { ApplicationCommandPermissionType } = require('discord.js');
 
 module.exports.delete = fastify => ({
 	handler: async (req, res) => {
@@ -71,6 +72,7 @@ module.exports.patch = fastify => ({
 		const client = res.context.config.client;
 		const guildId = req.params.guild;
 		const categoryId = Number(req.params.category);
+		/** @type {import('discord.js').Guild} */
 		const guild = client.guilds.cache.get(req.params.guild);
 		const data = req.body;
 
@@ -144,6 +146,37 @@ module.exports.patch = fastify => ({
 		await client.tickets.getCategory(categoryId, true);
 		await updateStaffRoles(guild);
 
+		if (req.user.payload.accessToken && JSON.stringify(category.staffRoles) !== JSON.stringify(original.staffRoles)) {
+			Promise.all([
+				'Create ticket for user',
+				'claim',
+				'force-close',
+				'move',
+				'priority',
+				'release',
+			].map(name =>
+				client.application.commands.permissions.set({
+					command: client.application.commands.cache.find(cmd => cmd.name === name),
+					guild,
+					permissions: [
+						{
+							id: guild.id, // @everyone
+							permission: false,
+							type: ApplicationCommandPermissionType.Role,
+						},
+						...category.staffRoles.map(id => ({
+							id,
+							permission: true,
+							type: ApplicationCommandPermissionType.Role,
+						})),
+					],
+					token: req.user.payload.accessToken,
+				}),
+			))
+				.then(() => client.log.success('Updated application command permissions in "%s"', guild.name))
+				.catch(error => client.log.error(error));
+		}
+
 		logAdminEvent(client, {
 			action: 'update',
 			diff: {
diff --git a/src/routes/api/admin/guilds/[guild]/categories/index.js b/src/routes/api/admin/guilds/[guild]/categories/index.js
index 318914f..08f896a 100644
--- a/src/routes/api/admin/guilds/[guild]/categories/index.js
+++ b/src/routes/api/admin/guilds/[guild]/categories/index.js
@@ -1,7 +1,10 @@
 const { logAdminEvent } = require('../../../../../../lib/logging');
 const { updateStaffRoles } = require('../../../../../../lib/users');
 const emoji = require('node-emoji');
-const { ChannelType: { GuildCategory } } = require('discord.js');
+const {
+	ApplicationCommandPermissionType,
+	ChannelType: { GuildCategory },
+} = require('discord.js');
 const ms = require('ms');
 
 module.exports.get = fastify => ({
@@ -98,6 +101,37 @@ module.exports.post = fastify => ({
 		await client.tickets.getCategory(category.id, true);
 		await updateStaffRoles(guild);
 
+		if (req.user.payload.accessToken) {
+			Promise.all([
+				'Create ticket for user',
+				'claim',
+				'force-close',
+				'move',
+				'priority',
+				'release',
+			].map(name =>
+				client.application.commands.permissions.set({
+					command: client.application.commands.cache.find(cmd => cmd.name === name),
+					guild,
+					permissions: [
+						{
+							id: guild.id, // @everyone
+							permission: false,
+							type: ApplicationCommandPermissionType.Role,
+						},
+						...category.staffRoles.map(id => ({
+							id,
+							permission: true,
+							type: ApplicationCommandPermissionType.Role,
+						})),
+					],
+					token: req.user.payload.accessToken,
+				}),
+			))
+				.then(() => client.log.success('Updated application command permissions in "%s"', guild.name))
+				.catch(error => client.log.error(error));
+		}
+
 		logAdminEvent(client, {
 			action: 'create',
 			guildId: guild.id,