brute_cli 0.1.2 → 0.1.3

data/lib/brute_cli/stream_formatter.rb

@@ -0,0 +1,114 @@
+# frozen_string_literal: true
+
+require "tty-markdown"
+
+module BruteCLI
+  # Streaming markdown renderer powered by TTY::Markdown.
+  #
+  # Tokens arrive one-at-a-time from the LLM via SSE. The StreamFormatter
+  # accumulates them into a buffer, printing raw characters for
+  # immediate feedback. On each newline, the entire buffer is
+  # re-rendered through TTY::Markdown.parse and the previous output
+  # is overwritten via ANSI cursor save/restore.
+  #
+  # Uses DEC save/restore cursor (\e7 / \e8) — the terminal
+  # tracks its own position, so we don't need fragile manual
+  # row/column accounting.
+  #
+  #   fmt = BruteCLI::StreamFormatter.new(width: 80)
+  #   streamer << "# He"          # prints raw "# He"
+  #   streamer << "llo World\n"   # re-renders as styled "Hello World"
+  #   streamer.flush              # finalize any partial line
+  #   streamer.reset              # ready for next turn
+  #
+  class StreamFormatter
+    SAVE_CURSOR    = "\e7"
+    RESTORE_CURSOR = "\e8"
+    CLEAR_TO_END   = "\e[J"
+
+    def initialize(output: nil, width: nil)
+      @output = output
+      @width  = width || TTY::Screen.width
+      reset
+    end
+
+    # Accept a chunk of streamed text. May contain zero, one, or many
+    # newlines — each is handled correctly.
+    def <<(text)
+      text.each_char { |ch| consume(ch) }
+    end
+
+    # Finalize any partial (unterminated) line still in the buffer.
+    # Called when the response ends or a tool call interrupts.
+    #
+    # After flushing, all state is cleared so the next block of
+    # content (after tool frames, etc.) starts fresh.
+    def flush
+      return if @buffer.empty? && @line_buf.empty?
+      @buffer << @line_buf unless @line_buf.empty?
+      restore_and_render
+      out.puts
+      @buffer       = +""
+      @line_buf     = +""
+      @origin_saved = false
+    end
+
+    # Reset all state for the next agent turn.
+    def reset
+      @buffer       = +""
+      @line_buf     = +""
+      @origin_saved = false
+    end
+
+    private
+
+    # Process a single character.
+    def consume(ch)
+      save_origin unless @origin_saved
+      @line_buf << ch
+      if ch == "\n"
+        finish_line
+      else
+        out.print ch
+      end
+    end
+
+    # Save the cursor position at the start of this output block.
+    # Called once before the first character is printed, then not
+    # again until after a flush or reset.
+    def save_origin
+      out.print SAVE_CURSOR
+      @origin_saved = true
+    end
+
+    # A complete line arrived — append to the buffer and re-render.
+    def finish_line
+      @buffer << @line_buf
+      @line_buf = +""
+      restore_and_render
+    end
+
+    # Restore cursor to the saved origin, clear everything after it,
+    # and print the fully rendered markdown.
+    def restore_and_render
+      rendered = render_markdown(@buffer)
+      out.print RESTORE_CURSOR
+      out.print CLEAR_TO_END
+      out.print rendered
+    end
+
+    def render_markdown(text)
+      TTY::Markdown.parse(text, width: @width, color: :always)
+    rescue => _e
+      # If TTY::Markdown chokes on partial markdown, return raw text
+      text
+    end
+
+    # Resolve the output stream. When no explicit output was provided,
+    # use $stdout dynamically so that test helpers like capture_stdout
+    # (which swap $stdout) work transparently.
+    def out
+      @output || $stdout
+    end
+  end
+end

data/lib/brute_cli/styles.rb

@@ -1,45 +1,46 @@
 # frozen_string_literal: true
 
-require 'lipgloss'
+require "colorize"
 
 module BruteCLI
-  module Styles
-    # Styles.foreground("#fff").bold(true) etc -- delegates to Lipgloss::Style.new
-    def self.method_missing(name, *args, &block)
-      Lipgloss::Style.new.send(name, *args, &block)
-    end
+  # Available themes — each maps to a colorize color symbol.
+  THEMES = {
+    "lemon-and-lime" => :yellow,
+    "ganja-king"     => :green,
+    "og-treacle"     => :magenta,
+    "blues-clues"    => :blue,
+    "hot-cheeto"     => :light_red,
+    "blue-ice-vape"  => :light_magenta,
+    "matrix"         => :light_green,
+    "mrs-jackson"    => :red,
+  }.freeze
 
-    def self.respond_to_missing?(name, include_private = false)
-      Lipgloss::Style.new.respond_to?(name) || super
-    end
+  DEFAULT_THEME = "lemon-and-lime"
 
-    # Brand colors
-    PURPLE  = '#6B50FF'
-    PINK    = '#FF60FF'
-    CYAN    = '#3EEFCF'
-    RED     = '#FF5F87'
-    DIM     = '#757575'
-    MUTED   = '#585858'
-    WHITE   = '#F1F1F1'
-    DARK_BG = '#1A1A2E'
+  # Global accent color — change this one constant to re-theme the entire CLI.
+  COLOR = THEMES.fetch(DEFAULT_THEME)
 
