lex-codegen 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 410e07dba313fed006a63a77643c2cdf1e03dbe41c17f895725ceb710f10f8fb
-  data.tar.gz: 45a9d37de9308cdfc113b3b83d0bd222b1f5879198d01fd8b8b152fbfafa83e0
+  metadata.gz: 16434586bacb8d4a8cab6c1d886144a76058c07a9cc5b9d75f6b7c2e94159bc8
+  data.tar.gz: 2d442d686396ce9beb409d8597532cc94b22f77fcfa39a802333fc7d64e61844
 SHA512:
-  metadata.gz: f57fbf2f6f9f899ee3cadf673ba72f5c47b58de1ad789a560aa1aaca9dedc314d9d5991f9af795ea58075c38ed6b9dc135df6ada5677de35dcd2c0add19a8972
-  data.tar.gz: 63a20b03008b9e1442b44170725615a3757222816c9ed983d33d88e6106c8a36c1d7ae3903a55e2588afaa3d1f0a5422ce19c274830843dfca4744b5bd1a273d
+  metadata.gz: e9ecfc2219050fc9fcdd70092d59272670d1c2be097c1ef813e401f8176278b952c8af22280892097efdf177f796f47c7d84e68726bef9a3adcf648f8b626357
+  data.tar.gz: d12994780bfb62571cafbc961f420cc7482ad9b0d987fa4050ebe36cf4d01f61219575c143f892f7379f71be31b91d32c73a186403325f515cbab17652888397

data/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [0.1.2] - 2026-03-20
+
+### Added
+- `codegen_fixes` table migration for self-healing pipeline tracking
+- `Runners::AutoFix` for LLM-powered automated extension repair with git branch creation and spec validation
+- Migration registration via `Legion::Data::Local.register_migrations`
+
 ## [0.1.0] - 2026-03-13
 
 ### Added

data/lib/legion/extensions/codegen/local_migrations/20260320000001_create_codegen_fixes.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    create_table?(:codegen_fixes) do
+      primary_key :id
+      String :fix_id, size: 64, null: false, unique: true, index: true
+      String :gem_name, size: 128, null: false, index: true
+      String :runner_class, size: 255
+      String :branch, size: 255
+      column :patch, :text
+      String :status, size: 32, default: 'pending', null: false, index: true
+      TrueClass :specs_passed, default: false
+      column :spec_output, :text
+      DateTime :created_at, null: false, default: Sequel::CURRENT_TIMESTAMP
+    end
+  end
+
+  down do
+    drop_table?(:codegen_fixes)
+  end
+end

data/lib/legion/extensions/codegen/runners/auto_fix.rb

