code_healer 0.1.26 → 0.1.32

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 20c5c060c65cfee68cf65f2478e6335b1927f7e810fdcf6dd1c6796e29b49f65
-  data.tar.gz: 38aad1427647fe628c1a0aa2824fe09b206d6ac752ef6b1407a41ddd15560383
+  metadata.gz: 2a36985eccac2d4e1f55a411b9872478f6ec9e2f5285ebe969f6a839e600e37a
+  data.tar.gz: 7f1c43c592b9b6f6ed953ac70cc13855048d7f62222b65ddbf56b047a711ece5
 SHA512:
-  metadata.gz: eb2f391d38dfb0e444ff60e66d92625409f09c8389705a08c3ca3783f9aa56ae1d10dffc36109472cf2158c1e44b3df1eaba6100263f57d7d82bec713d58bbe6
-  data.tar.gz: 7864c4f255da0a1682d8c5787009998fe0e554a29fadbaf486f84d3c7664cdb12596381d8affa303110dc68a7b42d83c00f10da94c7f5f7368bc8bed88d13189
+  metadata.gz: 790508da0467e26544b1d89bab7caece0bc0afd0e8bc10cf1aee0166704d75a5ef717cab75049b32b6c51e296339df1e8a036194684b100c7ca7e840de5a8230
+  data.tar.gz: a34f3ec6ba73155e4adba2ed08ed30a3581dc587ec1d91df2386a69653ec46a83351d2f03bfae02e525e432c64410ea119e9293c44493b08210e7e4a4ba824bd

data/CHANGELOG.md

@@ -1,3 +1,19 @@
+## [0.1.32] - 2025-09-15
+
+### Added
+- Targeted RSpec test-fix loop after code changes in `ClaudeCodeEvolutionHandler`:
+  - Runs specs for files related to recent modifications
+  - Parses failures and re-invokes Claude with failure summary
+  - Iterates up to `test_fix.max_iterations` (default 2)
+- Configuration: `test_fix.max_iterations` with sensible default in `ConfigManager` and `setup.rb`.
+
+### Changed
+- Removed demo mode code and comments across gem: prompts, setup script, logger phrasing, and workspace manager.
+- Switched class targeting to excluded-classes-only model; removed `allowed_classes` usage in config managers and setup generation.
+
+### Notes
+- Backwards compatible; no breaking API changes. Configure `test_fix.max_iterations` in `config/code_healer.yml` to tune retries.
+
 # Changelog
 
 All notable changes to this project will be documented in this file.

data/lib/code_healer/claude_code_evolution_handler.rb

@@ -12,31 +12,25 @@ module CodeHealer
         PresentationLogger.detail("File: #{file_path}")
         
         begin
-          # Build concise, demo-optimized prompt (no repo-wide scan, no tests)
+          # Build concise prompt
           prompt = BusinessContextManager.build_claude_code_prompt(
             error, class_name, method_name, file_path
           )
+
           prompt << "\n\nStrict instructions:" \
                    "\n- Do NOT scan the entire codebase." \
                    "\n- Work only with the provided file/method context and backtrace." \
                    "\n- Return a unified diff (no prose)." \
-                   "\n- Keep changes minimal and safe." \
-                   "\n- Do NOT create or run tests." if CodeHealer::ConfigManager.demo_mode?
+                   "\n- Keep changes minimal and safe."
           
           # Execute Claude Code command
           success = execute_claude_code_fix(prompt, class_name, method_name)
           
           if success
             PresentationLogger.success("Claude run completed")
-            # Reload modified files
             reload_modified_files
-            
-            # 🚀 Trigger Git operations (commit, push, PR creation)
-            # Note: Git operations are now handled by the isolated workspace manager
-            # to prevent duplication and ensure proper isolation
-            PresentationLogger.info("Git operations handled by workspace manager")
-            
-            return true
+            # Run targeted RSpec and iterate on failures
+            return run_tests_and_iterate_fixes(class_name, method_name)
           else
             PresentationLogger.error("Claude run failed")
             return false
@@ -68,6 +62,10 @@ module CodeHealer
               PresentationLogger.success("Response received from Claude")
               PresentationLogger.detail(stdout)
 
+
+
+              # Business context references are intentionally not logged
+
               # Check if Claude Code is asking for permission
               if stdout.include?("permission") || stdout.include?("grant") || stdout.include?("edit")
                 PresentationLogger.warn("Claude requested edit permissions. Ensure permissions are granted.")
@@ -114,25 +112,118 @@ module CodeHealer
         # Replace placeholder
         command = command_template.gsub('{prompt}', escaped_prompt)
     
-        # Add demo mode specific instructions
-        if CodeHealer::ConfigManager.demo_mode?
-          command += " --append-system-prompt 'DEMO MODE: Focus on quick fixes, skip tests, limit file changes to #{config['max_file_changes'] || 3} files maximum'"
-        else
-          if config['include_tests']
-            command += " --append-system-prompt 'Include tests when fixing the code'"
-          end
-          
-          if config['max_file_changes']
-            command += " --append-system-prompt 'Limit changes to #{config['max_file_changes']} files maximum'"
-          end
+        # Add limits and testing hints
+        if config['include_tests']
+          command += " --append-system-prompt 'Include tests when fixing the code'"
+        end
+        
+        if config['max_file_changes']
+          command += " --append-system-prompt 'Limit changes to #{config['max_file_changes']} files maximum'"
         end
     