-    PROMPT       = foreground(PURPLE).bold(true)
-    DIM_TEXT     = foreground(DIM)
-    SEPARATOR    = foreground(MUTED)
-    STAT_VALUE   = foreground(CYAN)
-    ERROR_BADGE  = foreground(WHITE).background(RED).bold(true).padding_left(1).padding_right(1)
-    ERROR_REASON = foreground(RED)
-    TOOL_BADGE   = foreground(DARK_BG).background(CYAN).bold(true).padding_left(1).padding_right(1)
-    TOOL_ARG_KEY = foreground(PURPLE)
-    TOOL_ARG_VAL = foreground(DIM)
-    TOOL_OK      = foreground(DARK_BG).background(CYAN).padding_left(1).padding_right(1)
-    TOOL_FAIL    = foreground(WHITE).background(RED).bold(true).padding_left(1).padding_right(1)
-    TOOL_FRAME   = border_style(:rounded).border_foreground(MUTED).padding_left(1).padding_right(1)
+  # Named color/style constants for use with "string".colorize(CONST).
+  # Compound styles (bold + color, background + foreground) use the hash form.
+  DIM         = :light_black
+  ACCENT      = COLOR
+  ACCENT_BOLD = { color: COLOR, mode: :bold }
+  ACCENT_BG   = { color: :black, background: COLOR, mode: :bold }
+  ACCENT_BG2  = { color: :black, background: COLOR }
+  ERROR_BG    = { color: :white, background: :red, mode: :bold }
+  ERROR_FG    = :red
+
+  # Re-derive all accent constants from a new color.
+  # Call this before any output (e.g. right after option parsing).
+  def self.apply_theme!(name)
+    color = THEMES.fetch(name) do
+      raise ArgumentError, "Unknown theme #{name.inspect}. Valid themes: #{THEMES.keys.join(', ')}"
+    end
 
-    DIFF_ADDED   = foreground('#4fd6be')
-    DIFF_REMOVED = foreground('#c53b53')
-    DIFF_HUNK    = foreground(CYAN).bold(true)
-    DIFF_CONTEXT = foreground(DIM)
-    TOOL_INLINE  = foreground(DIM)
+    remove_const(:COLOR)       ; const_set(:COLOR,       color)
+    remove_const(:ACCENT)      ; const_set(:ACCENT,      color)
+    remove_const(:ACCENT_BOLD) ; const_set(:ACCENT_BOLD, { color: color, mode: :bold })
+    remove_const(:ACCENT_BG)   ; const_set(:ACCENT_BG,   { color: :black, background: color, mode: :bold })
+    remove_const(:ACCENT_BG2)  ; const_set(:ACCENT_BG2,  { color: :black, background: color })
   end
 end

data/lib/brute_cli/version.rb

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

data/lib/brute_cli.rb

@@ -7,21 +7,34 @@ begin
 rescue LoadError
 end
 
+require 'tty-screen'
 require 'tty-spinner'
-require 'lipgloss'
-require 'glamour'
-
 require 'brute_cli/version'
 require 'brute_cli/styles'
 require 'brute_cli/emoji'
+require 'brute_cli/bat'
+require 'brute_cli/stream_formatter'
+require 'brute_cli/fzf_menu'
+require 'brute_cli/commands'
 require 'brute_cli/repl'
 
 module BruteCLI
+  LOGO = <<-LOGO
+ .o8                                .             
+"888                              .o8             
+ 888oooo.  oooo d8b oooo  oooo  .o888oo  .ooooo.  
+ d88' `88b `888""8P `888  `888    888   d88' `88b 
+ 888   888  888      888   888    888   888ooo888 
+ 888   888  888      888   888    888 . 888    .o 
+ `Y8bod8P' d888b     `V88V"V8P'   "888" `Y8bod8P' 
+  LOGO
+                                                  
+
   def self.error(message)
-    warn "#{Styles::ERROR_BADGE.render('ERROR')} #{Styles::ERROR_REASON.render(message)}"
+    $stderr.puts "#{"ERROR".colorize(ERROR_BG)} #{message.colorize(ERROR_FG)}"
   end
 
   def self.warn(message)
-    warn Styles::DIM_TEXT.render("warning: #{message}")
+    $stderr.puts "warning: #{message}".colorize(DIM)
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: brute_cli
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Brute Contributors
 bindir: exe
 cert_chain: []
-date: 1980-01-02 00:00:00.000000000 Z
+date: 1980-01-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: brute
@@ -52,19 +52,75 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '4.1'
 - !ruby/object:Gem::Dependency
-  name: glamour
+  name: colorize
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.2'
+        version: '1.1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.2'
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: reline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.5'
+- !ruby/object:Gem::Dependency
+  name: tty-markdown
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+- !ruby/object:Gem::Dependency
+  name: tty-screen
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+- !ruby/object:Gem::Dependency
+  name: bubbletea
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
 - !ruby/object:Gem::Dependency
   name: lipgloss
   requirement: !ruby/object:Gem::Requirement
@@ -79,6 +135,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.2'
+- !ruby/object:Gem::Dependency
+  name: bubbles
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
 - !ruby/object:Gem::Dependency
   name: tty-spinner
   requirement: !ruby/object:Gem::Requirement
@@ -93,6 +163,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
 description: Interactive command-line interface for the Brute coding agent. Supports
   single-prompt, interactive, piped, and session modes.
 executables:
@@ -102,8 +186,13 @@ extra_rdoc_files: []
 files:
 - exe/brute
 - lib/brute_cli.rb
+- lib/brute_cli/bat.rb
+- lib/brute_cli/commands.rb
 - lib/brute_cli/emoji.rb
+- lib/brute_cli/fzf_menu.rb
+- lib/brute_cli/question_screen.rb
 - lib/brute_cli/repl.rb
+- lib/brute_cli/stream_formatter.rb
 - lib/brute_cli/styles.rb
 - lib/brute_cli/version.rb
 licenses: