aidp 0.9.5 → 0.10.0

data/lib/aidp/analyze/steps.rb

@@ -5,37 +5,37 @@ module Aidp
     module Steps
       SPEC = {
         "01_REPOSITORY_ANALYSIS" => {
-          "templates" => ["01_repository_analysis.md"],
+          "templates" => ["01_REPOSITORY_ANALYSIS.md"],
           "description" => "Initial code-maat based repository mining",
           "outs" => ["docs/analysis/repository_analysis.md"],
           "gate" => false
         },
         "02_ARCHITECTURE_ANALYSIS" => {
-          "templates" => ["02_architecture_analysis.md"],
+          "templates" => ["02_ARCHITECTURE_ANALYSIS.md"],
           "description" => "Identify architectural patterns, dependencies, and violations",
           "outs" => ["docs/analysis/architecture_analysis.md"],
           "gate" => true
         },
         "03_TEST_ANALYSIS" => {
-          "templates" => ["03_test_analysis.md"],
+          "templates" => ["03_TEST_ANALYSIS.md"],
           "description" => "Analyze existing test coverage and identify gaps",
           "outs" => ["docs/analysis/test_analysis.md"],
           "gate" => false
         },
         "04_FUNCTIONALITY_ANALYSIS" => {
-          "templates" => ["04_functionality_analysis.md"],
+          "templates" => ["04_FUNCTIONALITY_ANALYSIS.md"],
           "description" => "Map features, identify dead code, analyze complexity",
           "outs" => ["docs/analysis/functionality_analysis.md"],
           "gate" => false
         },
         "05_DOCUMENTATION_ANALYSIS" => {
-          "templates" => ["05_documentation_analysis.md"],
+          "templates" => ["05_DOCUMENTATION_ANALYSIS.md"],
           "description" => "Identify missing documentation and generate what's needed",
           "outs" => ["docs/analysis/documentation_analysis.md"],
           "gate" => false
         },
         "06_STATIC_ANALYSIS" => {
-          "templates" => ["06_static_analysis.md"],
+          "templates" => ["06_STATIC_ANALYSIS.md"],
           "description" => "Check for existing tools and recommend improvements",
           "outs" => ["docs/analysis/static_analysis.md"],
           "gate" => false
@@ -47,7 +47,7 @@ module Aidp
           "gate" => false
         },
         "07_REFACTORING_RECOMMENDATIONS" => {
-          "templates" => ["07_refactoring_recommendations.md"],
+          "templates" => ["07_REFACTORING_RECOMMENDATIONS.md"],
           "description" => "Provide actionable refactoring guidance",
           "outs" => ["docs/analysis/refactoring_recommendations.md"],
           "gate" => true

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

@@ -2,10 +2,11 @@
 
 require "pathname"
 require "tree_sitter"
+require "tty-prompt"
 require "fileutils"
 
 module Aidp
-  module Analysis
+  module Analyze
     class TreeSitterGrammarLoader
       # Default grammar configurations
       GRAMMAR_CONFIGS = {
@@ -61,10 +62,11 @@ module Aidp
         }
       }.freeze
 
-      def initialize(project_dir = Dir.pwd)
+      def initialize(project_dir = Dir.pwd, prompt: TTY::Prompt.new)
         @project_dir = project_dir
         @grammars_dir = File.join(project_dir, ".aidp", "grammars")
         @loaded_grammars = {}
+        @prompt = prompt
       end
 
       # Load grammar for a specific language
@@ -92,7 +94,7 @@ module Aidp
         grammar_path = File.join(@grammars_dir, language)
 
         unless File.exist?(grammar_path)
-          puts "Installing Tree-sitter grammar for #{language}..."
+          display_message("Installing Tree-sitter grammar for #{language}...", type: :info)
           install_grammar(language, config)
         end
       end
@@ -110,7 +112,7 @@ module Aidp
         require "json"
         File.write(File.join(grammar_path, "grammar.json"), JSON.generate(config))
 
-        puts "Grammar for #{language} marked as available"
+        display_message("Grammar for #{language} marked as available", type: :success)
       end
 
       def create_parser(language, config)
@@ -140,7 +142,7 @@ module Aidp
           real: true
         }
       rescue TreeSitter::ParserNotFoundError => e
