code_sage 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 54bbfe11204be27b426808d1ee98e5ea40aed7c8a25d525c613044f75f35e9c6
-  data.tar.gz: 453c726fd2f5dc7f2a6c7a975ead2f36420a7d40f3d96c3eb9a7099ad7b710cd
+  metadata.gz: b2373e4df03b94ff9250c649d452c2c71b8b011bdb72c8323ade0c2050f93f9d
+  data.tar.gz: 19b5c35ee826b396a10d8534d3524731918c7dc179cfc37d47d6d25fa23d90fc
 SHA512:
-  metadata.gz: 49e79a62468d7dda3d7d027259802d2fa00d3049a096e885430e634b70384185cda168be105afc0c3842c9908751d4f18bdb761e4dfd15aa022791ada6f02430
-  data.tar.gz: 45280224e33c0dbf2c8c721a0def678388148b2bc294f714c2956d5234e2c7cc52a909f4f7340dc6b29595d6c96570259441e2bc5add7c240fe925040f86749d
+  metadata.gz: 2c288c141dba2e6c283196c0880bd9578e7f843712820ce2a291a60c3ed7a44f280f6dca63e275b224d9d2e5f92105e05878a68b51ceaa5feb0691e989fa4c86
+  data.tar.gz: 8585510f9d203134152740e1af4ea70727492c813e30eb58a3d86ecffecde2abcf7ba19519ffa8371f6c6917d60a67f673e09d73e7014f1e4469505cc32e9cb4

data/.code_sage.yml

@@ -0,0 +1,38 @@
+llm:
+  provider: qwen
+  model: qwen3:1.7b
+  temperature: 0.2
+
+git:
+  default_branch: main
+  include_patterns:
+    - "*.rb"
+    - "*.rake"
+    - "Gemfile"
+    - "Rakefile"
+  exclude_patterns:
+    - "spec/**/*"
+    - "test/**/*"
+
+review:
+  focus_areas:
+    - security
+    - performance
+    - maintainability
+    - best_practices
+  severity_levels:
+    - low
+    - medium
+    - high
+    - critical
+
+output:
+  format: json
+  verbose: false
+  colors: false
+
+auto_fix:
+  enabled: true
+  confirm_before_apply: true
+  create_backups: true
+  backup_extension: ".backup"

data/CHANGELOG.md

@@ -7,18 +7,48 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.1.1] - 2025-01-08
+
 ### Added
-- Initial gem structure
-- CLI interface with Thor
-- Git integration using Rugged
-- LLM-powered code review using llm_chain
-- Multiple output formats (console, JSON, markdown)
-- Ruby code analysis and review
-- Comprehensive reporting system
+- **Auto-fix functionality**: AI-powered automatic code correction
+  - `--auto-fix` flag to enable automatic code fixes
+  - `--confirm-fixes` / `--no-confirm-fixes` options for confirmation control
+  - Intelligent detection of files needing fixes based on review findings
+  - Automatic backup creation before applying fixes
+  - Safe code extraction and validation
+- **Enhanced configuration system**:
+  - Local configuration file support (`.code_sage.yml` in current directory)
+  - Improved config priority: CLI explicit > local config > global config > defaults
+  - New `auto_fix` configuration section with safety options
+  - Better config file detection and informative messages
+- **Improved output format handling**:
+  - Configuration-based default output format
+  - CLI format options properly override config settings
+  - More consistent format application across commands
+
+### Enhanced
+- **Reviewer system**: 
+  - Better LLM prompt engineering for code fixes
+  - More robust error handling in auto-fix workflow
+  - Enhanced verbose output with configuration information
+- **Configuration management**:
+  - `config_info` and `show_config_info` methods for better visibility
+  - Smarter config file resolution (local > global > defaults)
+  - More informative config-related messages
+
+### Fixed
+- Output format from configuration file now properly applied
+- Config file loading order and priority issues resolved
+- Better error handling in git analysis and LLM interactions
 
 ## [0.1.0] - 2025-01-XX
 
 ### Added
 - Initial release of CodeSage
 - Basic code review functionality
-- CLI interface for easy usage 
+- CLI interface for easy usage
+- Git integration using Rugged
+- LLM-powered code review using llm_chain  
+- Multiple output formats (console, JSON, markdown)
+- Ruby code analysis and review
+- Comprehensive reporting system 

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    code_sage (0.1.0)
+    code_sage (0.1.1)
       colorize (~> 0.8)
       llm_chain (~> 0.5.2)
       rugged (~> 1.0)

data/README.md

@@ -74,6 +74,11 @@ $ code_sage review -v
 
 # Review with custom configuration
 $ code_sage review -c ~/.my_code_sage_config.yml
