rails-ai-context 4.3.2 → 4.4.0

data/lib/rails_ai_context/introspectors/multi_database_introspector.rb

@@ -43,7 +43,8 @@ module RailsAiContext
         else
           parse_database_yml
         end
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] discover_databases failed: #{e.message}" if ENV["DEBUG"]
         parse_database_yml
       end
 
@@ -56,7 +57,8 @@ module RailsAiContext
         else
           []
         end
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] discover_replicas failed: #{e.message}" if ENV["DEBUG"]
         []
       end
 
@@ -65,8 +67,30 @@ module RailsAiContext
         return nil unless File.exist?(database_yml)
 
         content = File.read(database_yml)
-        { detected: true, note: "Sharding configuration found in database.yml" } if content.match?(/shard/i)
-      rescue
+        return nil unless content.match?(/shard/i)
+
+        result = { detected: true }
+        # Extract shard names from database.yml
+        shard_names = content.scan(/^\s{4,}(\w*shard\w*):/).flatten.uniq
+        result[:shard_names] = shard_names if shard_names.any?
+
+        # Extract shard config from model source (connects_to shards: { ... })
+        models_dir = File.join(root, "app/models")
+        if Dir.exist?(models_dir)
+          Dir.glob(File.join(models_dir, "**/*.rb")).each do |path|
+            src = File.read(path) rescue next
+            if (match = src.match(/connects_to\s+.*shards:\s*\{([^}]+)\}/m))
+              shard_keys = match[1].scan(/(\w+):/).flatten
+              result[:shard_keys] = shard_keys if shard_keys.any?
+              result[:shard_count] = shard_keys.size
+              break
+            end
+          end
+        end
+
+        result
+      rescue => e
+        $stderr.puts "[rails-ai-context] detect_sharding failed: #{e.message}" if ENV["DEBUG"]
         nil
       end
 
@@ -90,7 +114,8 @@ module RailsAiContext
           if content.match?(/connected_to\b/)
             connections << { model: model_name, uses_connected_to: true } unless connections.any? { |c| c[:model] == model_name }
           end
-        rescue
+        rescue => e
+          $stderr.puts "[rails-ai-context] detect_model_connections failed: #{e.message}" if ENV["DEBUG"]
           next
         end
 
@@ -105,10 +130,11 @@ module RailsAiContext
         databases = []
         current_env = defined?(Rails) ? Rails.env : "development"
         in_env = false
-        skip_keys = %w[adapter database host port username password encoding pool timeout socket url]
+        skip_keys = %w[adapter database host port username password encoding pool timeout socket url replica]
+        current_db = nil
 
         content.each_line do |line|
-          if line.match?(/\A#{current_env}:/)
+          if line.match?(/\A#{Regexp.escape(current_env)}:/)
             in_env = true
             next
           elsif line.match?(/\A\w+:/) && in_env
@@ -117,14 +143,32 @@ module RailsAiContext
 
           next unless in_env
 
-          if line.match?(/\A\s{2}(\w+):/) && !line.include?("<<")
-            db_name = line.strip.chomp(":")
-            databases << { name: db_name } unless skip_keys.include?(db_name)
+          # 2-space indent = database name (primary, cache, etc.) or flat config key
+          if (match = line.match(/\A\s{2}(\w+):\s*(.*)/)) && !line.include?("<<")
+            key = match[1]
+            value = match[2].strip
+            if skip_keys.include?(key)
+              # Flat config (single-db): extract adapter/database inline
+              if key == "adapter" && databases.empty?
+                databases << { name: "primary", adapter: value }
+              end
+            else
+              # This is a named database (multi-db config)
+              current_db = { name: key }
+              databases << current_db
+            end
+          # 4-space indent = settings under a named database
+          elsif current_db && (match = line.match(/\A\s{4}(\w+):\s*(.*)/))
+            key = match[1]
+            value = match[2].strip
+            current_db[:adapter] = value if key == "adapter"
+            current_db[:replica] = true if key == "replica" && value == "true"
           end
         end
 
         databases
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] parse_database_yml failed: #{e.message}" if ENV["DEBUG"]
         []
       end
 
@@ -136,7 +180,8 @@ module RailsAiContext
         else
           name
         end
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] anonymize_db_name failed: #{e.message}" if ENV["DEBUG"]
         "external"
       end
     end

