aidp 0.9.6 → 0.11.0

data/lib/aidp/harness/ui/error_handler.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require "tty-prompt"
 require "pastel"
 
 module Aidp
@@ -8,14 +9,19 @@ module Aidp
       # Centralized error handling for UI components
       class ErrorHandler
         class UIError < StandardError; end
+
         class ComponentError < UIError; end
+
         class ValidationError < UIError; end
+
         class DisplayError < UIError; end
+
         class InteractionError < UIError; end
 
         def initialize(ui_components = {})
           @logger = ui_components[:logger] || default_logger
           @formatter = ui_components[:formatter] || ErrorFormatter.new
+          @prompt = ui_components[:prompt] || TTY::Prompt.new
         end
 
         def handle_error(error, context = {})
@@ -67,8 +73,7 @@ module Aidp
         end
 
         def display_user_friendly_error(message)
-          pastel = Pastel.new
-          puts("#{pastel.red("Error:")} #{message}")
+          @prompt.say("Error: #{message}", color: :red)
         end
 
         def display_generic_error(error)

data/lib/aidp/harness/ui/frame_manager.rb

@@ -9,13 +9,16 @@ module Aidp
       # Handles nested framing using CLI UI frames
       class FrameManager < Base
         class FrameError < StandardError; end
+
         class InvalidFrameError < FrameError; end
+
         class DisplayError < FrameError; end
 
         def initialize(ui_components = {})
           super()
           @frame = ui_components[:frame] || TTY::Box
-          @formatter = ui_components[:formatter] || FrameFormatter.new
+          @formatter = ui_components[:formatter] || (defined?(FrameFormatter) ? FrameFormatter.new : nil)
+          @output = ui_components[:output]
           @frame_open = false
           @frame_stack = []
           @frame_history = []
@@ -26,6 +29,18 @@ module Aidp
           }
         end
 
+        private
+
+        def display_message(message)
+          if @output
+            @output.say(message)
+          else
+            puts message
+          end
+        end
+
+        public
+
         def open_frame(frame_type, title, frame_data = nil, &block)
           validate_frame_type(frame_type)
           validate_title(title)
@@ -43,11 +58,11 @@ module Aidp
             @frame_stats[:status_counts][frame_data[:status]] += 1
           end
 
-          if block_given?
+          if block
             content = yield
-            puts TTY::Box.frame(formatted_title, content, width: 80)
+            display_message(@frame.frame(formatted_title, content, width: 80))
           else
-            puts TTY::Box.frame(formatted_title, width: 80)
+            display_message(@frame.frame(formatted_title, width: 80))
           end
         rescue InvalidFrameError => e
           raise e
@@ -60,11 +75,18 @@ module Aidp
           validate_title(title)
           raise DisplayError, "No parent frame exists for nesting" if @frame_stack.empty?
 
-          formatted_title = @formatter.format_frame_title(frame_type, title, frame_data)
+          formatted_title = @formatter ? @formatter.format_frame_title(frame_type, title, frame_data) : title.to_s
           @frame_stack.push({type: frame_type, title: title, data: frame_data})
-          puts TTY::Box.frame(formatted_title, width: 80) do
-            yield if block_given?
+
+          frame_result = @frame.frame(formatted_title, width: 80) do
+            if block
+              yield || ""
+            else
+              ""
+            end
           end
+
+          display_message(frame_result)
         rescue InvalidFrameError => e
           raise e
         rescue => e
@@ -98,7 +120,7 @@ module Aidp
           when :failed then "Failed"
           else status.to_s.capitalize
           end
-          puts "Status: #{status_text}"
+          display_message("Status: #{status_text}")
         end
 
         def current_frame_status
@@ -125,18 +147,18 @@ module Aidp
         end
 
         def display_frame_summary
-          puts "\n📊 Frame Summary"
-          puts "=" * 50
+          display_message("\n📊 Frame Summary")
+          display_message("=" * 50)
 
           if @frame_stats[:total_frames] == 0
-            puts "No frames used"
+            display_message("No frames used")
             return
           end
 
-          puts "Total Frames: #{@frame_stats[:total_frames]}"
+          display_message("Total Frames: #{@frame_stats[:total_frames]}")
 
           unless @frame_stats[:frame_types].empty?
