aidp 0.9.5 → 0.10.0

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

@@ -15,7 +15,8 @@ module Aidp
         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 +27,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)
@@ -45,9 +58,9 @@ module Aidp
 
           if block_given?
             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 +73,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_given?
+              yield || ""
+            else
+              ""
+            end
           end
+
+          display_message(frame_result)
         rescue InvalidFrameError => e
           raise e
         rescue => e
@@ -98,7 +118,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 +145,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 +165,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 +178,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
@@ -182,13 +202,13 @@ module Aidp
           validate_title(title)
           raise ArgumentError, "Block required for frame_with_block" unless block_given?
 
-          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?
@@ -207,7 +227,7 @@ module Aidp
         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 +236,12 @@ module Aidp
         def section(title, &block)
           validate_title(title)
 
-          formatted_title = @formatter.format_section_title(title)
+          formatted_title = @formatter ? @formatter.format_section_title(title) : "📋 #{title}"
           if block_given?
             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 +250,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
+          formatted_title = @formatter ? @formatter.format_subsection_title(title) : "📝 #{title}"
+          display_message(@frame.frame(formatted_title, width: 80) do
             yield if block_given?
-          end
+          end)
         rescue => e
           raise DisplayError, "Failed to create subsection: #{e.message}"
         end
@@ -241,10 +261,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
+          formatted_title = @formatter ? @formatter.format_workflow_title(workflow_name) : "⚙️ #{workflow_name}"
+          display_message(@frame.frame(formatted_title, width: 80) do
             yield if block_given?
-          end
+          end)
         rescue => e
           raise DisplayError, "Failed to create workflow frame: #{e.message}"
         end
@@ -252,10 +272,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
+          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_given?
-          end
+          end)
         rescue => e
           raise DisplayError, "Failed to create step frame: #{e.message}"
         end

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

@@ -23,6 +23,7 @@ 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 = []
@@ -69,9 +70,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}"
@@ -133,7 +134,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 +143,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 +195,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 +221,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 +252,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/progress_display.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "tty-progressbar"
+require "tty-prompt"
 require "pastel"
 require_relative "base"
 
@@ -24,7 +25,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 +117,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
 
@@ -199,7 +201,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 +212,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 +253,27 @@ module Aidp
             timestamp: Time.now
           }
         end
+
+        private
+
+        def display_message(message, type: :info)
+          if @prompt
+            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)
+          elsif @output
+            @output.puts(message)
+          else
+            puts(message)
+          end
+        end
       end
 
       # Formats progress display text

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

@@ -18,6 +18,7 @@ module Aidp
           @spinner = ui_components[:spinner] || TTY::Spinner
           @pastel = Pastel.new
           @formatter = ui_components[:formatter] || StatusFormatter.new
+          @output = ui_components[:output]
           @status_history = []
           @current_spinner = nil
           @spinner_active = false
@@ -56,7 +57,7 @@ module Aidp
           validate_message(message)
 
           formatted_message = @formatter.format_success_message(message)
-          puts(formatted_message)
+          display_message(formatted_message)
         rescue => e
           raise DisplayError, "Failed to show success status: #{e.message}"
         end
@@ -65,7 +66,7 @@ module Aidp
           validate_message(message)
 
           formatted_message = @formatter.format_error_message(message)
-          puts(formatted_message)
+          display_message(formatted_message)
         rescue => e
           raise DisplayError, "Failed to show error status: #{e.message}"
         end
@@ -74,7 +75,7 @@ module Aidp
           validate_message(message)
 
           formatted_message = @formatter.format_warning_message(message)
-          puts(formatted_message)
+          display_message(formatted_message)
         rescue => e
           raise DisplayError, "Failed to show warning status: #{e.message}"
         end
@@ -181,6 +182,14 @@ module Aidp
 
         private
 
+        def display_message(message)
+          if @output
+            @output.say(message)
+          else
+            puts message
+          end
+        end
+
         def validate_message(message)
           raise InvalidStatusError, "Message cannot be empty" if message.to_s.strip.empty?
         end
@@ -211,22 +220,22 @@ module Aidp
         end
 
         def display_loading_status(message)
-          puts("⏳ #{message}")
+          display_message("⏳ #{message}")
         end
 
         def display_success_status(message)
-          puts("✅ #{message}")
+          display_message("✅ #{message}")
         end
 
         def display_error_status(message, error_data)
-          puts("❌ #{message}")
+          display_message("❌ #{message}")
           if error_data && error_data[:message]
-            puts("  #{error_data[:message]}")
+            display_message("  #{error_data[:message]}")
           end
         end
 
         def display_warning_status(message)
-          puts("⚠️ #{message}")
+          display_message("⚠️ #{message}")
         end
       end
 

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

@@ -27,6 +27,7 @@ module Aidp
           @status_manager = ui_components[:status_manager] || StatusManager.new
           @frame_manager = ui_components[:frame_manager] || FrameManager.new
           @formatter = ui_components[:formatter] || WorkflowControllerFormatter.new
+          @output = ui_components[:output]
 
           @current_state = :running
           @state_history = []
@@ -187,8 +188,23 @@ module Aidp
           @control_thread = nil
         end
 
+        # Testing helper method to set state
+        def set_state_for_testing(state)
+          @control_mutex.synchronize do
+            @current_state = state
+          end
+        end
+
         private
 
+        def display_message(message)
+          if @output
+            @output.say(message)
+          else
+            puts message
+          end
+        end
+
         def validate_state_transition(action)
           case action
           when :pause
@@ -229,19 +245,19 @@ module Aidp
         end
 
         def display_status_info(status)
-          puts("Current State: #{@formatter.format_state(status[:state])}")
-          puts("State Name: #{status[:state_name]}")
+          display_message("Current State: #{@formatter.format_state(status[:state])}")
+          display_message("State Name: #{status[:state_name]}")
 
           if status[:pause_time]
-            puts("Paused Since: #{status[:pause_time]}")
+            display_message("Paused Since: #{status[:pause_time]}")
           end
 
-          puts("\nAvailable Actions:")
-          puts("  Pause: #{status[:can_pause] ? "Yes" : "No"}")
-          puts("  Resume: #{status[:can_resume] ? "Yes" : "No"}")
-          puts("  Cancel: #{status[:can_cancel] ? "Yes" : "No"}")
-          puts("  Stop: #{status[:can_stop] ? "Yes" : "No"}")
-          puts("  Complete: #{status[:can_complete] ? "Yes" : "No"}")
+          display_message("\nAvailable Actions:")
+          display_message("  Pause: #{status[:can_pause] ? "Yes" : "No"}")
+          display_message("  Resume: #{status[:can_resume] ? "Yes" : "No"}")
+          display_message("  Cancel: #{status[:can_cancel] ? "Yes" : "No"}")
+          display_message("  Stop: #{status[:can_stop] ? "Yes" : "No"}")
+          display_message("  Complete: #{status[:can_complete] ? "Yes" : "No"}")
         end
 
         def control_interface_loop