parse rails model associations

Author: flyerhzm

lib/helpers/parse_rails.rb

@@ -30,5 +30,52 @@
     #   }
     # }
     save_data(:rails_tables, tables)
+
+    associations = []
+    context_stack = []
+
+    within_files Synvert::RAILS_MODEL_FILES do
+      add_callback :module_node, at: 'start' do |node|
+        name = node.constant_path.to_source
+        context_stack.push(name)
+      end
+
+      add_callback :module_node, at: 'end' do |node|
+        context_stack.pop
+      end
+
+      add_callback :class_node, at: 'start' do |node|
+        name = node.constant_path.to_source
+        context_stack.push(name)
+      end
+
+      add_callback :class_node, at: 'end' do |node|
+        context_stack.pop
+      end
+
+      add_callback :call_node, at: 'start' do |node|
+        if node.receiver.nil? && %i[belongs_to has_one has_many has_and_belongs_to_many].include?(node.name)
+          association_name = node.arguments.arguments.first.to_value
+          option_elements = node.arguments.arguments.second&.elements
+          options = {}
+          if option_elements
+            %i[foreign_key foreign_type polymorphic].each do |option_key|
+              option_element = option_elements.find { |element| element.key.value == option_key.to_s }
+              options[option_key] = option_element.value.to_value if option_element
+            end
+          end
+          associations << { class_name: context_stack.join('::'), name: association_name.to_s, type: node.name.to_s, **options }
+        end
+      end
+    end
+    # rails_models
+    # {
+    #   associations: [
+    #     { class_name: "Polymorphic", name: "imageable", type: "belongs_to", polymorphic: true }
+    #     { class_name: "User", name: "organization", type: "belongs_to" },
+    #     { class_name: "User", name: "posts", type: "has_many" }
+    #   ]
+    # }
+    save_data(:rails_models, associations: associations)
   end
 end

spec/helpers/parse_rails_spec.rb

@@ -43,4 +43,34 @@
       }
     })
   end
+
+  it 'saves associations' do
+    rewriter =
+      Synvert::Rewriter.new 'test', 'rails_parse_helper' do
+        call_helper 'rails/parse'
+      end
+
+    FileUtils.mkdir_p('app/models')
+    File.write('app/models/user.rb', <<~EOF)
+      class User < ApplicationRecord
+        belongs_to :organization
+        has_many :posts, dependent: :destroy
+      end
+    EOF
+    File.write('app/models/picture.rb', <<~EOF)
+      class Picture < ApplicationRecord
+        belongs_to :imageable, polymorphic: true
+      end
+    EOF
+
+    rewriter.process
+
+    expect(rewriter.load_data(:rails_models)).to eq({
+      associations: [
+        { class_name: "Picture", name: "imageable", type: "belongs_to", polymorphic: true },
+        { class_name: "User", name: "organization", type: "belongs_to" },
+        { class_name: "User", name: "posts", type: "has_many" }
+      ]
+    })
+  end
 end