-            puts "\nFrame Types:"
+            display_message("\nFrame Types:")
             @frame_stats[:frame_types].each do |type, count|
               emoji = case type
               when :section then "📋"
@@ -145,12 +167,12 @@ module Aidp
               when :step then "🔧"
               else "📋"
               end
-              puts "  #{emoji} #{type.to_s.capitalize}: #{count}"
+              display_message("  #{emoji} #{type.to_s.capitalize}: #{count}")
             end
           end
 
           unless @frame_stats[:status_counts].empty?
-            puts "\nStatus Counts:"
+            display_message("\nStatus Counts:")
             @frame_stats[:status_counts].each do |status, count|
               status_emoji = case status
               when :running then "🔄"
@@ -158,12 +180,12 @@ module Aidp
               when :failed then "❌"
               else "❓"
               end
-              puts "  #{status_emoji} #{status.to_s.capitalize}: #{count}"
+              display_message("  #{status_emoji} #{status.to_s.capitalize}: #{count}")
             end
           end
 
-          puts "\nCurrent Frame Depth: #{@frame_depth}"
-          puts "Frames in History: #{@frame_history.length}"
+          display_message("\nCurrent Frame Depth: #{@frame_depth}")
+          display_message("Frames in History: #{@frame_history.length}")
         end
 
         def clear_frame_history
@@ -180,15 +202,15 @@ module Aidp
         def frame_with_block(frame_type, title, frame_data = nil, &block)
           validate_frame_type(frame_type)
           validate_title(title)
-          raise ArgumentError, "Block required for frame_with_block" unless block_given?
+          raise ArgumentError, "Block required for frame_with_block" unless block
 
-          formatted_title = @formatter.format_frame_title(frame_type, title, frame_data)
+          formatted_title = @formatter ? @formatter.format_frame_title(frame_type, title, frame_data) : title.to_s
           @frame_open = true
           @frame_stack.push({type: frame_type, title: title, data: frame_data})
 
           begin
             content = yield
-            puts TTY::Box.frame(formatted_title, content, width: 80)
+            display_message(@frame.frame(formatted_title, content, width: 80))
 
             @frame_open = false
             @frame_stack.pop unless @frame_stack.empty?
@@ -200,14 +222,14 @@ module Aidp
           rescue => e
             @frame_open = false
             @frame_stack.pop unless @frame_stack.empty?
-            raise e  # Re-raise the original exception
+            raise e # Re-raise the original exception
           end
         end
 
         def divider(text)
           validate_text(text)
 
-          formatted_text = @formatter.format_divider_text(text)
+          formatted_text = @formatter ? @formatter.format_divider_text(text) : text
           @frame.divider(formatted_text)
         rescue => e
           raise DisplayError, "Failed to create divider: #{e.message}"
@@ -216,12 +238,12 @@ module Aidp
         def section(title, &block)
           validate_title(title)
 
-          formatted_title = @formatter.format_section_title(title)
-          if block_given?
+          formatted_title = @formatter ? @formatter.format_section_title(title) : "📋 #{title}"
+          if block
             content = yield
-            puts TTY::Box.frame(formatted_title, content, width: 80)
+            display_message(@frame.frame(formatted_title, content, width: 80))
           else
-            puts TTY::Box.frame(formatted_title, width: 80)
+            display_message(@frame.frame(formatted_title, width: 80))
           end
         rescue => e
           raise DisplayError, "Failed to create section: #{e.message}"
@@ -230,10 +252,10 @@ module Aidp
         def subsection(title, &block)
           validate_title(title)
 
-          formatted_title = @formatter.format_subsection_title(title)
-          puts TTY::Box.frame(formatted_title, width: 80) do
-            yield if block_given?
-          end
+          formatted_title = @formatter ? @formatter.format_subsection_title(title) : "📝 #{title}"
+          display_message(@frame.frame(formatted_title, width: 80) do
+            yield if block
+          end)
         rescue => e
           raise DisplayError, "Failed to create subsection: #{e.message}"
         end
@@ -241,10 +263,10 @@ module Aidp
         def workflow_frame(workflow_name, &block)
           validate_workflow_name(workflow_name)
 
