trains 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be756e07f687c79e259385beb680f5749c54a880273be056ba83a717aa821887
-  data.tar.gz: da3fc5403f7844919f9bfd3ac49ef37479e7f4149328e851d1b1e654353398af
+  metadata.gz: 2862732986e24d80f6fe47f3035e206cc033354767902260e7003c983b30ce3d
+  data.tar.gz: 0fd7fbb15a6e12d2bdc61f8753b0f9c7fbe85f9a2a43639089c166364255a27e
 SHA512:
-  metadata.gz: 3a389ed84aedba71b13061c4233389d3008bd895a532f506a9f492a6d6aaa54ae96708d27f0e4cba9269f1666ef8f614c5ebf467a8bbf70c63595b449e2bcc0f
-  data.tar.gz: be28d384e28be094161bcdb28105a4bef9dd2d7f86978936f554bf699a205c1953a88bd1de071c1005f2ec4d5c54e91bf6ace9e20f7f917bc8204b46ced1a431
+  metadata.gz: 151eeb6f6a3c4a1494d40e8e231c6dcaa2b5f66673e59fbf26623d5054a74e514f3d02385b752f2d1cb20145c3ce06794286de3f34b3b11b002d4991419b7d10
+  data.tar.gz: 555e8794465a0f4ea87e05ebad167d3248e4263cb4a9de02736b3003fb06eb40f8fd20a88cda681e78f1803294d025a815c9fed80c1f00a62a0799b3bfb5d39e

data/lib/trains/version.rb

@@ -1,3 +1,3 @@
 module Trains
-  VERSION = '0.0.8'
+  VERSION = '0.0.9'.freeze
 end

data/lib/trains/visitor/migration.rb

@@ -5,18 +5,26 @@ module Trains
     # Visitor that parses DB migration and associates them with Rails models
     class Migration < Base
       def_node_matcher :send_node?, '(send nil? ...)'
-      attr_reader :is_migration, :model
-
+      attr_reader :is_migration, :model, :result
+
+      ALLOWED_METHOD_NAMES = %i[change up down].freeze
+      ALLOWED_TABLE_MODIFIERS = %i[
+        create_table
+        change_table
+        update_column
+        add_index
+      ].freeze
+      COLUMN_MODIFIERS = %i[
+        add_column
+        change_column
+        remove_column
+      ].freeze
+
+      # skipcq: RB-LI1087
       def initialize
-        @model = nil
-        @table_modifier = nil
-        @table_name = nil
-        @is_class = false
-        @is_migration = false
-        @class_name = nil
-        @fields = []
-
-        @scope = { class: nil, method: nil, send: nil }
+        @result = []
+        @migration_class = nil
+        @migration_version = nil
       end
 
       def on_class(node)
@@ -27,16 +35,7 @@ module Trains
         @migration_class = node.children.first.source
         @migration_version = extract_version(node.parent_class.source)
 
-        process_node(node.body)
-      end
-
-      def result
-        DTO::Migration.new(
-          table_name: @table_name,
-          modifier: @table_modifier,
-          fields: @fields,
-          version: @migration_version
-        )
+        process_def_node(node.body)
       end
 
       private
@@ -48,75 +47,94 @@ module Trains
         match.to_s.to_f
       end
 
-      def process_node(node)
-        return unless node.def_type?
-
-        process_def_node(node)
-      end
-
       def process_def_node(node)
-        allowed_method_names = %i[change up down]
-        allowed_table_modifiers = %i[
-          create_table
-          change_table
-          update_column
-          add_column
-          remove_column
-          change_column
-          add_index
-        ]
-        column_modifiers = %i[
-          add_column
-          change_column
-          remove_column
-        ]
+        return unless node.def_type?
 
         method_name = node.method_name
-        return unless allowed_method_names.include? method_name
+        unless ALLOWED_METHOD_NAMES.include?(method_name) || COLUMN_MODIFIERS.include?(method_name)
+          return
+        end
         return if node.body.nil?
 
-        table_modifier =
-          # 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?
-            # 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
+        case node.body.type
+        when :send
+          @result << parse_one_liner_migration(node.body)
+        when :begin
+          if node.body.children.map(&:type).include?(:block)
+            migration = parse_block_migration(node)
+            @result = [*@result, *migration] if migration
+          else
+            node.body.each_descendant(:send) do |send_node|
+              migration = parse_one_liner_migration(send_node)
+              @result << migration if migration
+            end
           end
-        return unless allowed_table_modifiers.include? table_modifier
+        when :block
+          @result = [*@result, *parse_block_migration(node)]
+        else
+          LOGGER.debug(node)
+        end
+      end
 
-        @table_modifier = table_modifier
+      def parse_one_liner_migration(node)
+        return unless COLUMN_MODIFIERS.include?(node.method_name)
 
-        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
+        arguments = node.arguments
+        table_name = arguments[0].value.to_s.singularize.camelize
+        column_name = arguments[1].value
+        type = arguments[2].value unless node.method_name == :remove_column
 
-            next
-          end
+        DTO::Migration.new(
+          table_name,
+          node.method_name,
+          [DTO::Field.new(column_name, type)],
+          @migration_version
+        )
+      end
+
+      def parse_block_migration(node)
+        migrations = []
+        # Visit every send_node that performs DDL tasks
+        node.each_descendant do |ddl_node|
+          fields = []
 
-          next if allowed_table_modifiers.include?(send_node.method_name)
+          if ddl_node.block_type?
+            next unless ALLOWED_TABLE_MODIFIERS.include?(ddl_node.method_name)
 
-          parse_migration_field(send_node)
+            table_name = ddl_node.send_node.arguments[0].value.to_s.singularize.camelize
+            ddl_node.body.each_descendant(:send) do |send_node|
+              fields = [*fields, *parse_migration_field(send_node)]
+            end
+
+            migrations << DTO::Migration.new(
+              table_name,
+              ddl_node.method_name,
+              fields,
+              @migration_version
+            )
+          elsif ddl_node.send_type?
+            migrations = [*migrations, *parse_one_liner_migration(ddl_node)]
+          end
         end
+
+        migrations
       end
 
       def parse_migration_field(node)
+        fields = []
         if node.children[1] == :timestamps
-          @fields.append(DTO::Field.new(:created_at, :datetime))
-          @fields.append(DTO::Field.new(:updated_at, :datetime))
-          return
+          fields << DTO::Field.new(:created_at, :datetime)
+          fields << DTO::Field.new(:updated_at, :datetime)
+          return fields
         end
 
+        return [] if node.arguments.nil? || node.arguments.empty?
+
         type = node.children[1]
         value = node.children[2].value unless node.children[2].hash_type?
-        @fields.append(DTO::Field.new(value, type))
+        fields << DTO::Field.new(value, type)
+
+        fields
       end
     end
   end

metadata

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