rubycode 0.1.2 → 0.1.3

data/lib/rubycode/tools/write.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "fileutils"
+
+module RubyCode
+  module Tools
+    # Tool for creating new files
+    class Write < Base
+      private
+
+      def perform(params)
+        file_path = params["file_path"]
+        content = params["content"]
+        full_path = resolve_path(file_path)
+
+        validate_file_does_not_exist(file_path, full_path)
+        request_approval(file_path, content)
+        create_file(full_path, content)
+        build_result(file_path, content)
+      end
+
+      def validate_file_does_not_exist(file_path, full_path)
+        return unless File.exist?(full_path)
+
+        raise ToolError, I18n.t("rubycode.errors.file_exists", path: file_path)
+      end
+
+      def request_approval(file_path, content)
+        approval_handler = context[:approval_handler]
+        return if approval_handler.request_write_approval(file_path, content)
+
+        raise ToolError, I18n.t("rubycode.errors.user_cancelled_write")
+      end
+
+      def create_file(full_path, content)
+        dir = File.dirname(full_path)
+        FileUtils.mkdir_p(dir)
+        File.write(full_path, content)
+      end
+
+      def build_result(file_path, content)
+        ToolResult.new(
+          content: "Created '#{file_path}' (#{content.lines.count} lines)",
+          metadata: {
+            file_path: file_path,
+            lines: content.lines.count,
+            bytes: content.bytesize
+          }
+        )
+      end
+
+      def resolve_path(file_path)
+        File.absolute_path?(file_path) ? file_path : File.join(root_path, file_path)
+      end
+    end
+  end
+end

data/lib/rubycode/tools.rb

@@ -4,6 +4,8 @@ require_relative "tools/base"
 require_relative "tools/bash"
 require_relative "tools/read"
 require_relative "tools/search"
+require_relative "tools/write"
+require_relative "tools/update"
 require_relative "tools/done"
 
 module RubyCode
@@ -14,6 +16,8 @@ module RubyCode
       Bash,
       Read,
       Search,
+      Write,
+      Update,
       Done
     ].freeze
 

data/lib/rubycode/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyCode
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/lib/rubycode/views/agent_loop/adapter_error.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module RubyCode
+  module Views
+    module AgentLoop
+      # Builds adapter error message
+      class AdapterError
+        def self.build(message:)
+          "   ✗ Adapter Error: #{message}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/agent_loop/iteration_footer.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module AgentLoop
+      # Builds iteration footer
+      class IterationFooter
+        def self.build
+          pastel = Pastel.new
+          pastel.cyan("└────────────────────────────────────────────────")
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/agent_loop/iteration_header.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module AgentLoop
+      # Builds iteration header with tool list
+      class IterationHeader
+        def self.build(iteration:, tool_calls:)
+          pastel = Pastel.new
+
+          header = pastel.cyan("┌─ Iteration #{iteration} ─────────────────────────")
+          tool_list = tool_calls.each_with_index.map do |tool_call, idx|
+            tool_name = tool_call.dig("function", "name")
+            "  #{pastel.dim("#{idx + 1}.")} #{tool_name}"
+          end
+
+          [header, *tool_list].join("\n")
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/agent_loop/response_received.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module AgentLoop
+      # Builds response received status
+      class ResponseReceived
+        def self.build
+          pastel = Pastel.new
+          "#{pastel.green("✓")} Response received"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/agent_loop/retry_status.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module RubyCode
+  module Views
+    module AgentLoop
+      # Builds retry status message
+      class RetryStatus
+        def self.build(attempt:, max_retries:, delay:, error:)
+          "⚠  Request failed: #{error}. Retrying in #{delay.round(1)}s... (attempt #{attempt}/#{max_retries})"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/agent_loop/thinking_status.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module AgentLoop
+      # Builds thinking status indicator
+      class ThinkingStatus
+        def self.build
+          pastel = Pastel.new
+          "#{pastel.dim("→")} Thinking..."
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/agent_loop/tool_error.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module RubyCode
+  module Views
+    module AgentLoop
+      # Builds tool error message
+      class ToolError
+        def self.build(message:)
+          "   ✗ #{message}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/agent_loop.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+# Agent loop view components
+require_relative "agent_loop/thinking_status"
+require_relative "agent_loop/response_received"
+require_relative "agent_loop/iteration_header"
+require_relative "agent_loop/iteration_footer"
+require_relative "agent_loop/tool_error"