-          formatted_title = @formatter.format_workflow_title(workflow_name)
-          puts TTY::Box.frame(formatted_title, width: 80) do
-            yield if block_given?
-          end
+          formatted_title = @formatter ? @formatter.format_workflow_title(workflow_name) : "⚙️ #{workflow_name}"
+          display_message(@frame.frame(formatted_title, width: 80) do
+            yield if block
+          end)
         rescue => e
           raise DisplayError, "Failed to create workflow frame: #{e.message}"
         end
@@ -252,10 +274,10 @@ module Aidp
         def step_frame(step_name, step_number, total_steps, &block)
           validate_step_inputs(step_name, step_number, total_steps)
 
-          formatted_title = @formatter.format_step_title(step_name, step_number, total_steps)
-          puts TTY::Box.frame(formatted_title, width: 80) do
-            yield if block_given?
-          end
+          formatted_title = @formatter ? @formatter.format_step_title(step_name, step_number, total_steps) : "🔧 #{step_name} (#{step_number}/#{total_steps})"
+          display_message(@frame.frame(formatted_title, width: 80) do
+            yield if block
+          end)
         rescue => e
           raise DisplayError, "Failed to create step frame: #{e.message}"
         end

data/lib/aidp/harness/ui/job_monitor.rb

@@ -12,7 +12,9 @@ module Aidp
       # Real-time job monitoring and status tracking
       class JobMonitor < Base
         class JobMonitorError < StandardError; end
+
         class JobNotFoundError < JobMonitorError; end