-        # Add business context instructions
-        command += " --append-system-prompt 'CRITICAL: Before fixing any code, you MUST use the Atlassian MCP tools to search Confluence for business context. Search for documentation about the class and method you are fixing. Only proceed with the fix after reviewing the business requirements from Confluence. If no documentation is found, note this in your response.'"
-    
+        # Add business context instructions and require a delimited summary we can parse (Confluence only)
+        # command += " --append-system-prompt 'CRITICAL: Before fixing any code, use the Atlassian MCP tools to fetch business context from Confluence ONLY. Summarize the relevant findings in a concise, bullet list between the markers <<CONTEXT_START>> and <<CONTEXT_END>>. Include Confluence page titles and links, and key rules. Keep to <=5 bullets. Then proceed with the fix.'"
+
+        # Explicit Confluence page fetch (env override with default fallback)
+        explicit_confluence_page_id = (ENV['CONFLUENCE_PAGE_ID'] || '4949770295').to_s.strip
+        unless explicit_confluence_page_id.empty?
+          command += " --append-system-prompt 'CRITICAL: Before fixing any code, Explicitly fetch Confluence page ID #{explicit_confluence_page_id} using Atlassian MCP (mcp__atlassian), extract applicable business logic/rules, and APPLY those rules in the fix.'"
+        end
+          
         # Return command
         command
       end
+
+      # Run targeted specs for changed files and iterate fixes up to a configured limit
+      def run_tests_and_iterate_fixes(class_name, method_name)
+        max_iters = ConfigManager.max_test_fix_iterations
+        it = 0
+        loop do
+          it += 1
+          PresentationLogger.step("RSpec run #{it}/#{max_iters}")
+          failures = run_targeted_rspec_for_changes
+          if failures.nil?
+            PresentationLogger.warn("No RSpec detected or no changed files with specs. Skipping test loop.")
+            return true
+          end
+          if failures.empty?
+            PresentationLogger.success("All targeted specs passed")
+            return true
+          end
+          PresentationLogger.warn("Failures detected (#{failures.size}). Attempting fix iteration...")
+          break if it >= max_iters
+          attempt_fix_from_failures(failures, class_name, method_name)
+        end
+        PresentationLogger.warn("Reached max test-fix iterations (#{max_iters}).")
+        false
+      end
+
+      def run_targeted_rspec_for_changes
+        changed = get_recently_modified_files.select { |f| f.end_with?('.rb') }
+        spec_files = changed.map { |f| f.sub(%r{^app/}, 'spec/').sub(/\.rb\z/, '_spec.rb') }
+        spec_files.select! { |s| File.exist?(s) }
+        return nil if spec_files.empty? || !File.exist?('spec')
+        cmd = ["bundle exec rspec --format documentation --no-color", spec_files.map { |s| "'#{s}'" }.join(' ')].join(' ')
+        stdout, stderr, status = Open3.capture3(cmd)
+        PresentationLogger.detail(stdout) if stdout && !stdout.empty?
+        PresentationLogger.warn(stderr) if stderr && !stderr.empty?
+        parse_rspec_failures(stdout)
+      rescue => e
+        PresentationLogger.warn("RSpec execution failed: #{e.message}")
+        nil
+      end
+
+      def parse_rspec_failures(output)
+        return [] unless output
+        failures = []
+        current = nil
+        output.each_line do |line|
+          if line =~ /^\s*\d+\)\s+(.*)$/
+            current = { title: $1.strip, details: [] }
+            failures << current
+          elsif current
+            current[:details] << line
+          end
+        end
+        failures
+      end
+
+      def attempt_fix_from_failures(failures, class_name, method_name)
+        summary = failures.map { |f| "- #{f[:title]}\n  #{f[:details].first(5).join}" }.join("\n")
+        PresentationLogger.claude_action("Sending failure summary to Claude for iterative fix")
+        iterative_prompt = <<~PROMPT
+          The previous fix compiled, but targeted RSpec tests failed. Here is a concise failure summary:\n\n#{summary}\n\nUpdate the relevant code to make these tests pass. Return only a unified diff. Keep changes minimal and safe.
+        PROMPT
+        config = ConfigManager.claude_code_settings
+        command = build_claude_command(iterative_prompt, config)
+        stdout, stderr, status = Open3.capture3(command)
+        PresentationLogger.detail(stdout) if stdout && !stdout.empty?
+        PresentationLogger.warn(stderr) if stderr && !stderr.empty?
+        if status.success?
+          PresentationLogger.success("Iterative Claude run succeeded")
+          reload_modified_files
+          true
+        else
+          PresentationLogger.error("Iterative Claude run failed (status #{status.exitstatus})")
+          false
+        end
+      end
+      
+      # Parse Claude Terminal output for Confluence links/titles (Confluence only)
+      def extract_business_context_references(text)
+        refs = []
+        return refs unless text
+        
+        # Match Confluence URLs
+        confluence_regex = /(https?:\/\/[^\s]+confluence[^\s]+\/(display|spaces|pages)\/[^\s)"']+)/i
+        text.scan(confluence_regex).each do |match|
+          url = match[0]
+          title = url.split('/').last.gsub('-', ' ')[0..80]
+          refs << { source: 'Confluence', display: "#{title} (#{url})" }
+        end
+        
+        refs.uniq { |r| r[:display] }
+      end
       
       def reload_modified_files
         PresentationLogger.step("Reloading modified files")

data/lib/code_healer/config_manager.rb

@@ -34,8 +34,9 @@ module CodeHealer
         config['allowed_error_types'] || []
       end
 
+      # Deprecated: allowed classes are no longer used. We rely solely on excluded_classes.
       def allowed_classes
-        config['allowed_classes'] || []
+        []
       end
 
       def excluded_classes
@@ -45,8 +46,7 @@ module CodeHealer
       def can_evolve_class?(class_name)
         return false unless enabled?
         return false if excluded_classes.include?(class_name)
-        return true if allowed_classes.empty?
-        allowed_classes.include?(class_name)
+        true
       end
 
       def can_handle_error?(error)
@@ -82,6 +82,15 @@ module CodeHealer
         config['claude_code'] || {}
       end
 
+      # Test-fix iteration settings
+      def test_fix_settings
+        config['test_fix'] || {}
+      end
+
+      def max_test_fix_iterations
+        (test_fix_settings['max_iterations'] || 2).to_i
+      end
+
       def claude_persist_session?
         claude_code_settings['persist_session'] == true
       end
@@ -163,21 +172,6 @@ module CodeHealer
         config['api'] || {}
       end
 
-      # Demo Configuration
-      def demo_settings
-        config['demo'] || {}
-      end
-
-      def demo_mode?
-        demo_settings['enabled'] == true
-      end
-
-      def demo_skip_tests?
-        demo_mode? && demo_settings['skip_tests'] != false
-      end
-
-
-
       def git_settings
         config['git'] || {}
       end
@@ -337,7 +331,6 @@ module CodeHealer
           'max_evolutions_per_day' => 10,
           'auto_generate_tests' => true,
           'allowed_error_types' => ['ZeroDivisionError', 'NoMethodError', 'ArgumentError', 'TypeError'],
-          'allowed_classes' => ['User', 'Order', 'PaymentProcessor'],
           'excluded_classes' => ['ApplicationController', 'ApplicationRecord', 'ApplicationJob', 'ApplicationMailer'],
           'evolution_strategy' => {
             'method' => 'api',
@@ -355,6 +348,9 @@ module CodeHealer
               'spec/business_context_specs.rb'
             ]
           },
