MODTRIGGER

cfg/bot_state.json

@@ -1 +1 @@
-{"Async":["RefreshNP","RemindOnTime","URLCleaner","TempRoleNoMore","LoadAlko"],"Commands":["Help","FGOSummon","RollProbability","Register","Eval","Ping","Information","RoleAdd","RoleDel","Summon","ReminderAdd","Tester","NowPlaying","Alko","Alkolist","Chaos","Coinflip","Reminders","ReminderDel","TriggerAdd","TriggerDel","TriggerSearch","Triggers","Doubles","Triples","Quads","Quints","Spoiler","EncodeMsg","Quickie","GetSauce","Winner","Loser","EmojiList","Number","AskRin","AnswerAdd"],"Events":["TriggerEvent","URLCollector","RepostOverseer","DecodeMsg","SharedEmoji","AnotherRoll"]}
+{"Async":["RefreshNP","RemindOnTime","URLCleaner","TempRoleNoMore","LoadAlko"],"Commands":["Help","FGOSummon","RollProbability","Register","Eval","Ping","Information","RoleAdd","RoleDel","Summon","ReminderAdd","Tester","NowPlaying","Alko","Alkolist","Chaos","Coinflip","Reminders","ReminderDel","TriggerAdd","TriggerMod","TriggerDel","TriggerSearch","Triggers","Doubles","Triples","Quads","Quints","Spoiler","EncodeMsg","Quickie","GetSauce","Winner","Loser","EmojiList","Number","AskRin","AnswerAdd"],"Events":["TriggerEvent","URLCollector","RepostOverseer","DecodeMsg","SharedEmoji","AnotherRoll"]}

lib/commands/triggers/trigger_controller.rb

@@ -49,7 +49,7 @@ def store_trigger(reply: "", filename: "")
     #
     # @param reply [EventContainer] EventContainer for Message event
     # @return [String, nil] final filename with UUID
-    def download_reply_picture(reply)
+    def self.download_reply_picture(reply)
       file = reply.message.attachments.first
       if /https:\/\/cdn.discordapp.com.*/.match?(file.url)
 

lib/commands/triggers/trigger_mod.rb

@@ -0,0 +1,122 @@
+module TohsakaBot
+  module Commands
+    module TriggerMod
+      extend Discordrb::Commands::CommandContainer
+      command(:triggermod,
+              aliases: %i[triggermod triggermodify modtrigger modifytrigger edittrigger triggeredit],
+              description: 'Edits a trigger.',
+              usage: "Use 'triggermod -h|--help' for help.",
+              min_args: 1,
+              require_register: true,
+              enabled_in_pm: false) do |event, *msg|
+
+        admin = [73510616697929728, 73086349363650560]
+        discord_uid = event.author.id.to_i
+
+        extra_help = "Example: `triggermod -i 420 -p new trigger -r new response -c 100`"
+
+        options = TohsakaBot.command_parser(
+            event, msg, 'Usage: triggermod [-i id] [-p triggering phrase] [-r new reply] [-f new file] [-m new mode] [-c new chance]', extra_help,
+            [:id, 'Trigger id to edit', :type=> :string],
+            [:phrase, 'Edit the phrase the bot triggers on.', :type => :strings],
+            [:reply, 'Edit the text the bot responds with.', :type => :strings],
+            [:file, 'Edit the file the bot responds with.'],
+            [:mode, 'Edit the trigger mode. See `triggeradd --help` for more details.', :type => :string],
+            [:chance, 'Edit the chance of this trigger triggering [admin only; integer between 0 and 100]', :type => :integer]
+        )
+        break if options.nil?
+
+        if options.id.nil?
+          event.respond("Specify a trigger ID to edit")
+        end
+
+        triggers = TohsakaBot.db[:triggers]
+        trigger = triggers.where(:id => options.id.to_i).single_record!
+
+        if trigger.nil?
+          event.respond("Could not find trigger with that ID")
+          break
+        end
+
+        if trigger[:user_id] != TohsakaBot.get_user_id(discord_uid) and !admin.include?(discord_uid)
+          event.respond("No permissions to edit this trigger")
+          break
+        end
+
+        if options.phrase.nil? and options.reply.nil? and options.mode.nil? and options.file.nil? and options.chance.nil?
+          event.respond("Specify an action")
+          break
+        elsif !options.file.nil? and !options.reply.nil?
+          event.respond("Cannot edit both file and reply at the same time.")
+          break;
+        end
+
+        phrase = options.phrase.nil? ? nil : options.phrase.join(' ')
+        unless phrase.nil?
+          trigger[:phrase] = phrase
+        end
+
+        unless options.mode.nil?
+          trigger[:mode] = /e.*/s.match?(options.mode) ? 0 : 1
+        end
+
+        unless options.reply.nil?
+          reply = options.reply.join(' ')
+          trigger[:reply] = reply
+
+          File.delete("data/triggers/#{trigger[:file]}") unless trigger[:file].blank?
+          trigger[:file] = nil
+        end
+
+        if not options.chance.nil? and admin.include?(discord_uid)
+          chance = options.chance
+          if chance >= 0 and chance <= 100
+            trigger[:chance] = chance
+          end
+        end
+
+        unless options.file.nil?
+          old_file = trigger[:file]
+
+          if !event.message.attachments.first.nil?
+            filename = TriggerController.download_reply_picture(event)
+            if filename.nil?
+              event.respond('File too large. Max: ~8MiB or 8388119 bytes.')
+              break
+            end
+            trigger[:reply] = nil
+            trigger[:file] = filename
+            File.delete("data/triggers/#{old_file}") unless old_file.blank?
+          else
+            event.respond('Tell me the response (10s remaining).')
+            response = event.message.await!(timeout: 10)
+
+            if response
+              if !response.message.attachments.first.nil?
+                filename = TriggerController.download_reply_picture(response)
+                if filename.nil?
+                  event.respond('File too large. Max: ~8MiB or 8388119 bytes.')
+                  break
+                end
+                trigger[:reply] = nil
+                trigger[:file] = filename
+                File.delete("data/triggers/#{old_file}") unless old_file.blank?
+              end
+            else
+              event.respond('You took too long!')
+              break
+            end
+          end
+        end
+
+        event.respond("Trigger modified `<ID #{options.id}>`.")
+
+        TohsakaBot.db.transaction do
+          triggers.where(:id => options.id.to_i).update(trigger)
+        end
+
+        TohsakaBot.trigger_data.reload_active
+      end
+    end
+  end
+end