data/lib/rails_ai_context/introspectors/performance_introspector.rb

@@ -91,7 +91,8 @@ module RailsAiContext
             scopes_with_includes: content.scan(/scope\s+:\w+.*\.includes\(/).any?,
             content: content
           }
-        rescue
+        rescue => e
+          $stderr.puts "[rails-ai-context] load_model_data failed: #{e.message}" if ENV["DEBUG"]
           nil
         end
       end
@@ -188,7 +189,9 @@ module RailsAiContext
         missing = []
 
         schema_data.each do |table_name, table|
-          table[:columns].each do |col|
+          columns = table[:columns]
+
+          columns.each do |col|
             next unless col[:name].end_with?("_id")
 
             indexed = table[:indexes].any? { |idx| idx.include?(col[:name]) }
@@ -196,11 +199,31 @@ module RailsAiContext
             next if col[:type] == "references"
             next if indexed
 
-            missing << {
-              table: table_name,
-              column: col[:name],
-              suggestion: "add_index :#{table_name}, :#{col[:name]}"
-            }
+            # Check for polymorphic association (_type column alongside _id)
+            base_name = col[:name].sub(/_id\z/, "")
+            type_col = columns.find { |c| c[:name] == "#{base_name}_type" }
+
+            if type_col
+              # Polymorphic: need compound index on [type, id]
+              compound_indexed = table[:indexes].any? { |idx|
+                idx_str = idx.to_s
+                idx_str.include?("#{base_name}_type") && idx_str.include?("#{base_name}_id")
+              }
+              unless compound_indexed
+                missing << {
+                  table: table_name,
+                  column: "#{base_name}_type, #{base_name}_id",
+                  polymorphic: true,
+                  suggestion: "add_index :#{table_name}, [:#{base_name}_type, :#{base_name}_id]"
+                }
+              end
+            else
+              missing << {
+                table: table_name,
+                column: col[:name],
+                suggestion: "add_index :#{table_name}, :#{col[:name]}"
+              }
+            end
           end
         end
 
@@ -218,7 +241,8 @@ module RailsAiContext
             content = File.read(path)
             match = content.match(/class\s+(\w+)\s*<\s*ApplicationRecord/)
             match[1] if match
-          rescue
+          rescue => e
+            $stderr.puts "[rails-ai-context] detect_model_all_in_controllers failed: #{e.message}" if ENV["DEBUG"]
             nil
           end
         else
@@ -270,7 +294,8 @@ module RailsAiContext
             associations: has_many_assocs,
             suggestion: "Consider eager loading when rendering #{class_name} with associations: #{has_many_assocs.join(", ")}"
           }
-        rescue
+        rescue => e
+          $stderr.puts "[rails-ai-context] detect_eager_load_candidates failed: #{e.message}" if ENV["DEBUG"]
           next
         end
 

data/lib/rails_ai_context/introspectors/rake_task_introspector.rb

@@ -50,11 +50,21 @@ module RailsAiContext
 
           if (t_match = line.match(/^\s*task\s+:(\w+)/))
             name = (current_namespace + [ t_match[1] ]).join(":")
