rubyn-code 0.2.2 → 0.3.0

data/lib/rubyn_code/cli/commands/new_session.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module RubynCode
+  module CLI
+    module Commands
+      # Saves the current session and starts a fresh conversation.
+      # Like pressing Escape in Claude Code — clears context without quitting.
+      class NewSession < Base
+        def self.command_name = '/new'
+        def self.description = 'Save current session and start a fresh conversation'
+        def self.aliases = ['/reset'].freeze
+
+        def execute(_args, ctx)
+          save_current_session(ctx)
+          clear_conversation(ctx)
+          new_session_id = generate_session_id
+
+          ctx.renderer.info('Session saved. Starting fresh.')
+          ctx.renderer.info("New session: #{new_session_id[0..7]}")
+
+          { action: :new_session, session_id: new_session_id }
+        end
+
+        private
+
+        def save_current_session(ctx)
+          ctx.session_persistence.save_session(
+            session_id: ctx.session_id,
+            project_path: ctx.project_root,
+            messages: ctx.conversation.messages,
+            model: Config::Defaults::DEFAULT_MODEL
+          )
+        end
+
+        def clear_conversation(ctx)
+          ctx.conversation.clear!
+        end
+
+        def generate_session_id
+          SecureRandom.hex(16)
+        end
+      end
+    end
+  end
+end

data/lib/rubyn_code/cli/daemon_runner.rb

@@ -18,14 +18,9 @@ module RubynCode
       end
 
       def run
-        ensure_home_dir!
-        ensure_auth!
-        setup_database!
-        display_banner!
-
+        bootstrap!
         daemon = build_daemon
         daemon.start!
-
         display_shutdown_summary(daemon)
       rescue Interrupt
         @renderer.info("\nShutting down daemon...")
@@ -38,6 +33,13 @@ module RubynCode
 
       private
 
