enhancement: Detect polymorphic associations in generator

lib/generators/avo/resource_generator.rb

@@ -177,9 +177,26 @@
           end
         end
 
-        def fields_from_model_tags
-          tags.each do |name, _|
-            fields[(remove_last_word_from name).pluralize] = {field: "tags"}
+        def detect_polymorphic_associations
+          polymorphic_fields = model_db_columns.keys.group_by { |col| col.gsub(/_type|_id$/, '') }
+                                               .select { |_field, group| group.size == 2 }
+
+          polymorphic_fields.each_key do |field|
+            # Check if the model has a reflection for the polymorphic association
+            if model.reflections.key?(field)
+              reflection = model.reflections[field]
+              associated_classes = reflection.polymorphic? ? reflection.class_name : nil
+            else
+              associated_classes = nil
+            end
+
+            fields[field] = {
+              field: "polymorphic",
+              options: {
+                types: associated_classes ? [associated_classes] : [],
+                comment: associated_classes ? nil : "This is a potential polymorphic association, but the associated classes could not be determined. Please configure manually."
+              }
+            }
           end
         end
 
@@ -256,6 +273,7 @@
           fields_from_model_associations
           fields_from_model_rich_texts
           fields_from_model_tags
+          detect_polymorphic_associations
 
           generated_fields_template
         end
@@ -272,7 +290,14 @@
             options = ""
             field_options[:options].each { |k, v| options += ", #{k}: #{v}" } if field_options[:options].present?
 
-            fields_string += "\n    #{field_string field_name, field_options[:field], options}"
+            if field_options[:field] == "polymorphic"
+              types = field_options[:options][:types] || []
+              comment = field_options[:options][:comment]
+              fields_string += "\n    # #{comment}" if comment
+              fields_string += "\n    # field :#{field_name}, as: :polymorphic, types: #{types.inspect}#{options}"
+            else
+              fields_string += "\n    #{field_string field_name, field_options[:field], options}"
+            end
           end
 
           fields_string

spec/features/avo/generators/resource_generator_spec.rb

@@ -46,6 +46,28 @@
       end
     end
   end
+
+  context "when generating resources for Comment model" do
+    it "generates fields for polymorphic associations" do
+      files = [
+        Rails.root.join("app", "avo", "resources", "comment.rb").to_s,
+        Rails.root.join("app", "controllers", "avo", "comments_controller.rb").to_s
+      ]
+
+      keeping_original_files(files) do
+        Rails::Generators.invoke("avo:resource", ["comment", "-q"], {destination_root: Rails.root})
+
+        expect(File.exist?(files[0])).to be true
+
+        generated_content = File.read(files[0])
+
+        expect(generated_content).to include("# This is a potential polymorphic association, but the associated classes could not be determined. Please configure manually.")
+        expect(generated_content).to include("field :commentable, as: :polymorphic")
+
+        check_files_and_clean_up files
+      end
+    end
+  end
 end
 
 def keeping_original_files(files)