+
         class MonitorError < JobMonitorError; end
 
         JOB_STATUSES = {

data/lib/aidp/harness/ui/navigation/main_menu.rb

@@ -14,7 +14,9 @@ module Aidp
         # Main hierarchical navigation menu system
         class MainMenu < Base
           class MenuError < StandardError; end
+
           class InvalidMenuError < MenuError; end
+
           class NavigationError < MenuError; end
 
           def initialize(ui_components = {}, prompt: nil)
@@ -23,10 +25,11 @@ module Aidp
             @pastel = Pastel.new
             @formatter = ui_components[:formatter] || MenuFormatter.new
             @state_manager = ui_components[:state_manager] || MenuState.new
+            @output = ui_components[:output]
             @menu_items = []
             @current_level = 0
             @breadcrumb = []
-            @navigation_history = []  # Track all navigation actions
+            @navigation_history = [] # Track all navigation actions
           end
 
           def add_menu_item(item)
@@ -69,9 +72,9 @@ module Aidp
 
             # Add navigation instructions if items exist
             if menu_items.any?
-              puts "Use arrow keys to navigate, Enter to select"
+              display_message "Use arrow keys to navigate, Enter to select"
             else
-              puts "No options available"
+              display_message "No options available"
             end
           rescue => e
             raise NavigationError, "Failed to display menu: #{e.message}"
@@ -112,7 +115,7 @@ module Aidp
             @menu_items = menu_items
             display_menu_items
 
-            max_attempts = 10  # Prevent infinite loops
+            max_attempts = 10 # Prevent infinite loops
             attempts = 0
 
             loop do
@@ -133,7 +136,7 @@ module Aidp
                 if index >= 0 && index < menu_items.length
                   return menu_items[index]
                 else
-                  puts "Invalid selection"
+                  display_message "Invalid selection"
                   next
                 end
               else
@@ -142,14 +145,14 @@ module Aidp
                 if selected_item
                   return selected_item
                 else
-                  puts "Invalid selection"
+                  display_message "Invalid selection"
                   next
                 end
               end
             end
 
             # If we get here, we've exceeded max attempts
-            puts "Too many invalid attempts. Returning first option."
+            display_message "Too many invalid attempts. Returning first option."
             menu_items.first
           end
 
@@ -194,10 +197,10 @@ module Aidp
           end
 
           def display_navigation_help
-            puts "Navigation Help"
-            puts "Use arrow keys to navigate"
-            puts "Press Enter to select"
-            puts "Press Escape to go back"
+            display_message "Navigation Help"
+            display_message "Use arrow keys to navigate"
+            display_message "Press Enter to select"
+            display_message "Press Escape to go back"
           end
 
           # Additional methods expected by tests
@@ -220,15 +223,23 @@ module Aidp
           # Make display_breadcrumb public for testing
           def display_breadcrumb
             if @breadcrumb.empty?
-              puts "Home"
+              display_message "Home"
             else
               breadcrumb_text = @formatter.format_breadcrumb(@breadcrumb)
-              puts(breadcrumb_text)
+              display_message(breadcrumb_text)
             end
           end
 
           private
 
+          def display_message(message)
+            if @output
+              @output.say(message)
+            else
+              puts message
+            end
+          end
+
           def validate_menu_item(item)
             raise InvalidMenuError, "Menu item must be a MenuItem" unless item.is_a?(MenuItem)
           end
@@ -243,14 +254,14 @@ module Aidp
 
           def display_menu_header(title)
             formatted_title = @formatter.format_menu_title(title)
-            puts(formatted_title)
-            puts(@formatter.format_separator)
+            display_message(formatted_title)
+            display_message(@formatter.format_separator)
           end
 
           def display_menu_items
             @menu_items.each_with_index do |item, index|
               formatted_item = @formatter.format_menu_item(item, index + 1)
-              puts(formatted_item)
+              display_message(formatted_item)
             end
           end
 

data/lib/aidp/harness/ui/navigation/menu_item.rb

@@ -7,6 +7,7 @@ module Aidp
         # Represents a single menu item in the navigation system
         class MenuItem
           class MenuItemError < StandardError; end
+
           class InvalidTypeError < MenuItemError; end
 
           VALID_TYPES = [:action, :submenu, :workflow, :separator].freeze

data/lib/aidp/harness/ui/navigation/menu_state.rb

@@ -7,6 +7,7 @@ module Aidp
         # Manages navigation state and history
         class MenuState
           class StateError < StandardError; end
+
           class InvalidStateError < StateError; end
 
           def initialize

data/lib/aidp/harness/ui/navigation/submenu.rb

@@ -9,6 +9,7 @@ module Aidp
         # Specialized submenu for drill-down functionality
         class SubMenu < MainMenu
           class SubMenuError < MenuError; end
+
           class InvalidSubMenuError < SubMenuError; end
 
           def initialize(title, parent_menu = nil, ui_components = {})

data/lib/aidp/harness/ui/navigation/workflow_selector.rb

@@ -12,7 +12,9 @@ module Aidp
         # Handles workflow mode selection (simple vs advanced)
         class WorkflowSelector < Base
           class WorkflowError < StandardError; end
+
           class InvalidModeError < WorkflowError; end
+
           class SelectionError < WorkflowError; end
 
           WORKFLOW_MODES = {

data/lib/aidp/harness/ui/progress_display.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "tty-progressbar"
+require "tty-prompt"
 require "pastel"
 require_relative "base"
 
@@ -9,8 +10,12 @@ module Aidp
     module UI
       # Handles progress display using CLI UI progress bars
       class ProgressDisplay < Base
+        include Aidp::MessageDisplay
+
         class ProgressError < StandardError; end
+
         class InvalidProgressError < ProgressError; end
+
         class DisplayError < ProgressError; end
 
         attr_reader :refresh_interval
@@ -24,7 +29,8 @@ module Aidp
           @auto_refresh_enabled = false
           @refresh_interval = 1.0
           @refresh_thread = nil
-          @output = ui_components[:output] || $stdout
+          @output = ui_components[:output]
+          @prompt = ui_components[:prompt] || TTY::Prompt.new
           @spinner_class = begin
             ui_components[:spinner] || TTY::Spinner
           rescue
@@ -115,7 +121,7 @@ module Aidp
           raise ArgumentError, "Progress items must be an array" unless progress_items.is_a?(Array)
 
           if progress_items.empty?
-            @output.puts @pastel.dim("No progress items to display.")
+            display_message(@pastel.dim("No progress items to display."), type: :muted)
             return
           end
 
@@ -176,7 +182,7 @@ module Aidp
 
         def execute_progress_steps(bar, total_steps, &block)
           total_steps.times do
-            yield(bar) if block_given?
+            yield(bar) if block
             bar.tick
           end
         end
@@ -186,7 +192,7 @@ module Aidp
           total_substeps.times do |index|
             substep_title = @formatter.format_substep_title(title, index + 1, total_substeps)
             bar.update_title(substep_title)
-            yield(bar, index) if block_given?
+            yield(bar, index) if block
             bar.tick
           end
         end
@@ -199,7 +205,7 @@ module Aidp
             " (Step: #{progress_data[:current_step]})"
           task_id = progress_data[:id] ? "[#{progress_data[:id]}] " : ""
 
-          @output.puts "#{task_id}#{progress}% #{message}#{step_info}"
+          display_message("#{task_id}#{progress}% #{message}#{step_info}", type: :info)
         end
 
         def display_detailed_progress(progress_data)
@@ -210,13 +216,13 @@ module Aidp
           started_at = progress_data[:started_at] ? progress_data[:started_at].strftime("%H:%M:%S") : "N/A"
           eta = progress_data[:eta] || "N/A"
 
-          @output.puts "Progress: #{progress}% - #{message} (Step: #{current_step}/#{total_steps}, Started: #{started_at}, ETA: #{eta})"
+          display_message("Progress: #{progress}% - #{message} (Step: #{current_step}/#{total_steps}, Started: #{started_at}, ETA: #{eta})", type: :info)
         end
 
         def display_minimal_progress(progress_data)
           progress = progress_data[:progress] || 0
           message = progress_data[:message] || "Processing..."
-          @output.puts "#{@pastel.blue("Progress:")} #{progress}% - #{message}"
+          display_message("Progress: #{progress}% - #{message}", type: :info)
         end
 
         def create_progress_bar(progress)
@@ -251,6 +257,19 @@ module Aidp
             timestamp: Time.now
           }
         end
+
+        private
+
+        # Use mixin display_message; fallback to stdout if no prompt
+        def display_message(message, type: :info)
+          if @prompt
+            super
+          elsif @output
+            @output.puts(message)
+          else
+            puts(message)
+          end
+        end
       end
 
       # Formats progress display text

data/lib/aidp/harness/ui/question_collector.rb

@@ -9,7 +9,9 @@ module Aidp
       # Handles interactive question collection using CLI UI prompts
       class QuestionCollector < Base
         class QuestionError < StandardError; end
+
         class ValidationError < QuestionError; end
+
         class CollectionError < QuestionError; end
 
         def initialize(ui_components = {}, prompt: nil)

data/lib/aidp/harness/ui/spinner_group.rb

@@ -9,7 +9,9 @@ module Aidp
       # Handles concurrent operations using CLI UI spinner groups
       class SpinnerGroup < Base
         class SpinnerGroupError < StandardError; end
+
         class InvalidOperationError < SpinnerGroupError; end
+
         class ExecutionError < SpinnerGroupError; end
 
         def initialize(ui_components = {})

data/lib/aidp/harness/ui/spinner_helper.rb

@@ -18,7 +18,7 @@ module Aidp
 
         # Main method: automatically manages spinner around a block
         def with_spinner(message, format: :dots, success_message: nil, error_message: nil, &block)
-          raise ArgumentError, "Block required for with_spinner" unless block_given?
+          raise ArgumentError, "Block required for with_spinner" unless block
 
           spinner = create_spinner(message, format)
           start_spinner(spinner)

data/lib/aidp/harness/ui/status_manager.rb

@@ -12,7 +12,9 @@ module Aidp
       # Real-time status updates using CLI UI spinners
       class StatusManager < Base
         class StatusError < StandardError; end
+
         class InvalidStatusError < StatusError; end
+
         class UpdateError < StatusError; end
 
         def initialize(ui_components = {})
@@ -164,7 +166,7 @@ module Aidp
         end
 
         def track_workflow_status(workflow_name, spinner, &block) # Will be updated dynamically
-          yield(spinner) if block_given?
+          yield(spinner) if block
           @status_widget.show_success_status("Completed #{workflow_name}")
         rescue => e
           @status_widget.show_error_status("Failed #{workflow_name}: #{e.message}")
@@ -172,7 +174,7 @@ module Aidp
         end
 
         def track_step_status(step_name, spinner, &block)
-          yield(spinner) if block_given?
+          yield(spinner) if block
           @status_widget.show_success_status("Completed #{step_name}")
         rescue => e
           @status_widget.show_error_status("Failed #{step_name}: #{e.message}")