-            tasks << {
+            entry = {
               name: name,
               description: last_desc,
               file: relative
-            }.compact
+            }
+            # Extract task dependencies (=> [:dep1, :dep2] or => :dep)
+            if (dep_match = line.match(/=>\s*(?:\[([^\]]+)\]|:(\w+))/))
+              deps = dep_match[1] ? dep_match[1].scan(/:(\w+)/).flatten : [ dep_match[2] ]
+              entry[:dependencies] = deps if deps.any?
+            end
+            # Extract task arguments (task :name, [:arg1, :arg2])
+            if (args_match = line.match(/task\s+:#{Regexp.escape(t_match[1])}\s*,\s*\[([^\]]+)\]/))
+              entry[:args] = args_match[1].scan(/:(\w+)/).flatten
+            end
+            tasks << entry.compact
             last_desc = nil
           end
         end

data/lib/rails_ai_context/introspectors/route_introspector.rb

@@ -14,12 +14,14 @@ module RailsAiContext
       # @return [Hash] routes grouped by controller
       def call
         routes = extract_routes
+        root = routes.find { |r| r[:path] == "/" && r[:verb]&.include?("GET") }
 
         {
           total_routes: routes.size,
           by_controller: group_by_controller(routes),
           api_namespaces: detect_api_namespaces(routes),
-          mounted_engines: detect_mounted_engines
+          mounted_engines: detect_mounted_engines,
+          root_route: root ? "#{root[:controller]}##{root[:action]}" : nil
         }
       end
 
@@ -33,28 +35,42 @@ module RailsAiContext
           next if route.respond_to?(:internal?) && route.internal?
           next if route.defaults[:controller].blank?
 
-          {
+          route_path = route.path.spec.to_s.gsub("(.:format)", "")
+          action = route.defaults[:action]
+
+          entry = {
             verb: route.verb.presence || "ANY",
-            path: route.path.spec.to_s.gsub("(.:format)", ""),
+            path: route_path,
             controller: route.defaults[:controller],
-            action: route.defaults[:action],
+            action: action,
             name: route.name,
             constraints: extract_constraints(route)
-          }.compact
+          }
+
+          params = route_path.scan(/:(\w+)/).flatten
+          entry[:params] = params if params.any?
+
+          entry[:restful] = %w[index show new create edit update destroy].include?(action)
+
+          entry.compact
         end
       end
 
       def extract_constraints(route)
         constraints = route.constraints.to_s
         constraints.empty? ? nil : constraints
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_constraints failed: #{e.message}" if ENV["DEBUG"]
         nil
       end
 
       def group_by_controller(routes)
         routes.group_by { |r| r[:controller] }.transform_values do |controller_routes|
           controller_routes.map do |r|
-            { verb: r[:verb], path: r[:path], action: r[:action], name: r[:name] }.compact
+            entry = { verb: r[:verb], path: r[:path], action: r[:action], name: r[:name] }
+            entry[:params] = r[:params] if r[:params]
+            entry[:restful] = r[:restful] unless r[:restful].nil?
+            entry.compact
           end
         end
       end
@@ -77,7 +93,8 @@ module RailsAiContext
               engine: engine_class.name,
               path: r.path.spec.to_s
             }
-          rescue
+          rescue => e
+            $stderr.puts "[rails-ai-context] detect_mounted_engines failed: #{e.message}" if ENV["DEBUG"]
             nil
           end
       end

data/lib/rails_ai_context/introspectors/schema_introspector.rb

@@ -33,13 +33,15 @@ module RailsAiContext
 
       def active_record_connected?
         defined?(ActiveRecord::Base) && ActiveRecord::Base.connected?
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] active_record_connected? failed: #{e.message}" if ENV["DEBUG"]
         false
       end
 
       def adapter_name
         ActiveRecord::Base.connection.adapter_name
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] adapter_name failed: #{e.message}" if ENV["DEBUG"]
         "unknown"
       end
 
@@ -106,7 +108,8 @@ module RailsAiContext
             on_update: fk.on_update
           }.compact
         end
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_foreign_keys failed: #{e.message}" if ENV["DEBUG"]
         [] # Some adapters don't support foreign_keys
       end
 
@@ -136,7 +139,8 @@ module RailsAiContext
         end
 
         defaults
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] parse_schema_defaults_for_table failed: #{e.message}" if ENV["DEBUG"]
         {}
       end
 
@@ -207,12 +211,20 @@ module RailsAiContext
               col[:default] = raw.start_with?('"') ? raw[1..-2] : raw
             end
             col[:array] = true if line.include?("array: true")
