diff options
Diffstat (limited to 'controllers')
| -rw-r--r-- | controllers/instance.rb | 32 | ||||
| -rw-r--r-- | controllers/interfaces/cli.rb | 4 | ||||
| -rw-r--r-- | controllers/session.rb | 15 |
3 files changed, 40 insertions, 11 deletions
diff --git a/controllers/instance.rb b/controllers/instance.rb index f3cbd74..243e266 100644 --- a/controllers/instance.rb +++ b/controllers/instance.rb @@ -12,12 +12,14 @@ require_relative './session' module NanoBot module Controllers class Instance - def initialize(cartridge_path:, state: nil) + def initialize(cartridge_path:, stream:, state: nil) + @stream = stream + load_cartridge!(cartridge_path) provider = Components::Provider.new(@cartridge[:provider]) - @session = Session.new(provider:, cartridge: @cartridge, state:) + @session = Session.new(provider:, cartridge: @cartridge, state:, stream: @stream) end def debug @@ -26,18 +28,42 @@ module NanoBot def eval(input) Interfaces::Eval.evaluate(input, @cartridge, @session) + + return unless @stream.is_a?(StringIO) + + @stream.flush + result = @stream.string.clone + @stream.truncate(0) + @stream.rewind + result end def repl + if @stream.is_a?(StringIO) + @stream.flush + @stream = $stdout + @session.stream = @stream + end Interfaces::REPL.start(@cartridge, @session) end private def load_cartridge!(path) + elected_path = if path.strip == '-' + File.expand_path('../static/cartridges/default.yml', __dir__) + else + Components::Storage.cartridge_path(path) + end + + if elected_path.nil? + @stream.write("Cartridge file not found: \"#{path}\"\n") + raise StandardError, "Cartridge file not found: \"#{path}\"" + end + @cartridge = Logic::Helpers::Hash.symbolize_keys( YAML.safe_load( - File.read(Components::Storage.cartridge_path(path)), + File.read(elected_path), permitted_classes: [Symbol] ) ) diff --git a/controllers/interfaces/cli.rb b/controllers/interfaces/cli.rb index 3574a80..92e468b 100644 --- a/controllers/interfaces/cli.rb +++ b/controllers/interfaces/cli.rb @@ -26,7 +26,9 @@ module NanoBot params = { cartridge_path: ARGV[0], state: ARGV[1], command: ARGV[2] } - bot = Instance.new(cartridge_path: params[:cartridge_path], state: params[:state]) + bot = Instance.new( + cartridge_path: params[:cartridge_path], state: params[:state], stream: $stdout + ) case params[:command] when 'eval' diff --git a/controllers/session.rb b/controllers/session.rb index ff3b8ad..ad640c8 100644 --- a/controllers/session.rb +++ b/controllers/session.rb @@ -12,12 +12,13 @@ module NanoBot STREAM_TIMEOUT_IN_SECONDS = 5 class Session - def initialize(provider:, cartridge:, state: nil) + attr_accessor :stream + + def initialize(provider:, cartridge:, state: nil, stream: $stdout) + @stream = stream @provider = provider @cartridge = cartridge - @output = $stdout - @stateless = state.nil? || state.strip == '-' || state.strip.empty? if @stateless @@ -67,7 +68,7 @@ module NanoBot process(input, mode:) end - def stream(interface) + def streaming(interface) provider = @provider.settings.key?(:stream) ? @provider.settings[:stream] : true interface = interface.key?(:stream) ? interface[:stream] : true @@ -77,7 +78,7 @@ module NanoBot def process(input, mode:) interface = Logic::Helpers::Hash.fetch(@cartridge, [:interfaces, mode.to_sym]) || {} - streaming = stream(interface) + streaming = streaming(interface) input[:interface] = interface @@ -105,11 +106,11 @@ module NanoBot end def flush - @output.flush + @stream.flush end def print(content) - @output.write(content) + @stream.write(content) end end end |