rails-ai-context 4.3.0 → 4.3.2

data/lib/rails_ai_context/tools/get_partial_interface.rb

@@ -212,7 +212,7 @@ module RailsAiContext
         prefixed_basename = basename.start_with?("_") ? basename : "_#{basename}"
         unprefixed_basename = basename.delete_prefix("_")
 
-        extensions = %w[.html.erb .erb .html.haml .haml .html.slim .slim]
+        extensions = %w[.html.erb .erb .html.haml .haml .html.slim .slim .rb .json.jbuilder .jbuilder .turbo_stream.erb]
         candidates = []
 
         # Try prefixed name with various extensions
@@ -332,18 +332,24 @@ module RailsAiContext
       private_class_method def self.find_render_sites(views_dir, partial, root)
         sites = []
         # Build search names: the partial can be referenced multiple ways
+        # Normalize: strip underscore prefix from basename and extensions
         parts = partial.split("/")
         basename = parts.last.delete_prefix("_").sub(/\..*\z/, "")
         dir_prefix = parts[0...-1].join("/")
 
+        # Build the canonical render name (how Rails references partials in render calls)
+        # "shared/_status_badge.html.erb" → "shared/status_badge"
+        # "_status_badge" → "status_badge"
+        canonical = (dir_prefix.empty? ? basename : "#{dir_prefix}/#{basename}")
+
         # Possible render references:
         # render "shared/status_badge"
         # render partial: "shared/status_badge"
         # render "status_badge" (from same directory)
         search_patterns = [
-          partial.delete_prefix("_").sub(/\..*\z/, ""),          # shared/status_badge
-          basename                                                 # status_badge
-        ]
+          canonical,                                                # shared/status_badge
+          basename                                                  # status_badge
+        ].uniq
 
         view_files = Dir.glob(File.join(views_dir, "**", "*.{erb,haml,slim}")).sort
 
@@ -357,7 +363,8 @@ module RailsAiContext
           content.each_line.with_index(1) do |line, line_num|
             search_patterns.each do |search_name|
               # Match render "partial_name" or render partial: "partial_name"
