ruby_rich 0.3.1 → 0.4.1

data/lib/ruby_rich/theme.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+
+module RubyRich
+  class Theme
+    DEFAULT_ROLES = {
+      accent: { color: :blue, bright: true, bold: true },
+      body: { color: :white, bright: true },
+      muted: { color: :black, bright: true },
+      dim: { color: :black, bright: true },
+      thinking: { color: :white, bright: true, italic: true },
+      success: { color: :green, bright: true },
+      warning: { color: :yellow, bright: true },
+      error: { color: :red, bright: true },
+      status: { color: :cyan, bright: true }
+    }.freeze
+
+    attr_reader :roles, :border, :focused_border
+
+    def self.auto
+      ENV["COLORFGBG"].to_s.split(";").last.to_i >= 8 ? light : agent_dark
+    rescue
+      agent_dark
+    end
+
+    def self.agent_dark
+      new(
+        border: :blue,
+        focused_border: :cyan,
+        roles: {
+          accent: { color: :blue, bright: true, bold: true },
+          body: { color: :white, bright: true },
+          muted: { color: :black, bright: true },
+          dim: { color: :black, bright: true },
+          thinking: { color: :white, bright: true, italic: true },
+          success: { color: :green, bright: true },
+          warning: { color: :yellow, bright: true },
+          error: { color: :red, bright: true },
+          status: { color: :cyan, bright: true }
+        }
+      )
+    end
+
+    def self.light
+      new(
+        border: :blue,
+        focused_border: :cyan,
+        roles: {
+          accent: { color: :blue, bright: false, bold: true },
+          body: { color: :black, bright: false },
+          muted: { color: :black, bright: true },
+          dim: { color: :black, bright: true },
+          success: { color: :green, bright: false },
+          warning: { color: :yellow, bright: false },
+          error: { color: :red, bright: false },
+          status: { color: :cyan, bright: false }
+        }
+      )
+    end
+
+    def self.no_color
+      new(roles: DEFAULT_ROLES.transform_values { { color: :white } }, border: :white, focused_border: :white)
+    end
+
+    def initialize(roles: {}, border: :blue, focused_border: :cyan)
+      @roles = DEFAULT_ROLES.merge(roles)
+      @border = border
+      @focused_border = focused_border
+    end
+
+    def style(text, role = :body)
+      options = @roles.fetch(role, @roles[:body])
+      "#{style_code(options)}#{text}#{AnsiCode.reset}"
+    end
+
+    def color(name, bright: false)
+      AnsiCode.color(name, bright)
+    end
+
+    def panel_border(focused: false)
+      focused ? @focused_border : @border
+    end
+
+    private
+
+    def style_code(options)
+      code = AnsiCode.font(
+        options.fetch(:color, :white),
+        font_bright: options.fetch(:bright, false),
+        bold: options[:bold],
+        italic: options[:italic]
+      )
+      code += AnsiCode.faint if options[:faint]
+      code
+    end
+  end
+end

data/lib/ruby_rich/tool_block.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+module RubyRich
+  class ToolBlock
+    STATES = [:pending, :running, :done, :error, :cancelled, :denied].freeze
+
+    def initialize(entry, width: 80)
+      @entry = entry
+      @width = [width, 20].max
+    end
+
+    def render
+      name = @entry.name || @entry.metadata[:name] || "tool"
+      status = normalize_status(@entry.status || :pending)
+      header = "#{AnsiCode.color(color(status), true)}• #{status_marker(status)} #{name} #{status}#{AnsiCode.reset}"
+      return [header, "  #{summary(@entry.content)}", "  details collapsed; press Ctrl+O for full output"] if @entry.collapsed
+
+      [header] + details(@entry.content)
+    end
+
+    private
+
+    def normalize_status(status)
+      status = status.to_sym
+      return :error if status == :failed || status == :issue
+      return status if STATES.include?(status)
+
+      :pending
+    end
+
+    def color(status)
+      case status
+      when :done then :green
+      when :error, :denied then :red
+      when :cancelled then :yellow
+      else :blue
+      end
+    end
+
+    def status_marker(status)
+      case status
+      when :done then "✓"
+      when :error then "!"
+      when :cancelled then "-"
+      when :denied then "x"
+      else "▸"
+      end
+    end
+
+    def summary(content)
+      plain = content.to_s.gsub(/\e\[[0-9;:]*m/, "").split("\n").first.to_s
+      plain.empty? ? "<no output>" : plain[0, [@width - 4, 20].max]
+    end
+
+    def details(content)
+      text = content.to_s
+      return ["  <no output>"] if text.empty?
+
+      text.split("\n").flat_map { |line| wrap(line).map { |part| "  #{part}" } }
+    end
+
+    def wrap(line)
+      max_width = [@width - 2, 20].max
+      result = []
+      current = +""
+      width = 0
+      in_escape = false
+      line.each_char do |char|
+        if in_escape
+          current << char
+          in_escape = false if char == "m"
+          next
+        elsif char.ord == 27
+          current << char
+          in_escape = true
+          next
+        end
+
+        char_width = Unicode::DisplayWidth.of(char)
+        if width + char_width > max_width
+          result << current
+          current = +""
+          width = 0
+        end
+        current << char
+        width += char_width
+      end
+      result << current unless current.empty?
+      result
+    end
+  end
+end