rubyn-code 0.1.0 → 0.2.0

data/lib/rubyn_code/cli/spinner.rb

@@ -1,50 +1,50 @@
 # frozen_string_literal: true
 
-require "tty-spinner"
+require 'tty-spinner'
 
 module RubynCode
   module CLI
     class Spinner
       THINKING_MESSAGES = [
-        "Massaging the hash...",
-        "Refactoring in my head...",
-        "Consulting Matz...",
-        "Freezing strings...",
-        "Monkey-patching reality...",
-        "Yielding to the block...",
-        "Enumerating possibilities...",
-        "Injecting dependencies...",
-        "Guard clause-ing my thoughts...",
-        "Sharpening the gems...",
-        "Duck typing furiously...",
-        "Reducing complexity...",
-        "Mapping it out...",
-        "Selecting the right approach...",
-        "Running the mental specs...",
-        "Composing a module...",
-        "Memoizing the answer...",
-        "Digging through the hash...",
-        "Pattern matching on this...",
-        "Raising my standards...",
-        "Rescuing the situation...",
-        "Benchmarking my thoughts...",
-        "Sending :think to self...",
-        "Evaluating the proc...",
-        "Opening the eigenclass...",
-        "Calling .new on an idea...",
-        "Plucking the good bits...",
-        "Finding each solution...",
-        "Requiring more context...",
-        "Bundling my thoughts...",
+        'Massaging the hash...',
+        'Refactoring in my head...',
+        'Consulting Matz...',
+        'Freezing strings...',
+        'Monkey-patching reality...',
+        'Yielding to the block...',
+        'Enumerating possibilities...',
+        'Injecting dependencies...',
+        'Guard clause-ing my thoughts...',
+        'Sharpening the gems...',
+        'Duck typing furiously...',
+        'Reducing complexity...',
+        'Mapping it out...',
+        'Selecting the right approach...',
+        'Running the mental specs...',
+        'Composing a module...',
+        'Memoizing the answer...',
+        'Digging through the hash...',
+        'Pattern matching on this...',
+        'Raising my standards...',
+        'Rescuing the situation...',
+        'Benchmarking my thoughts...',
+        'Sending :think to self...',
+        'Evaluating the proc...',
+        'Opening the eigenclass...',
+        'Calling .new on an idea...',
+        'Plucking the good bits...',
+        'Finding each solution...',
+        'Requiring more context...',
+        'Bundling my thoughts...'
       ].freeze
 
       SUB_AGENT_MESSAGES = [
-        "Sub-agent is spelunking...",
-        "Agent exploring the codebase...",
-        "Reading all the things...",
-        "Sub-agent doing the legwork...",
-        "Agent grepping through files...",
-        "Dispatching the intern...",
+        'Sub-agent is spelunking...',
+        'Agent exploring the codebase...',
+        'Reading all the things...',
+        'Sub-agent doing the legwork...',
+        'Agent grepping through files...',
+        'Dispatching the intern...'
       ].freeze
 
       def initialize
@@ -62,7 +62,7 @@ module RubynCode
       end
 
       def start_sub_agent(tool_count = 0)
-        msg = if tool_count > 0
+        msg = if tool_count.positive?
                 "#{SUB_AGENT_MESSAGES.sample} (#{tool_count} tools)"
               else
                 SUB_AGENT_MESSAGES.sample
@@ -77,12 +77,12 @@ module RubynCode
         start(message)
       end
 
-      def success(message = "Done")
+      def success(message = 'Done')
         @spinner&.success("(#{message})")
         @spinner = nil
       end
 
-      def error(message = "Failed")
+      def error(message = 'Failed')
         @spinner&.error("(#{message})")
         @spinner = nil
       end

data/lib/rubyn_code/cli/stream_formatter.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require "pastel"
-require "rouge"
+require 'pastel'
+require 'rouge'
 
 module RubynCode
   module CLI
@@ -9,13 +9,13 @@ module RubynCode
     # Buffers code blocks until they close, then syntax-highlights them.
     # Applies inline formatting (bold, code, headers) as text arrives.
     class StreamFormatter
