rubyn-code 0.7.0 → 0.9.0

data/lib/rubyn_code/tools/todo_write.rb

@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+require_relative 'registry'
+
+module RubynCode
+  module Tools
+    # Update the in-turn task checklist. The model uses this to keep the user
+    # informed of progress while it works. The store is shared between the
+    # Agent::Loop (which exposes it for the renderer) and this tool.
+    class TodoWrite < Base
+      TOOL_NAME = 'TodoWrite'
+      DESCRIPTION = 'Update the in-turn task checklist. ' \
+                    'Use this to keep the user informed of what you plan to do, ' \
+                    'what you are currently working on, and what you have finished.'
+      PARAMETERS = {
+        todos: { type: :array, required: true,
+                 description: 'The full set of tasks currently on the checklist. ' \
+                              'Replace the existing list with this. Each task is ' \
+                              '{ "content": "...", "status": "pending|in_progress|completed", ' \
+                              '"active_form": "..." }.' }
+      }.freeze
+      RISK_LEVEL = :read
+      REQUIRES_CONFIRMATION = false
+
+      VALID_STATUS = %w[pending in_progress completed].freeze
+
+      def initialize(project_root:, store: nil)
+        super(project_root: project_root)
+        @store = store
+      end
+
+      def execute(todos:)
+        items = Array(todos)
+        validated = []
+
+        items.each_with_index do |item, i|
+          return "TodoWrite: item #{i} is not a hash — got #{item.class}" unless item.is_a?(Hash)
+
+          content = item[:content] || item['content']
+          status  = (item[:status] || item['status']).to_s
+          active_form = item[:active_form] || item['active_form']
+
+          return "TodoWrite: item #{i} missing 'content'" if content.to_s.empty?
+          unless VALID_STATUS.include?(status)
+            return "TodoWrite: item #{i} status must be one of #{VALID_STATUS.join('/')} (got #{status.inspect})"
+          end
+
+          validated << {
+            'content' => content.to_s,
+            'status' => status,
+            'active_form' => active_form.to_s.empty? ? content.to_s : active_form.to_s
+          }
+        end
+
+        @store&.replace(validated)
+        format(validated)
+      end
+
+      def self.summarize(_output, args)
+        count = Array(args[:todos] || args['todos']).size
+        if count.zero?
+          'cleared checklist'
+        else
+          "checklist: #{count} item#{'s' unless count == 1}"
+        end
+      end
+
+      private
+
+      def format(items)
+        return 'Checklist cleared.' if items.empty?
+
+        items.map do |item|
+          mark =
+            case item['status']
+            when 'completed' then '[x]'
+            when 'in_progress' then '[~]'
+            else '[ ]'
+            end
+          "#{mark} #{item['content']}"
+        end.join("\n")
+      end
+    end
+
+    Registry.register(TodoWrite)
+  end
+end

data/lib/rubyn_code/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubynCode
-  VERSION = '0.7.0'
+  VERSION = '0.9.0'
 end

data/lib/rubyn_code.rb

@@ -47,6 +47,11 @@ module RubynCode
   module LLM
     autoload :Client, 'rubyn_code/llm/client'
     autoload :MessageBuilder, 'rubyn_code/llm/message_builder'
+    autoload :TextBlock,     'rubyn_code/llm/message_builder'
+    autoload :ThinkingBlock, 'rubyn_code/llm/message_builder'
+    autoload :ImageBlock,    'rubyn_code/llm/message_builder'
+    autoload :ToolUseBlock,  'rubyn_code/llm/message_builder'
+    autoload :ImageReader, 'rubyn_code/llm/image_reader'
     autoload :ModelRouter, 'rubyn_code/llm/model_router'
 
     # Adapters (provider-specific implementations)
@@ -54,7 +59,6 @@ module RubynCode
       autoload :Base, 'rubyn_code/llm/adapters/base'
       autoload :JsonParsing, 'rubyn_code/llm/adapters/json_parsing'
       autoload :PromptCaching, 'rubyn_code/llm/adapters/prompt_caching'
