aidp 0.9.6 → 0.11.0

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

@@ -10,7 +10,9 @@ module Aidp
       # Handles status display using CLI UI spinners
       class StatusWidget < Base
         class StatusError < StandardError; end
+
         class InvalidStatusError < StatusError; end
+
         class DisplayError < StatusError; end
 
         def initialize(ui_components = {})
@@ -18,6 +20,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
@@ -28,7 +31,7 @@ module Aidp
 
           formatted_message = @formatter.format_status_message(message)
           @spinner.spin(formatted_message) do |spinner|
-            yield(spinner) if block_given?
+            yield(spinner) if block
           end
         rescue => e
           raise DisplayError, "Failed to show status: #{e.message}"
@@ -56,7 +59,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 +68,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 +77,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 +184,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 +222,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

@@ -11,7 +11,9 @@ module Aidp
       # Workflow control interface for pause/resume/cancel/stop
       class WorkflowController < Base
         class WorkflowError < StandardError; end
+
         class InvalidStateError < WorkflowError; end
+
         class ControlError < WorkflowError; end
 
         WORKFLOW_STATES = {
@@ -27,6 +29,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 +190,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 +247,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