aidp 0.9.5 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 12e70b298f80b189c33c7db0a67027086a8441a4fdd5c0ae45045b491c5ca825
-  data.tar.gz: 1f6445fd7ed137c78e4c6000aac0418d15d00856beea9db9afd4d843aad11b4a
+  metadata.gz: 2358e7598866519630169447a6ba6fd60704571ca81afc8b6e9ec856916cacc1
+  data.tar.gz: a33aba24daef5e8492d36954b4a36c44fff738dc9b9a54762c81ac28708f2cfa
 SHA512:
-  metadata.gz: 5fab5f97f7874314f0991cd75260175b7aab803d199bb25ab1223fae0d280a7b6a97a11ff160d90b5bc263fc312bc64bb3c95c32e34e31d960c8ab7a403a8778
-  data.tar.gz: 9bf30d28286d09e7d25bb23bd3109e422e91518f7e46113949ed27a87cc722046dc8cfd881f00279c6f849173a6a9f4c503d07b519d7cd7ca53d7cd53a71cf67
+  metadata.gz: c37341e9a7a26a5bb23141133f08a67f26f5a88dcf1d5e12b36f638359f6e738cb4ffd86e15b0d8cc1c8282ecea31803faa6b0ff82c55bb1d9b63aaf987cc82b
+  data.tar.gz: 8b78de9305399181898a47542880840487ab667ccb7e151c4ff52d497437a5fd780440844c3b2585d6356c91c25378c7e73b15eebb1fa8f5e321bb46d4a79773

data/lib/aidp/analyze/error_handler.rb

@@ -8,7 +8,8 @@ module Aidp
     class ErrorHandler
       attr_reader :logger, :error_counts, :recovery_strategies
 
-      def initialize(log_file: nil, verbose: false)
+      def initialize(log_file: nil, verbose: false, output: nil)
+        @output = output
         @logger = setup_logger(log_file, verbose)
         @error_counts = Hash.new(0)
         @recovery_strategies = setup_recovery_strategies
@@ -87,7 +88,8 @@ module Aidp
       private
 
       def setup_logger(log_file, verbose)
-        logger = Logger.new(log_file || $stdout)
+        output_stream = log_file || @output || $stdout
+        logger = Logger.new(output_stream)
         logger.level = verbose ? Logger::DEBUG : Logger::INFO
         logger.formatter = proc do |severity, datetime, progname, msg|
           "#{datetime.strftime("%Y-%m-%d %H:%M:%S")} [#{severity}] #{msg}\n"

data/lib/aidp/{analysis → analyze}/kb_inspector.rb

@@ -2,15 +2,32 @@
 
 require "json"
 require "tty-box"
+require "tty-prompt"
 
 module Aidp
-  module Analysis
+  module Analyze
     class KBInspector
-      def initialize(kb_dir = ".aidp/kb")
+      def initialize(kb_dir = ".aidp/kb", prompt: TTY::Prompt.new)
         @kb_dir = File.expand_path(kb_dir)
+        @prompt = prompt
         @data = load_kb_data
       end
 
+      # Helper method for consistent message display using TTY::Prompt
+      def display_message(message, type: :info)
+        color = case type
+        when :error then :red
+        when :success then :green
+        when :warning then :yellow
+        when :info then :blue
+        when :highlight then :cyan
+        when :muted then :bright_black
+        else :white
+        end
+
+        @prompt.say(message, color: color)
+      end
+
       def show(type, format: "summary")
         case type
         when "seams"
@@ -28,8 +45,8 @@ module Aidp
         when "summary"
           show_summary(format)
         else
-          puts "Unknown KB type: #{type}"
-          puts "Available types: seams, hotspots, cycles, apis, symbols, imports, summary"
+          display_message("Unknown KB type: #{type}", type: :error)
+          display_message("Available types: seams, hotspots, cycles, apis, symbols, imports, summary", type: :info)
         end
       end
 
@@ -42,8 +59,8 @@ module Aidp
         when "cycles"
           generate_cycle_graph(format, output)
         else