data/lib/rubycode/views/bash_approval.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    # Builds bash command approval prompt display
+    class BashApproval
+      def self.build(command:, base_command:, safe_commands:)
+        pastel = Pastel.new
+
+        [
+          "",
+          pastel.red("━" * 80),
+          pastel.bold.red("⚠ WARNING: Non-Whitelisted Command"),
+          "#{pastel.cyan("Command:")} #{command}",
+          "#{pastel.cyan("Base command:")} #{base_command}",
+          pastel.red("─" * 80),
+          pastel.yellow("This command is not in the safe whitelist:"),
+          pastel.dim("Safe commands: #{safe_commands.join(", ")}"),
+          "",
+          pastel.yellow("⚠ Only approve if you trust this command will not cause harm"),
+          pastel.red("━" * 80)
+        ].join("\n")
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/configuration_table.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require "pastel"
+require "tty-table"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds configuration table display
+      class ConfigurationTable
+        def self.build(directory:, model:, debug_mode:)
+          pastel = Pastel.new
+
+          table = TTY::Table.new(
+            header: [pastel.bold("Setting"), pastel.bold("Value")],
+            rows: [
+              ["Directory", directory],
+              ["Model", model],
+              ["Debug Mode", debug_mode ? pastel.green("ON") : pastel.dim("OFF")]
+            ]
+          )
+
+          "\n#{table.render(:unicode, padding: [0, 1])}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/error_display.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds error display with backtrace
+      class ErrorDisplay
+        def self.build(error:)
+          pastel = Pastel.new
+          backtrace_preview = error.backtrace.first(3).join("\n")
+
+          "\n#{pastel.red("[ERROR]")} #{error.message}\n#{pastel.dim(backtrace_preview)}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/error_message.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds error message display
+      class ErrorMessage
+        def self.build(message:)
+          pastel = Pastel.new
+          "\n#{pastel.red("[ERROR]")} #{message}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/exit_message.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds goodbye/exit message
+      class ExitMessage
+        def self.build
+          pastel = Pastel.new
+          "\n#{pastel.green(I18n.t("rubycode.cli.exit"))}\n"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/interrupt_message.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds interrupt notification message
+      class InterruptMessage
+        def self.build
+          pastel = Pastel.new
+          "\n#{pastel.yellow("[INTERRUPTED]")} Type 'exit' to quit or continue chatting."
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/memory_cleared_message.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds memory cleared confirmation message
+      class MemoryClearedMessage
+        def self.build
+          pastel = Pastel.new
+          "#{pastel.yellow("✓")} #{I18n.t("rubycode.cli.memory_cleared")}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/ready_message.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds ready status message
+      class ReadyMessage
+        def self.build
+          pastel = Pastel.new
+          "\n#{pastel.green("✓")} #{pastel.bold("Ready!")} #{I18n.t("rubycode.cli.ready")}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli/response_box.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require "pastel"
+require "tty-markdown"
+
+module RubyCode
+  module Views
+    module Cli
+      # Builds response box with markdown rendering
+      class ResponseBox
+        def self.build(response:)
+          pastel = Pastel.new
+
+          header = "\n#{pastel.magenta("╔═══ Agent Response ═══")}"
+          footer = pastel.magenta("╚══════════════════════")
+
+          # Try to parse as markdown, fallback to plain
+          content = begin
+            TTY::Markdown.parse(response, width: 80)
+          rescue StandardError
+            response
+          end
+
+          "#{header}\n#{content}\n#{footer}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/cli.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+# CLI view components
+require_relative "cli/error_message"
+require_relative "cli/configuration_table"
+require_relative "cli/ready_message"
+require_relative "cli/exit_message"
+require_relative "cli/memory_cleared_message"
+require_relative "cli/response_box"
+require_relative "cli/interrupt_message"
+require_relative "cli/error_display"