-      autoload :TokenCaching, 'rubyn_code/llm/adapters/token_caching'
       autoload :Anthropic, 'rubyn_code/llm/adapters/anthropic'
       autoload :AnthropicCompatible, 'rubyn_code/llm/adapters/anthropic_compatible'
       autoload :AnthropicStreaming, 'rubyn_code/llm/adapters/anthropic_streaming'
@@ -63,9 +67,6 @@ module RubynCode
       autoload :OpenAICompatible, 'rubyn_code/llm/adapters/openai_compatible'
       autoload :OpenAIMessageTranslator, 'rubyn_code/llm/adapters/openai_message_translator'
     end
-
-    # Backward-compat: LLM::Streaming → Adapters::AnthropicStreaming
-    autoload :Streaming, 'rubyn_code/llm/streaming'
   end
 
   # Layer 1: Agent Loop
@@ -81,6 +82,7 @@ module RubynCode
   module Tools
     autoload :Base, 'rubyn_code/tools/base'
     autoload :Registry, 'rubyn_code/tools/registry'
+    autoload :TodoStore, 'rubyn_code/tools/todo_store'
     autoload :Schema, 'rubyn_code/tools/schema'
     autoload :Executor, 'rubyn_code/tools/executor'
     autoload :ReadFile, 'rubyn_code/tools/read_file'
@@ -88,6 +90,7 @@ module RubynCode
     autoload :EditFile, 'rubyn_code/tools/edit_file'
     autoload :Glob, 'rubyn_code/tools/glob'
     autoload :Grep, 'rubyn_code/tools/grep'
+    autoload :CodeGraph, 'rubyn_code/tools/code_graph'
     autoload :Bash, 'rubyn_code/tools/bash'
     autoload :RailsGenerate, 'rubyn_code/tools/rails_generate'
     autoload :DbMigrate, 'rubyn_code/tools/db_migrate'
@@ -95,6 +98,7 @@ module RubynCode
     autoload :BundleInstall, 'rubyn_code/tools/bundle_install'
     autoload :BundleAdd, 'rubyn_code/tools/bundle_add'
     autoload :Compact, 'rubyn_code/tools/compact'
+    autoload :TodoWrite, 'rubyn_code/tools/todo_write'
     autoload :LoadSkill, 'rubyn_code/tools/load_skill'
     autoload :Task, 'rubyn_code/tools/task'
     autoload :MemorySearch, 'rubyn_code/tools/memory_search'
@@ -103,6 +107,7 @@ module RubynCode
     autoload :ReadInbox, 'rubyn_code/tools/read_inbox'
     autoload :ReviewPr, 'rubyn_code/tools/review_pr'
     autoload :SpawnAgent, 'rubyn_code/tools/spawn_agent'
+    autoload :PhoneAFriend, 'rubyn_code/tools/phone_a_friend'
     autoload :BackgroundRun, 'rubyn_code/tools/background_run'
     autoload :WebSearch, 'rubyn_code/tools/web_search'
     autoload :WebFetch, 'rubyn_code/tools/web_fetch'
@@ -133,8 +138,6 @@ module RubynCode
     autoload :AutoCompact, 'rubyn_code/context/auto_compact'
     autoload :ManualCompact, 'rubyn_code/context/manual_compact'
     autoload :ContextCollapse, 'rubyn_code/context/context_collapse'
-    autoload :ContextBudget, 'rubyn_code/context/context_budget'
-    autoload :SchemaFilter, 'rubyn_code/context/schema_filter'
     autoload :DecisionCompactor, 'rubyn_code/context/decision_compactor'
   end
 
@@ -201,7 +204,6 @@ module RubynCode
   # Layer 10: Protocols
   module Protocols
     autoload :ShutdownHandshake, 'rubyn_code/protocols/shutdown_handshake'
-    autoload :PlanApproval, 'rubyn_code/protocols/plan_approval'
     autoload :InterruptHandler, 'rubyn_code/protocols/interrupt_handler'
   end
 
@@ -258,6 +260,7 @@ module RubynCode
     autoload :ToolBridge, 'rubyn_code/mcp/tool_bridge'
     autoload :ServerExtrasBridge, 'rubyn_code/mcp/server_extras_bridge'
     autoload :Config, 'rubyn_code/mcp/config'