-        puts "Warning: Tree-sitter parser not found for #{language}: #{e.message}"
+        display_message("Warning: Tree-sitter parser not found for #{language}: #{e.message}", type: :warn)
         create_mock_parser(language)
       end
 
@@ -476,6 +478,21 @@ module Aidp
 
         nodes
       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/{analysis → analyze}/tree_sitter_scan.rb

@@ -4,19 +4,21 @@ require "json"
 require "fileutils"
 require "digest"
 require "etc"
+require "tty-prompt"
 
 require_relative "tree_sitter_grammar_loader"
 require_relative "seams"
 
 module Aidp
-  module Analysis
+  module Analyze
     class TreeSitterScan
-      def initialize(root: Dir.pwd, kb_dir: ".aidp/kb", langs: %w[ruby], threads: Etc.nprocessors)
+      def initialize(root: Dir.pwd, kb_dir: ".aidp/kb", langs: %w[ruby], threads: Etc.nprocessors, prompt: TTY::Prompt.new)
         @root = File.expand_path(root)
         @kb_dir = File.expand_path(kb_dir, @root)
         @langs = Array(langs)
         @threads = threads
-        @grammar_loader = TreeSitterGrammarLoader.new(@root)
+        @prompt = prompt
+        @grammar_loader = TreeSitterGrammarLoader.new(@root, prompt: @prompt)
 
         # Data structures to accumulate analysis results
         @symbols = []
@@ -34,14 +36,14 @@ module Aidp
       end
 
       def run
-        puts "🔍 Starting Tree-sitter static analysis..."
-        puts "📁 Root: #{@root}"
-        puts "🗂️  KB Directory: #{@kb_dir}"
-        puts "🌐 Languages: #{@langs.join(", ")}"
-        puts "🧵 Threads: #{@threads}"
+        display_message("🔍 Starting Tree-sitter static analysis...", type: :highlight)
+        display_message("📁 Root: #{@root}", type: :info)
+        display_message("🗂️  KB Directory: #{@kb_dir}", type: :info)
+        display_message("🌐 Languages: #{@langs.join(", ")}", type: :info)
+        display_message("🧵 Threads: #{@threads}", type: :info)
 
         files = discover_files
-        puts "📄 Found #{files.length} files to analyze"
+        display_message("📄 Found #{files.length} files to analyze", type: :info)
 
         prepare_kb_dir
         load_cache
@@ -49,8 +51,8 @@ module Aidp
         parallel_parse(files)
         write_kb_files
 
-        puts "✅ Tree-sitter analysis complete!"
-        puts "📊 Generated KB files in #{@kb_dir}"
+        display_message("✅ Tree-sitter analysis complete!", type: :success)
+        display_message("📊 Generated KB files in #{@kb_dir}", type: :success)
       end
 
       private
@@ -144,14 +146,14 @@ module Aidp
       end
 
       def parallel_parse(files)
-        puts "🔄 Parsing files in parallel..."
+        display_message("🔄 Parsing files in parallel...", type: :info)
 
         # Group files by language for efficient processing
         files_by_lang = files.group_by { |file| detect_language(file) }
 
         # Process each language group
         files_by_lang.each do |lang, lang_files|
-          puts "📝 Processing #{lang_files.length} #{lang} files..."
+          display_message("📝 Processing #{lang_files.length} #{lang} files...", type: :info)
 
           # Load grammar for this language
           grammar = @grammar_loader.load_grammar(lang)
@@ -614,7 +616,7 @@ module Aidp
       end
 
       def write_kb_files
-        puts "💾 Writing knowledge base files..."
+        display_message("💾 Writing knowledge base files...", type: :info)
 
         prepare_kb_dir
 
@@ -637,7 +639,7 @@ module Aidp
       def write_json_file(filename, data)
         file_path = File.join(@kb_dir, filename)
         File.write(file_path, JSON.pretty_generate(data))
-        puts "📄 Written #{filename} (#{data.length} entries)"
+        display_message("📄 Written #{filename} (#{data.length} entries)", type: :success)
       end
 
       def generate_hotspots
@@ -681,6 +683,21 @@ module Aidp
         # This would analyze test file naming and content
         []
       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/cli/first_run_wizard.rb