-      def initialize(renderer = nil)
+      def initialize(_renderer = nil)
         @pastel = Pastel.new
         @rouge_formatter = Rouge::Formatters::Terminal256.new(theme: Rouge::Themes::Monokai.new)
-        @buffer = +""
+        @buffer = +''
         @in_code_block = false
         @code_lang = nil
-        @code_buffer = +""
+        @code_buffer = +''
       end
 
       # Feed a chunk of streamed text
@@ -29,11 +29,11 @@ module RubynCode
         end
 
         # Print any remaining partial line (no newline yet) if not in a code block
-        unless @in_code_block || @buffer.empty?
-          $stdout.print format_inline(@buffer)
-          $stdout.flush
-          @buffer = +""
-        end
+        return if @in_code_block || @buffer.empty?
+
+        $stdout.print format_inline(@buffer)
+        $stdout.flush
+        @buffer = +''
       end
 
       # Flush any remaining buffered content
@@ -45,7 +45,7 @@ module RubynCode
           else
             $stdout.print format_inline(@buffer)
           end
-          @buffer = +""
+          @buffer = +''
         end
 
         # Flush unclosed code block
@@ -69,8 +69,8 @@ module RubynCode
             # Opening fence
             @in_code_block = true
             @code_lang = stripped.match(/```(\w*)/)[1]
-            @code_lang = "ruby" if @code_lang.empty?
-            @code_buffer = +""
+            @code_lang = 'ruby' if @code_lang.empty?
+            @code_buffer = +''
             $stdout.puts @pastel.dim("  ┌─ #{@code_lang}")
           end
           return
@@ -89,50 +89,51 @@ module RubynCode
       def render_code_block
         return if @code_buffer.empty?
 
-        lexer = Rouge::Lexer.find(@code_lang || "ruby") || Rouge::Lexers::PlainText.new
+        lexer = Rouge::Lexer.find(@code_lang || 'ruby') || Rouge::Lexers::PlainText.new
         highlighted = @rouge_formatter.format(lexer.lex(@code_buffer))
-        border = @pastel.dim("  │ ")
+        border = @pastel.dim('  │ ')
 
         highlighted.each_line do |l|
           $stdout.print "#{border}#{l}"
         end
-        $stdout.puts @pastel.dim("  └─")
+        $stdout.puts @pastel.dim('  └─')
         $stdout.flush
 
-        @code_buffer = +""
+        @code_buffer = +''
       rescue StandardError
         # Fallback: print unformatted
         @code_buffer.each_line { |l| $stdout.print "  #{l}" }
         $stdout.puts
-        @code_buffer = +""
+        @code_buffer = +''
       end
 
       def format_line(line)
         stripped = line.rstrip
 
         # Headers
-        if stripped.match?(/\A\#{1,6}\s/)
+        case stripped
+        when /\A\#{1,6}\s/
           level = stripped.match(/\A(\#{1,6})\s/)[1].length
-          text = stripped.sub(/\A\#{1,6}\s+/, "")
+          text = stripped.sub(/\A\#{1,6}\s+/, '')
           case level
           when 1 then "#{@pastel.bold.underline(text)}\n"
           when 2 then "\n#{@pastel.bold(text)}\n"
           else "#{@pastel.bold(text)}\n"
           end
         # Bullet lists
-        elsif stripped.match?(/\A\s*[-*]\s/)
+        when /\A\s*[-*]\s/
           indent = stripped.match(/\A(\s*)/)[1]
-          content = stripped.sub(/\A\s*[-*]\s+/, "")
-          "#{indent}  #{@pastel.cyan("•")} #{format_inline(content)}\n"
+          content = stripped.sub(/\A\s*[-*]\s+/, '')
+          "#{indent}  #{@pastel.cyan('•')} #{format_inline(content)}\n"
         # Numbered lists
-        elsif stripped.match?(/\A\s*\d+\.\s/)
+        when /\A\s*\d+\.\s/
           indent = stripped.match(/\A(\s*)/)[1]
           num = stripped.match(/(\d+)\./)[1]
-          content = stripped.sub(/\A\s*\d+\.\s+/, "")
-          "#{indent}  #{@pastel.cyan(num + ".")} #{format_inline(content)}\n"
+          content = stripped.sub(/\A\s*\d+\.\s+/, '')
+          "#{indent}  #{@pastel.cyan("#{num}.")} #{format_inline(content)}\n"
         # Horizontal rules
-        elsif stripped.match?(/\A-{3,}\z/)
-          "#{@pastel.dim("─" * 40)}\n"
+        when /\A-{3,}\z/
+          "#{@pastel.dim('─' * 40)}\n"
         else
           "#{format_inline(line.chomp)}\n"
         end

data/lib/rubyn_code/cli/version_check.rb

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+require 'faraday'
+require 'json'
+
+module RubynCode
+  module CLI
+    # Non-blocking version check against RubyGems.
+    # Runs in a background thread so it never delays startup.
+    # Caches the result for 24 hours to avoid hammering the API.
+    class VersionCheck
+      RUBYGEMS_API = 'https://rubygems.org/api/v1/versions/rubyn-code/latest.json'
+      CACHE_FILE = File.join(Config::Defaults::HOME_DIR, '.version_check')
+      CACHE_TTL = 86_400 # 24 hours
+
+      def initialize(renderer:)
+        @renderer = renderer
+        @thread = nil
+      end
+
+      # Kicks off a background check. Call `notify` later to display results.
+      def start
+        return if ENV['RUBYN_NO_UPDATE_CHECK']
+
+        @thread = Thread.new { check }
+        @thread.abort_on_exception = false
+      end
+
+      # Waits briefly for the check to finish and prints a message if outdated.
+      def notify(timeout: 2)
+        return unless @thread
+
+        @thread.join(timeout)
+        return unless @result
+
+        return unless newer?(@result, RubynCode::VERSION)
+
+        @renderer.warning(
+          "Update available: #{RubynCode::VERSION} -> #{@result}  " \
+          '(gem install rubyn-code)'
+        )
+      end
+
+      private
+
+      def check
+        cached = read_cache
+        if cached
+          @result = cached
+          return
+        end
+
+        conn = Faraday.new do |f|
+          f.options.timeout = 5
+          f.options.open_timeout = 3
+        end
+        response = conn.get(RUBYGEMS_API)
+        return unless response.success?
+
+        data = JSON.parse(response.body)
+        latest = data['version']
+        return unless latest
+        return unless latest.match?(/\A\d+\.\d+/)
+        return unless Gem::Version.correct?(latest)
+
+        write_cache(latest)
+        @result = latest
+      rescue StandardError
+        # Silent — never interrupt startup for a version check
+      end
+
+      def newer?(remote, local)
+        Gem::Version.new(remote) > Gem::Version.new(local)
+      rescue ArgumentError
+        false
+      end
+
+      def read_cache
+        return nil unless File.exist?(CACHE_FILE)
+        return nil if (Time.now - File.mtime(CACHE_FILE)) > CACHE_TTL
+
+        File.read(CACHE_FILE).strip
+      rescue StandardError
+        nil
+      end
+
+      def write_cache(version)
+        File.write(CACHE_FILE, version)
+      rescue StandardError
+        # Best effort
+      end
+    end
+  end
+end

data/lib/rubyn_code/config/RUBYN.md

@@ -0,0 +1,14 @@
+# Config Layer
+
+Application settings with per-project overrides.
+
+## Classes
+
+- **`Settings`** — Main configuration module. Merges defaults with user overrides from
+  `~/.rubyn-code/config.yml` and project-level `.rubyn-code.yml`.
+
+- **`Defaults`** — Frozen constants for all default values: `MAX_ITERATIONS`, model names,
+  token limits, budget caps, permission tiers, etc.
+
+- **`ProjectConfig`** — Loads project-specific configuration from `.rubyn-code.yml` in the
+  working directory. Supports custom permission tiers, deny lists, and hook definitions.

data/lib/rubyn_code/config/defaults.rb

@@ -3,19 +3,28 @@
 module RubynCode
   module Config
     module Defaults
-      HOME_DIR = File.expand_path("~/.rubyn-code")
-      CONFIG_FILE = File.join(HOME_DIR, "config.yml")
-      DB_FILE = File.join(HOME_DIR, "rubyn_code.db")
-      TOKENS_FILE = File.join(HOME_DIR, "tokens.yml")
-      SESSIONS_DIR = File.join(HOME_DIR, "sessions")
-      MEMORIES_DIR = File.join(HOME_DIR, "memories")
-
-      DEFAULT_MODEL = "claude-opus-4-6"
+      HOME_DIR = File.expand_path('~/.rubyn-code')
+      CONFIG_FILE = File.join(HOME_DIR, 'config.yml')
+      DB_FILE = File.join(HOME_DIR, 'rubyn_code.db')
+      TOKENS_FILE = File.join(HOME_DIR, 'tokens.yml')
+      SESSIONS_DIR = File.join(HOME_DIR, 'sessions')
+      MEMORIES_DIR = File.join(HOME_DIR, 'memories')
+
+      DEFAULT_MODEL = 'claude-opus-4-6'
       MAX_ITERATIONS = 200
-      MAX_SUB_AGENT_ITERATIONS = 30
-      MAX_OUTPUT_CHARS = 50_000
-      CONTEXT_THRESHOLD_TOKENS = 50_000
-      MICRO_COMPACT_KEEP_RECENT = 3
+      MAX_SUB_AGENT_ITERATIONS = 200
+      MAX_EXPLORE_AGENT_ITERATIONS = 200
+
+      # Output token management (3-tier recovery, matches Claude Code)
+      CAPPED_MAX_OUTPUT_TOKENS = 8_000 # Default cap — keeps prompt cache efficient
+      ESCALATED_MAX_OUTPUT_TOKENS = 32_000   # Silent escalation on first max_tokens hit
+      MAX_OUTPUT_TOKENS_RECOVERY_LIMIT = 3   # Multi-turn recovery attempts after escalation
+
+      MAX_OUTPUT_CHARS = 10_000
+      MAX_TOOL_RESULT_CHARS = 10_000          # Per-tool result cap
+      MAX_MESSAGE_TOOL_RESULTS_CHARS = 50_000 # Aggregate cap for all tool results in one message
+      CONTEXT_THRESHOLD_TOKENS = 80_000
+      MICRO_COMPACT_KEEP_RECENT = 2
 
       POLL_INTERVAL = 5
       IDLE_TIMEOUT = 60
@@ -23,15 +32,15 @@ module RubynCode
       SESSION_BUDGET_USD = 5.00
       DAILY_BUDGET_USD = 10.00
 
-      OAUTH_CLIENT_ID = "rubyn-code"
-      OAUTH_REDIRECT_URI = "http://localhost:19275/callback"
-      OAUTH_AUTHORIZE_URL = "https://claude.ai/oauth/authorize"
-      OAUTH_TOKEN_URL = "https://claude.ai/oauth/token"
-      OAUTH_SCOPES = "user:read model:read model:write"
+      OAUTH_CLIENT_ID = 'rubyn-code'
+      OAUTH_REDIRECT_URI = 'http://localhost:19275/callback'
+      OAUTH_AUTHORIZE_URL = 'https://claude.ai/oauth/authorize'
+      OAUTH_TOKEN_URL = 'https://claude.ai/oauth/token'
+      OAUTH_SCOPES = 'user:read model:read model:write'
 
       DANGEROUS_PATTERNS = [
-        "rm -rf /", "sudo rm", "shutdown", "reboot",
-        "> /dev/", "mkfs", "dd if=", ":(){:|:&};:"
+        'rm -rf /', 'sudo rm', 'shutdown', 'reboot',
+        '> /dev/', 'mkfs', 'dd if=', ':(){:|:&};:'
       ].freeze
 
       SCRUB_ENV_VARS = %w[

data/lib/rubyn_code/config/project_config.rb

@@ -1,17 +1,17 @@
 # frozen_string_literal: true
 
-require "yaml"
-require "fileutils"
-require_relative "defaults"
-require_relative "settings"
+require 'yaml'
+require 'fileutils'
+require_relative 'defaults'
+require_relative 'settings'
 
 module RubynCode
   module Config
     class ProjectConfig
       class LoadError < StandardError; end
 
-      PROJECT_DIR_NAME = ".rubyn-code"
-      CONFIG_FILENAME = "config.yml"
+      PROJECT_DIR_NAME = '.rubyn-code'
+      CONFIG_FILENAME = 'config.yml'
 
       attr_reader :project_root, :config_path, :data
 
@@ -73,9 +73,7 @@ module RubynCode
 
         loop do
           candidate = File.join(dir, PROJECT_DIR_NAME, CONFIG_FILENAME)
-          if File.exist?(candidate)
-            return new(project_root: dir, global_settings: global_settings)
-          end
+          return new(project_root: dir, global_settings: global_settings) if File.exist?(candidate)
 
           parent = File.dirname(dir)
           break if parent == dir # filesystem root reached

data/lib/rubyn_code/config/settings.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "yaml"
-require "fileutils"
-require_relative "defaults"
+require 'yaml'
+require 'fileutils'
+require_relative 'defaults'
 
 module RubynCode
   module Config

data/lib/rubyn_code/context/RUBYN.md

@@ -0,0 +1,20 @@
+# Layer 4: Context Management
+
+Manages the conversation context window to stay within Claude's token limits.
+
+## Classes
+
+- **`Manager`** — Orchestrates context strategy. Tracks token usage, decides when
+  compaction is needed, selects the right compaction strategy.
+
+- **`Compactor`** — Base compaction logic. Sends the conversation to Claude with a
+  summarization prompt, replaces old messages with the summary.
+
+- **`AutoCompact`** — Triggers automatically when token usage exceeds a threshold
+  (e.g. 80% of the context window). Runs transparently mid-conversation.
+
+- **`MicroCompact`** — Lightweight compaction that trims tool results (large file
+  contents, long bash outputs) without summarizing the whole conversation.
+
+- **`ManualCompact`** — Triggered by the user via `/compact`. Lets you specify a
+  focus area for the summary (e.g. "focus on the auth refactor").

data/lib/rubyn_code/context/auto_compact.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require "json"
-require "fileutils"
+require 'json'
+require 'fileutils'
 
 module RubynCode
   module Context
@@ -33,13 +33,13 @@ module RubynCode
         transcript_text = serialize_tail(messages, MAX_TRANSCRIPT_CHARS)
         summary = request_summary(transcript_text, llm_client)
 
-        [{ role: "user", content: "[Context compacted]\n\n#{summary}" }]
+        [{ role: 'user', content: "[Context compacted]\n\n#{summary}" }]
       end
 
       # Persists the full conversation to a timestamped JSON file.
       def self.save_transcript(messages, dir)
         FileUtils.mkdir_p(dir)
-        timestamp = Time.now.strftime("%Y%m%d_%H%M%S")
+        timestamp = Time.now.strftime('%Y%m%d_%H%M%S')
         path = File.join(dir, "transcript_#{timestamp}.json")
         File.write(path, JSON.pretty_generate(messages))
         path
@@ -57,19 +57,19 @@ module RubynCode
       def self.request_summary(transcript_text, llm_client)
         summary_messages = [
           {
-            role: "user",
+            role: 'user',
             content: "#{SUMMARY_INSTRUCTION}\n\n---\n\n#{transcript_text}"
           }
         ]
 
         options = {}
-        options[:model] = "claude-sonnet-4-20250514" if llm_client.respond_to?(:chat)
+        options[:model] = 'claude-sonnet-4-20250514' if llm_client.respond_to?(:chat)
 
         response = llm_client.chat(messages: summary_messages, **options)
 
         case response
         when String then response
-        when Hash then response[:content] || response["content"] || response.to_s
+        when Hash then response[:content] || response['content'] || response.to_s
         else
           response.respond_to?(:text) ? response.text : response.to_s
         end

data/lib/rubyn_code/context/compactor.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "json"
+require 'json'
 
 module RubynCode
   module Context
@@ -82,7 +82,7 @@ module RubynCode
       def ensure_llm_client!
         return if @llm_client
 
-        raise RubynCode::Error, "LLM client is required for summarization-based compaction"
+        raise RubynCode::Error, 'LLM client is required for summarization-based compaction'
       end
     end
   end

data/lib/rubyn_code/context/context_collapse.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module RubynCode
+  module Context
+    # Lightweight context reduction that removes old conversation turns without
+    # calling the LLM. Runs before auto-compact — if collapse alone brings the
+    # context under threshold, the expensive LLM summarization is skipped.
+    #
+    # Keeps the first message (initial user request), the most recent N exchanges,
+    # and replaces everything in between with a "[earlier conversation snipped]" marker.
+    module ContextCollapse
+      SNIP_MARKER = '[%d earlier messages snipped for context efficiency]'
+      CHARS_PER_TOKEN = 4
+
+      # Returns a collapsed copy of messages if doing so brings the estimated
+      # token count under threshold. Returns nil if collapse isn't sufficient
+      # (caller should fall through to full auto-compact).
+      #
+      # @param messages [Array<Hash>] conversation messages
+      # @param threshold [Integer] target token count
+      # @param keep_recent [Integer] number of recent messages to preserve
+      # @return [Array<Hash>, nil] collapsed messages or nil if not sufficient
+      def self.call(messages, threshold:, keep_recent: 6)
+        return nil if messages.size <= keep_recent + 2
+
+        # Keep first message + last N messages, snip the middle
+        first = messages.first
+        recent = messages.last(keep_recent)
+        snipped_count = messages.size - keep_recent - 1
+
+        collapsed = [
+          first,
+          { role: 'user', content: format(SNIP_MARKER, snipped_count) },
+          *recent
+        ]
+
+        # Only use collapse if it gets us under threshold
+        estimated = (JSON.generate(collapsed).length.to_f / CHARS_PER_TOKEN).ceil
+        estimated <= threshold ? collapsed : nil
+      rescue JSON::GeneratorError
+        nil
+      end
+    end
+  end
+end

data/lib/rubyn_code/context/manager.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "json"
+require 'json'
 
 module RubynCode
   module Context
@@ -13,7 +13,7 @@ module RubynCode
       attr_reader :total_input_tokens, :total_output_tokens
 
       # @param threshold [Integer] estimated token count that triggers auto-compaction
-      def initialize(threshold: 50_000)
+      def initialize(threshold: Config::Defaults::CONTEXT_THRESHOLD_TOKENS)
         @threshold = threshold
         @total_input_tokens = 0
         @total_output_tokens = 0
@@ -53,13 +53,30 @@ module RubynCode
       #
       # @param conversation [#messages, #messages=] conversation wrapper
       # @return [void]
+      # Fraction of the compaction threshold at which micro-compact kicks in.
+      # Running it too early busts the prompt cache prefix (mutated messages
+      # change the hash, invalidating server-side cached tokens).
+      MICRO_COMPACT_RATIO = 0.7
+
       def check_compaction!(conversation)
         messages = conversation.messages
 
-        MicroCompact.call(messages)
+        # Step 1: Zero-cost micro-compact — but only when we're approaching
+        # the compaction threshold. Running it every turn mutates old messages,
+        # which invalidates the prompt cache prefix and wastes tokens.
+        est = estimated_tokens(messages)
+        MicroCompact.call(messages) if est > (@threshold * MICRO_COMPACT_RATIO)
 
         return unless needs_compaction?(messages)
 
+        # Step 2: Try context collapse (snip old messages, no LLM call)
+        collapsed = ContextCollapse.call(messages, threshold: @threshold)
+        if collapsed
+          apply_compacted_messages(conversation, collapsed)
+          return
+        end
+
+        # Step 3: Full LLM-driven auto-compact (expensive, last resort)
         compactor = Compactor.new(
           llm_client: conversation.respond_to?(:llm_client) ? conversation.llm_client : nil,
           threshold: @threshold