+    autoload :Discovery, 'rubyn_code/mcp/discovery'
   end
 
   # Layer 16: Learning
@@ -266,7 +269,6 @@ module RubynCode
     autoload :Instinct, 'rubyn_code/learning/instinct'
     autoload :InstinctMethods, 'rubyn_code/learning/instinct'
     autoload :Injector, 'rubyn_code/learning/injector'
-    autoload :Shortcut, 'rubyn_code/learning/shortcut'
     autoload :Porter, 'rubyn_code/learning/porter'
   end
 
@@ -286,6 +288,7 @@ module RubynCode
   # Codebase Index
   module Index
     autoload :CodebaseIndex, 'rubyn_code/index/codebase_index'
+    autoload :PrismExtractor, 'rubyn_code/index/prism_extractor'
   end
 
   # CLI
@@ -347,6 +350,9 @@ module RubynCode
       autoload :ChiselAudit, 'rubyn_code/cli/commands/chisel_audit'
       autoload :ChiselDebt, 'rubyn_code/cli/commands/chisel_debt'
       autoload :ChiselGain, 'rubyn_code/cli/commands/chisel_gain'
+      autoload :Think, 'rubyn_code/cli/commands/think'
+      autoload :Effort, 'rubyn_code/cli/commands/effort'
+      autoload :Export, 'rubyn_code/cli/commands/export'
     end
   end
 

data/skills/rubyn_self_test.md