@@ -10,40 +10,53 @@ module Aidp
     class FirstRunWizard
       TEMPLATES_DIR = File.expand_path(File.join(__dir__, "..", "..", "..", "templates"))
 
-      def self.ensure_config(project_dir, input: $stdin, output: $stdout, non_interactive: false, prompt: TTY::Prompt.new)
+      def self.ensure_config(project_dir, non_interactive: false, prompt: TTY::Prompt.new)
         return true if Aidp::Config.config_exists?(project_dir)
 
-        wizard = new(project_dir, input: input, output: output, prompt: prompt)
+        wizard = new(project_dir, prompt: prompt)
 
-        if non_interactive || !input.tty? || !output.tty?
+        if non_interactive
           # Non-interactive environment - create minimal config silently
           path = wizard.send(:write_minimal_config, project_dir)
-          output.puts "Created minimal configuration at #{wizard.send(:relative, path)} (non-interactive default)"
+          wizard.send(:display_message, "Created minimal configuration at #{wizard.send(:relative, path)} (non-interactive default)", type: :success)
           return true
         end
 
         wizard.run
       end
 
-      def self.setup_config(project_dir, input: $stdin, output: $stdout, non_interactive: false, prompt: TTY::Prompt.new)
-        wizard = new(project_dir, input: input, output: output, prompt: prompt)
+      def self.setup_config(project_dir, non_interactive: false, prompt: TTY::Prompt.new)
+        wizard = new(project_dir, prompt: prompt)
 
-        if non_interactive || !input.tty? || !output.tty?
+        if non_interactive
           # Non-interactive environment - skip setup
-          output.puts "Configuration setup skipped in non-interactive environment"
+          wizard.send(:display_message, "Configuration setup skipped in non-interactive environment", type: :info)
           return true
         end
 
         wizard.run_setup_config
       end
 
-      def initialize(project_dir, input: $stdin, output: $stdout, prompt: TTY::Prompt.new)
+      def initialize(project_dir, prompt: TTY::Prompt.new)
         @project_dir = project_dir
-        @input = input
-        @output = output
         @prompt = prompt
       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 run
         banner
         loop do
@@ -51,10 +64,10 @@ module Aidp
           case choice
           when "1" then return finish(write_quick_config(@project_dir))
           when "2" then return finish(run_custom)
-          when "q", "Q" then @output.puts("Exiting without creating configuration.")
+          when "q", "Q" then display_message("Exiting without creating configuration.")
                              return false
           else
-            @output.puts "Invalid selection. Please choose one of the listed options."
+            display_message("Invalid selection. Please choose one of the listed options.", type: :warning)
           end
         end
       end
@@ -81,14 +94,14 @@ module Aidp
       private
 
       def banner
-        @output.puts "\n🚀 First-time setup detected"
-        @output.puts "No 'aidp.yml' configuration file found in #{relative(@project_dir)}."
-        @output.puts "Let's create one so you can start using AI Dev Pipeline."
-        @output.puts
+        display_message("\n🚀 First-time setup detected", type: :highlight)
+        display_message("No 'aidp.yml' configuration file found in #{relative(@project_dir)}.")
+        display_message("Let's create one so you can start using AI Dev Pipeline.")
+        display_message("")
       end
 
       def ask_choice