-          puts "Unknown graph type: #{type}"
-          puts "Available types: imports, calls, cycles"
+          display_message("Unknown graph type: #{type}", type: :error)
+          display_message("Available types: imports, calls, cycles", type: :info)
         end
       end
 
@@ -73,7 +90,7 @@ module Aidp
           border: :thick,
           padding: [1, 2]
         )
-        puts box
+        display_message(box)
       end
 
       def load_kb_data
@@ -87,7 +104,7 @@ module Aidp
             rescue JSON::ParserError => e
               # Suppress warnings in test mode to avoid CI failures
               unless ENV["RACK_ENV"] == "test" || defined?(RSpec)
-                puts "Warning: Could not parse #{file_path}: #{e.message}"
+                display_message("Warning: Could not parse #{file_path}: #{e.message}", type: :warn)
               end
               data[type.to_sym] = []
             end
@@ -100,42 +117,42 @@ module Aidp
       end
 
       def show_summary(_format)
-        puts "\n📊 Knowledge Base Summary"
-        puts "=" * 50
-
-        puts "📁 KB Directory: #{@kb_dir}"
-        puts "📄 Files analyzed: #{count_files}"
-        puts "🏗️  Symbols: #{@data[:symbols]&.length || 0}"
-        puts "📦 Imports: #{@data[:imports]&.length || 0}"
-        puts "🔗 Calls: #{@data[:calls]&.length || 0}"
-        puts "📏 Metrics: #{@data[:metrics]&.length || 0}"
-        puts "🔧 Seams: #{@data[:seams]&.length || 0}"
-        puts "🔥 Hotspots: #{@data[:hotspots]&.length || 0}"
-        puts "🧪 Tests: #{@data[:tests]&.length || 0}"
-        puts "🔄 Cycles: #{@data[:cycles]&.length || 0}"
+        display_message("\n📊 Knowledge Base Summary", type: :highlight)
+        display_message("=" * 50, type: :info)
+
+        display_message("📁 KB Directory: #{@kb_dir}", type: :info)
+        display_message("📄 Files analyzed: #{count_files}", type: :info)
+        display_message("🏗️  Symbols: #{@data[:symbols]&.length || 0}", type: :info)
+        display_message("📦 Imports: #{@data[:imports]&.length || 0}", type: :info)
+        display_message("🔗 Calls: #{@data[:calls]&.length || 0}", type: :info)
+        display_message("📏 Metrics: #{@data[:metrics]&.length || 0}", type: :info)
+        display_message("🔧 Seams: #{@data[:seams]&.length || 0}", type: :info)
+        display_message("🔥 Hotspots: #{@data[:hotspots]&.length || 0}", type: :info)
+        display_message("🧪 Tests: #{@data[:tests]&.length || 0}", type: :info)
+        display_message("🔄 Cycles: #{@data[:cycles]&.length || 0}", type: :info)
 
         if @data[:seams]&.any?
-          puts "\n🔧 Seam Types:"
+          display_message("\n🔧 Seam Types:", type: :info)
           seam_types = @data[:seams].group_by { |s| s[:kind] }
           seam_types.each do |type, seams|
-            puts "  #{type}: #{seams.length}"
+            display_message("  #{type}: #{seams.length}", type: :info)
           end
         end
 
         if @data[:hotspots]&.any?
-          puts "\n🔥 Top 5 Hotspots:"
+          display_message("\n🔥 Top 5 Hotspots:", type: :info)
           @data[:hotspots].first(5).each_with_index do |hotspot, i|
-            puts "  #{i + 1}. #{hotspot[:file]}:#{hotspot[:method]} (score: #{hotspot[:score]})"
+            display_message("  #{i + 1}. #{hotspot[:file]}:#{hotspot[:method]} (score: #{hotspot[:score]})", type: :info)
           end
         end
       end
 
       def show_seams(format)
-        return puts "No seams data available" unless @data[:seams]&.any?
+        return display_message("No seams data available") unless @data[:seams]&.any?
 
         case format
         when "json"
-          puts JSON.pretty_generate(@data[:seams])
+          display_message(JSON.pretty_generate(@data[:seams]))
         when "table"
           show_seams_table
         else
@@ -144,8 +161,8 @@ module Aidp
       end
 
       def show_seams_table