data/lib/rubycode/views/formatter/debug_tool_info.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Formatter
+      # Builds debug mode tool information display
+      class DebugToolInfo
+        def self.build(tool_name:, arguments:)
+          pastel = Pastel.new
+          "\n#{pastel.yellow("[TOOL]")} #{tool_name}\n   #{pastel.dim("Args:")} #{arguments.inspect}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/formatter/info_message.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Formatter
+      # Builds info message display
+      class InfoMessage
+        def self.build(message:)
+          pastel = Pastel.new
+          "   #{pastel.dim("ℹ #{message}")}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/formatter/minimal_tool_info.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require "pastel"
+require "tty-table"
+
+module RubyCode
+  module Views
+    module Formatter
+      # Builds minimal tool info table display
+      class MinimalToolInfo
+        def self.build(label:, value:)
+          pastel = Pastel.new
+
+          table = TTY::Table.new(rows: [
+                                   [
+                                     pastel.cyan(label),
+                                     value
+                                   ]
+                                 ])
+
+          "  #{table.render(:basic, padding: [0, 1])}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/formatter/tool_result.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module Formatter
+      # Builds tool result summary display
+      class ToolResult
+        def self.build(result:)
+          pastel = Pastel.new
+          first_line = result.lines.first&.strip || "(empty)"
+          suffix = result.lines.count > 1 ? "... (#{result.lines.count} lines)" : ""
+
+          "   #{pastel.green("✓")} #{pastel.dim("Result:")} #{first_line}#{suffix}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/formatter.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+# Display formatter view components
+require_relative "formatter/debug_tool_info"
+require_relative "formatter/tool_result"
+require_relative "formatter/info_message"
+require_relative "formatter/minimal_tool_info"

data/lib/rubycode/views/response_handler/agent_finished.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "tty-box"
+
+module RubyCode
+  module Views
+    module ResponseHandler
+      # Builds agent finished success box
+      class AgentFinished
+        def self.build(iteration:, total_tool_calls:)
+          success_box = TTY::Box.frame(
+            title: { top_left: " #{I18n.t("rubycode.response_handler.agent_finished.title")} " },
+            border: :light,
+            padding: 1,
+            style: {
+              fg: :green,
+              border: {
+                fg: :green
+              }
+            }
+          ) do
+            I18n.t("rubycode.response_handler.agent_finished.message",
+                   iterations: iteration,
+                   tool_calls: total_tool_calls)
+          end
+          "\n#{success_box}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/response_handler/complete_message.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "tty-box"
+
+module RubyCode
+  module Views
+    module ResponseHandler
+      # Builds task completion success box
+      class CompleteMessage
+        def self.build(iteration:, total_tool_calls:)
+          success_box = TTY::Box.frame(
+            title: { top_left: " #{I18n.t("rubycode.response_handler.complete.title")} " },
+            border: :light,
+            padding: 1,
+            style: {
+              fg: :green,
+              border: {
+                fg: :green
+              }
+            }
+          ) do
+            I18n.t("rubycode.response_handler.complete.message",
+                   iterations: iteration,
+                   tool_calls: total_tool_calls)
+          end
+          "\n#{success_box}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/response_handler/max_iterations.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require "tty-box"
+
+module RubyCode
+  module Views
+    module ResponseHandler
+      # Builds max iterations warning box
+      class MaxIterations
+        def self.build(max_iterations:)
+          error_box = TTY::Box.frame(
+            title: { top_left: " #{I18n.t("rubycode.response_handler.max_iterations.title")} " },
+            border: :thick,
+            padding: 1,
+            style: {
+              fg: :yellow,
+              border: {
+                fg: :yellow
+              }
+            }
+          ) do
+            I18n.t("rubycode.response_handler.max_iterations.message", max: max_iterations)
+          end
+          "\n#{error_box}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/response_handler/max_tool_calls.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require "tty-box"
+
+module RubyCode
+  module Views
+    module ResponseHandler
+      # Builds max tool calls warning box
+      class MaxToolCalls
+        def self.build(max_tool_calls:)
+          error_box = TTY::Box.frame(
+            title: { top_left: " ⚠ WARNING " },
+            border: :thick,
+            padding: 1,
+            style: {
+              fg: :yellow,
+              border: {
+                fg: :yellow
+              }
+            }
+          ) do
+            "Reached maximum tool calls (#{max_tool_calls})\nStopping to prevent excessive operations."
+          end
+          "\n#{error_box}"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/response_handler/tool_injection_warning.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "pastel"
+
+module RubyCode
+  module Views
+    module ResponseHandler
+      # Builds tool injection warning message
+      class ToolInjectionWarning
+        def self.build(iteration:)
+          pastel = Pastel.new
+          "   #{pastel.yellow("[WARNING]")} No tool calls - injecting reminder (iteration #{iteration})"
+        end
+      end
+    end
+  end
+end

data/lib/rubycode/views/response_handler.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+# Response handler view components
+require_relative "response_handler/max_iterations"
+require_relative "response_handler/max_tool_calls"
+require_relative "response_handler/complete_message"
+require_relative "response_handler/agent_finished"
+require_relative "response_handler/tool_injection_warning"