@@ -449,6 +449,146 @@ fast and need no API calls.
   '
   ```
 
+#### 17k. Extended thinking + `/think` toggle (PRs #132, #138)
+- **bash**: confirms `LLM::ThinkingBlock` is defined, the Anthropic adapter
+  translates `thinking: {budget_tokens}` into the request body, and `/think`
+  toggles state on `LLM::Client`. PASS if the final line is `THINKING: PASS`.
+
+  ```bash
+  bundle exec ruby -Ilib -rrubyn_code -rrubyn_code/llm/message_builder -e '
+    tb = RubynCode::LLM::ThinkingBlock.new(text: "plan")
+    adapter = RubynCode::LLM::Adapters::Anthropic.allocate
+    body = {}
+    adapter.send(:apply_thinking, body, budget_tokens: 4096)
+    client = RubynCode::LLM::Client.allocate
+    client.thinking_budget_tokens = 4096
+    ok = (tb.type == "thinking") &&
+         (body[:thinking] == { type: "enabled", budget_tokens: 4096 }) &&
+         (client.thinking_budget_tokens == 4096)
+    puts(ok ? "THINKING: PASS" : "THINKING: FAIL")
+  '
+  ```
+
+#### 17l. Image / vision input (`@image.png`) (PRs #133, #140, #141)
+- **bash**: confirms `LLM::ImageBlock`, `LLM::ImageReader`, and
+  `MentionExpander#expand_images` are wired and base64-encode PNGs.
+  PASS if the final line is `IMAGES: PASS`.
+
+  ```bash
+  bundle exec ruby -Ilib -rrubyn_code -rrubyn_code/llm/message_builder -e '
+    require "tmpdir"
+    Dir.mktmpdir do |dir|
+      path = File.join(dir, "p.png")
+      File.binwrite(path, "\x89PNG\r\n\x1a\n".dup)
+      block = RubynCode::LLM::ImageReader.for_path(path)
+      ex = RubynCode::CLI::MentionExpander.new(project_root: dir).expand_images("look @p.png")
+      ok = (block.is_a?(RubynCode::LLM::ImageBlock)) &&
+           (block.media_type == "image/png") &&
+           (ex.size == 1)
+      puts(ok ? "IMAGES: PASS" : "IMAGES: FAIL")
+    end
+  '
+  ```
+
+#### 17m. TodoWrite live checklist (PR #134)
+- **bash**: confirms `Tools::TodoWrite` is registered and mutates a shared
+  `Tools::TodoStore`. PASS if the final line is `TODOWRITE: PASS`.
+
+  ```bash
+  bundle exec ruby -Ilib -rrubyn_code -rrubyn_code/tools/todo_write -rrubyn_code/tools/todo_store -e '
+    store = RubynCode::Tools::TodoStore.new
+    tool  = RubynCode::Tools::TodoWrite.new(project_root: Dir.pwd, store: store)
+    out   = tool.execute(todos: [{ "content" => "a", "status" => "in_progress", "active_form" => "A" }])
+    ok = (out.include?("[~] a")) && (store.current.first[:status] == "in_progress") &&
+         (RubynCode::Tools::Registry.get("TodoWrite") == RubynCode::Tools::TodoWrite)
+    puts(ok ? "TODOWRITE: PASS" : "TODOWRITE: FAIL")
+  '
+  ```
+
+#### 17n. Custom-command frontmatter (argument-hint, allowed-tools, model:) (PRs #135, #138)
+- **bash**: confirms `CustomCommand` exposes all three frontmatter keys and
+  `Agent::Loop` honors the per-prompt override setters.
+  PASS if the final line is `FRONTMATTER: PASS`.
+
+  ```bash
+  bundle exec ruby -Ilib -rrubyn_code -rrubyn_code/agent/loop -e '
+    loop = RubynCode::Agent::Loop.allocate
+    loop.allowed_tools_override = %w[bash read]
+    loop.model_override = "claude-opus-4-8"
+    at = loop.instance_variable_get(:@allowed_tools_override)
+    mo = loop.instance_variable_get(:@model_override)
+    cmd = RubynCode::CLI::Commands::CustomCommand.new(
+      name: "d", description: "d", body: "b",
+      argument_hint: "[env]", allowed_tools: %w[bash read], model: "claude-opus-4-8"
+    )
+    ok = (at == %w[bash read]) && (mo == "claude-opus-4-8") &&
+         (cmd.argument_hint == "[env]") && (cmd.allowed_tools == %w[bash read]) &&
+         (cmd.model == "claude-opus-4-8")
+    puts(ok ? "FRONTMATTER: PASS" : "FRONTMATTER: FAIL")
+  '
+  ```
+
+#### 17o. `.mcp.json` auto-discovery (PRs #136, #138, #139)
+- **bash**: writes a project-root `.mcp.json`, calls `MCP::Discovery.discover`,
+  and asserts the entry is tagged `:project`.
+  PASS if the final line is `MCP-DISCOVERY: PASS`.
+
+  ```bash
+  bundle exec ruby -Ilib -rrubyn_code -rrubyn_code/mcp/discovery -e '
+    require "tmpdir"; require "json"
+    Dir.mktmpdir do |dir|
+      File.write(File.join(dir, ".mcp.json"),
+                 JSON.generate(mcpServers: { "p" => { command: "p" } }))
+      entries = RubynCode::MCP::Discovery.discover(dir)
+      ok = (entries.any? { |e| e.name == "p" && e.source == :project })
+      puts(ok ? "MCP-DISCOVERY: PASS" : "MCP-DISCOVERY: FAIL")
+    end
+  '
+  ```
+
+#### 17p. `/export` transcript (markdown / jsonl) (PR #137)
+- **bash**: drives `Commands::Export` against a synthetic conversation with
+  a `thinking` block and a `tool_use`. Writes a markdown file and asserts
+  the file contains role sections, the thinking `<details>`, and the
+  fenced-JSON tool_use. PASS if the final line is `EXPORT: PASS`.
+
+  ```bash
+  bundle exec ruby -Ilib -rrubyn_code -rrubyn_code/cli/commands/export -e '
+    require "tmpdir"
+    Dir.mktmpdir do |dir|
+      msgs = [
+        { role: "user", content: "hi" },
+        { role: "assistant", content: [
+          { type: "thinking", text: "plan" },
+          { type: "text",     text: "ok" },
+          { type: "tool_use", name: "bash", input: { cmd: "ls" } }
+        ] }
+      ]
+      conv = Object.new
+      conv.define_singleton_method(:to_a) { msgs }
+      renderer = Object.new
+      renderer.define_singleton_method(:info)    { |*_| nil }
+      renderer.define_singleton_method(:warning) { |*_| nil }
+      renderer.define_singleton_method(:ask)     { |*_| true }
+      ctx = RubynCode::CLI::Commands::Context.new(
+        renderer: renderer, conversation: conv, agent_loop: nil,
+        context_manager: nil, budget_enforcer: nil, llm_client: nil,
+        db: nil, session_id: nil, project_root: nil, skill_loader: nil,
+        session_persistence: nil, background_worker: nil, permission_tier: nil,
+        plan_mode: false, message_handler: nil, hook_registry: nil,
+        checkpoint_manager: nil
+      )
+      path = File.join(dir, "x.md")
+      RubynCode::CLI::Commands::Export.new.execute([path], ctx)
+      body = File.read(path)
+      ok = body.include?("## User") &&
+           body.include?("<details><summary>thinking</summary>") &&
+           body.include?("[tool: bash]")
+      puts(ok ? "EXPORT: PASS" : "EXPORT: FAIL")
+    end
+  '
+  ```
+
 ### 18. Chisel — Minimal-Code Enforcement (opt-in)
 
 Chisel is rubyn-code's "write the minimum that works" layer. It is **off by

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rubyn-code
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.9.0
 platform: ruby
 authors:
 - fadedmaturity