+          'test_fix' => {
+            'max_iterations' => 2
+          },
           'business_context' => {
             'enabled' => true,
             'sources' => ['docs/business_rules.md']
@@ -365,10 +361,7 @@ module CodeHealer
             'max_tokens' => 2000,
             'temperature' => 0.1
           },
-          'demo' => {
-            'enabled' => false,
-            'skip_tests' => true
-          },
+          
           'git' => {
             'auto_commit' => true,
             'auto_push' => true,

data/lib/code_healer/core.rb

@@ -1,6 +1,7 @@
 require 'logger'
 require 'git'
 require 'octokit'
+require_relative 'presentation_logger'
 
 module CodeHealer
   class Core
@@ -180,15 +181,25 @@ module CodeHealer
       end
       
       def patch_classes_for_evolution
-        # Get allowed classes from config
-        allowed_classes = CodeHealer::ConfigManager.allowed_classes
-        
-        allowed_classes.each do |class_name|
-          begin
-            klass = class_name.constantize
-            patch_class_for_evolution(klass)
-          rescue NameError => e
-            puts "⚠️  Could not load class #{class_name}: #{e.message}"
+        # Patch all autoloaded classes except excluded ones
+        excluded = CodeHealer::ConfigManager.excluded_classes
+        # Attempt to iterate over loaded application classes
+        if defined?(Rails) && Rails.application
+          app_paths = [Rails.root.join('app', 'models'), Rails.root.join('app', 'controllers')]
+          app_paths.each do |path|
+            Dir.glob(File.join(path.to_s, '**', '*.rb')).each do |file|
+              begin
+                relative = file.sub(Rails.root.join('app').to_s + '/', '')
+                class_name = relative.gsub('.rb', '').split('/').map(&:classify).join('::')
+                next if class_name.nil? || class_name.empty?
+                next if excluded.include?(class_name)
+                klass = class_name.constantize rescue nil
+                next unless klass.is_a?(Class) || klass.is_a?(Module)
+                patch_class_for_evolution(klass)
+              rescue => e
+                puts "⚠️  Skipping #{file}: #{e.message}"
+              end
+            end
           end
         end
       end
@@ -229,26 +240,26 @@ module CodeHealer
         def extract_from_backtrace(backtrace)
           return [nil, nil] unless backtrace
           
-          puts "🔍 DEBUG: Starting backtrace analysis..."
-          puts "🔍 DEBUG: First 5 backtrace lines:"
-          backtrace.first(5).each_with_index { |line, i| puts "  #{i}: #{line}" }
+          PresentationLogger.detail("Starting backtrace analysis...")
+          PresentationLogger.detail("First 5 backtrace lines:")
+          backtrace.first(5).each_with_index { |line, i| PresentationLogger.detail("  #{i}: #{line}") }
           
           # Use the exact working implementation from SelfRuby
           core_methods = %w[* + - / % ** == != < > <= >= <=> === =~ !~ & | ^ ~ << >> [] []= `]
           app_file_line = backtrace.find { |line| line.include?('/app/') }
           return [nil, nil] unless app_file_line
           
-          puts "🔍 DEBUG: Found app file line: #{app_file_line}"
+          PresentationLogger.detail("Found app file line: #{app_file_line}")
           
           if app_file_line =~ /(.+):(\d+):in `(.+)'/
             file_path = $1
             method_name = $3
             
-            puts "🔍 DEBUG: Extracted file_path=#{file_path}, method_name=#{method_name}"
+            PresentationLogger.detail("Extracted file_path=#{file_path}, method_name=#{method_name}")
             
             # If it's a core method, look deeper in the backtrace
             if core_methods.include?(method_name)
-              puts "🔍 DEBUG: #{method_name} is a core method, looking deeper..."
+              PresentationLogger.detail("#{method_name} is a core method, looking deeper...")
               deeper_app_line = backtrace.find do |line| 
                 line.include?('/app/') && 
                 line =~ /in `(.+)'/ && 
@@ -260,14 +271,14 @@ module CodeHealer
               end
               
               if deeper_app_line
-                puts "🔍 DEBUG: Found deeper app line: #{deeper_app_line}"
+                PresentationLogger.detail("Found deeper app line: #{deeper_app_line}")
                 if deeper_app_line =~ /(.+):(\d+):in `(.+)'/
                   file_path = $1
                   method_name = $3
-                  puts "🔍 DEBUG: Updated to file_path=#{file_path}, method_name=#{method_name}"
+                  PresentationLogger.detail("Updated to file_path=#{file_path}, method_name=#{method_name}")
                 end
               else
-                puts "🔍 DEBUG: No deeper app line found"
+                PresentationLogger.detail("No deeper app line found")
               end
             end
             
@@ -279,7 +290,7 @@ module CodeHealer
               method_name.include?('reduce') ||
               method_name.include?('sum')
             )
-              puts "🔍 DEBUG: #{method_name} is a block/iterator, looking for containing method..."
+              PresentationLogger.detail("#{method_name} is a block/iterator, looking for containing method...")
               # Look for the FIRST valid method in the backtrace, not just any method
               containing_line = backtrace.find do |line|
                 line.include?('/app/') && 
@@ -294,14 +305,14 @@ module CodeHealer
               end
               
               if containing_line
-                puts "🔍 DEBUG: Found containing line: #{containing_line}"
+                PresentationLogger.detail("Found containing line: #{containing_line}")
                 if containing_line =~ /(.+):(\d+):in `(.+)'/
                   file_path = $1
                   method_name = $3
-                  puts "🔍 DEBUG: Updated to file_path=#{file_path}, method_name=#{method_name}"
+                  PresentationLogger.detail("Updated to file_path=#{file_path}, method_name=#{method_name}")
                 end
               else
-                puts "🔍 DEBUG: No containing line found"
+                PresentationLogger.detail("No containing line found")
               end
             end
             
@@ -320,13 +331,13 @@ module CodeHealer
                 end
               end
               
-              puts "🔍 DEBUG: Final result - class_name=#{class_name}, method_name=#{method_name}"
-              puts "🔍 Extracted: #{class_name}##{method_name} from #{file_path}"
+              PresentationLogger.detail("Final result - class_name=#{class_name}, method_name=#{method_name}")
+              PresentationLogger.detail("Extracted: #{class_name}##{method_name} from #{file_path}")
               return [class_name, method_name]
             end
           end
           
-          puts "🔍 DEBUG: No valid method found in backtrace"
+          PresentationLogger.detail("No valid method found in backtrace")
           [nil, nil]
         end
         

data/lib/code_healer/healing_workspace_manager.rb

@@ -62,43 +62,43 @@ module CodeHealer
       end
       
       def apply_fixes_in_workspace(workspace_path, fixes, class_name, method_name)
-        puts "🔧 [WORKSPACE] Starting fix application..."
-        puts "🔧 [WORKSPACE] Workspace: #{workspace_path}"
-        puts "🔧 [WORKSPACE] Class: #{class_name}, Method: #{method_name}"
-        puts "🔧 [WORKSPACE] Fixes to apply: #{fixes.inspect}"
+        PresentationLogger.detail("Starting fix application...")
+        PresentationLogger.detail("Workspace: #{workspace_path}")
+        PresentationLogger.detail("Class: #{class_name}, Method: #{method_name}")
+        PresentationLogger.detail("Fixes to apply: #{fixes.inspect}")
         
         begin
-          puts "🔧 [WORKSPACE] Processing #{fixes.length} fixes..."
+          PresentationLogger.detail("Processing #{fixes.length} fixes...")
           # Apply each fix to the workspace
           fixes.each_with_index do |fix, index|
-            puts "🔧 [WORKSPACE] Processing fix #{index + 1}: #{fix.inspect}"
+            PresentationLogger.detail("Processing fix #{index + 1}: #{fix.inspect}")
             file_path = File.join(workspace_path, fix[:file_path])
-            puts "🔧 [WORKSPACE] Target file: #{file_path}"
-            puts "🔧 [WORKSPACE] File exists: #{File.exist?(file_path)}"
+            PresentationLogger.detail("Target file: #{file_path}")
+            PresentationLogger.detail("File exists: #{File.exist?(file_path)}")
             
             next unless File.exist?(file_path)
             
-            puts "🔧 [WORKSPACE] Creating backup..."
+            PresentationLogger.detail("Creating backup...")
             # Backup original file
             backup_file(file_path)
             
-            puts "🔧 [WORKSPACE] Applying fix to file..."
+            PresentationLogger.detail("Applying fix to file...")
             # Apply the fix
             apply_fix_to_file(file_path, fix[:new_code], class_name, method_name)
           end
           
           # Show workspace Git status after applying fixes
           Dir.chdir(workspace_path) do
-            puts "🔧 [WORKSPACE] Git status after fixes:"
+            PresentationLogger.detail("Git status after fixes:")
             system("git status --porcelain")
-            puts "🔧 [WORKSPACE] Git diff after fixes:"
+            PresentationLogger.detail("Git diff after fixes:")
             system("git diff")
           end
           
-          puts "✅ Fixes applied successfully in workspace"
+          PresentationLogger.success("Fixes applied successfully in workspace")
           true
         rescue => e
-          puts "❌ Failed to apply fixes in workspace: #{e.message}"
+          PresentationLogger.error("Failed to apply fixes in workspace: #{e.message}")
           false
         end
       end
@@ -118,8 +118,7 @@ module CodeHealer
               return false
             end
             
-            # Optionally skip heavy tests in demo mode
-            unless CodeHealer::ConfigManager.demo_skip_tests?
+            # Run tests if available
               # Run tests if available
               if File.exist?('Gemfile')
                 bundle_check = system("bundle check >/dev/null 2>&1")
@@ -147,7 +146,7 @@ module CodeHealer
       end
       
       def validate_workspace_for_commit(workspace_path)
-        puts "🔍 [WORKSPACE] Validating workspace for commit..."
+        PresentationLogger.detail("Validating workspace for commit...")
         
         Dir.chdir(workspace_path) do
           # Check for any temporary files that might have been added
@@ -156,30 +155,30 @@ module CodeHealer
           
           all_files = (staged_files + working_files).uniq.reject(&:empty?)
           
-          puts "🔍 [WORKSPACE] Files to be committed: #{all_files.join(', ')}"
+          PresentationLogger.detail("Files to be committed: #{all_files.join(', ')}")
           
           # Check for any temporary files
           temp_files = all_files.select { |file| should_skip_file?(file) }
           
           if temp_files.any?
-            puts "⚠️ [WORKSPACE] WARNING: Temporary files detected in commit:"
-            temp_files.each { |file| puts "   - #{file}" }
+            PresentationLogger.warn("Temporary files detected in commit:")
+            temp_files.each { |file| PresentationLogger.detail("   - #{file}") }
             
             # Remove them from staging
             temp_files.each do |file|
-              puts "🗑️ [WORKSPACE] Removing temporary file from staging: #{file}"
+              PresentationLogger.detail("Removing temporary file from staging: #{file}")
               system("git reset HEAD '#{file}' 2>/dev/null || true")
             end
             
-            puts "🔍 [WORKSPACE] Temporary files removed from staging"
+            PresentationLogger.detail("Temporary files removed from staging")
             return false
           end
           
-          puts "✅ [WORKSPACE] Workspace validation passed - no temporary files detected"
+          PresentationLogger.detail("Workspace validation passed - no temporary files detected")
           return true
         end
       rescue => e
-        puts "❌ [WORKSPACE] Workspace validation failed: #{e.message}"
+        PresentationLogger.error("Workspace validation failed: #{e.message}")
         return false
       end
       
@@ -333,7 +332,23 @@ module CodeHealer
           
           puts "🔗 Creating PR for repository: #{repo_name}"
           
-          client = Octokit::Client.new(access_token: github_token)
+          # Configure Octokit with better error handling
+          client = Octokit::Client.new(
+            access_token: github_token,
+            api_endpoint: 'https://api.github.com',
+            web_endpoint: 'https://github.com',
+            auto_paginate: true,
+            per_page: 100
+          )
+          
+          # Test the connection first
+          begin
+            user = client.user
+            puts "✅ GitHub authentication successful for user: #{user.login}"
+          rescue => auth_error
+            puts "❌ GitHub authentication failed: #{auth_error.message}"
+            return nil
+          end
           
           # Create pull request
           pr = client.create_pull_request(
@@ -348,8 +363,21 @@ module CodeHealer
           
           puts "✅ Pull request created successfully: #{pr.html_url}"
           pr.html_url
+        rescue Octokit::Unauthorized => e
+          puts "❌ GitHub authentication failed: #{e.message}"
+          puts "💡 Check your GitHub token permissions and validity"
+          nil
+        rescue Octokit::NotFound => e
+          puts "❌ Repository not found: #{e.message}"
+          puts "💡 Check repository name and access permissions"
+          nil
+        rescue Octokit::UnprocessableEntity => e
+          puts "❌ Invalid pull request data: #{e.message}"
+          puts "💡 Check branch names and repository state"
+          nil
         rescue => e
           puts "❌ Failed to create pull request: #{e.message}"
+          puts "💡 Error class: #{e.class}"
           puts "💡 Check your GitHub token and repository access"
           nil
         end
@@ -398,66 +426,66 @@ module CodeHealer
       end
 
       def create_persistent_workspace(repo_path, workspace_path, branch_name)
-        puts "🏥 [WORKSPACE] Creating new persistent workspace..."
+        PresentationLogger.detail("Creating new persistent workspace...")
         
         # Get the GitHub remote URL
         Dir.chdir(repo_path) do
           remote_url = `git config --get remote.origin.url`.strip
           if remote_url.empty?
-            puts "❌ [WORKSPACE] No remote origin found in #{repo_path}"
+            PresentationLogger.error("No remote origin found in #{repo_path}")
             return false
           end
           
-          puts "🏥 [WORKSPACE] Cloning from: #{remote_url}"
-          puts "🏥 [WORKSPACE] To workspace: #{workspace_path}"
+          PresentationLogger.detail("Cloning from: #{remote_url}")
+          PresentationLogger.detail("To workspace: #{workspace_path}")
           
           # Clone the full repository for persistent use
           result = system("git clone #{remote_url} #{workspace_path}")
           if result
-            puts "🏥 [WORKSPACE] Repository cloned successfully"
+            PresentationLogger.detail("Repository cloned successfully")
             # Now checkout to the target branch
             checkout_to_branch(workspace_path, branch_name, repo_path)
           else
-            puts "❌ [WORKSPACE] Failed to clone repository"
+            PresentationLogger.error("Failed to clone repository")
             return false
           end
         end
       end
 
       def checkout_to_branch(workspace_path, branch_name, repo_path)
-        puts "🏥 [WORKSPACE] Checking out to target branch..."
+        PresentationLogger.detail("Checking out to target branch...")
         
         # Determine target branch
         target_branch = branch_name || CodeHealer::ConfigManager.pr_target_branch || get_default_branch(repo_path)
-        puts "🏥 [WORKSPACE] Target branch: #{target_branch}"
+        PresentationLogger.detail("Target branch: #{target_branch}")
         
         Dir.chdir(workspace_path) do
           # Fetch latest changes
-          puts "🏥 [WORKSPACE] Fetching latest changes..."
+          PresentationLogger.detail("Fetching latest changes...")
           system("git fetch origin")
           
           # Check if branch exists locally
           local_branch_exists = system("git show-ref --verify --quiet refs/heads/#{target_branch}")
           
           if local_branch_exists
-            puts "🏥 [WORKSPACE] Checking out existing local branch: #{target_branch}"
+            PresentationLogger.detail("Checking out existing local branch: #{target_branch}")
             system("git checkout #{target_branch}")
           else
-            puts "🏥 [WORKSPACE] Checking out remote branch: #{target_branch}"
+            PresentationLogger.detail("Checking out remote branch: #{target_branch}")
             system("git checkout -b #{target_branch} origin/#{target_branch}")
           end
           
           # Pull latest changes
-          puts "🏥 [WORKSPACE] Pulling latest changes..."
+          PresentationLogger.detail("Pulling latest changes...")
           system("git pull origin #{target_branch}")
           
           # Ensure workspace is clean
-          puts "🏥 [WORKSPACE] Ensuring workspace is clean..."
+          PresentationLogger.detail("Ensuring workspace is clean...")
           system("git reset --hard HEAD")
           system("git clean -fd")
           
           # Remove any tracked temporary files that shouldn't be committed - AGGRESSIVE cleanup
-          puts "🏥 [WORKSPACE] Removing tracked temporary files..."
+          PresentationLogger.detail("Removing tracked temporary files...")
           
           # Remove root level temporary directories
           system("git rm -r --cached tmp/ 2>/dev/null || true")
@@ -477,7 +505,7 @@ module CodeHealer
           system("find . -name '*.log' -exec git rm --cached {} + 2>/dev/null || true")
           system("find . -name '*.cache' -exec git rm --cached {} + 2>/dev/null || true")
           
-          puts "🏥 [WORKSPACE] Successfully checked out to: #{target_branch}"
+          PresentationLogger.detail("Successfully checked out to: #{target_branch}")
         end
       end
 
@@ -503,25 +531,26 @@ module CodeHealer
       end
 
       def add_only_relevant_files(workspace_path)
-        puts "📁 [WORKSPACE] Adding only relevant files, respecting .gitignore..."
+        PresentationLogger.detail("Adding only relevant files, respecting .gitignore...")
         
         Dir.chdir(workspace_path) do
           # First, ensure .gitignore is respected
           if File.exist?('.gitignore')
-            puts "📁 [WORKSPACE] Using repository's .gitignore file"
+            PresentationLogger.detail("Using repository's .gitignore file")
           else
-            puts "📁 [WORKSPACE] No .gitignore found, using default patterns"
+            PresentationLogger.detail("No .gitignore found, using default patterns")
           end
           
           # Get list of modified files
           modified_files = `git status --porcelain | grep '^ M\\|^M \\|^A ' | awk '{print $2}'`.strip.split("\n")
+
           
           if modified_files.empty?
-            puts "📁 [WORKSPACE] No modified files to add"
+            PresentationLogger.detail("No modified files to add")
             return
           end
           
-          puts "📁 [WORKSPACE] Modified files: #{modified_files.join(', ')}"
+          PresentationLogger.detail("Modified files: #{modified_files.join(', ')}")
           
           # Add each modified file individually
           modified_files.each do |file|
@@ -529,15 +558,15 @@ module CodeHealer
             
             # Skip temporary and generated files
             if should_skip_file?(file)
-              puts "📁 [WORKSPACE] Skipping temporary file: #{file}"
+              PresentationLogger.detail("Skipping temporary file: #{file}")
               next
             end
             
-            puts "📁 [WORKSPACE] Adding file: #{file}"
+            PresentationLogger.detail("Adding file: #{file}")
             system("git add '#{file}'")
           end
           
-          puts "📁 [WORKSPACE] File addition completed"
+          PresentationLogger.detail("File addition completed")
         end
       end
 
@@ -594,7 +623,7 @@ module CodeHealer
         
         # Additional check: if path contains 'tmp' or 'log' anywhere, skip it
         if file_path.include?('tmp') || file_path.include?('log')
-          puts "📁 [WORKSPACE] Skipping file containing 'tmp' or 'log': #{file_path}"
+          PresentationLogger.detail("Skipping file containing 'tmp' or 'log': #{file_path}")
           return true
         end
         

data/lib/code_healer/mcp_server.rb

@@ -1,5 +1,6 @@
 require_relative 'mcp_tools'
 require_relative 'mcp_prompts'
+require_relative 'presentation_logger'
 
 module CodeHealer
   class McpServer
@@ -45,7 +46,7 @@ module CodeHealer
       end
       
       def analyze_error(error, context)
-        puts "🧠 MCP analyzing error: #{error.class} - #{error.message}"
+        PresentationLogger.detail("MCP analyzing error: #{error.class} - #{error.message}")
         
         # Extract class and method names from context
         class_name = context[:class_name] || 'UnknownClass'
@@ -61,11 +62,11 @@ module CodeHealer
             server_context: { codebase_context: context }
           )
           
-          puts "✅ MCP analysis complete"
+          PresentationLogger.detail("MCP analysis complete")
           # Parse the JSON response from MCP tool
           JSON.parse(result.content.first[:text])
         else
-          puts "⚠️  ErrorAnalysisTool not available, using fallback analysis"
+          PresentationLogger.detail("ErrorAnalysisTool not available, using fallback analysis")
           # Fallback analysis
           {
             severity: 'medium',
@@ -78,13 +79,13 @@ module CodeHealer
       end
       
       def generate_contextual_fix(error, analysis, context)
-        puts "🧠 MCP generating contextual fix..."
+        PresentationLogger.detail("MCP generating contextual fix...")
         
         # Extract class and method names from context
         class_name = context[:class_name] || 'UnknownClass'
         method_name = context[:method_name] || 'unknown_method'
         
-        puts "🔍 Debug: class_name = #{class_name}, method_name = #{method_name}"
+        PresentationLogger.detail("Debug: class_name = #{class_name}, method_name = #{method_name}")
         
         # Use MCP tool to generate fix
         if defined?(CodeFixTool)
@@ -101,11 +102,11 @@ module CodeHealer
             }
           )
           
-          puts "✅ MCP generated intelligent fix"
+          PresentationLogger.detail("MCP generated intelligent fix")
           # Parse the JSON response from MCP tool
           JSON.parse(result.content.first[:text])
         else
-          puts "⚠️  CodeFixTool not available, using fallback fix generation"
+          PresentationLogger.detail("CodeFixTool not available, using fallback fix generation")
           # Fallback fix generation
           generate_fallback_fix(error, class_name, method_name)
         end

data/lib/code_healer/presentation_logger.rb

@@ -1,5 +1,5 @@
 module CodeHealer
-  # Presentation-focused logger for conference demos and clean operator output
+  # Presentation-focused logger for clean operator output
   class PresentationLogger
     class << self
       def verbose?
@@ -60,9 +60,9 @@ module CodeHealer
         # Show only the first 3 relevant lines for presentation
         relevant_lines = backtrace_array.first(3).map do |line|
           # Extract just the file and line number for cleaner display
-          if line.match?(/^(.+\.rb):(\d+):in/)
-            file = File.basename($1)
-            line_num = $2
+          if (m = line.match(/^(.+\.rb):(\d+):in/))
+            file = File.basename(m[1])
+            line_num = m[2]
             method = line.match(/in `(.+)'/)&.[](1) || 'unknown'
             "#{file}:#{line_num} in #{method}"
           else

data/lib/code_healer/setup.rb

@@ -414,34 +414,7 @@ evolution_method = case evolution_method.downcase
 
 fallback_to_api = ask_for_yes_no("Fallback to API if Claude Code fails?", default: true)
 
-# Demo Mode Configuration
-puts
-puts "🎭 Demo Mode Configuration:"
-puts "Demo mode optimizes CodeHealer for fast demonstrations and presentations:"
-puts "- Skips test generation for faster response times"
-puts "- Skips pull request creation for immediate results"
-puts "- Uses optimized Claude prompts for quick fixes"
-puts
-
-enable_demo_mode = ask_for_yes_no("Enable demo mode for fast demonstrations?", default: false)
-
-demo_config = {}
-if enable_demo_mode
-  demo_config[:skip_tests] = ask_for_yes_no("Skip test generation in demo mode?", default: true)
-
-  
-  puts
-  puts "🚀 Demo mode will significantly speed up healing operations!"
-  puts "   Perfect for conference talks and live demonstrations."
-  
-  # Add demo-specific instructions
-  puts
-  puts "📋 Demo Mode Features:"
-  puts "   - Timeout reduced to 60 seconds for quick responses"
-  puts "   - Sticky workspace enabled for faster context loading"
-  puts "   - Claude session persistence for better performance"
-  puts "   - Tests skipped for immediate results (PRs still created)"
-end
+ 
 
 # Create configuration files
 puts
@@ -481,13 +454,6 @@ create_file_with_content('.env', env_content, dry_run: options[:dry_run])
     # CodeHealer Configuration
     enabled: true
     
-    # Allowed classes for healing (customize as needed)
-    allowed_classes:
-      - User
-      - Order
-      - PaymentProcessor
-      - OrderProcessor
-    
     # Excluded classes (never touch these)
     excluded_classes:
       - ApplicationController
@@ -513,9 +479,9 @@ create_file_with_content('.env', env_content, dry_run: options[:dry_run])
     # Claude Code Terminal Configuration
     claude_code:
       enabled: #{evolution_method == 'claude_code_terminal' || evolution_method == 'hybrid'}
-      timeout: #{enable_demo_mode ? 60 : 300}  # Shorter timeout for demo mode
+      timeout: 300
       max_file_changes: 10
-      include_tests: #{!enable_demo_mode || !demo_config[:skip_tests]}
+      include_tests: true
       persist_session: true  # Keep Claude session alive for faster responses
       ignore:
         - "tmp/"
@@ -608,15 +574,15 @@ create_file_with_content('.env', env_content, dry_run: options[:dry_run])
     max_evolutions_per_day: 10
     
     # Notification Configuration (optional)
+    # Test-Fix Iteration Configuration
+    test_fix:
+      max_iterations: 2
     notifications:
       enabled: false
       slack_webhook: ""
       email_notifications: false
     
-    # Demo Mode Configuration
-    demo:
-      enabled: #{enable_demo_mode}
-      skip_tests: #{demo_config[:skip_tests] || false}
+    
     
     # Performance Configuration
     performance:
@@ -631,7 +597,7 @@ create_file_with_content('.env', env_content, dry_run: options[:dry_run])
       cleanup_after_hours: #{cleanup_after_hours}
       max_workspaces: 10
       clone_strategy: "branch"  # Options: branch, full_repo
-      sticky_workspace: #{enable_demo_mode}  # Reuse workspace for faster demo responses
+      sticky_workspace: false
 YAML
 
 create_file_with_content('config/code_healer.yml', config_content, dry_run: options[:dry_run])
@@ -711,9 +677,7 @@ puts
   puts "   - Your code will be cloned to: #{code_heal_directory}"
   puts "   - This ensures safe, isolated healing without affecting your running server"
   puts "   - Workspaces are automatically cleaned up after #{cleanup_after_hours} hours"
-  if enable_demo_mode
-    puts "   - Demo mode: Sticky workspace enabled for faster context loading"
-  end
+  
 puts
 puts "⚙️  Configuration:"
 puts "   - code_healer.yml contains comprehensive settings with sensible defaults"

data/lib/code_healer/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CodeHealer
-  VERSION = "0.1.26"
+  VERSION = "0.1.32"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: code_healer
 version: !ruby/object:Gem::Version
-  version: 0.1.26
+  version: 0.1.32
 platform: ruby
 authors:
 - Deepan Kumar
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-09-03 00:00:00.000000000 Z
+date: 2025-09-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails