trains 0.0.7 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 553d908187ec4b9cc7e3f06616e096e8a341061ec3a345bce5018c6e1578e4ae
-  data.tar.gz: 2201e5acb6d148926f4602a885d3588cb902496b788d8565330c9ed3588e3243
+  metadata.gz: be756e07f687c79e259385beb680f5749c54a880273be056ba83a717aa821887
+  data.tar.gz: da3fc5403f7844919f9bfd3ac49ef37479e7f4149328e851d1b1e654353398af
 SHA512:
-  metadata.gz: e4b2ee11d9e1af06232169f886a3695f14e1d2c9418d4449706af4a872932f0efa408a43b3a4f3cd2237a133f3720adc1e94ad6e313facdefe917a401b154134
-  data.tar.gz: dc2e1d251fc0cdc28d2cc340484eeae228039deab55fbca8b1c61988417d01ce5aa921a7a84367fd661f841e839718ea74048fb7738e740328bb0b969e3907a0
+  metadata.gz: 3a389ed84aedba71b13061c4233389d3008bd895a532f506a9f492a6d6aaa54ae96708d27f0e4cba9269f1666ef8f614c5ebf467a8bbf70c63595b449e2bcc0f
+  data.tar.gz: be28d384e28be094161bcdb28105a4bef9dd2d7f86978936f554bf699a205c1953a88bd1de071c1005f2ec4d5c54e91bf6ace9e20f7f917bc8204b46ced1a431

data/lib/trains/version.rb

@@ -1,3 +1,3 @@
 module Trains
-  VERSION = '0.0.7'
+  VERSION = '0.0.8'
 end

data/lib/trains/visitor/migration.rb

@@ -1,10 +1,10 @@
-require "yaml"
+require 'yaml'
 
 module Trains
   module Visitor
     # Visitor that parses DB migration and associates them with Rails models
     class Migration < Base
-      def_node_matcher :send_node?, "(send nil? ...)"
+      def_node_matcher :send_node?, '(send nil? ...)'
       attr_reader :is_migration, :model
 
       def initialize
@@ -20,7 +20,7 @@ module Trains
       end
 
       def on_class(node)
-        unless node.parent_class.source.include? "ActiveRecord::Migration"
+        unless node.parent_class.source.include? 'ActiveRecord::Migration'
           return
         end
 
@@ -63,49 +63,51 @@ module Trains
           add_column
           remove_column
           change_column
+          add_index
+        ]
+        column_modifiers = %i[
+          add_column
+          change_column
+          remove_column
         ]
-        block_type_modifier = false
 
         method_name = node.method_name
         return unless allowed_method_names.include? method_name
         return if node.body.nil?
 
         table_modifier =
-          if node.body.children[0] == nil
-            block_type_modifier = false
-            # if table modifier is a one-liner method call
+          # if table modifier is a one-liner method call
+          if node.body.children[0].nil?
             node.body.children[1]
           elsif node.body.children[0].block_type?
-            block_type_modifier = true
             # if table modifier is in a block
             node.body.children[0].method_name
+          elsif node.body.children[0].send_type?
+            node.body.children[0].method_name
           end
         return unless allowed_table_modifiers.include? table_modifier
 
         @table_modifier = table_modifier
 
-        # Get the name of the table being modified
-        if block_type_modifier
-          raw_table_name =
-            node.body.children[0].children[0].children[2].value.to_s
-          @table_name = raw_table_name.singularize.camelize
+        node.each_descendant(:send) do |send_node|
+          if allowed_table_modifiers.include?(send_node.method_name)
+            raw_table_name = send_node.arguments[0]
+            @table_name = raw_table_name.value.to_s.singularize.camelize
+            if column_modifiers.include?(send_node.method_name)
+              @fields.append(DTO::Field.new(send_node.arguments[1].value,
+                                            send_node.arguments[2]&.value))
+            end
 
-          node.body.children[0].children[2].each_child_node do |child|
-            process_migration_field(child)
+            next
           end
-        else
-          raw_table_name = node.body.children[2].value.to_s
-          @table_name = raw_table_name.singularize.camelize
 
-          field_name = node.body.children[3]&.value
-          field_type = node.body.children[4]&.value
-          @fields.append(DTO::Field.new(field_name, field_type))
+          next if allowed_table_modifiers.include?(send_node.method_name)
+
+          parse_migration_field(send_node)
         end
       end
 
-      def process_migration_field(node)
-        return unless node.send_type?
-
+      def parse_migration_field(node)
         if node.children[1] == :timestamps
           @fields.append(DTO::Field.new(:created_at, :datetime))
           @fields.append(DTO::Field.new(:updated_at, :datetime))

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trains
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-22 00:00:00.000000000 Z
+date: 2023-05-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport