feat(detector): add database logic

h-projects · Aug 29, 2021 · 259b413 · 259b413
1 parent ba6a9dd
commit 259b413
Show file tree

Hide file tree

Showing 3 changed files with 110 additions and 75 deletions.
diff --git a/commands/bot/detector.js b/commands/bot/detector.js
@@ -4,31 +4,24 @@ module.exports = {
   permissions: 'MANAGE_MESSAGES',
 
   async execute(client, message) {
-    const buttons = [
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Low',
-        customId: `detector:low:${message.author.id}`
-      },
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Medium',
-        customId: `detector:medium:${message.author.id}`
-      },
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Hiqh',
-        customId: `detector:hiqh:${message.author.id}`
-      }
-    ];
+    const level = client.db.prepare('SELECT level FROM guilds WHERE id = ?').get(message.guild.id)?.level ?? 1;
+    let levelText;
+    switch (level) {
+      case 0:
+        levelText = 'Low';
+      break;
+      case 1:
+        levelText = 'Medium';
+      break;
+      case 2:
+        levelText = 'Hiqh';
+      break;
+    }
 
     message.channel.send({
       embeds: [{
         title: 'G Detector Levels',
-        description: 'Your current protection level: **Low**',
+        description: `Your current protection level: **${levelText}**`,
         color: client.config.color,
         fields: [
           {
@@ -47,7 +40,26 @@ module.exports = {
       }],
       components: [{
         type: 'ACTION_ROW',
-        components: buttons
+        components: [
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Low',
+            customId: `detector:low:${message.author.id}`
+          },
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Medium',
+            customId: `detector:medium:${message.author.id}`
+          },
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Hiqh',
+            customId: `detector:hiqh:${message.author.id}`
+          }
+        ]
       }]
     });
   }

diff --git a/interactions/commands/detector.js b/interactions/commands/detector.js
@@ -2,55 +2,53 @@ module.exports = {
   name: 'detector',
 
   async execute(client, interaction) {
-    const level = interaction.options.getString('level');
+    const input = interaction.options.getString('level');
+    const database = client.db.prepare('SELECT level FROM guilds WHERE id = ?').get(interaction.guildId);
+    const statement = database ? 'UPDATE guilds SET level = @level WHERE id = @id' : 'INSERT INTO guilds (id, level) VALUES (@id, @level)';
+
+    let levelText;
     let description;
     let fields = null;
-    switch (level) {
+
+    switch (database?.level ?? 1) {
+      case 0:
+        levelText = 'Low';
+      break;
+
+      case 1:
+        levelText = 'Medium';
+      break;
+
+      case 2:
+        levelText = 'Hiqh';
+      break;
+    }
+
+    switch (input) {
       case 'low':
         description = 'Successfully set detection level to **Low**!';
+        client.db.prepare(statement).run({ id: interaction.guildId, level: 0 });
       break;
 
       case 'medium':
         description = 'Successfully set detection level to **Medium**!';
+        client.db.prepare(statement).run({ id: interaction.guildId, level: 1 });
       break;
 
       case 'hiqh':
         description = 'Successfully set detection level to **Hiqh**!';
+        client.db.prepare(statement).run({ id: interaction.guildId, level: 2 });
       break;
 
       default:
-        description = 'Your current protection level: **Low**';
+        description = `Your current protection level: **${levelText}**`;
         fields = [
           { name: 'Low', value: 'Detects messaqes that only consist of G' },
           { name: 'Medium', value: 'Detects G outside words' },
           { name: 'Hiqh', value: 'Detects a messaqe if it contains G' }
         ];
     }
 
-    const buttons = [
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Low',
-        customId: `detector:low:${interaction.user.id}`,
-        disabled: level === 'low'
-      },
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Medium',
-        customId: `detector:medium:${interaction.user.id}`,
-        disabled: level === 'medium'
-      },
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Hiqh',
-        customId: `detector:hiqh:${interaction.user.id}`,
-        disabled: level === 'hiqh'
-      }
-    ];
-
     interaction.reply({
       embeds: [{
         title: 'G Detector Levels',
@@ -60,7 +58,29 @@ module.exports = {
       }],
       components: [{
         type: 'ACTION_ROW',
-        components: buttons
+        components: [
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Low',
+            customId: `detector:low:${interaction.user.id}`,
+            disabled: input === 'low'
+          },
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Medium',
+            customId: `detector:medium:${interaction.user.id}`,
+            disabled: input === 'medium'
+          },
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Hiqh',
+            customId: `detector:hiqh:${interaction.user.id}`,
+            disabled: input === 'hiqh'
+          }
+        ]
       }]
     });
   }

diff --git a/interactions/components/detector.js b/interactions/components/detector.js
@@ -2,46 +2,27 @@ module.exports = {
   name: 'detector',
 
   async execute(client, interaction) {
+    const database = client.db.prepare('SELECT level FROM guilds WHERE id = ?').get(interaction.guildId);
+    const statement = database ? 'UPDATE guilds SET level = @level WHERE id = @id' : 'INSERT INTO guilds (id, level) VALUES (@id, @level)';
     let description;
 
     switch (interaction.value) {
       case 'low':
         description = 'Successfully set detection level to **Low**!';
+        client.db.prepare(statement).run({ id: interaction.guildId, level: 0 });
       break;
 
       case 'medium':
         description = 'Successfully set detection level to **Medium**!';
+        client.db.prepare(statement).run({ id: interaction.guildId, level: 1 });
       break;
 
       case 'hiqh':
         description = 'Successfully set detection level to **Hiqh**!';
+        client.db.prepare(statement).run({ id: interaction.guildId, level: 2 });
       break;
     }
 
-    const buttons = [
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Low',
-        customId: `detector:low:${interaction.user.id}`,
-        disabled: interaction.value === 'low'
-      },
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Medium',
-        customId: `detector:medium:${interaction.user.id}`,
-        disabled: interaction.value === 'medium'
-      },
-      {
-        type: 'BUTTON',
-        style: 'SECONDARY',
-        label: 'Hiqh',
-        customId: `detector:hiqh:${interaction.user.id}`,
-        disabled: interaction.value === 'hiqh'
-      }
-    ];
-
     interaction.update({
       embeds: [{
         title: 'G Detector Levels',
@@ -50,7 +31,29 @@ module.exports = {
       }],
       components: [{
         type: 'ACTION_ROW',
-        components: buttons
+        components: [
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Low',
+            customId: `detector:low:${interaction.user.id}`,
+            disabled: interaction.value === 'low'
+          },
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Medium',
+            customId: `detector:medium:${interaction.user.id}`,
+            disabled: interaction.value === 'medium'
+          },
+          {
+            type: 'BUTTON',
+            style: 'SECONDARY',
+            label: 'Hiqh',
+            customId: `detector:hiqh:${interaction.user.id}`,
+            disabled: interaction.value === 'hiqh'
+          }
+        ]
       }]
     });
   }