trains 0.0.18 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7a1893b50977fd1adda92c2268df81b65f47fa4218961835c11c5534274774db
-  data.tar.gz: aa0c28905214fa2d776d144e3e58bfa952ad154b91a4f007e0f9d506e6fe723d
+  metadata.gz: 760e5b8af816d5480e1ba069af6b42c3af88ca0833322049a730345403ddc348
+  data.tar.gz: 77ee8712a8934224accf6d522623613e8e264d1de92bac6f58ace205f20cd12c
 SHA512:
-  metadata.gz: 87bf0d7266a6347d452434b83f19f3798ed68f97b0be036baed26390442894ccf725a0e54bb233cbb25523b9864e9bbc3d246bae5341c489121e1ef80b7195b4
-  data.tar.gz: 377afeb54297eca4feb98a5c482a348250319f7d61b84650fb33910c91faa86d1c9eade41c34b82cf217c9833ebc5e238af7cc04c86c53255e29480fa4e0d605
+  metadata.gz: 260b68afd11589fa470c4475cd759b521ccd0b82bd3ceb5668acbb7dffd63bcc348a45f72c889e1d2d1902740f5bd9aef6f4aa0cc5ba14032bb8f0242abbf6ca
+  data.tar.gz: f04a57a248c8c657e34866106273e06bb2eb1e5d2bbc282fe2762422704f11521f2d97ce12a73f4dfa8ada78725091c3e27bc75b5fb0695cf536eef322ce4252

data/lib/trains/dto/model.rb

@@ -1,5 +1,12 @@
 module Trains
   module DTO
-    Model = Data.define(:name, :fields, :version)
+    Model = Data.define(
+      :name, :fields, :version, :renamed_columns, :removed_columns, :ignored_columns
+    ) do
+      def initialize(name:, fields:, version:, renamed_columns: [],
+                     removed_columns: [], ignored_columns: [])
+        super(name:, fields:, version:, renamed_columns:, removed_columns:, ignored_columns:)
+      end
+    end
   end
 end

data/lib/trains/dto/rename.rb

@@ -0,0 +1,5 @@
+module Trains
+  module DTO
+    Rename = Data.define(:from, :to)
+  end
+end

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|
@@ -16,14 +19,66 @@ module Trains
             end
           when :add_column, :add_column_with_default, :add_reference
             models[mig.table_name].fields.push(*mig.fields)
+          when :ignore_column
+            models[mig.table_name].ignored_columns.push(*mig.fields.map(&:name))
           when :remove_column
             column =
               models[mig.table_name].fields.find do |field|
                 field.name == mig.fields.first.name
               end
             models[mig.table_name].fields.delete(column)
+            models[mig.table_name].removed_columns.push(mig.fields.first.name)
+          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)
+            models[mig.table_name].renamed_columns.push(
+              Trains::DTO::Rename.new(
+                from: mig.fields.first.name.to_sym, to: mig.fields.first.type.to_sym
+              )
+            )
           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)
+                models[mig.table_name].removed_columns.push(field.name)
+              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].to_sym,
+                    type: column.type
+                  )
+                )
+                # Delete the field
+                models[mig.table_name].fields.delete(column)
+                models[mig.table_name].renamed_columns.push(
+                  Trains::DTO::Rename.new(
+                    from: field.name[0], to: field.name[1]
+                  )
+                )
+              else
+                models[mig.table_name].fields.push(field)
+              end
+            end
           when :change_column
             # get column
             column =
@@ -34,6 +89,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.18'.freeze
+  VERSION = '0.1.0'.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,6 +9,7 @@ module Trains
         belongs_to
         has_one
         has_and_belongs_to_many
+        ignored_columns=
       ].freeze
       MODEL_PARENT_CLASSES = %w[
         ApplicationRecord
@@ -32,10 +33,29 @@ module Trains
       def parse_model(node)
         return unless POSSIBLE_ASSOCIATIONS.include?(node.method_name)
 
+        if node.method_name == :ignored_columns=
+          return if node.arguments.nil?
+          return unless node.arguments.first.is_a? RuboCop::AST::ArrayNode
+
+          arguments = node.arguments.first.to_a
+          arguments = arguments.select do |child|
+            child.is_a?(RuboCop::AST::SymbolNode) || child.is_a?(RuboCop::AST::StrNode)
+          end.map(&:value)
+
+          @result << DTO::Migration.new(
+            @model_class,
+            :ignore_column,
+            [*arguments.map { |arg| DTO::Field.new(arg.to_sym, nil) }],
+            nil
+          )
+
+          return
+        end
+
         @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.18
+  version: 0.1.0
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-07-31 00:00:00.000000000 Z
+date: 2023-08-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -91,6 +91,7 @@ files:
 - lib/trains/dto/method.rb
 - lib/trains/dto/migration.rb
 - lib/trains/dto/model.rb
+- lib/trains/dto/rename.rb
 - lib/trains/dto/route.rb
 - lib/trains/scanner.rb
 - lib/trains/utils/args.rb