ruby_rich 0.3.1 → 0.4.1

data/lib/ruby_rich/app_shell.rb

@@ -0,0 +1,374 @@
+# frozen_string_literal: true
+
+module RubyRich
+  class AppShell
+    attr_reader :layout, :transcript, :viewport, :sidebar, :composer, :focus_manager, :theme, :live, :token_usage, :progress_manager
+
+    DEFAULT_COMMANDS = [
+      { label: "/help", value: "/help", description: "Show commands" },
+      { label: "/plan", value: "/plan", description: "Append a plan note" },
+      { label: "/thinking", value: "/thinking", description: "Add a thinking block" },
+      { label: "/tool", value: "/tool", description: "Add a tool call" },
+      { label: "/quit", value: "/quit", description: "Exit demo" }
+    ].freeze
+
+    def initialize(title: "Agent", subtitle: nil, model: "deepseek-v4-pro", theme: Theme.agent_dark, commands: DEFAULT_COMMANDS, on_submit: nil)
+      @title = title
+      @subtitle = subtitle || "DeepSeek-TUI · #{model}"
+      @model = model
+      @theme = theme
+      @on_submit = on_submit
+      @status = "agent · #{model}"
+      @token_usage = nil
+      @progress_text = nil
+
+      @transcript = Transcript.new
+      @progress_manager = ProgressManager.new(on_change: ->(text) { @progress_text = text })
+      @viewport = Viewport.new(@transcript, scrollbar: true, auto_scroll: true)
+      @sidebar = Sidebar.new
+      @composer = Composer.new(
+        placeholder: "编写任务或使用 /。",
+        commands: commands,
+        on_submit: method(:handle_submit),
+        on_select: method(:handle_select)
+      )
+      @focus_manager = FocusManager.new
+      @layout = build_layout
+      attach_components
+    end
+
+    def add_user(text)
+      @transcript.add_user(text)
+      @viewport.scroll_to_bottom
+      self
+    end
+
+    def add_assistant(text)
+      @transcript.add_assistant(text)
+      @viewport.scroll_to_bottom
+      self
+    end
+
+    def add_thinking(text, status: "idle", collapsed: true)
+      @transcript.add_thinking(text, status: status, collapsed: collapsed)
+      @viewport.scroll_to_bottom
+      self
+    end
+
+    def add_tool(name, status: :running, result: nil, collapsed: false)
+      @transcript.add_tool(name, status: status, result: result, collapsed: collapsed)
+      @viewport.scroll_to_bottom
+      self
+    end
+
+    def add_separator(label = nil)
+      @transcript.add_separator(label)
+      @viewport.scroll_to_bottom
+      self
+    end
+
+    def add_markdown(text)
+      @transcript.add_markdown(text)
+      @viewport.scroll_to_bottom
+      self
+    end
+
+    def add_diff(title: nil, content:, language: "diff")
+      label = title ? "#{title}\n#{content}" : content
+      @transcript.add_block(:diff, label, language: language)
+      @viewport.scroll_to_bottom
+      self
+    end
+
+    def update_plan(text)
+      @sidebar.update_plan(text)
+      self
+    end
+
+    def set_tasks(tasks)
+      @sidebar.set_tasks(tasks)
+      self
+    end
+
+    def status=(text)
+      @status = text.to_s
+    end
+
+    def update_status(text)
+      self.status = text
+      self
+    end
+
+    def show_token_usage(input: nil, output: nil, total: nil, **extra)
+      @token_usage = { input: input, output: output, total: total }.merge(extra).compact
+      self
+    end
+
+    def start_progress(message = nil, owner: Thread.current.object_id, style: :primary, quiet_on_fast_finish: false)
+      _ = style
+      _ = quiet_on_fast_finish
+      @progress_manager.start(message, owner: owner)
+    end
+
+    def with_progress(message = nil, style: :primary, quiet_on_fast_finish: false, &block)
+      _ = style
+      _ = quiet_on_fast_finish
+      @progress_manager.with_progress(message, &block)
+    end
+
+    def confirm(title:, message:, choices:, default: nil, &callback)
+      result = default || choices.first&.fetch(:key)
+      callback.call(result) if callback
+      result
+    end
+
+    def form(title:, fields:, &callback)
+      values = {}
+      fields.each do |field|
+        name = field.fetch(:name).to_sym
+        values[name] = field.key?(:default) ? field[:default] : default_field_value(field)
+      end
+      callback.call(values) if callback
+      values
+    end
+
+    def open_pager(text, command: ENV.fetch("PAGER", "less -R"))
+      Terminal.with_cooked(mouse: true) do
+        IO.popen(command, "w") { |io| io.write(text.to_s) }
+      end
+      true
+    rescue
+      false
+    end
+
+    def start(refresh_rate: 24, mouse: true, alt_screen: false)
+      Live.start(@layout, refresh_rate: refresh_rate, mouse: mouse, alt_screen: alt_screen, autowrap: false) do |live|
+        @live = live
+        live.listening = true
+      end
+    ensure
+      @live = nil
+    end
+
+    def stop
+      return false unless @live
+
+      @live.post { |live| live.stop } || false
+    end
+
+    private
+
+    def build_layout
+      root = Layout.new(name: :root)
+      root.split_column(
+        Layout.new(name: :header, size: 1),
+        Layout.new(name: :body, ratio: 1),
+        Layout.new(name: :composer, size: 6),
+        Layout.new(name: :status, size: 1)
+      )
+
+      root[:body].split_row(
+        Layout.new(name: :transcript, ratio: 1),
+        Layout.new(name: :sidebar, size: 36)
+      )
+
+      root[:header].content = HeaderView.new(self)
+      root[:transcript].content = @viewport
+      root[:sidebar].content = @sidebar
+      root[:composer].content = FramedView.new(@composer, title: "Composer", theme: @theme) { @composer.focused? }
+      root[:status].content = StatusView.new(self)
+      root
+    end
+
+    def attach_components
+      @viewport.attach(@layout[:transcript])
+      @transcript.attach(@layout[:transcript])
+      @composer.focus.attach(@layout[:composer])
+
+      @focus_manager
+        .register(:transcript, @layout[:transcript], FocusTarget.new(@transcript, @viewport))
+        .register(:composer, @layout[:composer], @composer)
+        .attach(@layout)
+      @focus_manager.focus(:composer)
+
+      @layout.key(:ctrl_c, 1_000) do |_event, live|
+        live.stop if @stop_on_ctrl_c != false
+        false
+      end
+    end
+
+    def handle_select(command, _live)
+      case command[:value]
+      when "/plan"
+        @status = "plan command selected"
+      when "/thinking"
+        @status = "thinking command selected"
+      when "/tool"
+        @status = "tool command selected"
+      end
+    end
+
+    def handle_submit(value, live, attachments = [])
+      case value.strip
+      when "/quit"
+        live&.stop
+      when "/help"
+        add_assistant("Commands: /help, /plan, /thinking, /tool, /quit")
+      when "/plan"
+        @sidebar.add_task("Plan updated #{Time.now.strftime('%H:%M:%S')}", status: :in_progress)
+      when "/thinking"
+        add_thinking("Let me inspect the current state and keep the details collapsible.", status: "idle", collapsed: false)
+      when "/tool"
+        add_tool("read_file", status: :done, result: "name: read_file\nresult: <demo output>", collapsed: false)
+      else
+        add_user(value)
+      end
+
+      @on_submit&.call(value, live, self, attachments)
+    end
+
+    class FocusTarget
+      def initialize(*targets)
+        @targets = targets
+      end
+
+      def focus
+        @targets.each { |target| target.focus if target.respond_to?(:focus) }
+      end
+
+      def blur
+        @targets.each { |target| target.blur if target.respond_to?(:blur) }
+      end
+    end
+
+    class HeaderView
+      attr_accessor :width, :height
+
+      def initialize(shell)
+        @shell = shell
+        @width = 0
+        @height = 1
+      end
+
+      def render
+        theme = @shell.theme
+        left = "#{theme.style(@shell.instance_variable_get(:@title), :accent)}  #{theme.style(@shell.instance_variable_get(:@subtitle), :muted)}"
+        usage = @shell.token_usage
+        usage_text = if usage && !usage.empty?
+                       total = usage[:total] || usage[:tokens]
+                       total ? "#{total} tok" : usage.map { |key, value| "#{key}=#{value}" }.join(" ")
+                     else
+                       "tokens --"
+                     end
+        right = "#{theme.style(@shell.instance_variable_get(:@model), :status)}  #{theme.style('● Live', :body)}  #{theme.style(usage_text, :status)}"
+        [join_edges(left, right, @width)]
+      end
+
+      private
+
+      def join_edges(left, right, width)
+        space = [width - visible_width(left) - visible_width(right), 1].max
+        truncate_display(left + (" " * space) + right, width)
+      end
+
+      def visible_width(text)
+        text.gsub(/\e\[[0-9;:]*m/, "").display_width
+      end
+
+      def truncate_display(text, width)
+        return text if visible_width(text) <= width
+
+        result = +""
+        used = 0
+        in_escape = false
+        text.each_char do |char|
+          if in_escape
+            result << char
+            in_escape = false if char == "m"
+            next
+          elsif char.ord == 27
+            result << char
+            in_escape = true
+            next
+          end
+
+          char_width = Unicode::DisplayWidth.of(char)
+          break if used + char_width > width
+
+          result << char
+          used += char_width
+        end
+        result
+      end
+    end
+
+    class StatusView
+      attr_accessor :width, :height
+
+      def initialize(shell)
+        @shell = shell
+        @width = 0
+        @height = 1
+      end
+
+      def render
+        theme = @shell.theme
+        focus = @shell.focus_manager.focused_name || :none
+        progress = @shell.instance_variable_get(:@progress_text)
+        status = progress || @shell.instance_variable_get(:@status)
+        line = "#{theme.style(status, :accent)}  #{theme.style('focus: ' + focus.to_s, :muted)}  #{theme.style('Tab focus · Ctrl+C quit · /quit', :dim)}"
+        [line]
+      end
+    end
+
+    class FramedView
+      attr_accessor :width, :height
+
+      def initialize(component, title:, theme:, &focused)
+        @component = component
+        @title = title
+        @theme = theme
+        @focused = focused
+        @width = 0
+        @height = 0
+      end
+
+      def render
+        sync_component_dimensions
+        panel = Panel.new(rendered_content, title: @title, border_style: @theme.panel_border(focused: @focused.call), title_align: :left)
+        panel.width = @width
+        panel.height = @height
+        panel.render
+      end
+
+      def desired_height
+        return @height unless @component.respond_to?(:desired_height)
+
+        @component.desired_height + 2
+      end
+
+      private
+
+      def sync_component_dimensions
+        inner_width = [@width - 2, 1].max
+        inner_height = [@height - 2, 1].max
+        @component.width = inner_width if @component.respond_to?(:width=)
+        @component.height = inner_height if @component.respond_to?(:height=)
+      end
+
+      def rendered_content
+        rendered = @component.render
+        rendered.is_a?(Array) ? rendered.join("\n") : rendered.to_s
+      end
+    end
+
+    def default_field_value(field)
+      case field[:type]
+      when :boolean then false
+      when :multi_select then []
+      when :number then nil
+      else ""
+      end
+    end
+  end
+end

data/lib/ruby_rich/attachment.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module RubyRich
+  class Attachment
+    attr_accessor :type, :path, :mime_type, :display_name, :metadata
+
+    def initialize(type:, path:, mime_type: nil, display_name: nil, metadata: {})
+      @type = type.to_sym
+      @path = path.to_s
+      @mime_type = mime_type
+      @display_name = display_name || File.basename(@path)
+      @metadata = metadata.dup
+    end
+
+    def to_h
+      {
+        type: @type,
+        path: @path,
+        mime_type: @mime_type,
+        display_name: @display_name,
+        metadata: @metadata
+      }.compact
+    end
+  end
+end

data/lib/ruby_rich/columns.rb

@@ -0,0 +1,244 @@
+module RubyRich
+  class Columns
+    class Column
+      attr_accessor :content, :width, :align, :padding, :title
+
+      def initialize(width: nil, align: :left, padding: 1, title: nil)
+        @content = []
+        @width = width
+        @align = align
+        @padding = padding
+        @title = title
+      end
+
+      def add(text)
+        @content << text.to_s
+        self
+      end
+
+      def <<(text)
+        add(text)
+      end
+
+      def clear
+        @content.clear
+        self
+      end
+
+      def lines
+        @content
+      end
+
+      def height
+        @content.length
+      end
+    end
+
+    attr_reader :columns, :total_width, :gutter_width
+
+    def initialize(total_width: 80, gutter_width: 2)
+      @columns = []
+      @total_width = total_width
+      @gutter_width = gutter_width
+    end
+
+    # 添加列
+    def add_column(width: nil, align: :left, padding: 1, title: nil)
+      column = Column.new(width: width, align: align, padding: padding, title: title)
+      @columns << column
+      
+      # 如果没有指定宽度，自动计算平均宽度
+      calculate_column_widths if width.nil?
+      
+      column
+    end
+
+    # 删除列
+    def remove_column(index)
+      @columns.delete_at(index) if index >= 0 && index < @columns.length
+      calculate_column_widths
+    end
+
+    # 清空所有列的内容
+    def clear_all
+      @columns.each(&:clear)
+      self
+    end
+
+    # 渲染列布局
+    def render(show_headers: true, show_borders: false, equal_height: true)
+      return "" if @columns.empty?
+
+      calculate_column_widths
+      lines = []
+
+      # 渲染标题行
+      if show_headers && @columns.any? { |col| col.title }
+        header_line = render_header_line(show_borders)
+        lines << header_line unless header_line.empty?
+        
+        if show_borders
+          separator_line = render_separator_line
+          lines << separator_line
+        end
+      end
+
+      # 准备内容行
+      max_height = equal_height ? @columns.map(&:height).max : 0
+      
+      # 填充较短的列以达到相同高度
+      if equal_height && max_height > 0
+        @columns.each do |column|
+          while column.height < max_height
+            column.add("")
+          end
+        end
+      end
+
+      # 渲染内容行
+      content_height = @columns.map(&:height).max || 0
+      content_height.times do |row_index|
+        line = render_content_line(row_index, show_borders)
+        lines << line
+      end
+
+      # 渲染底部边框
+      if show_borders
+        bottom_line = render_separator_line
+        lines << bottom_line
+      end
+
+      lines.join("\n")
+    end
+
+    # 按比例设置列宽
+    def set_ratios(*ratios)
+      return if ratios.length != @columns.length
+
+      total_ratio = ratios.sum.to_f
+      available_width = @total_width - (@gutter_width * (@columns.length - 1))
+      
+      @columns.each_with_index do |column, index|
+        column.width = (available_width * ratios[index] / total_ratio).to_i
+      end
+      
+      self
+    end
+
+    # 设置等宽列
+    def equal_widths
+      calculate_column_widths
+      self
+    end
+
+    private
+
+    def calculate_column_widths
+      return if @columns.empty?
+
+      # 计算可用宽度（总宽度减去间隔）
+      available_width = @total_width - (@gutter_width * (@columns.length - 1))
+      
+      # 为每列分配相等的宽度
+      base_width = available_width / @columns.length
+      remainder = available_width % @columns.length
+      
+      @columns.each_with_index do |column, index|
+        column.width = base_width + (index < remainder ? 1 : 0)
+      end
+    end
+
+    def render_header_line(show_borders)
+      line_parts = []
+      
+      @columns.each_with_index do |column, index|
+        header_text = column.title || ""
+        
+        # 根据列的对齐方式格式化标题
+        formatted_header = format_text(header_text, column.width, column.align)
+        
+        if show_borders
+          formatted_header = "│ #{formatted_header} │"
+        end
+        
+        line_parts << formatted_header
+        
+        # 添加间隔（除了最后一列）
+        if index < @columns.length - 1 && !show_borders
+          line_parts << " " * @gutter_width
+        end
+      end
+      
+      line_parts.join("")
+    end
+
+    def render_content_line(row_index, show_borders)
+      line_parts = []
+      
+      @columns.each_with_index do |column, index|
+        content_text = column.lines[row_index] || ""
+        
+        # 根据列的对齐方式格式化内容
+        formatted_content = format_text(content_text, column.width, column.align)
+        
+        if show_borders
+          formatted_content = "│ #{formatted_content} │"
+        end
+        
+        line_parts << formatted_content
+        
+        # 添加间隔（除了最后一列）
+        if index < @columns.length - 1 && !show_borders
+          line_parts << " " * @gutter_width
+        end
+      end
+      
+      line_parts.join("")
+    end
+
+    def render_separator_line
+      line_parts = []
+      
+      @columns.each_with_index do |column, index|
+        separator = "─" * (column.width + 2)  # +2 for padding
+        line_parts << "├#{separator}┤"
+        
+        if index < @columns.length - 1
+          line_parts << "┬"
+        end
+      end
+      
+      line_parts.join("")
+    end
+
+    def format_text(text, width, align)
+      # 移除 ANSI 转义序列计算实际显示宽度
+      display_text = text.gsub(/\e\[[0-9;]*m/, '')
+      
+      if display_text.length > width
+        # 截断过长的文本
+        truncated = display_text[0, width - 3] + "..."
+        # 保持原有的 ANSI 样式
+        if text.include?("\e[")
+          style_start = text.match(/\e\[[0-9;]*m/)&.to_s || ""
+          truncated = style_start + truncated + "\e[0m"
+        end
+        truncated
+      else
+        # 根据对齐方式填充空格
+        padding_needed = width - display_text.length
+        
+        case align
+        when :center
+          left_padding = padding_needed / 2
+          right_padding = padding_needed - left_padding
+          " " * left_padding + text + " " * right_padding
+        when :right
+          " " * padding_needed + text
+        else # :left
+          text + " " * padding_needed
+        end
+      end
+    end
+  end
+end