-              next unless line.match?(/render\s.*["']#{Regexp.escape(search_name)}["']/)
+              # Allow content before search_name (e.g. "shared/status_badge" matches "status_badge")
+              next unless line.match?(/render\s.*["'][^"']*#{Regexp.escape(search_name)}["']/)
 
               # For short basename matches, verify directory context
               if search_name == basename && dir_prefix.length > 0

data/lib/rails_ai_context/tools/get_schema.rb

@@ -43,6 +43,7 @@ module RailsAiContext
       )
 
       def self.call(table: nil, detail: "standard", limit: nil, offset: 0, format: "markdown", server_context: nil)
+        set_call_params(table: table, detail: detail)
         schema = cached_context[:schema]
         return text_response("Schema introspection not available. Add :schema to introspectors.") unless schema
         return text_response("Schema introspection not available: #{schema[:error]}") if schema[:error]

data/lib/rails_ai_context/tools/get_stimulus.rb

@@ -17,7 +17,7 @@ module RailsAiContext
           detail: {
             type: "string",
             enum: %w[summary standard full],
-            description: "Detail level. summary: names + counts. standard: names + targets + actions (default). full: everything including values, outlets, classes."
+            description: "Detail level. summary: names + counts. standard: targets + values + actions (default). full: everything including outlets, classes, HTML usage."
           },
           limit: {
             type: "integer",
@@ -73,14 +73,19 @@ module RailsAiContext
 
         case detail
         when "summary"
-          active = controllers.select { |c| (c[:targets] || []).any? || (c[:actions] || []).any? }
-          empty = controllers.reject { |c| (c[:targets] || []).any? || (c[:actions] || []).any? }
+          active = controllers.select { |c| (c[:targets] || []).any? || (c[:actions] || []).any? || (c[:values].is_a?(Hash) ? c[:values] : {}).any? }
+          empty = controllers.reject { |c| (c[:targets] || []).any? || (c[:actions] || []).any? || (c[:values].is_a?(Hash) ? c[:values] : {}).any? }
 
           lines = [ "# Stimulus Controllers (#{total})", "" ]
           active.each do |ctrl|
             targets = (ctrl[:targets] || []).size
+            values = (ctrl[:values].is_a?(Hash) ? ctrl[:values] : {}).size
             actions = (ctrl[:actions] || []).size
-            lines << "- **#{ctrl[:name]}** — #{targets} targets, #{actions} actions"
+            parts = []
+            parts << "#{targets} targets" if targets > 0
+            parts << "#{values} values" if values > 0
+            parts << "#{actions} actions" if actions > 0
+            lines << "- **#{ctrl[:name]}** — #{parts.join(', ')}"
           end
           if empty.any?
             names = empty.map { |c| c[:name] }.join(", ")
@@ -90,13 +95,14 @@ module RailsAiContext
           text_response(lines.join("\n"))
 
         when "standard"
-          active = controllers.select { |c| (c[:targets] || []).any? || (c[:actions] || []).any? }
-          empty = controllers.reject { |c| (c[:targets] || []).any? || (c[:actions] || []).any? }
+          active = controllers.select { |c| (c[:targets] || []).any? || (c[:actions] || []).any? || (c[:values].is_a?(Hash) ? c[:values] : {}).any? }
+          empty = controllers.reject { |c| (c[:targets] || []).any? || (c[:actions] || []).any? || (c[:values].is_a?(Hash) ? c[:values] : {}).any? }
 
           lines = [ "# Stimulus Controllers (#{total})", "" ]
           active.each do |ctrl|
             lines << "## #{ctrl[:name]}"
             lines << "- Targets: #{(ctrl[:targets] || []).join(', ')}" if ctrl[:targets]&.any?
+            lines << "- Values: #{(ctrl[:values].is_a?(Hash) ? ctrl[:values] : {}).map { |k, v| "#{k} (#{v})" }.join(', ')}" if (ctrl[:values].is_a?(Hash) ? ctrl[:values] : {}).any?
             lines << "- Actions: #{(ctrl[:actions] || []).join(', ')}" if ctrl[:actions]&.any?
             if ctrl[:complexity].is_a?(Hash)
               parts = []
@@ -110,7 +116,7 @@ module RailsAiContext
           end
           if empty.any?
             names = empty.map { |c| c[:name] }.join(", ")
-            lines << "_Lifecycle only (no targets/actions): #{names}_"
+            lines << "_Lifecycle only (no targets/values/actions): #{names}_"
           end
 
           # Cross-controller composition

data/lib/rails_ai_context/tools/get_turbo_map.rb

@@ -105,7 +105,11 @@ module RailsAiContext
 
       private_class_method def self.format_summary(model_broadcasts, rb_broadcasts, view_subscriptions, view_frames, warnings, filter_label: nil)
         total_broadcasts = model_broadcasts.size + rb_broadcasts.size
+        turbo_data = cached_context[:turbo]
+        turbo_stream_response_count = turbo_data.is_a?(Hash) && !turbo_data[:error] ? turbo_data[:turbo_stream_responses]&.size.to_i : 0
+
         lines = [ "# Turbo Map", "" ]
+        lines << "- **Turbo Stream responses:** #{turbo_stream_response_count} (controller `.turbo_stream.erb` templates)" if turbo_stream_response_count > 0
         lines << "- **Model broadcasts:** #{model_broadcasts.size} (via `broadcasts`, `broadcasts_to`, etc.)"
         lines << "- **Explicit broadcasts:** #{rb_broadcasts.size} (via `broadcast_*_to` calls in .rb files)"
         lines << "- **Stream subscriptions:** #{view_subscriptions.size} (`turbo_stream_from` in views)"
@@ -195,7 +199,9 @@ module RailsAiContext
           lines << ""
         end
 
-        if model_broadcasts.empty? && rb_broadcasts.empty? && view_subscriptions.empty? && view_frames.empty?
+        has_turbo_stream_responses = turbo_data.is_a?(Hash) && turbo_data[:turbo_stream_responses]&.any?
+
+        if model_broadcasts.empty? && rb_broadcasts.empty? && view_subscriptions.empty? && view_frames.empty? && !has_turbo_stream_responses
           if filter_label
             lines << "_No Turbo usage matching #{filter_label}. Try without filter to see all Turbo Streams and Frames._"
           else
@@ -211,6 +217,31 @@ module RailsAiContext
       private_class_method def self.format_full(model_broadcasts, rb_broadcasts, view_subscriptions, view_frames, warnings, filter_label: nil)
         lines = [ "# Turbo Map (Full Detail)", "" ]
 
+        # Turbo Drive Configuration & Stream Responses
+        turbo_data = cached_context[:turbo]
+        if turbo_data.is_a?(Hash) && !turbo_data[:error]
+          drive_parts = []
+          drive_parts << "morph: #{turbo_data[:morph_meta] ? 'yes' : 'no'}" unless turbo_data[:morph_meta].nil?
+          drive_parts << "permanent elements: #{turbo_data[:permanent_elements].size}" if turbo_data[:permanent_elements]&.any?
+          if turbo_data[:turbo_drive_settings].is_a?(Hash) && turbo_data[:turbo_drive_settings].any?
+            turbo_data[:turbo_drive_settings].each { |k, v| drive_parts << "#{k}: #{v}" }
+          end
+          if drive_parts.any?
+            lines << "## Turbo Drive Configuration"
+            drive_parts.each { |p| lines << "- #{p}" }
+            lines << ""
+          end
+
+          # Turbo Stream responses
+          if turbo_data[:turbo_stream_responses]&.any?
+            lines << "## Turbo Stream Responses (#{turbo_data[:turbo_stream_responses].size})"
+            turbo_data[:turbo_stream_responses].each do |resp|
+              lines << "- `#{resp}`"
+            end
+            lines << ""
+          end
+        end
+
         # Model broadcasts with full context
         if model_broadcasts.any?
           lines << "## Model Broadcasts (#{model_broadcasts.size})"
@@ -289,7 +320,9 @@ module RailsAiContext
           lines << ""
         end
 
-        if model_broadcasts.empty? && rb_broadcasts.empty? && view_subscriptions.empty? && view_frames.empty?
+        has_turbo_stream_responses = turbo_data.is_a?(Hash) && turbo_data[:turbo_stream_responses]&.any?
+
+        if model_broadcasts.empty? && rb_broadcasts.empty? && view_subscriptions.empty? && view_frames.empty? && !has_turbo_stream_responses
           if filter_label
             lines << "_No Turbo usage matching #{filter_label}. Try without filter to see all Turbo Streams and Frames._"
           else

data/lib/rails_ai_context/tools/get_view.rb

@@ -87,7 +87,9 @@ module RailsAiContext
             ctrl_templates.sort.each do |name, meta|
               parts = meta[:partials]&.any? ? " renders: #{meta[:partials].join(', ')}" : ""
               stim = meta[:stimulus]&.any? ? " stimulus: #{meta[:stimulus].join(', ')}" : ""
-              lines << "- #{name} (#{meta[:lines]} lines)#{parts}#{stim}"
+              comps = meta[:components]&.any? ? " components: #{meta[:components].join(', ')}" : ""
+              phlex_tag = meta[:phlex] ? " [phlex]" : ""
+              lines << "- #{name} (#{meta[:lines]} lines#{phlex_tag})#{parts}#{comps}#{stim}"
             end
             ctrl_partials.sort.each do |name, meta|
               lines << "- #{name} (#{meta[:lines]} lines)"
@@ -118,13 +120,29 @@ module RailsAiContext
             lines << "## #{ctrl}/" unless controller && all_dirs.size == 1
             ctrl_templates.sort.each do |name, meta|
               detail_parts = []
-              detail_parts << "renders: #{meta[:partials].join(', ')}" if meta[:partials]&.any?
-              detail_parts << "stimulus: #{meta[:stimulus].join(', ')}" if meta[:stimulus]&.any?
               extra = extract_view_metadata(name)
-              detail_parts << "ivars: #{extra[:ivars].join(', ')}" if extra[:ivars]&.any?
-              detail_parts << "turbo: #{extra[:turbo].join(', ')}" if extra[:turbo]&.any?
+
+              if meta[:phlex]
+                # Phlex views: show components, helpers, stimulus, ivars
+                # Prefer introspector-level data, fall back to extract_view_metadata
+                components = meta[:components]&.any? ? meta[:components] : extra[:components]
+                helpers = meta[:helpers]&.any? ? meta[:helpers] : extra[:helpers]
+                detail_parts << "ivars: #{extra[:ivars].join(', ')}" if extra[:ivars]&.any?
+                detail_parts << "components: #{components.join(', ')}" if components&.any?
+                detail_parts << "helpers: #{helpers.join(', ')}" if helpers&.any?
+                detail_parts << "stimulus: #{meta[:stimulus].join(', ')}" if meta[:stimulus]&.any?
+                detail_parts << "turbo: #{extra[:turbo].join(', ')}" if extra[:turbo]&.any?
+              else
+                # ERB/Haml/Slim views: existing behavior
+                detail_parts << "renders: #{meta[:partials].join(', ')}" if meta[:partials]&.any?
+                detail_parts << "stimulus: #{meta[:stimulus].join(', ')}" if meta[:stimulus]&.any?
+                detail_parts << "ivars: #{extra[:ivars].join(', ')}" if extra[:ivars]&.any?
+                detail_parts << "turbo: #{extra[:turbo].join(', ')}" if extra[:turbo]&.any?
+              end
+
+              phlex_tag = meta[:phlex] ? " [phlex]" : ""
               details = detail_parts.any? ? " — #{detail_parts.join(' | ')}" : ""
-              lines << "- **#{name}** (#{meta[:lines]} lines)#{details}"
+              lines << "- **#{name}** (#{meta[:lines]} lines#{phlex_tag})#{details}"
             end
             ctrl_partials.sort.each do |name, meta|
               fields = meta[:fields]&.any? ? " fields: #{meta[:fields].join(', ')}" : ""
@@ -269,7 +287,7 @@ module RailsAiContext
       # Extract instance variables and Turbo wiring from a view template
       private_class_method def self.extract_view_metadata(relative_path)
         content = read_view_content(relative_path)
-        return { ivars: [], turbo: [] } if content.nil? || content.include?("(file not found)")
+        return { ivars: [], turbo: [], components: [], helpers: [] } if content.nil? || content.include?("(file not found)")
 
         # Instance variables used in template
         ivars = content.scan(/@(\w+)/).flatten.uniq.reject { |v| %w[output_buffer virtual_path _request].include?(v) }.sort
@@ -284,9 +302,47 @@ module RailsAiContext
           turbo << "stream:#{val}" unless val.start_with?('"') || val.start_with?("'") || turbo.any? { |t| t.include?(val) }
         end
 
-        { ivars: ivars, turbo: turbo.uniq }
+        result = { ivars: ivars, turbo: turbo.uniq }
+
+        # For Phlex views (.rb), extract component renders and helper calls
+        if relative_path.end_with?(".rb") && phlex_view_content?(content)
+          result[:components] = extract_phlex_components(content)
+          result[:helpers] = extract_phlex_helpers(content)
+        end
+
+        result
       rescue
-        { ivars: [], turbo: [] }
+        { ivars: [], turbo: [], components: [], helpers: [] }
+      end
+
+      # Detect if content is a Phlex view class
+      private_class_method def self.phlex_view_content?(content)
+        content.match?(/class\s+\S+\s*<\s*\S+/) && content.match?(/def\s+view_template\b/)
+      end
+
+      # Extract component render calls from Phlex Ruby DSL
+      private_class_method def self.extract_phlex_components(content)
+        components = Set.new
+        content.scan(/render[\s(]+([A-Z]\w+(?:::\w+)*)\.new/).each do |match|
+          components << match[0]
+        end
+        components.to_a.sort
+      end
+
+      # Extract helper method calls from Phlex views
+      PHLEX_HELPERS = %w[
+        link_to image_tag content_for button_to form_with form_for
+        content_tag tag number_to_currency number_to_human
+        time_ago_in_words distance_of_time_in_words
+        truncate pluralize raw sanitize dom_id
+      ].freeze
+
+      private_class_method def self.extract_phlex_helpers(content)
+        helpers = []
+        PHLEX_HELPERS.each do |method|
+          helpers << method if content.match?(/\b#{method}\b/)
+        end
+        helpers
       end
 
       # Scan templates that render a partial to extract locals keys

data/lib/rails_ai_context/tools/migration_advisor.rb

@@ -108,7 +108,8 @@ module RailsAiContext
         private
 
         def migration_class_name(action, table, column = nil)
-          parts = [ action.camelize, column&.camelize, "To", table.camelize ].compact
+          preposition = action == "remove" ? "From" : "To"
+          parts = [ action.camelize, column&.camelize, preposition, table.camelize ].compact
           parts.join
         end
 
@@ -130,6 +131,8 @@ module RailsAiContext
           opts = options ? ", #{options}" : ""
           class_name = migration_class_name("add", table, column)
 
+          lines << "**Run:** `bin/rails generate migration #{class_name} #{column}:#{type}`"
+          lines << ""
           lines << "```ruby"
           lines << "# rails generate migration #{class_name} #{column}:#{type}"
           lines << "class #{class_name} < ActiveRecord::Migration[#{rails_version}]"
@@ -165,6 +168,8 @@ module RailsAiContext
           # Check if column is referenced
           col_type = find_column_type(table, column, schema) || type || "string"
 
+          lines << "**Run:** `bin/rails generate migration #{class_name} #{column}:#{col_type}`"
+          lines << ""
           lines << "**Warning:** `remove_column` is irreversible without specifying the column type."
           lines << ""
           lines << "```ruby"
@@ -282,6 +287,9 @@ module RailsAiContext
 
           opts = options ? ", #{options}" : ""
 
+          # Detect original column type from schema for a reversible down method
+          original_type = find_column_type(table, column, cached_context[:schema]) || "string"
+
           lines << "**Warning:** Changing column type may cause data loss if types are incompatible."
           lines << ""
           lines << "```ruby"
@@ -291,8 +299,7 @@ module RailsAiContext
           lines << "  end"
           lines << ""
           lines << "  def down"
-          lines << "    # Specify the original type here"
-          lines << "    change_column :#{table}, :#{column}, :original_type"
+          lines << "    change_column :#{table}, :#{column}, :#{original_type}"
           lines << "  end"
           lines << "end"
           lines << "```"