-        puts "\n🔧 Seams Analysis"
-        puts "=" * 80
+        display_message("\n🔧 Seams Analysis")
+        display_message("=" * 80)
 
         create_table(
           ["Type", "File", "Line", "Symbol", "Suggestion"],
@@ -162,34 +179,34 @@ module Aidp
       end
 
       def show_seams_summary
-        puts "\n🔧 Seams Analysis"
-        puts "=" * 50
+        display_message("\n🔧 Seams Analysis")
+        display_message("=" * 50)
 
         seam_types = @data[:seams].group_by { |s| s[:kind] }
 
         seam_types.each do |type, seams|
-          puts "\n📌 #{type.upcase} (#{seams.length} found)"
-          puts "-" * 30
+          display_message("\n📌 #{type.upcase} (#{seams.length} found)")
+          display_message("-" * 30)
 
           seams.first(10).each do |seam|
-            puts "  #{seam[:file]}:#{seam[:line]}"
-            puts "    Symbol: #{seam[:symbol_id]&.split(":")&.last}"
-            puts "    Suggestion: #{seam[:suggestion]}"
-            puts
+            display_message("  #{seam[:file]}:#{seam[:line]}")
+            display_message("    Symbol: #{seam[:symbol_id]&.split(":")&.last}")
+            display_message("    Suggestion: #{seam[:suggestion]}")
+            display_message("")
           end
 
           if seams.length > 10
-            puts "  ... and #{seams.length - 10} more"
+            display_message("  ... and #{seams.length - 10} more")
           end
         end
       end
 
       def show_hotspots(format)
-        return puts "No hotspots data available" unless @data[:hotspots]&.any?
+        return display_message("No hotspots data available") unless @data[:hotspots]&.any?
 
         case format
         when "json"
-          puts JSON.pretty_generate(@data[:hotspots])
+          display_message(JSON.pretty_generate(@data[:hotspots]))
         when "table"
           show_hotspots_table
         else
@@ -198,8 +215,8 @@ module Aidp
       end
 
       def show_hotspots_table
-        puts "\n🔥 Code Hotspots"
-        puts "=" * 80
+        display_message("\n🔥 Code Hotspots")
+        display_message("=" * 80)
 
         create_table(
           ["Rank", "File", "Method", "Score", "Complexity", "Touches"],
@@ -217,81 +234,81 @@ module Aidp
       end
 
       def show_hotspots_summary
-        puts "\n🔥 Code Hotspots (Top 20)"
-        puts "=" * 50
+        display_message("\n🔥 Code Hotspots (Top 20)")
+        display_message("=" * 50)
 
         @data[:hotspots].each_with_index do |hotspot, i|
-          puts "#{i + 1}. #{hotspot[:file]}:#{hotspot[:method]}"
-          puts "   Score: #{hotspot[:score]} (Complexity: #{hotspot[:complexity]}, Touches: #{hotspot[:touches]})"
-          puts
+          display_message("#{i + 1}. #{hotspot[:file]}:#{hotspot[:method]}")
+          display_message("   Score: #{hotspot[:score]} (Complexity: #{hotspot[:complexity]}, Touches: #{hotspot[:touches]})")
+          display_message("")
         end
       end
 
       def show_cycles(format)
-        return puts "No cycles data available" unless @data[:cycles]&.any?
+        return display_message("No cycles data available") unless @data[:cycles]&.any?
 
         case format
         when "json"
-          puts JSON.pretty_generate(@data[:cycles])
+          display_message(JSON.pretty_generate(@data[:cycles]))
         else
           show_cycles_summary
         end
       end
 
       def show_cycles_summary
-        puts "\n🔄 Import Cycles"
-        puts "=" * 50
+        display_message("\n🔄 Import Cycles")
+        display_message("=" * 50)
 
         @data[:cycles].each_with_index do |cycle, i|
-          puts "Cycle #{i + 1}:"
+          display_message("Cycle #{i + 1}:")
           cycle[:members].each do |member|
-            puts "  - #{member}"
+            display_message("  - #{member}")
           end
-          puts "  Weight: #{cycle[:weight]}" if cycle[:weight]
-          puts
+          display_message("  Weight: #{cycle[:weight]}") if cycle[:weight]
+          display_message("")
         end
       end
 
       def show_apis(format)
-        return puts "No APIs data available" unless @data[:tests]&.any?
+        return display_message("No APIs data available") unless @data[:tests]&.any?
 
         untested_apis = @data[:tests].select { |t| t[:tests].empty? }
 
         case format
         when "json"
-          puts JSON.pretty_generate(untested_apis)
+          display_message(JSON.pretty_generate(untested_apis))
         else
           show_apis_summary(untested_apis)
         end
       end
 
       def show_apis_summary(untested_apis)
-        puts "\n🧪 Untested Public APIs"
-        puts "=" * 50
+        display_message("\n🧪 Untested Public APIs")
+        display_message("=" * 50)
 
         if untested_apis.empty?
-          puts "✅ All public APIs have associated tests!"
+          display_message("✅ All public APIs have associated tests!")
         else
-          puts "Found #{untested_apis.length} untested public APIs:"
-          puts
+          display_message("Found #{untested_apis.length} untested public APIs:")
+          display_message("")
 
           untested_apis.each do |api|
             symbol = @data[:symbols]&.find { |s| s[:id] == api[:symbol_id] }
             if symbol
-              puts "  #{symbol[:file]}:#{symbol[:line]} - #{symbol[:name]}"
-              puts "    Suggestion: Create characterization tests"
-              puts
+              display_message("  #{symbol[:file]}:#{symbol[:line]} - #{symbol[:name]}")
+              display_message("    Suggestion: Create characterization tests")
+              display_message("")
             end
           end
         end
       end
 
       def show_symbols(format)
-        return puts "No symbols data available" unless @data[:symbols]&.any?
+        return display_message("No symbols data available") unless @data[:symbols]&.any?
 
         case format
         when "json"
-          puts JSON.pretty_generate(@data[:symbols])
+          display_message(JSON.pretty_generate(@data[:symbols]))
         when "table"
           show_symbols_table
         else
@@ -300,8 +317,8 @@ module Aidp
       end
 
       def show_symbols_table
-        puts "\n🏗️  Symbols"
-        puts "=" * 80
+        display_message("\n🏗️  Symbols")
+        display_message("=" * 80)
 
         create_table(
           ["Type", "Name", "File", "Line", "Visibility"],
@@ -318,22 +335,22 @@ module Aidp
       end
 
       def show_symbols_summary
-        puts "\n🏗️  Symbols Summary"
-        puts "=" * 50
+        display_message("\n🏗️  Symbols Summary")
+        display_message("=" * 50)
 
         symbol_types = @data[:symbols].group_by { |s| s[:kind] }
 
         symbol_types.each do |type, symbols|
-          puts "#{type.capitalize}: #{symbols.length}"
+          display_message("#{type.capitalize}: #{symbols.length}")
         end
       end
 
       def show_imports(format)
-        return puts "No imports data available" unless @data[:imports]&.any?
+        return display_message("No imports data available") unless @data[:imports]&.any?
 
         case format
         when "json"
-          puts JSON.pretty_generate(@data[:imports])
+          display_message(JSON.pretty_generate(@data[:imports]))
         when "table"
           show_imports_table
         else
@@ -342,8 +359,8 @@ module Aidp
       end
 
       def show_imports_table
-        puts "\n📦 Imports"
-        puts "=" * 80
+        display_message("\n📦 Imports")
+        display_message("=" * 80)
 
         create_table(
           ["Type", "Target", "File", "Line"],
@@ -359,18 +376,18 @@ module Aidp
       end
 
       def show_imports_summary
-        puts "\n📦 Imports Summary"
-        puts "=" * 50
+        display_message("\n📦 Imports Summary")
+        display_message("=" * 50)
 
         import_types = @data[:imports].group_by { |i| i[:kind] }
 
         import_types.each do |type, imports|
-          puts "#{type.capitalize}: #{imports.length}"
+          display_message("#{type.capitalize}: #{imports.length}")
         end
       end
 
       def generate_import_graph(format, output)
-        puts "Generating import graph in #{format} format..."
+        display_message("Generating import graph in #{format} format...")
 
         case format
         when "dot"
@@ -380,7 +397,7 @@ module Aidp
         when "json"
           generate_json_graph(output)
         else
-          puts "Unsupported graph format: #{format}"
+          display_message("Unsupported graph format: #{format}")
         end
       end
 
@@ -399,9 +416,9 @@ module Aidp
 
         if output
           File.write(output, content.join("\n"))
-          puts "Graph written to #{output}"
+          display_message("Graph written to #{output}")
         else
-          puts content.join("\n")
+          display_message(content.join("\n"))
         end
       end
 
@@ -416,9 +433,9 @@ module Aidp
 
         if output
           File.write(output, content.join("\n"))
-          puts "Graph written to #{output}"
+          display_message("Graph written to #{output}")
         else
-          puts content.join("\n")
+          display_message(content.join("\n"))
         end
       end
 
@@ -447,20 +464,20 @@ module Aidp
 
         if output
           File.write(output, JSON.pretty_generate(graph_data))
-          puts "Graph written to #{output}"
+          display_message("Graph written to #{output}")
         else
-          puts JSON.pretty_generate(graph_data)
+          display_message(JSON.pretty_generate(graph_data))
         end
       end
 
       def generate_call_graph(_format, _output)
         # Similar to import graph but for method calls
-        puts "Call graph generation not yet implemented"
+        display_message("Call graph generation not yet implemented")
       end
 
       def generate_cycle_graph(_format, _output)
         # Generate graph showing only the cycles
-        puts "Cycle graph generation not yet implemented"
+        display_message("Cycle graph generation not yet implemented")
       end
 
       def count_files

data/lib/aidp/analyze/prioritizer.rb

@@ -1,14 +1,15 @@
 # frozen_string_literal: true
 
+require "tty-prompt"
 require_relative "ruby_maat_integration"
 require_relative "feature_analyzer"
 
 module Aidp
   module Analyze
     class Prioritizer
-      def initialize(project_dir = Dir.pwd)
+      def initialize(project_dir = Dir.pwd, prompt: TTY::Prompt.new)
         @project_dir = project_dir
-        @code_maat = Aidp::Analyze::RubyMaatIntegration.new(project_dir)
+        @code_maat = Aidp::Analyze::RubyMaatIntegration.new(project_dir, prompt: prompt)
         @feature_analyzer = Aidp::Analyze::FeatureAnalyzer.new(project_dir)
       end
 

data/lib/aidp/analyze/ruby_maat_integration.rb

@@ -1,14 +1,16 @@
 # frozen_string_literal: true
 
 require "tty-command"
+require "tty-prompt"
 require "json"
 require "fileutils"
 
 module Aidp
   module Analyze
     class RubyMaatIntegration
-      def initialize(project_dir = Dir.pwd)
+      def initialize(project_dir = Dir.pwd, prompt: TTY::Prompt.new)
         @project_dir = project_dir
+        @prompt = prompt
       end
 
       # Check if RubyMaat gem is available and accessible
@@ -80,7 +82,7 @@ module Aidp
 
       # Run analysis on large repositories using chunking
       def run_chunked_analysis(git_log_file)
-        puts "Large repository detected. Running chunked analysis..."
+        display_message("Large repository detected. Running chunked analysis...", type: :info)
 
         # Split analysis into chunks
         chunks = create_analysis_chunks(git_log_file)
@@ -93,7 +95,7 @@ module Aidp
         }
 
         chunks.each_with_index do |chunk, index|
-          puts "Processing chunk #{index + 1}/#{chunks.length}..."
+          display_message("Processing chunk #{index + 1}/#{chunks.length}...", type: :info)
 
           chunk_results = analyze_chunk(chunk)
 
@@ -453,6 +455,21 @@ module Aidp
 
         result.success? && !result.out.strip.empty?
       end
+
+      private
+
+      # Helper method for consistent message display using TTY::Prompt
+      def display_message(message, type: :info)
+        color = case type
+        when :error then :red
+        when :warn then :yellow
+        when :success then :green
+        when :highlight then :cyan
+        else :white
+        end
+
+        @prompt.say(message, color: color)
+      end
     end
   end
 end

data/lib/aidp/analyze/runner.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require "tty-prompt"
 require_relative "steps"
 require_relative "progress"
 require_relative "../storage/file_manager"
@@ -10,17 +11,34 @@ module Aidp
     class Runner
       include Aidp::DebugMixin
 
-      def initialize(project_dir, harness_runner = nil)
+      def initialize(project_dir, harness_runner = nil, prompt: TTY::Prompt.new)
         @project_dir = project_dir
         @harness_runner = harness_runner
         @is_harness_mode = !harness_runner.nil?
         @file_manager = Aidp::Storage::FileManager.new(File.join(project_dir, ".aidp"))
+        @prompt = prompt
       end
 
       def progress
         @progress ||= Aidp::Analyze::Progress.new(@project_dir)
       end
 
+      private
+
+      def display_message(message, type: :info)
+        color = case type
+        when :error then :red
+        when :success then :green
+        when :warning then :yellow
+        when :info then :blue
+        when :highlight then :cyan
+        else :white
+        end
+        @prompt.say(message, color: color)
+      end
+
+      public
+
       def run_step(step_name, options = {})
         # Always validate step exists first
         step_spec = Aidp::Analyze::Steps::SPEC[step_name]
@@ -42,7 +60,7 @@ module Aidp
       # Harness-aware step execution
       def run_step_with_harness(step_name, options = {})
         # Get current provider from harness
-        current_provider = @harness_runner.instance_variable_get(:@current_provider)
+        current_provider = @harness_runner.current_provider
         provider_type = current_provider || "cursor"
 
         debug_step(step_name, "Harness execution", {
@@ -72,7 +90,7 @@ module Aidp
 
       # Standalone step execution (simplified - synchronous)
       def run_step_standalone(step_name, options = {})
-        puts "🚀 Running step synchronously: #{step_name}"
+        display_message("🚀 Running step synchronously: #{step_name}", type: :info)
 
         start_time = Time.now
         prompt = composed_prompt(step_name, options)
@@ -85,7 +103,7 @@ module Aidp
         # Store execution metrics
         @file_manager.record_step_execution(step_name, "cursor", duration, result[:status] == "completed")
 
-        puts "✅ Step completed in #{duration.round(2)}s"
+        display_message("✅ Step completed in #{duration.round(2)}s", type: :success)
         result
       end
 
@@ -223,11 +241,11 @@ module Aidp
         # Add current execution context
         context_parts << "## Analysis Context"
         context_parts << "Project Directory: #{@project_dir}"
-        context_parts << "Current Step: #{@harness_runner.instance_variable_get(:@current_step)}"
-        context_parts << "Current Provider: #{@harness_runner.instance_variable_get(:@current_provider)}"
+        context_parts << "Current Step: #{@harness_runner.current_step}"
+        context_parts << "Current Provider: #{@harness_runner.current_provider}"
 
         # Add user input context
-        user_input = @harness_runner.instance_variable_get(:@user_input)
+        user_input = @harness_runner.user_input
         if user_input && !user_input.empty?
           context_parts << "\n## Previous User Input"
           user_input.each do |key, value|
@@ -236,7 +254,7 @@ module Aidp
         end
 
         # Add execution history context
-        execution_log = @harness_runner.instance_variable_get(:@execution_log)
+        execution_log = @harness_runner.execution_log
         if execution_log && !execution_log.empty?
           context_parts << "\n## Analysis History"
           recent_logs = execution_log.last(5) # Last 5 entries
@@ -251,7 +269,7 @@ module Aidp
       # Execute step with harness provider management
       def execute_with_harness_provider(provider_type, prompt, step_name, _options)
         # Get provider manager from harness
-        provider_manager = @harness_runner.instance_variable_get(:@provider_manager)
+        provider_manager = @harness_runner.provider_manager
 
         # Execute with provider
         provider_manager.execute_with_provider(provider_type, prompt, {

data/lib/aidp/{analysis → analyze}/seams.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Aidp
-  module Analysis
+  module Analyze
     module Seams
       # I/O and OS integration patterns
       IO_PATTERNS = [