From 0bd83dad8772c417a04003bd1e74695c46950071 Mon Sep 17 00:00:00 2001
From: Braeden Sowinski <sowinskibraeden@gmail.com>
Date: Wed, 16 Mar 2022 20:15:50 -0700
Subject: [PATCH] update/remove role and exists func

---
 commands/removeRole.js | 94 ++++++++++++++++++++++++++++++++++++++++++
 commands/setRole.js    |  5 ++-
 util/Exists.js         |  4 ++
 3 files changed, 101 insertions(+), 2 deletions(-)
 create mode 100644 commands/removeRole.js
 create mode 100644 util/Exists.js

diff --git a/commands/removeRole.js b/commands/removeRole.js
new file mode 100644
index 0000000..0555196
--- /dev/null
+++ b/commands/removeRole.js
@@ -0,0 +1,94 @@
+const { MessageEmbed } = require('discord.js');
+
+module.exports = {
+  name: "removerole",
+  description: "Remove a role allowed to use the bot",
+  usage: "[role]",
+  permissions: {
+    channel: ["VIEW_CHANNEL", "SEND_MESSAGES", "EMBED_LINKS"],
+    member: ["MANAGE_GUILD"],
+  },
+  aliases: ["remove_role"],
+  /**
+   *
+   * @param {require("../structures/LinesPoliceCadBot")} client
+   * @param {import("discord.js").MessageCreate} message
+   * @param {string[]} args
+   * @param {*} param3
+  */
+  run: async (client, message, args) => {
+    if (args.length==0) return message.channel.send(`You must provide a role.`);
+    let roleid = args[0].replace('<@&', '').replace('>', '');
+    let role = message.guild.roles.cache.find(x => x.id == roleid);
+    if (role == undefined) {
+      return message.channel.send(`Uh Oh! The role ${args[0]} connot be found.`);
+    } else {
+      let guild = await client.dbo.collection("prefixes").findOne({"server.serverID":message.guild.id}).then(guild => guild);
+      if (guild.server.hasCustomRoles==false) return message.channel.send(`There are no roles to be removed.`);
+      if (!guild.server.allowedRoles.includes(roleid)) return message.channel.send(`The role ${args[0]} is not added to your roles.`);
+      for (let i = 0; i < guild.server.allowedRoles.length; i++) {
+        if (guild.server.allowedRoles[i]==roleid) {
+          if ((guild.server.allowedRoles.length-1)==0) {
+            client.dbo.collection("prefixes").updateOne({"server.serverID":message.guild.id},{$pull:{"server.allowedRoles":roleid},$set:{"server.hasCustomRoles":false}},function(err, res) {
+              if (err) throw err;
+              return message.channel.send(`Successfully removed ${args[0]} from allowed roles! There are no more allowed roles.`);
+            });  
+          } else if ((guild.server.allowedRoles.length-1)>0) {
+            client.dbo.collection("prefixes").updateOne({"server.serverID":message.guild.id},{$pull:{"server.allowedRoles":roleid}},function(err, res) {
+              if (err) throw err;
+              return message.channel.send(`Successfully removed ${args[0]} from allowed roles.`);
+            });
+          }
+        }
+      }
+    }
+  },
+  SlashCommand: {
+    options: [
+      {
+        name: "role",
+        description: "Role to remove from allowed roles",
+        value: "role",
+        type: 8,
+        required: true,
+      },
+    ],  
+    /**
+     *
+     * @param {require("../structures/LinesPoliceCadBot")} client
+     * @param {import("discord.js").MessageCreate} message
+     * @param {string[]} args
+     * @param {*} param3
+    */
+    run: async (client, interaction, args, { GuildDB }) => {
+      if (GuildDB.customChannelStatus==true&&!GuildDB.allowedChannels.includes(interaction.channel_id)) {
+        return interaction.send(`You are not allowed to use the bot in this channel.`);
+      }
+      
+      let roleid = args[0].value;
+      let role = interaction.guild.roles.cache.find(x => x.id == roleid);
+      if (role == undefined) {
+        return interaction.send(`Uh Oh! The role <@&${args[0].value}> connot be found.`);
+      } else {
+        let guild = await client.dbo.collection("prefixes").findOne({"server.serverID":interaction.guild.id}).then(guild => guild);
+        if (guild.server.hasCustomRoles==false) return interaction.send(`There are no roles to be removed.`);
+        if (!guild.server.allowedRoles.includes(roleid)) return interaction.send(`The role <@&${args[0].value}> is not added to your roles.`);
+        for (let i = 0; i < guild.server.allowedRoles.length; i++) {
+          if (guild.server.allowedRoles[i]==roleid) {
+            if ((guild.server.allowedRoles.length-1)==0) {
+              client.dbo.collection("prefixes").updateOne({"server.serverID":interaction.guild.id},{$pull:{"server.allowedRoles":roleid},$set:{"server.hasCustomRoles":false}},function(err, res) {
+                if (err) throw err;
+                return interaction.send(`Successfully removed <@&${args[0].value}> from allowed roles! There are no more allowed roles.`);
+              });  
+            } else if ((guild.server.allowedRoles.length-1)>0) {
+              client.dbo.collection("prefixes").updateOne({"server.serverID":interaction.guild.id},{$pull:{"server.allowedRoles":roleid}},function(err, res) {
+                if (err) throw err;
+                return interaction.send(`Successfully removed <@&${args[0].value}> from allowed roles.`);
+              });
+            }
+          }
+        }
+      }
+    },
+  },
+}
\ No newline at end of file
diff --git a/commands/setRole.js b/commands/setRole.js
index 53602ba..cef54dd 100644
--- a/commands/setRole.js
+++ b/commands/setRole.js
@@ -1,4 +1,5 @@
 const { MessageEmbed } = require('discord.js');