@@ -240,6 +240,8 @@ files:
 - lib/rubyn_code/cli/commands/custom_loader.rb
 - lib/rubyn_code/cli/commands/diff.rb
 - lib/rubyn_code/cli/commands/doctor.rb
+- lib/rubyn_code/cli/commands/effort.rb
+- lib/rubyn_code/cli/commands/export.rb
 - lib/rubyn_code/cli/commands/goal.rb
 - lib/rubyn_code/cli/commands/help.rb
 - lib/rubyn_code/cli/commands/install_skills.rb
@@ -262,6 +264,7 @@ files:
 - lib/rubyn_code/cli/commands/skills.rb
 - lib/rubyn_code/cli/commands/spawn.rb
 - lib/rubyn_code/cli/commands/tasks.rb
+- lib/rubyn_code/cli/commands/think.rb
 - lib/rubyn_code/cli/commands/tokens.rb
 - lib/rubyn_code/cli/commands/undo.rb
 - lib/rubyn_code/cli/commands/version.rb
@@ -289,13 +292,11 @@ files:
 - lib/rubyn_code/context/RUBYN.md
 - lib/rubyn_code/context/auto_compact.rb
 - lib/rubyn_code/context/compactor.rb
-- lib/rubyn_code/context/context_budget.rb
 - lib/rubyn_code/context/context_collapse.rb
 - lib/rubyn_code/context/decision_compactor.rb
 - lib/rubyn_code/context/manager.rb
 - lib/rubyn_code/context/manual_compact.rb
 - lib/rubyn_code/context/micro_compact.rb
-- lib/rubyn_code/context/schema_filter.rb
 - lib/rubyn_code/db/RUBYN.md
 - lib/rubyn_code/db/connection.rb
 - lib/rubyn_code/db/migrator.rb
@@ -338,12 +339,12 @@ files:
 - lib/rubyn_code/ide/protocol.rb
 - lib/rubyn_code/ide/server.rb
 - lib/rubyn_code/index/codebase_index.rb
+- lib/rubyn_code/index/prism_extractor.rb
 - lib/rubyn_code/learning/RUBYN.md
 - lib/rubyn_code/learning/extractor.rb
 - lib/rubyn_code/learning/injector.rb
 - lib/rubyn_code/learning/instinct.rb
 - lib/rubyn_code/learning/porter.rb
-- lib/rubyn_code/learning/shortcut.rb
 - lib/rubyn_code/llm/RUBYN.md
 - lib/rubyn_code/llm/adapters/anthropic.rb
 - lib/rubyn_code/llm/adapters/anthropic_compatible.rb
@@ -355,14 +356,14 @@ files:
 - lib/rubyn_code/llm/adapters/openai_message_translator.rb
 - lib/rubyn_code/llm/adapters/openai_streaming.rb
 - lib/rubyn_code/llm/adapters/prompt_caching.rb
-- lib/rubyn_code/llm/adapters/token_caching.rb
 - lib/rubyn_code/llm/client.rb