+      def bootstrap!
+        ensure_home_dir!
+        ensure_auth!
+        setup_database!
+        display_banner!
+      end
+
       def build_daemon
         Autonomous::Daemon.new(
           agent_name: @daemon_opts[:agent_name],
@@ -79,13 +81,16 @@ module RubynCode
       end
 
       def ensure_auth!
-        unless Auth::TokenStore.valid?
-          @renderer.error('No valid authentication found.')
-          @renderer.info('Run `rubyn-code --auth` or set ANTHROPIC_API_KEY first.')
+        provider = @daemon_opts.fetch(:provider, Config::Defaults::DEFAULT_PROVIDER)
+
+        unless Auth::TokenStore.load_for_provider(provider)
+          @renderer.error("No valid authentication found for provider '#{provider}'.")
+          env_key = Config::Defaults::PROVIDER_ENV_KEYS.fetch(provider, "#{provider.upcase}_API_KEY")
+          @renderer.info("Set #{env_key} or run `rubyn-code --auth`.")
           exit(1)
         end
 
-        @llm_client = LLM::Client.new
+        @llm_client = LLM::Client.new(provider: provider)
       end
 
       def setup_database!
@@ -96,9 +101,18 @@ module RubynCode
       end
 
       def display_banner!
+        display_banner_header!
+        display_banner_details!
+        display_banner_footer!
+      end
+
+      def display_banner_header!
         @renderer.info('╔══════════════════════════════════════╗')
         @renderer.info('║        GOLEM Daemon Starting         ║')
         @renderer.info('╚══════════════════════════════════════╝')
+      end
+
+      def display_banner_details!
         @renderer.info("  Agent:        #{@daemon_opts[:agent_name]}")
         @renderer.info("  Role:         #{@daemon_opts[:role]}")
         @renderer.info("  Project:      #{@project_root}")
@@ -106,6 +120,9 @@ module RubynCode
         @renderer.info("  Max cost:     $#{@daemon_opts[:max_cost]}")
         @renderer.info("  Idle timeout: #{@daemon_opts[:idle_timeout]}s")
         @renderer.info("  Poll interval: #{@daemon_opts[:poll_interval]}s")
+      end
+
+      def display_banner_footer!
         @renderer.info('')
         @renderer.info('Waiting for tasks... (Ctrl-C to stop)')
         @renderer.info('Seed tasks via the REPL: /tasks or the task tool.')
@@ -122,7 +139,7 @@ module RubynCode
         @renderer.info('╚══════════════════════════════════════╝')
         @renderer.info("  Final state:    #{status[:state]}")
         @renderer.info("  Tasks completed: #{status[:runs_completed]}")
-        @renderer.info("  Total cost:     $#{'%.4f' % status[:total_cost]}")
+        @renderer.info(format('  Total cost:     $%.4f', status[:total_cost]))
       end
     end
   end

data/lib/rubyn_code/cli/renderer.rb

@@ -33,14 +33,23 @@ module RubynCode
         puts @pastel.cyan("  > #{name}: #{format_params(params)}")
       end
 
-      def tool_result(_name, output)
-        truncated = output.to_s[0...300]
-        lines = truncated.lines
-        if lines.length > 6
-          puts @pastel.dim("    #{lines[0..4].map(&:strip).join("\n    ")}")
-          puts @pastel.dim("    ... (#{lines.length - 5} more lines)")
+      DIFF_TOOLS = %w[edit_file write_file].freeze
+      DIFF_RESULT_LIMIT = 2000
+      DEFAULT_RESULT_LIMIT = 500
+
+      def tool_result(name, output)
+        raw = output.to_s
+
+        if DIFF_TOOLS.include?(name.to_s)
+          render_diff_result(raw[0...DIFF_RESULT_LIMIT].lines)
         else
-          puts @pastel.dim("    #{truncated.strip.gsub("\n", "\n    ")}")
+          truncated = raw[0...DEFAULT_RESULT_LIMIT]
+          lines = truncated.lines
+          if lines.length > 6
+            render_truncated_result(lines)
+          else
+            puts @pastel.dim("    #{truncated.strip.gsub("\n", "\n    ")}")
+          end
         end
       end
 
@@ -80,54 +89,93 @@ module RubynCode
 
       def cost_summary(session_cost:, daily_cost:, tokens:)
         puts @pastel.bold('Cost Summary:')
-        puts "  Session: $#{'%.4f' % session_cost}"
-        puts "  Today:   $#{'%.4f' % daily_cost}"
+        puts format('  Session: $%.4f', session_cost)
+        puts format('  Today:   $%.4f', daily_cost)
         puts "  Tokens:  #{tokens[:input]} in / #{tokens[:output]} out"
       end
 
       def prompt
-        if @yolo
-          @pastel.bold.green('rubyn ') + @pastel.bold.red('YOLO') + @pastel.bold.green(' > ')
-        else
-          @pastel.bold.green('rubyn > ')
-        end
+        return @pastel.bold.green('rubyn ') + @pastel.bold.red('YOLO') + @pastel.bold.green(' > ') if @yolo
+
+        @pastel.bold.green('rubyn > ')
       end
 
+      DIFF_COLORS = {
+        /\A  \+ / => :green,
+        /\A  - / => :red,
+        /\A  @@ / => :cyan,
+        /\A(?:Created|Updated|Edited) / => :yellow
+      }.freeze
+
       private
 
+      def render_diff_result(lines)
+        lines.each do |line|
+          stripped = line.rstrip
+          puts "    #{colorize_diff_line(stripped)}"
+        end
+      end
+
+      def colorize_diff_line(line)
+        DIFF_COLORS.each do |pattern, color|
+          return @pastel.decorate(line, color) if line.match?(pattern)
+        end
+        @pastel.dim(line)
+      end
+
+      def render_truncated_result(lines)
+        puts @pastel.dim("    #{lines[0..4].map(&:strip).join("\n    ")}")
+        puts @pastel.dim("    ... (#{lines.length - 5} more lines)")
+      end
+
       def render_markdown(text)
         lines = text.lines
         result = []
-
         in_code_block = false
         code_lang = nil
         code_buffer = []
 
         lines.each do |line|
-          if line.strip.match?(/\A```(\w*)/)
-            if in_code_block
-              result << render_code_block(code_buffer.join, code_lang)
-              code_buffer = []
-              in_code_block = false
-              code_lang = nil
-            else
-              in_code_block = true
-              code_lang = line.strip.match(/\A```(\w*)/)[1]
-              code_lang = 'ruby' if code_lang.empty?
-            end
-          elsif in_code_block
-            code_buffer << line
-          else
-            result << render_line(line)
-          end
+          in_code_block, code_lang, code_buffer = process_markdown_line(
+            line, in_code_block, code_lang, code_buffer, result
+          )
         end
 
         # Flush any unclosed code block
-        result << render_code_block(code_buffer.join, code_lang || 'ruby') unless code_buffer.empty?
+        flush_code_buffer(code_buffer, code_lang, result)
 
         result.join
       end
 
+      def process_markdown_line(line, in_code_block, code_lang, code_buffer, result)
+        if line.strip.match?(/\A```(\w*)/)
+          handle_code_fence(line, in_code_block, code_lang, code_buffer, result)
+        elsif in_code_block
+          code_buffer << line
+          [in_code_block, code_lang, code_buffer]
+        else
+          result << render_line(line)
+          [in_code_block, code_lang, code_buffer]
+        end
+      end
+
+      def handle_code_fence(line, in_code_block, code_lang, code_buffer, result)
+        if in_code_block
+          result << render_code_block(code_buffer.join, code_lang)
+          [false, nil, []]
+        else
+          lang = line.strip.match(/\A```(\w*)/)[1]
+          lang = 'ruby' if lang.empty?
+          [true, lang, []]
+        end
+      end
+
+      def flush_code_buffer(code_buffer, code_lang, result)
+        return if code_buffer.empty?
+
+        result << render_code_block(code_buffer.join, code_lang || 'ruby')
+      end
+
       def render_code_block(code, lang)
         lexer = Rouge::Lexer.find(lang) || Rouge::Lexers::PlainText.new
         highlighted = @rouge_formatter.format(lexer.lex(code))
@@ -139,37 +187,38 @@ module RubynCode
       end
 
       def render_line(line)
-        # Headers
-        if line.match?(/\A\s*\#{1,6}\s/)
-          level = line.match(/\A\s*(\#{1,6})\s/)[1].length
-          text = line.sub(/\A\s*\#{1,6}\s+/, '').rstrip
-          case level
-          when 1 then "#{@pastel.bold.underline(text)}\n"
-          when 2 then "#{@pastel.bold(text)}\n"
-          else "#{@pastel.bold(text)}\n"
-          end
-        # Bullet lists
-        elsif line.match?(/\A\s*[-*]\s/)
-          indent = line.match(/\A(\s*)/)[1]
-          content = line.sub(/\A\s*[-*]\s+/, '')
-          "#{indent}  #{@pastel.cyan('•')} #{render_inline(content)}"
-        # Numbered lists
-        elsif line.match?(/\A\s*\d+\.\s/)
-          indent = line.match(/\A(\s*)/)[1]
-          num = line.match(/(\d+)\./)[1]
-          content = line.sub(/\A\s*\d+\.\s+/, '')
-          "#{indent}  #{@pastel.cyan("#{num}.")} #{render_inline(content)}"
-        # Horizontal rules
-        elsif line.strip.match?(/\A-{3,}\z/)
-          "#{@pastel.dim('─' * [terminal_width - 4, 40].min)}\n"
-        # Table rows
-        elsif line.include?('|')
-          render_table_row(line)
-        else
-          render_inline(line)
+        case line
+        when /\A\s*\#{1,6}\s/   then render_header(line)
+        when /\A\s*[-*]\s/      then render_bullet(line)
+        when /\A\s*\d+\.\s/     then render_numbered_item(line)
+        when ->(l) { l.strip.match?(/\A-{3,}\z/) } then "#{@pastel.dim('─' * [terminal_width - 4, 40].min)}\n"
+        when /\|/ then render_table_row(line)
+        else render_inline(line)
+        end
+      end
+
+      def render_header(line)
+        level = line.match(/\A\s*(\#{1,6})\s/)[1].length
+        text = line.sub(/\A\s*\#{1,6}\s+/, '').rstrip
+        case level
+        when 1 then "#{@pastel.bold.underline(text)}\n"
+        else "#{@pastel.bold(text)}\n"
         end
       end
 
+      def render_bullet(line)
+        indent = line.match(/\A(\s*)/)[1]
+        content = line.sub(/\A\s*[-*]\s+/, '')
+        "#{indent}  #{@pastel.cyan('•')} #{render_inline(content)}"
+      end
+
+      def render_numbered_item(line)
+        indent = line.match(/\A(\s*)/)[1]
+        num = line.match(/(\d+)\./)[1]
+        content = line.sub(/\A\s*\d+\.\s+/, '')
+        "#{indent}  #{@pastel.cyan("#{num}.")} #{render_inline(content)}"
+      end
+
       def render_inline(text)
         text
           .gsub(/\*\*(.+?)\*\*/) { @pastel.bold(Regexp.last_match(1)) }