+const exists = require('../util/Exists')
 
 module.exports = {
   name: "setrole",
@@ -24,7 +25,7 @@ module.exports = {
       return message.channel.send(`Uh Oh! The role ${args[0]} connot be found.`);
     } else {
       let guild = await client.dbo.collection("prefixes").findOne({"server.serverID":message.guild.id}).then(guild => guild);
-      if (guild.server.allowedRole!=undefined&&guild.server.allowedRoles.includes(roleid)) return message.channel.send(`The role ${args[0]} has already been added.`);
+      if (exists(guild.server.allowedRoles)&&guild.server.allowedRoles.includes(roleid)) return message.channel.send(`The role ${args[0]} has already been added.`);
       client.dbo.collection("prefixes").updateOne({"server.serverID":message.guild.id},{$push:{"server.allowedRoles":roleid},$set:{"server.hasCustomRoles":true}},function(err, res) {
         if (err) throw err;
         return message.channel.send(`Successfully added ${args[0]} to allowed roles.`);
@@ -59,7 +60,7 @@ module.exports = {
         return interaction.send(`Uh Oh! The role <@&${args[0].value}> connot be found.`);
       } else {
         let guild = await client.dbo.collection("prefixes").findOne({"server.serverID":interaction.guild.id}).then(guild => guild);
-        if (guild.server.allowedRole!=undefined&&guild.server.allowedRoles.includes(roleid)) return interaction.send(`The role <@&${args[0].value}> has already been added.`);
+        if (exists(guild.server.allowedRoles)&&guild.server.allowedRoles.includes(roleid)) return interaction.send(`The role <@&${args[0].value}> has already been added.`);
         client.dbo.collection("prefixes").updateOne({"server.serverID":interaction.guild.id},{$push:{"server.allowedRoles":roleid},$set:{"server.hasCustomRoles":true}},function(err, res) {
           if (err) throw err;
           return interaction.send(`Successfully added <@&${args[0].value}> to allowed roles.`);
diff --git a/util/Exists.js b/util/Exists.js
new file mode 100644
index 0000000..3cdeea0
--- /dev/null
+++ b/util/Exists.js
@@ -0,0 +1,4 @@
+module.exports = n => {
+	if (n == null || n == undefined || n == "") return false;
+	return true;
+}
\ No newline at end of file