+- lib/rubyn_code/llm/image_reader.rb
 - lib/rubyn_code/llm/message_builder.rb
 - lib/rubyn_code/llm/model_router.rb
-- lib/rubyn_code/llm/streaming.rb
 - lib/rubyn_code/mcp/RUBYN.md
 - lib/rubyn_code/mcp/client.rb
 - lib/rubyn_code/mcp/config.rb
+- lib/rubyn_code/mcp/discovery.rb
 - lib/rubyn_code/mcp/server_extras_bridge.rb
 - lib/rubyn_code/mcp/sse_transport.rb
 - lib/rubyn_code/mcp/stdio_transport.rb
@@ -393,7 +394,6 @@ files:
 - lib/rubyn_code/permissions/tier.rb
 - lib/rubyn_code/protocols/RUBYN.md
 - lib/rubyn_code/protocols/interrupt_handler.rb
-- lib/rubyn_code/protocols/plan_approval.rb
 - lib/rubyn_code/protocols/shutdown_handshake.rb
 - lib/rubyn_code/self_test.rb
 - lib/rubyn_code/skills/RUBYN.md
@@ -430,6 +430,7 @@ files:
 - lib/rubyn_code/tools/bash.rb
 - lib/rubyn_code/tools/bundle_add.rb
 - lib/rubyn_code/tools/bundle_install.rb
+- lib/rubyn_code/tools/code_graph.rb
 - lib/rubyn_code/tools/compact.rb
 - lib/rubyn_code/tools/db_migrate.rb
 - lib/rubyn_code/tools/edit_file.rb
@@ -447,6 +448,7 @@ files:
 - lib/rubyn_code/tools/memory_search.rb
 - lib/rubyn_code/tools/memory_write.rb
 - lib/rubyn_code/tools/output_compressor.rb
+- lib/rubyn_code/tools/phone_a_friend.rb
 - lib/rubyn_code/tools/rails_generate.rb
 - lib/rubyn_code/tools/read_file.rb
 - lib/rubyn_code/tools/read_inbox.rb
@@ -459,6 +461,8 @@ files:
 - lib/rubyn_code/tools/spawn_teammate.rb
 - lib/rubyn_code/tools/spec_output_parser.rb
 - lib/rubyn_code/tools/task.rb
+- lib/rubyn_code/tools/todo_store.rb
+- lib/rubyn_code/tools/todo_write.rb
 - lib/rubyn_code/tools/web_fetch.rb
 - lib/rubyn_code/tools/web_search.rb
 - lib/rubyn_code/tools/write_file.rb

data/lib/rubyn_code/context/context_budget.rb

