adding support for confirm

2 files changed, 55 insertions, 33 deletions

diff --git a/controllers/interfaces/tools.rb b/controllers/interfaces/tools.rb
index 1136600..d32afed 100644
--- a/controllers/interfaces/tools.rb
+++ b/controllers/interfaces/tools.rb
@@ -10,6 +10,17 @@ module NanoBot
   module Controllers
     module Interfaces
       module Tool
+        def self.confirm(session, cartridge, mode, feedback)
+          yeses = Logic::Cartridge::Safety.yeses(cartridge)
+          default_answer = Logic::Cartridge::Safety.default_answer(cartridge)
+          dispatch_feedback(session, cartridge, mode, feedback)
+          session.flush
+          answer = $stdin.gets.chomp.to_s.downcase.strip
+          answer = default_answer if answer == ''
+          session.print("\n")
+          yeses.include?(answer)
+        end
+
         def self.adapt(feedback, adapter, cartridge)
           call = {
             parameters: %w[id name parameters parameters-as-json output],
@@ -41,6 +52,8 @@ module NanoBot
         def self.dispatch_feedback(session, cartridge, mode, feedback)
           enabled = Logic::Cartridge::Tools.feedback?(cartridge, mode.to_sym, feedback[:action].to_sym)
 
+          enabled = true if feedback[:action].to_sym == :confirm
+
           return unless enabled
 
           color = Logic::Cartridge::Tools.fetch_from_interface(
diff --git a/controllers/session.rb b/controllers/session.rb
index 546a891..b3cc7ef 100644
--- a/controllers/session.rb
+++ b/controllers/session.rb
@@ -6,6 +6,7 @@ require 'fileutils'
 require 'rainbow'
 
 require_relative '../logic/helpers/hash'
+require_relative '../logic/cartridge/safety'
 require_relative '../logic/cartridge/streaming'
 require_relative '../logic/cartridge/interaction'
 require_relative '../logic/cartridge/fetch'
@@ -102,10 +103,9 @@ module NanoBot
         prefix = Logic::Cartridge::Affixes.get(@cartridge, mode.to_sym, :output, :prefix)
         suffix = Logic::Cartridge::Affixes.get(@cartridge, mode.to_sym, :output, :suffix)
 
-        color = Logic::Cartridge::Fetch.cascate(@cartridge, [
-                                                  [:interfaces, mode.to_sym, :output, :color],
-                                                  %i[interfaces output color]
-                                                ])
+        color = Logic::Cartridge::Fetch.cascate(
+          @cartridge, [[:interfaces, mode.to_sym, :output, :color], %i[interfaces output color]]
+        )
 
         color = color.to_sym if color
 
@@ -118,44 +118,53 @@ module NanoBot
         needs_another_round = false
 
         @provider.evaluate(input, @cartridge) do |feedback|
+          needs_another_round = true if feedback[:needs_another_round]
+
           updated_at = Time.now
 
-          needs_another_round = true if feedback[:needs_another_round]
+          if feedback[:interaction] &&
+             feedback.dig(:interaction, :meta, :tool, :action) &&
+             feedback[:interaction][:meta][:tool][:action] == 'confirm'
+            Interfaces::Tool.confirm(self, @cartridge, mode, feedback[:interaction][:meta][:tool])
+          else
 
-          if feedback[:interaction] && feedback.dig(:interaction, :meta, :tool, :action)
-            Interfaces::Tool.dispatch_feedback(self, @cartridge, mode, feedback[:interaction][:meta][:tool])
-          end
+            if feedback[:interaction] && feedback.dig(:interaction, :meta, :tool, :action)
+              Interfaces::Tool.dispatch_feedback(
+                self, @cartridge, mode, feedback[:interaction][:meta][:tool]
+              )
+            end
 
-          if feedback[:interaction]
-            event = Marshal.load(Marshal.dump(feedback[:interaction]))
-            event[:mode] = mode.to_s
-            event[:output] = nil
-
-            if feedback[:interaction][:who] == 'AI' && feedback[:interaction][:message]
-              event[:output] = feedback[:interaction][:message]
-              unless streaming
-                output = Logic::Cartridge::Interaction.output(
-                  @cartridge, mode.to_sym, feedback[:interaction], streaming, feedback[:finished]
-                )
-                output[:message] = Components::Adapter.apply(output[:message], @cartridge)
-                event[:output] = (output[:message]).to_s
+            if feedback[:interaction]
+              event = Marshal.load(Marshal.dump(feedback[:interaction]))
+              event[:mode] = mode.to_s
+              event[:output] = nil
+
+              if feedback[:interaction][:who] == 'AI' && feedback[:interaction][:message]
+                event[:output] = feedback[:interaction][:message]
+                unless streaming
+                  output = Logic::Cartridge::Interaction.output(
+                    @cartridge, mode.to_sym, feedback[:interaction], streaming, feedback[:finished]
+                  )
+                  output[:message] = Components::Adapter.apply(output[:message], @cartridge)
+                  event[:output] = (output[:message]).to_s
+                end
               end
-            end
 
-            @state[:history] << event if feedback[:should_be_stored]
+              @state[:history] << event if feedback[:should_be_stored]
 
-            if event[:output] && ((!feedback[:finished] && streaming) || (!streaming && feedback[:finished]))
-              self.print(color ? Rainbow(event[:output]).send(color) : event[:output])
-            end
+              if event[:output] && ((!feedback[:finished] && streaming) || (!streaming && feedback[:finished]))
+                self.print(color ? Rainbow(event[:output]).send(color) : event[:output])
+              end
 
-            # The `print` function already outputs a prefix and a suffix, so
-            # we should add them afterwards to avoid printing them twice.
-            event[:output] = "#{prefix}#{event[:output]}#{suffix}"
-          end
+              # The `print` function already outputs a prefix and a suffix, so
+              # we should add them afterwards to avoid printing them twice.
+              event[:output] = "#{prefix}#{event[:output]}#{suffix}"
+            end
 
-          if feedback[:finished]
-            flush
-            ready = true
+            if feedback[:finished]
+              flush
+              ready = true
+            end
           end
         end