Add bot restart

MrShoopa · Feb 7, 2020 · 92030d2 · 92030d2
1 parent c377fa4
commit 92030d2
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/src/bot_functions/TriggerHandlers.ts b/src/bot_functions/TriggerHandlers.ts
@@ -50,6 +50,8 @@ export default class TriggerHandlers {
 
         TriggerHandlers.checkForDiceRollRequest,
 
+        TriggerHandlers.checkForBotKillRequest,
+
         HelpTriggers.checkForHelpInfoRequest
 
     ]
@@ -245,6 +247,14 @@ export default class TriggerHandlers {
         //  Get anime recommendation [from My Anime List (JikanTS)]
     }
 
+    /* ----  Admin-only functions --- */
+    private static checkForBotKillRequest(message = TriggerHandlers.message) {
+        for (const trigger of TRIGGERS.kill_trigger)
+            if (message.toString().toLowerCase().includes(trigger))
+                return BotGeneralCommands.killBot(true, trigger)
+        //  Redo last command
+    }
+
     //  All else comes around
     private static replyGeneralDefault(message = TriggerHandlers.message) {
         for (const trigger of TRIGGERS.main_trigger)

diff --git a/src/bot_functions/general/GeneralCommands.ts b/src/bot_functions/general/GeneralCommands.ts
@@ -1,6 +1,7 @@
 import Discord from 'discord.js'
 import Bot from "../../Bot"
 
+import AUTH from '../../user_creds.json'
 export default class BotGeneralCommands {
 
     static redoLastAction(trigger: string) {
@@ -19,4 +20,21 @@ export default class BotGeneralCommands {
         return bot.context.channel.send('redoin, ' + bot.lastMessage.toString())
     }
 
+    static async killBot(adminOnly = true, trigger: string) {
+        let bot: Bot = globalThis.bot
+        bot.preliminary(trigger, 'Bot kill', true)
+
+        if (adminOnly) {
+            if (bot.context.author.id === AUTH.you.admin_user_id) {
+                await bot.context.channel.send('Yes master. ✅')
+                process.exit(69)
+            } else {
+                bot.context.channel.send(`You can't quite tell me to, ${bot.context.author.username}. `)
+            }
+        } else {
+            await bot.context.channel.send('Restarting.')
+            process.exit(100)
+        }
+    }
+
 }
diff --git a/src/bot_knowledge/triggers/triggers.json b/src/bot_knowledge/triggers/triggers.json
@@ -13,6 +13,14 @@
         "megarewind",
         "megaredo"
     ],
+    "kill_trigger": [
+        "megadork go kill yourself",
+        "megadork reset please",
+        "megadork do me a favor and kill yourself",
+        "megadork restart",
+        "megadork reboot",
+        "megadie"
+    ],
     "url_trigger": {
         "any": "."
     },