trains 0.0.17 → 0.0.19

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d9b0cdd1bee0a6ed5d222184662bb45dced14a141f00c166633ee177c3f83501
-  data.tar.gz: 75132b13c33a5cc68005cdbc6d525dfda2b3d97aad8144cbb923f759c299bd1e
+  metadata.gz: 93ee7909ca18c2d768daa77d9bfce0096ac1f1965fc3590d6d3cd4fd281698cd
+  data.tar.gz: 6e935fff8f51fdda8e95d49f3520aff2d0ce3d71a67e4267a385c1d915e0cd5f
 SHA512:
-  metadata.gz: 754df75dcf500b11b1d45df65c5bfde168be9dc160912cbd4a2dc0da37f9be9a380158a078b5aafa457bf73315e10f96d47fc50d6d2b849f29c2dc0c59196aee
-  data.tar.gz: 4e8d8dbaad75395f686d5547cb633d162730858f8cc9fccb48fa4e84756ee942affdc2ec0aceaec5e755e523661bbfa944acdc6578344606630add7cbf68c767
+  metadata.gz: 77468c5ea8f22d4eba7daae1b188a1535a6de55cf21d4fc210cbaf8637e6afb587c21ed9a274eba27d35f674849863e125f413dcdbcc4927405ed39056ea7590
+  data.tar.gz: 20db296bfda78eee13f8affb92c1c4635ec5dd0bc35e24015e6e4b28de64faaef0277ce34cad88e3dde4c9e76eb9513ad3198a22571f1e3a1212c9b163f28c93

data/lib/trains/scanner.rb

@@ -66,9 +66,10 @@ module Trains
 
     def parse_schema
       result_hash = {}
-      schema_file = [File.join(@dir, 'db', 'schema.rb')]
-      models_results = parse_util(schema_file, Visitor::Schema)
+      schema_file = File.join(@dir, 'db', 'schema.rb')
+      return result_hash unless File.exist? schema_file
 
+      models_results = parse_util([schema_file], Visitor::Schema)
       models_results
         .select { |result| result.error.nil? }
         .map(&:data)

data/lib/trains/utils/migration_tailor.rb

@@ -1,5 +1,8 @@
+# frozen_string_literal: true
+
 module Trains
   module Utils
+    # Combine multiple migrations into models
     module MigrationTailor
       def self.stitch(models = {}, migrations)
         migrations.each do |mig|
@@ -22,8 +25,46 @@ module Trains
                 field.name == mig.fields.first.name
               end
             models[mig.table_name].fields.delete(column)
+          when :rename_column
+            column =
+              models[mig.table_name].fields.find do |field|
+                field.name == mig.fields.first.name
+              end
+            models[mig.table_name].fields.push(
+              Trains::DTO::Field.new(
+                name: mig.fields.first.type.to_sym,
+                type: column.type
+              )
+            )
+            models[mig.table_name].fields.delete(column)
           when :change_table
-          # TODO: handle renaming columns
+            mig.fields.each do |field|
+              case field.type
+              when :remove
+                column =
+                  models[mig.table_name].fields.find do |mod_field|
+                    mod_field.name == field.name
+                  end
+                models[mig.table_name].fields.delete(column)
+              when :rename
+                # find the field and store temporarily
+                column =
+                  models[mig.table_name].fields.find do |mod_field|
+                    mod_field.name == field.name[0]
+                  end
+                # Create new field from temp with new name
+                models[mig.table_name].fields.push(
+                  Trains::DTO::Field.new(
+                    name: field.name[1],
+                    type: column.type
+                  )
+                )
+                # Delete the field
+                models[mig.table_name].fields.delete(column)
+              else
+                models[mig.table_name].fields.push(field)
+              end
+            end
           when :change_column
             # get column
             column =
@@ -34,6 +75,8 @@ module Trains
             models[mig.table_name].fields.delete(column)
 
             models[mig.table_name].fields << mig.fields.first
+          else
+            next
           end
 
         rescue NoMethodError

data/lib/trains/version.rb

@@ -1,3 +1,3 @@
 module Trains
-  VERSION = '0.0.17'.freeze
+  VERSION = '0.0.19'.freeze
 end

data/lib/trains/visitor/migration.rb

@@ -22,6 +22,7 @@ module Trains
         add_column_with_default
         change_column
         add_reference
+        rename_column
         remove_column
       ].freeze
 
@@ -181,7 +182,11 @@ module Trains
           # t.references
           type = :bigint
           value = "#{node.children[2].value}_id".to_sym
-          fields << DTO::Field.new(value.to_sym, type)
+          fields << DTO::Field.new(value, type)
+        when :rename
+          type = :rename
+          value = node.children[2..3].map { |field| field.value.to_sym }
+          fields << DTO::Field.new(value, type)
         when :index
         else
           # t.string, t.integer etc.

data/lib/trains/visitor/model.rb

@@ -9,7 +9,11 @@ module Trains
         belongs_to
         has_one
         has_and_belongs_to_many
-      ]
+      ].freeze
+      MODEL_PARENT_CLASSES = %w[
+        ApplicationRecord
+        ActiveRecord::Base
+      ].freeze
       attr_reader :result
 
       # skipcq: RB-LI1087
@@ -19,7 +23,7 @@ module Trains
 
       def on_class(node)
         return unless node.parent_class
-        return unless node.parent_class.source == 'ApplicationRecord'
+        return unless MODEL_PARENT_CLASSES.include? node.parent_class.source
 
         @model_class = node.identifier.source
         node.each_descendant(:send) { |send_node| parse_model(send_node) }
@@ -31,7 +35,7 @@ module Trains
         @result << DTO::Migration.new(
           @model_class,
           :add_column,
-          [DTO::Field.new(node.arguments.first.value, :bigint)],
+          [DTO::Field.new(node.arguments.first.value.to_sym, :bigint)],
           nil
         )
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trains
 version: !ruby/object:Gem::Version
-  version: 0.0.17
+  version: 0.0.19
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-07-12 00:00:00.000000000 Z
+date: 2023-08-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -130,7 +130,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.3
+rubygems_version: 3.4.16
 signing_key:
 specification_version: 4
 summary: Collect metadata about your Rails app by statically analyzing it