+
+# Auto-fix functionality (NEW in v0.1.1)
+$ code_sage review --auto-fix                          # With confirmation
+$ code_sage review --auto-fix --no-confirm-fixes       # Without confirmation
+$ code_sage review -f lib/my_file.rb --auto-fix -v     # Verbose auto-fix
 ```
 
 ### Configuration
@@ -117,6 +122,12 @@ output:
   format: console
   verbose: false
   colors: true
+
+auto_fix:
+  enabled: false
+  confirm_before_apply: true
+  create_backups: true
+  backup_extension: ".backup"
 ```
 
 #### Configuration Examples
@@ -162,6 +173,58 @@ $ code_sage config --key llm.model --value qwen2:7b
 $ code_sage config --reset
 ```
 
+### Auto-Fix Functionality (NEW in v0.1.1)
+
+CodeSage can now automatically apply AI-suggested fixes to your Ruby code based on review findings.
+
+#### How it works:
+1. **Analysis**: CodeSage reviews your code and identifies issues
+2. **Detection**: Files with fixable issues are automatically detected
+3. **AI Fixing**: LLM generates corrected code maintaining original functionality
+4. **Safety**: Creates backup files before applying any changes
+5. **Confirmation**: Optional interactive confirmation before applying fixes
+
+#### Usage Examples:
+
+```bash
+# Basic auto-fix with confirmation
+$ code_sage review -f lib/my_file.rb --auto-fix
+
+# Auto-fix without confirmation (use with caution)
+$ code_sage review -f lib/my_file.rb --auto-fix --no-confirm-fixes
+
+# Auto-fix multiple files with verbose output
+$ code_sage review -f lib/*.rb --auto-fix -v
+
+# Auto-fix with custom format output
+$ code_sage review --auto-fix --format json
+```
+
+#### Configuration:
+
+Enable auto-fix by default in your configuration:
+
+```yaml
+auto_fix:
+  enabled: true                    # Enable auto-fix by default
+  confirm_before_apply: false      # Skip confirmation prompts
+  create_backups: true             # Always create backups (recommended)
+  backup_extension: ".bak"         # Custom backup file extension
+```
+
+#### Safety Features:
+- **Automatic Backups**: Original files are backed up before modifications
+- **Confirmation Prompts**: Interactive confirmation before applying changes (by default)
+- **Validation**: Generated code is validated before application
+- **Verbose Logging**: Detailed information about fixes applied
+
+#### What gets fixed:
+- Nil safety issues in string interpolation
+- Missing error handling (division by zero, etc.)
+- Security vulnerabilities (unsafe eval usage)
+- Performance improvements (inefficient algorithms)
+- Ruby best practices and idioms
+
 ### Programmatic Usage
 
 You can also use CodeSage programmatically in your Ruby code:
@@ -196,6 +259,8 @@ Perform a code review on your repository.
 - `-c, --config PATH` - Path to configuration file
 - `-v, --verbose` - Verbose output
 - `--rag` - Enable RAG (Retrieval Augmented Generation) functionality
+- `--auto-fix` - Automatically apply AI-suggested fixes to files (NEW in v0.1.1)
+- `--confirm-fixes` / `--no-confirm-fixes` - Confirm before applying fixes (default: true)
 
 ### `config`
 
@@ -246,34 +311,4 @@ ollama pull llama2:7b
 
 # Start Ollama server
 ollama serve
-```
-
-#### Other Providers
-- **Anthropic**: Set `ANTHROPIC_API_KEY`
-- **Google**: Set `GOOGLE_API_KEY`
-- **Qwen**: Available through Ollama
-
-## Dependencies
-
-- `llm_chain` - For LLM integration
-- `thor` - CLI framework
-- `colorize` - Terminal colors
-- `rugged` - Git integration
-
-## Development
-
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
-
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
-
-## Contributing
-
-Bug reports and pull requests are welcome on GitHub at https://github.com/FuryCow/code_sage.
-
-## License
-
-The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
-
-## Changelog
-
-See [CHANGELOG.md](CHANGELOG.md) for version history and changes.
+```

data/lib/code_sage/cli.rb

@@ -11,6 +11,8 @@ module CodeSage
     option :verbose, aliases: '-v', type: :boolean, desc: "Verbose output"
     option :rag, type: :boolean, 
                  desc: "Enable RAG (Retrieval Augmented Generation) functionality (requires vector database)"
+    option :auto_fix, aliases: '--auto-fix', type: :boolean, desc: "Automatically apply suggested fixes to files"
+    option :confirm_fixes, aliases: '--confirm-fixes', type: :boolean, default: true, desc: "Confirm before applying fixes (default: true)"
     def review
       puts "🔮 CodeSage - Wisdom for your code".colorize(:cyan)
       puts
@@ -20,9 +22,12 @@ module CodeSage
           branch: options[:branch] || 'main',
           files: options[:files],
           format: options[:format],
+          format_explicit: ARGV.include?('--format') || ARGV.any? { |arg| arg.start_with?('--format=') },
           config_path: options[:config],
           verbose: options[:verbose],
-          enable_rag: options[:rag] || false
+          enable_rag: options[:rag] || false,
+          auto_fix: options[:auto_fix] || false,
+          confirm_fixes: options[:confirm_fixes] != false
         )
         
         result = reviewer.review
@@ -50,6 +55,7 @@ module CodeSage
       config_instance = Config.new
       
       if options[:show]
+        config_instance.show_config_info
         puts "📋 CodeSage Configuration".colorize(:cyan).bold
         puts "=" * 50
         puts YAML.dump(config_instance.data)
@@ -61,7 +67,7 @@ module CodeSage
         config_instance.save!
         puts "✅ Configuration updated: #{options[:key]} = #{options[:value]}".colorize(:green)
       else
-        puts "📋 Current configuration file: #{config_instance.config_path}".colorize(:cyan)
+        config_instance.show_config_info
         puts "Use --show to display configuration"
         puts "Use --key KEY --value VALUE to update settings"
         puts "Use --reset to restore defaults"

data/lib/code_sage/config.rb

@@ -23,6 +23,12 @@ module CodeSage
         'format' => 'console',
         'verbose' => false,
         'colors' => true
+      },
+      'auto_fix' => {
+        'enabled' => false,
+        'confirm_before_apply' => true,
+        'create_backups' => true,
+        'backup_extension' => '.backup'
       }
     }.freeze
     
@@ -53,11 +59,32 @@ module CodeSage
       @data = DEFAULT_CONFIG.dup
       save!
     end
+
+    def config_info
+      if File.exist?(@config_path)
+        "Using config file: #{@config_path}"
+      else
+        "No config file found, using defaults"
+      end
+    end
+
+    def show_config_info
+      puts "📋 #{config_info}".colorize(:cyan)
+    end
     
     private
     
     def default_config_path
-      File.expand_path('~/.code_sage.yml')
+      # Сначала ищем в текущей директории
+      local_config = File.expand_path('.code_sage.yml')
+      return local_config if File.exist?(local_config)
+      
+      # Если не найден, ищем в домашней папке
+      global_config = File.expand_path('~/.code_sage.yml')
+      return global_config if File.exist?(global_config)
+      
+      # Если ни один не найден, возвращаем путь к локальному для создания
+      local_config
     end
     
     def load_config
@@ -68,7 +95,7 @@ module CodeSage
         DEFAULT_CONFIG.dup
       end
     rescue => e
-      puts "Warning: Could not load config file #{@config_path}: #{e.message}".colorize(:yellow)
+      puts "Warning: Could not load config file #{@config_path}: #{e.message}".colorize(:red)
       DEFAULT_CONFIG.dup
     end
     

data/lib/code_sage/reviewer.rb

@@ -9,13 +9,31 @@ module CodeSage
         branch: 'main',
         files: nil,
         format: 'console',
+        format_explicit: false,
         config_path: nil,
         verbose: false,
-        enable_rag: false  # Отключаем RAG по умолчанию
+        enable_rag: false,  # Отключаем RAG по умолчанию
+        auto_fix: false,
+        confirm_fixes: true
       }.merge(options)
       
+      # Load configuration first
+      @config = @options[:config_path] ? Config.new(@options[:config_path]) : Config.new
+      
+      # Show config info in verbose mode
+      if @options[:verbose]
+        @config.show_config_info
+      end
+      
+      # Determine output format: CLI explicit > config > CLI default
+      output_format = if @options[:format_explicit]
+                        @options[:format]
+                      else
+                        @config.data['output']['format'] || @options[:format]
+                      end
+      
       @git_analyzer = GitAnalyzer.new(@options)
-      @formatter = ReportFormatter.new(@options[:format])
+      @formatter = ReportFormatter.new(output_format)
       @llm_chain = setup_llm_chain
     end
     
@@ -42,6 +60,11 @@ module CodeSage
       report = generate_report(reviews)
       output_report(report)
       
+      # Apply auto-fixes if requested
+      if @options[:auto_fix]
+        apply_fixes(reviews)
+      end
+      
       { success: true, reviews: reviews, report: report }
     rescue => e
       { success: false, error: e.message }
@@ -50,8 +73,8 @@ module CodeSage
     private
     
     def setup_llm_chain
-      # Load configuration
-      config = @options[:config_path] ? Config.new(@options[:config_path]) : Config.new
+      # Use already loaded configuration
+      config = @config
       
       llm_config = config.data['llm']
       provider = llm_config['provider'] || 'openai'
@@ -222,5 +245,128 @@ module CodeSage
       formatted_report = @formatter.format(report)
       puts formatted_report
     end
+    
+    def apply_fixes(reviews)
+      puts "\n🔧 Auto-fixing mode enabled".colorize(:cyan).bold
+      
+      files_to_fix = []
+      
+      reviews.each do |review|
+        next unless review[:review].include?('issue') || review[:review].include?('problem') || 
+                   review[:review].include?('fix') || review[:review].include?('improvement')
+        
+        files_to_fix << review[:file]
+      end
+      
+      if files_to_fix.empty?
+        puts "✅ No files need auto-fixing".colorize(:green)
+        return
+      end
+      
+      puts "📝 Files to fix: #{files_to_fix.join(', ')}"
+      
+      if @options[:confirm_fixes]
+        print "Do you want to apply auto-fixes? (y/N): "
+        response = STDIN.gets.chomp.downcase
+        return unless response == 'y' || response == 'yes'
+      end
+      
+      files_to_fix.each do |file_path|
+        puts "🔧 Fixing #{file_path}..." if @options[:verbose]
+        
+        begin
+          fixed_content = get_fixed_content(file_path)
+          if fixed_content && fixed_content != File.read(file_path)
+            apply_fix_to_file(file_path, fixed_content)
+            puts "✅ Fixed #{file_path}".colorize(:green)
+          else
+            puts "⚠️  No changes needed for #{file_path}".colorize(:yellow)
+          end
+        rescue => e
+          puts "❌ Error fixing #{file_path}: #{e.message}".colorize(:red)
+        end
+      end
+    end
+    
+    def get_fixed_content(file_path)
+      return nil unless File.exist?(file_path)
+      
+      content = File.read(file_path)
+      prompt = build_fix_prompt(file_path, content)
+      
+      puts "🤖 Generating fixes for #{file_path}..." if @options[:verbose]
+      
+      # Use LLM to get fixed content
+      full_prompt = "#{build_fix_system_message}\n\n#{prompt}"
+      fixed_content = @llm_chain.ask(full_prompt)
+      
+      # Extract Ruby code from the response
+      extract_ruby_code(fixed_content)
+    end
+    
+    def apply_fix_to_file(file_path, fixed_content)
+      # Create backup
+      backup_path = "#{file_path}.backup.#{Time.now.to_i}"
+      File.write(backup_path, File.read(file_path))
+      
+      # Apply fixes
+      File.write(file_path, fixed_content)
+      
+      puts "💾 Backup created: #{backup_path}" if @options[:verbose]
+    end
+    
+    def build_fix_system_message
+      <<~PROMPT
+        You are CodeSage, an expert Ruby developer specializing in code fixes and improvements.
+        
+        Your task is to:
+        1. Analyze the provided Ruby code
+        2. Apply all necessary fixes for issues found
+        3. Improve code quality, security, and performance
+        4. Return ONLY the corrected Ruby code
+        
+        Guidelines:
+        - Fix syntax errors and bugs
+        - Improve Ruby idioms and best practices
+        - Enhance security and performance
+        - Maintain original functionality
+        - Keep the same file structure and class/module names
+        - Add necessary require statements if missing
+        
+        IMPORTANT: Return ONLY the corrected Ruby code without any explanations, 
+        markdown formatting, or additional text. The response should be valid Ruby code that can be saved directly to a file.
+      PROMPT
+    end
+    
+    def build_fix_prompt(file_path, content)
+      <<~PROMPT
+        Please fix the following Ruby file:
+        
+        File: #{file_path}
+        
+        Current code:
+        ```ruby
+        #{content}
+        ```
+        
+        Apply all necessary fixes and improvements while maintaining the original functionality.
+        Return only the corrected Ruby code.
+      PROMPT
+    end
+    
+    def extract_ruby_code(response)
+      # Remove markdown code blocks if present
+      cleaned = response.gsub(/```ruby\s*\n/, '').gsub(/```\s*$/, '')
+      
+      # Remove leading/trailing whitespace
+      cleaned = cleaned.strip
+      
+      # Validate that it looks like Ruby code (basic check)
+      if cleaned.include?('class ') || cleaned.include?('module ') || cleaned.include?('def ') || cleaned.include?('require')
+        cleaned
+      else
+        nil
+      end
+    end
   end
 end 

data/lib/code_sage/version.rb

@@ -1,3 +1,3 @@
 module CodeSage
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: code_sage
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - FuryCow
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-07-01 00:00:00.000000000 Z
+date: 2025-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: llm_chain
@@ -130,6 +130,7 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- ".code_sage.yml"
 - ".github/workflows/gem.yml"
 - ".gitignore"
 - ".rspec"