+            if (comment_match = line.match(/comment:\s*"([^"]+)"/))
+              col[:comment] = comment_match[1]
+            end
             tables[current_table][:columns] << col
           elsif current_table && (match = line.match(/t\.index\s+\[([^\]]*)\]/))
             cols = match[1].scan(/["'](\w+)["']/).flatten
             unique = line.include?("unique: true")
             idx_name = line.match(/name:\s*["']([^"']+)["']/)&.send(:[], 1)
             tables[current_table][:indexes] << { name: idx_name, columns: cols, unique: unique }.compact if cols.any?
+          elsif current_table && (match = line.match(/t\.index\s+"([^"]+)"/))
+            expression = match[1]
+            idx_name = line.match(/name:\s*["']([^"']+)["']/)&.send(:[], 1)
+            unique = line.include?("unique: true")
+            tables[current_table][:indexes] << { name: idx_name, columns: [ expression ], unique: unique, expression: true }.compact
           elsif (match = line.match(/add_index\s+"(\w+)",\s+(.+)/))
             table_name = match[1]
             rest = match[2]
@@ -333,7 +345,8 @@ module RailsAiContext
         end
 
         constraints
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] parse_check_constraints failed: #{e.message}" if ENV["DEBUG"]
         []
       end
 
@@ -352,7 +365,8 @@ module RailsAiContext
         end
 
         enums
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] parse_enum_types failed: #{e.message}" if ENV["DEBUG"]
         []
       end
 
@@ -382,7 +396,8 @@ module RailsAiContext
         end
 
         columns
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] parse_generated_columns failed: #{e.message}" if ENV["DEBUG"]
         []
       end
 
@@ -496,6 +511,29 @@ module RailsAiContext
               col[:type] = new_type if col
             end
 