@@ -0,0 +1,165 @@
+# frozen_string_literal: true
+
+require 'securerandom'
+
+module Legion
+  module Extensions
+    module Codegen
+      module Runners
+        module AutoFix
+          extend self
+
+          def auto_fix(gem_name:, runner_class:, error_class:, backtraces:, **)
+            return { success: false, reason: :llm_unavailable } unless defined?(Legion::LLM)
+
+            spec = begin
+              Gem::Specification.find_by_name(gem_name)
+            rescue LoadError
+              nil
+            end
+            return { success: false, reason: :gem_not_found } unless spec
+
+            source_file = locate_source(spec, runner_class)
+            return { success: false, reason: :source_not_found } unless source_file && ::File.exist?(source_file)
+
+            source = ::File.read(source_file)
+            prompt = build_fix_prompt(source, error_class, backtraces)
+            fix_response = Legion::LLM.chat(messages: [{ role: 'user', content: prompt }])
+
+            patch = extract_patch(fix_response)
+            return { success: false, reason: :no_patch_generated } unless patch
+
+            branch = "fix/#{gem_name}-#{::Time.now.to_i}"
+            apply_result = apply_and_test(spec.gem_dir, branch, source_file, patch)
+
+            fix_id = save_fix(gem_name: gem_name, runner_class: runner_class,
+                              branch: branch, patch: patch,
+                              specs_passed: apply_result[:specs_passed],
+                              spec_output: apply_result[:output])
+
+            { success: apply_result[:specs_passed], fix_id: fix_id, branch: branch,
+              specs_passed: apply_result[:specs_passed] }
+          rescue StandardError => e
+            { success: false, reason: :error, message: e.message }
+          end
+
+          def approve_fix(fix_id:, **)
+            update_fix_status(fix_id, 'approved')
+          end
+
+          def reject_fix(fix_id:, **)
+            update_fix_status(fix_id, 'rejected')
+          end
+
+          def list_fixes(status: nil, **)
+            return { fixes: [], count: 0 } unless defined?(Legion::Data::Local)
+
+            ds = Legion::Data::Local.connection[:codegen_fixes]
+            ds = ds.where(status: status) if status
+            fixes = ds.order(Sequel.desc(:created_at)).limit(50).all
+            { fixes: fixes, count: fixes.size }
+          rescue StandardError
+            { fixes: [], count: 0 }
+          end
+
+          private
+
+          def locate_source(spec, runner_class)
+            relative = runner_class.gsub('::', '/').gsub(%r{^Legion/Extensions/}, '')
+                                   .downcase.gsub(/([a-z])([A-Z])/, '\1_\2').downcase
+            path = ::File.join(spec.gem_dir, 'lib', "legion/extensions/#{relative}.rb")
+            ::File.exist?(path) ? path : nil
+          end
+
+          def build_fix_prompt(source, error_class, backtraces)
+            <<~PROMPT
+              You are a Ruby debugging expert. A Legion extension runner is failing with the following error.
+
+              **Error class**: #{error_class}
+
+              **Backtrace**:
+              #{backtraces.first(10).join("\n")}
+
+              **Source code**:
+              ```ruby
+              #{source}
+              ```
+
+              Generate a minimal fix as a unified diff patch. Only change what is necessary to fix the error.
+              Output ONLY the unified diff, no explanation. Start with --- and +++.
+            PROMPT
+          end
+
+          def extract_patch(response)
+            content = response.is_a?(Hash) ? (response[:content] || response[:text]) : response.to_s
+            return nil if content.nil? || content.empty?
+
+            lines = content.lines
+            start = lines.index { |l| l.start_with?('---') }
+            return nil unless start
+
+            lines[start..].join
+          end
+
+          def apply_and_test(gem_dir, branch, source_file, patch)
+            require 'open3'
+
+            ::Dir.chdir(gem_dir) do
+              Open3.capture3('git', 'checkout', '-b', branch)
+              ::File.write("#{source_file}.patch", patch)
+              _out, _err, status = Open3.capture3('git', 'apply', "#{source_file}.patch")
+              ::FileUtils.rm_f("#{source_file}.patch")
+
+              unless status.success?
+                Open3.capture3('git', 'checkout', '-')
+                Open3.capture3('git', 'branch', '-D', branch)
+                return { specs_passed: false, output: 'Patch failed to apply' }
+              end
+
+              stdout, stderr, spec_status = Open3.capture3('bundle', 'exec', 'rspec', '--format', 'progress')
+              output = (stdout + stderr).slice(0, 10_240)
+
+              unless spec_status.success?
+                Open3.capture3('git', 'checkout', '-')
+                Open3.capture3('git', 'branch', '-D', branch)
+              end
+
+              { specs_passed: spec_status.success?, output: output }
+            end
+          rescue StandardError => e
+            { specs_passed: false, output: e.message }
+          end
+
+          def save_fix(gem_name:, branch:, patch:, specs_passed:, runner_class: nil, spec_output: nil)
+            fix_id = SecureRandom.uuid
+            if defined?(Legion::Data::Local)
+              Legion::Data::Local.connection[:codegen_fixes].insert(
+                fix_id: fix_id, gem_name: gem_name, runner_class: runner_class,
+                branch: branch, patch: patch, status: 'pending',
+                specs_passed: specs_passed, spec_output: spec_output&.slice(0, 10_240)
+              )
+            end
+            fix_id
+          rescue StandardError
+            fix_id
+          end
+
+          def update_fix_status(fix_id, new_status)
+            return { success: false, reason: :data_unavailable } unless defined?(Legion::Data::Local)
+
+            updated = Legion::Data::Local.connection[:codegen_fixes]
+                                         .where(fix_id: fix_id)
+                                         .update(status: new_status)
+            if updated.positive?
+              { success: true, fix_id: fix_id, status: new_status }
+            else
+              { success: false, reason: :not_found }
+            end
+          rescue StandardError => e
+            { success: false, reason: :error, message: e.message }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/codegen/version.rb

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module Codegen
-      VERSION = '0.1.1'
+      VERSION = '0.1.2'
     end
   end
 end

data/lib/legion/extensions/codegen.rb

@@ -9,6 +9,7 @@ require_relative 'codegen/helpers/file_writer'
 require_relative 'codegen/runners/generate'
 require_relative 'codegen/runners/template'
 require_relative 'codegen/runners/validate'
+require_relative 'codegen/runners/auto_fix'
 require_relative 'codegen/client'
 
 module Legion
@@ -18,3 +19,10 @@ module Legion
     end
   end
 end
+
+if defined?(Legion::Data::Local)
+  Legion::Data::Local.register_migrations(
+    name: :codegen,
+    path: File.join(__dir__, 'codegen', 'local_migrations')
+  )
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-codegen
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Esity
@@ -130,6 +130,8 @@ files:
 - lib/legion/extensions/codegen/helpers/file_writer.rb
 - lib/legion/extensions/codegen/helpers/spec_generator.rb
 - lib/legion/extensions/codegen/helpers/template_engine.rb
+- lib/legion/extensions/codegen/local_migrations/20260320000001_create_codegen_fixes.rb
+- lib/legion/extensions/codegen/runners/auto_fix.rb
 - lib/legion/extensions/codegen/runners/generate.rb
 - lib/legion/extensions/codegen/runners/template.rb
 - lib/legion/extensions/codegen/runners/validate.rb