-        @output.puts "Choose a configuration style:" unless @asking
+        display_message("Choose a configuration style:") unless @asking
 
         options = {
           "Quick setup (cursor + macos, no API keys needed)" => "1",
@@ -101,11 +114,11 @@ module Aidp
 
       def finish(path)
         if path
-          @output.puts "\n✅ Configuration created at #{relative(path)}"
-          @output.puts "You can edit this file anytime. Continuing startup...\n"
+          display_message("\n✅ Configuration created at #{relative(path)}", type: :success)
+          display_message("You can edit this file anytime. Continuing startup...\n")
           true
         else
-          @output.puts "❌ Failed to create configuration file."
+          display_message("❌ Failed to create configuration file.", type: :error)
           false
         end
       end
@@ -113,7 +126,7 @@ module Aidp
       def copy_template(filename)
         src = File.join(TEMPLATES_DIR, filename)
         unless File.exist?(src)
-          @output.puts "Template not found: #{filename}"
+          display_message("Template not found: #{filename}", type: :error)
           return nil
         end
         dest = File.join(@project_dir, "aidp.yml")
@@ -310,14 +323,10 @@ module Aidp
 
       def ask(prompt, default: nil)
         if default
-          @output.print "#{prompt} [#{default}]: "
+          @prompt.ask("#{prompt}:", default: default)
         else
-          @output.print "#{prompt}: "
+          @prompt.ask("#{prompt}:")
         end
-        @output.flush
-        ans = @input.gets&.strip
-        return default if (ans.nil? || ans.empty?) && default
-        ans
       end
 
       def relative(path)

data/lib/aidp/cli/jobs_command.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require "tty-prompt"
 require "tty-box"
 require "pastel"
 require "io/console"
@@ -10,8 +11,9 @@ require_relative "../storage/file_manager"
 module Aidp
   class CLI
     class JobsCommand
-      def initialize(input: $stdin, output: $stdout)
-        @io = TerminalIO.new(input, output)
+      def initialize(input: nil, output: nil, prompt: TTY::Prompt.new)
+        @io = TerminalIO.new(input: input, output: output)
+        @prompt = prompt
         @pastel = Pastel.new
         @running = true
         @view_mode = :list
@@ -21,18 +23,35 @@ module Aidp
         @screen_width = 80  # Default screen width
       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
+        when :muted then :bright_black
+        else :white
+        end
+        @prompt.say(message, color: color)
+      end
+
+      public
+
       def run
         # Simple harness jobs display
         jobs = fetch_harness_jobs
 
         if jobs.empty?
-          @io.puts "Harness Jobs"
-          @io.puts "-" * @screen_width
-          @io.puts
-          @io.puts "No harness jobs found"
-          @io.puts
-          @io.puts "Harness jobs are background tasks that run during harness mode."
-          @io.puts "They are stored as JSON files in the .aidp/harness_logs/ directory."
+          display_message("Harness Jobs", type: :info)
+          display_message("-" * @screen_width, type: :muted)
+          display_message("")
+          display_message("No harness jobs found", type: :info)
+          display_message("")
+          display_message("Harness jobs are background tasks that run during harness mode.", type: :info)
+          display_message("They are stored as JSON files in the .aidp/harness_logs/ directory.", type: :info)
         else
           render_harness_jobs(jobs)
         end
@@ -64,7 +83,7 @@ module Aidp
 
           jobs << job_info
         rescue JSON::ParserError => e
-          @io.puts "Warning: Could not parse harness log #{log_file}: #{e.message}" if ENV["AIDP_DEBUG"]
+          display_message("Warning: Could not parse harness log #{log_file}: #{e.message}", type: :warning) if ENV["AIDP_DEBUG"]
         end
 
         # Sort by creation time (newest first)
@@ -91,9 +110,9 @@ module Aidp
 
       # Render harness jobs in a simple table
       def render_harness_jobs(jobs)
-        @io.puts "Harness Jobs"
-        @io.puts "-" * @screen_width
-        @io.puts
+        display_message("Harness Jobs", type: :info)
+        display_message("-" * @screen_width, type: :muted)
+        display_message("")
 
         # Create job content for TTY::Box
         job_content = []
@@ -121,12 +140,12 @@ module Aidp
           border: :thick,
           padding: [1, 2]
         )
-        puts box
+        display_message(box)
 
-        @io.puts
-        @io.puts "Total: #{jobs.length} harness job(s)"
-        @io.puts
-        @io.puts "Note: Harness jobs are stored as JSON files in .aidp/harness_logs/"
+        display_message("")
+        display_message("Total: #{jobs.length} harness job(s)", type: :info)
+        display_message("")
+        display_message("Note: Harness jobs are stored as JSON files in .aidp/harness_logs/", type: :muted)
       end
 
       # Format timestamp for display

data/lib/aidp/cli/terminal_io.rb

@@ -5,9 +5,9 @@ require "stringio"
 module Aidp
   class CLI
     class TerminalIO
-      def initialize(input = $stdin, output = $stdout)
-        @input = input
-        @output = output
+      def initialize(input: nil, output: nil)
+        @input = input || $stdin
+        @output = output || $stdout
       end
 
       def ready?