+          # change_column_default :table, :column, default_value
+          elsif (match = stripped.match(/change_column_default\s+[:"'](\w+)['"']?,\s*[:"'](\w+)/))
+            table_name, col_name = match[1], match[2]
+            if tables[table_name]
+              col = tables[table_name][:columns].find { |c| c[:name] == col_name }
+              if col
+                default_match = stripped.match(/,\s*(?:from:\s*[^,]+,\s*)?to:\s*("[^"]*"|\d+(?:\.\d+)?|true|false|nil)/)
+                default_match ||= stripped.match(/,\s*[:"']\w+['"']?,\s*("[^"]*"|\d+(?:\.\d+)?|true|false|nil)\s*\z/)
+                if default_match
+                  raw = default_match[1]
+                  col[:default] = raw == "nil" ? nil : (raw.start_with?('"') ? raw[1..-2] : raw)
+                end
+              end
+            end
+
+          # change_column_null :table, :column, nullable
+          elsif (match = stripped.match(/change_column_null\s+[:"'](\w+)['"']?,\s*[:"'](\w+)['"']?,\s*(true|false)/))
+            table_name, col_name, nullable = match[1], match[2], match[3]
+            if tables[table_name]
+              col = tables[table_name][:columns].find { |c| c[:name] == col_name }
+              col[:null] = (nullable == "true") if col
+            end
+
           # rename_table :old, :new
           elsif (match = stripped.match(/rename_table\s+[:"'](\w+)['"']?,\s*[:"'](\w+)/))
             old_name, new_name = match[1], match[2]

data/lib/rails_ai_context/introspectors/seeds_introspector.rb

@@ -42,8 +42,10 @@ module RailsAiContext
           uses_insert_all: content.match?(/\.insert_all/),
           uses_faker: content.match?(/Faker::/),
           uses_factory_bot: content.match?(/FactoryBot/),
+          uses_csv: content.match?(/CSV\.|require.*csv/i),
           loads_directory: content.match?(/Dir\[|Dir\.glob|load.*seeds/),
-          environment_conditional: content.match?(/Rails\.env/)
+          environment_conditional: content.match?(/Rails\.env/),
+          has_ordering: content.match?(/load.*order|require.*order|seeds.*\d+/i)
         }
       rescue => e
         { exists: false, error: e.message }
@@ -78,7 +80,8 @@ module RailsAiContext
             model_name = match[0]
             models << model_name unless non_models.include?(model_name)
           end
-        rescue
+        rescue => e
+          $stderr.puts "[rails-ai-context] detect_seeded_models failed: #{e.message}" if ENV["DEBUG"]
           next
         end
 

data/lib/rails_ai_context/introspectors/stimulus_introspector.rb

@@ -19,6 +19,17 @@ module RailsAiContext
           parse_controller(path, controllers_dir)
         end
 
+        # Merge action bindings from views into each controller's data
+        bindings = extract_action_bindings
+        if bindings.any?
+          controllers.each do |ctrl|
+            next unless ctrl[:name]
+            if (ctrl_bindings = bindings[ctrl[:name]])
+              ctrl[:action_bindings] = ctrl_bindings
+            end
+          end
+        end
+
         {
           controllers: controllers,
           cross_controller_composition: extract_cross_controller_composition(root)
@@ -34,18 +45,22 @@ module RailsAiContext
         name = relative.sub(/_controller\.(js|ts)\z/, "").tr("/", "--")
         content = File.read(path)
 
+        outlets = extract_outlets(content)
+
         {
           name: name,
           file: relative,
           targets: extract_targets(content),
           values: extract_values(content),
           actions: extract_actions(content),
-          outlets: extract_outlets(content),
+          outlets: outlets,
+          outlet_controllers: outlets.any? ? outlets.each_with_object({}) { |o, h| h[o] = "#{o}-controller" } : nil,
           classes: extract_classes(content),
+          lifecycle: extract_lifecycle(content),
           import_graph: extract_import_graph(content),
           complexity: extract_complexity(content),
           turbo_event_listeners: extract_turbo_event_listeners(content)
-        }
+        }.compact
       rescue => e
         { name: File.basename(path), error: e.message }
       end
@@ -127,7 +142,8 @@ module RailsAiContext
           end
         end
         imports
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_import_graph failed: #{e.message}" if ENV["DEBUG"]
         []
       end
 
@@ -138,17 +154,53 @@ module RailsAiContext
         methods = content.scan(/^\s+(?:async\s+)?(\w+)\s*\([^)]*\)\s*\{/).flatten
         method_count = methods.count { |m| !JS_KEYWORDS.include?(m) }
         { loc: loc, method_count: method_count }
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_complexity failed: #{e.message}" if ENV["DEBUG"]
         { loc: 0, method_count: 0 }
       end
 
       def extract_turbo_event_listeners(content)
         events = content.scan(/["']turbo:([\w:-]+)["']/).flatten.uniq
         events.map { |e| "turbo:#{e}" }
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_turbo_event_listeners failed: #{e.message}" if ENV["DEBUG"]
         []
       end
 
+      def extract_lifecycle(content)
+        hooks = content.scan(/\b(connect|disconnect|initialize)\s*\(\s*\)/).flatten.uniq
+        hooks.any? ? hooks : nil
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_lifecycle failed: #{e.message}" if ENV["DEBUG"]
+        nil
+      end
+
+      def extract_action_bindings
+        bindings = Hash.new { |h, k| h[k] = [] }
+        view_dirs = [ File.join(app.root, "app", "views"), File.join(app.root, "app", "components") ]
+        view_dirs.each do |dir|
+          next unless Dir.exist?(dir)
+          Dir.glob(File.join(dir, "**", "*.{erb,haml,slim}")).each do |path|
+            content = File.read(path, encoding: "UTF-8", invalid: :replace, undef: :replace)
+            content.scan(/data-action=["']([^"']+)["']/).each do |match|
+              match[0].split(/\s+/).each do |binding_str|
+                # Format: event->controller#method
+                if (m = binding_str.match(/(?:(\w+)->)?(\w[\w-]*)#(\w+)/))
+                  controller = m[2]
+                  method = m[3]
+                  event = m[1]
+                  bindings[controller] << { event: event, method: method }.compact
+                end
+              end
+            end
+          end
+        end
+        bindings.transform_values(&:uniq)
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_action_bindings failed: #{e.message}" if ENV["DEBUG"]
+        {}
+      end
+
       def extract_cross_controller_composition(root)
         views_dir = File.join(root, "app/views")
         return [] unless Dir.exist?(views_dir)
@@ -166,7 +218,8 @@ module RailsAiContext
         end
 
         compositions.uniq
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] extract_cross_controller_composition failed: #{e.message}" if ENV["DEBUG"]
         []
       end
     end

data/lib/rails_ai_context/introspectors/test_introspector.rb

@@ -26,7 +26,9 @@ module RailsAiContext
           ci_config: detect_ci,
           coverage: detect_coverage,
           factory_traits: detect_factory_traits,
-          test_count_by_category: detect_test_count_by_category
+          test_count_by_category: detect_test_count_by_category,
+          shared_examples: detect_shared_examples,
+          database_cleaner: detect_database_cleaner
         }
       rescue => e
         { error: e.message }
@@ -113,7 +115,8 @@ module RailsAiContext
             file = path.sub("#{root}/", "")
             factories = File.read(path).scan(/factory\s+:(\w+)/).flatten
             names[file] = factories if factories.any?
-          rescue
+          rescue => e
+            $stderr.puts "[rails-ai-context] detect_factory_names failed: #{e.message}" if ENV["DEBUG"]
             next
           end
           return names if names.any?
@@ -200,7 +203,8 @@ module RailsAiContext
         content = File.read(gemfile_lock)
         return "simplecov" if content.include?("simplecov (")
         nil
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] detect_coverage failed: #{e.message}" if ENV["DEBUG"]
         nil
       end
 
@@ -219,7 +223,47 @@ module RailsAiContext
           return traits if traits.any?
         end
         nil
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] detect_factory_traits failed: #{e.message}" if ENV["DEBUG"]
+        nil
+      end
+
+      def detect_shared_examples
+        shared = []
+        %w[spec test].each do |base|
+          support_dir = File.join(root, base, "support")
+          next unless Dir.exist?(support_dir)
+          Dir.glob(File.join(support_dir, "**/*.rb")).each do |path|
+            content = File.read(path) rescue next
+            content.scan(/(?:shared_examples|shared_context|shared_examples_for)\s+["']([^"']+)["']/).each do |m|
+              shared << { name: m[0], file: path.sub("#{root}/", "") }
+            end
+          end
+        end
+        shared.sort_by { |s| s[:name] }
+      rescue => e
+        $stderr.puts "[rails-ai-context] detect_shared_examples failed: #{e.message}" if ENV["DEBUG"]
+        []
+      end
+
+      def detect_database_cleaner
+        gemfile_lock = File.join(root, "Gemfile.lock")
+        return nil unless File.exist?(gemfile_lock)
+        content = File.read(gemfile_lock)
+        if content.include?("database_cleaner")
+          strategy = nil
+          %w[spec/rails_helper.rb spec/spec_helper.rb test/test_helper.rb].each do |helper|
+            path = File.join(root, helper)
+            next unless File.exist?(path)
+            helper_content = File.read(path)
+            if (match = helper_content.match(/DatabaseCleaner\.strategy\s*=\s*:(\w+)/))
+              strategy = match[1]
+            end
+          end
+          { detected: true, strategy: strategy }.compact
+        end
+      rescue => e
+        $stderr.puts "[rails-ai-context] detect_database_cleaner failed: #{e.message}" if ENV["DEBUG"]
         nil
       end
 
@@ -234,7 +278,8 @@ module RailsAiContext
           end
         end
         counts
-      rescue
+      rescue => e
+        $stderr.puts "[rails-ai-context] detect_test_count_by_category failed: #{e.message}" if ENV["DEBUG"]
         {}
       end
     end