@@ -1,183 +0,0 @@
-# frozen_string_literal: true
-
-module RubynCode
-  module Context
-    # Budget-aware context loader that prioritizes which related files
-    # to load fully vs. as signatures-only. Prevents context bloat by
-    # capping auto-loaded context at a configurable token budget.
-    class ContextBudget
-      CHARS_PER_TOKEN = 4
-      DEFAULT_BUDGET = 4000 # tokens
-
-      # Rails convention-based priority for related files.
-      # Lower number = higher priority = loaded first.
-      PRIORITY_MAP = {
-        'spec' => 1, # tests for the file
-        'factory' => 2,  # FactoryBot factories
-        'service' => 3,  # service objects
-        'model' => 4, # related models
-        'controller' => 5,  # controllers
-        'serializer' => 6,  # serializers
-        'concern' => 7, # concerns/mixins
-        'helper' => 8, # helpers
-        'migration' => 9 # migrations
-      }.freeze
-
-      attr_reader :loaded_files, :signature_files, :tokens_used
-
-      def initialize(budget: DEFAULT_BUDGET, codebase_index: nil)
-        @budget = budget
-        @codebase_index = codebase_index
-        @loaded_files = []
-        @signature_files = []
-        @tokens_used = 0
-      end
-
-      # Load context for a primary file, filling budget with related files.
-      # Returns array of { file:, content:, mode: :full|:signatures }
-      #
-      # When a codebase_index is available and no related_files are supplied,
-      # uses impact_analysis to auto-discover related files (specs,
-      # associated models, controllers, etc.).
-      def load_for(file_path, related_files: [])
-        results = []
-
-        # Primary file always loads fully
-        primary_content = safe_read(file_path)
-        return results unless primary_content
-
-        primary_tokens = estimate_tokens(primary_content)
-        @tokens_used = primary_tokens
-        @loaded_files << file_path
-        results << { file: file_path, content: primary_content, mode: :full }
-
-        # Auto-discover related files from the index when none supplied
-        related_files = discover_related_files(file_path) if related_files.empty? && @codebase_index
-
-        # Sort related files by priority and fill remaining budget
-        sorted = prioritize(related_files)
-        remaining = @budget - @tokens_used
-        remaining = load_full_files(sorted, results, remaining)
-        load_signature_files(sorted, results, remaining)
-
-        results
-      end
-
-      # Extract method signatures and class structure without method bodies.
-      # Much more compact than full source — typically 10-20% of original size.
-      def extract_signatures(content)
-        signatures = []
-        indent_stack = []
-
-        content.lines.each do |line|
-          process_signature_line(line, signatures, indent_stack)
-        end
-
-        signatures.join
-      end
-
-      # Returns budget utilization stats.
-      def stats
-        {
-          budget: @budget,
-          tokens_used: @tokens_used,
-          utilization: @budget.positive? ? (@tokens_used.to_f / @budget).round(3) : 0.0,
-          full_files: @loaded_files.size,
-          signature_files: @signature_files.size
-        }
-      end
-
-      private
-
-      def discover_related_files(file_path)
-        analysis = @codebase_index.impact_analysis(file_path)
-        analysis[:affected_files].reject { |f| f == file_path }
-      rescue StandardError
-        []
-      end
-
-      def load_full_files(sorted, results, remaining)
-        sorted.each do |rel_path|
-          content = safe_read(rel_path)
-          next unless content
-
-          size = estimate_tokens(content)
-          next unless size <= remaining
-
-          results << { file: rel_path, content: content, mode: :full }
-          @loaded_files << rel_path
-          @tokens_used += size
-          remaining -= size
-        end
-        remaining
-      end
-
-      def load_signature_files(sorted, results, remaining)
-        sorted.each do |rel_path|
-          next if @loaded_files.include?(rel_path)
-
-          content = safe_read(rel_path)
-          next unless content
-
-          sigs = extract_signatures(content)
-          sig_size = estimate_tokens(sigs)
-          next unless sig_size <= remaining
-
-          results << { file: rel_path, content: sigs, mode: :signatures }
-          @signature_files << rel_path
-          @tokens_used += sig_size
-          remaining -= sig_size
-        end
-      end
-
-      # -- signature extraction dispatch
-      def process_signature_line(line, signatures, indent_stack)
-        stripped = line.strip
-        if signature_line?(stripped)
-          signatures << line
-          indent_stack << current_indent(line) if block_opener?(stripped)
-        elsif stripped == 'end' && indent_stack.any? && current_indent(line) <= indent_stack.last
-          signatures << line
-          indent_stack.pop
-        elsif class_or_module_line?(stripped)
-          signatures << line
-          indent_stack << current_indent(line)
-        end
-      end
-
-      def prioritize(files)
-        files.sort_by do |path|
-          basename = File.basename(path, '.*').downcase
-          priority = PRIORITY_MAP.find { |key, _| basename.include?(key) }&.last || 10
-          priority
-        end
-      end
-
-      def signature_line?(stripped)
-        stripped.match?(/\A\s*(def\s|attr_|include\s|extend\s|has_|belongs_|validates|scope\s|delegate\s)/)
-      end
-
-      def class_or_module_line?(stripped)
-        stripped.match?(/\A\s*(class|module)\s/)
-      end
-
-      def block_opener?(stripped)
-        stripped.match?(/\Adef\s/)
-      end
-
-      def current_indent(line)
-        line.match(/\A(\s*)/)[1].length
-      end
-
-      def safe_read(path)
-        File.read(path)
-      rescue StandardError
-        nil
-      end
-
-      def estimate_tokens(text)
-        (text.bytesize.to_f / CHARS_PER_TOKEN).ceil
-      end
-    end
-  end
-end