trains 0.0.6 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2b1af8d7a53681c21f561432534383da1d83bf4615bc8e85bdfa6c8ce7b210aa
-  data.tar.gz: aa4798c33b271c590c5b966078eb7a6f9fd603a069842ce454cadb14de8cc08d
+  metadata.gz: be756e07f687c79e259385beb680f5749c54a880273be056ba83a717aa821887
+  data.tar.gz: da3fc5403f7844919f9bfd3ac49ef37479e7f4149328e851d1b1e654353398af
 SHA512:
-  metadata.gz: f6faddb254499e05f02a13def98dc7ab1f7df69d2ff694617cbd24410e2cb23266c7fd4c067cb2e4e2dd4fd7e7552f7d5ba79c995f2f951ff452150b8d800222
-  data.tar.gz: 95748dab8e2f16aed324d9c55a8d14c0eeec265361d49794bf7d81f21f6b42b8a25ba7426f719fea673efc982864ff49c745d3cdeb6888780b63a3bdf81038b9
+  metadata.gz: 3a389ed84aedba71b13061c4233389d3008bd895a532f506a9f492a6d6aaa54ae96708d27f0e4cba9269f1666ef8f614c5ebf467a8bbf70c63595b449e2bcc0f
+  data.tar.gz: be28d384e28be094161bcdb28105a4bef9dd2d7f86978936f554bf699a205c1953a88bd1de071c1005f2ec4d5c54e91bf6ace9e20f7f917bc8204b46ced1a431

data/lib/trains/scanner.rb

@@ -55,8 +55,8 @@ module Trains
 
     def get_routes
       route_file = [File.join(@dir, 'config', 'routes.rb')]
-
       routes_results = parse_util(route_file, Visitor::Route)
+
       routes_results
         .select { |result| result.error.nil? }
         .map(&:data)

data/lib/trains/version.rb

@@ -1,3 +1,3 @@
 module Trains
-  VERSION = '0.0.6'
+  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))

data/lib/trains/visitor/route.rb

@@ -16,7 +16,7 @@ module Trains
         (send nil? %1 ...)
       PATTERN
 
-      ALLOWED_VERBS = %i[get put post update delete resources scope].freeze
+      ALLOWED_VERBS = %i[get put post update delete resources].freeze
 
       def initialize
         @route_list = []
@@ -28,15 +28,15 @@ module Trains
 
       def on_block(node)
         return unless route_parent?(node)
-        return if node.body.nil?
 
-        node.body.each_child_node do |child|
+        node.each_descendant(:send) do |child|
           ALLOWED_VERBS.each do |verb|
             if route_method?(child, verb)
               @route_list << parse_route(child, verb)
             end
           end
         end
+
       end
 
       private

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trains
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.8
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-08 00:00:00.000000000 Z
+date: 2023-05-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -66,7 +66,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.5'
-description: 
+description:
 email:
 - faraaz98@live.com
 executables: []
@@ -114,7 +114,7 @@ metadata:
   source_code_uri: https://github.com/faraazahmad/trains
   changelog_uri: https://github.com/faraazahmad/trains/blob/master/CHANGELOG.md
   rubygems_mfa_required: 'true'
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -129,8 +129,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
-signing_key: 
+rubygems_version: 3.4.3
+signing_key:
 specification_version: 4
 summary: Collect metadata about your Rails app by statically analyzing it
 test_files: []