trains 0.0.11 → 0.0.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 227d3fc953f2317d57846387bf0c358b0a1bfd6f70f89ddb760b70410616270e
-  data.tar.gz: edb13b82f62463858971853cd62bb21a7f6223df35625c2a88e651c9da279bbe
+  metadata.gz: 1ad58a1dcd643c69b2c278c861647ad0211807edbc2e660f1e09eae127ff6cf7
+  data.tar.gz: 3c4a2d0bb7a6604dd3ffe8f1ca80ea8764b72ae1b5e350e18425de3eb7409687
 SHA512:
-  metadata.gz: 9550034596435821878df66664809d21e01a0eced82a0f7579bf1f7d84d501d9043b93248527ab52d5c6b77d13d274449132f7f377ac285294361b4472662470
-  data.tar.gz: ddb3f8f34f5e7940cd98a8a05fae0ecdd600c18bea6a3206d9ef2c4910365f3b673946d8085340cddf4c96f782abe061f8cbfd967a3a4dbb7386a3737a0107ad
+  metadata.gz: 476e65ac7eaf350b8e7e15eee1a0d508677f8f5ad3b446503fd890a821fb24935f03de5f6fd7a4bb938a227ca9c1db60b338e9852284e802ac3a585ded01ae82
+  data.tar.gz: 726f8983354b206b1f881bea58182b74a0d462bf8c5898435547647b73eac9fe0b845bab206ce1f6da6cc4d9449e8664dadb011e9e76f8194d349d47abdeaf29

data/.deepsource.toml

@@ -2,3 +2,6 @@ version = 1
 
 [[analyzers]]
 name = "ruby"
+
+[[analyzers]]
+name = "shell"

data/lib/trains/scanner.rb

@@ -44,12 +44,10 @@ module Trains
 
     private
 
-    # Generate models from either db/schema.rb
-    # else stitch together migrations to create models
+    # Stitch together migrations to create models
     def generate_models
-      return get_models if File.exist?(File.join(@dir, 'db', 'schema.rb'))
-
       migrations = get_migrations.flatten.reject(&:nil?)
+      migrations = [*migrations, *parse_models.flatten.reject(&:nil?)]
       Utils::MigrationTailor.stitch(migrations)
     end
 
@@ -63,7 +61,7 @@ module Trains
         .flatten
     end
 
-    def get_models
+    def parse_schema
       result_hash = {}
       schema_file = [File.join(@dir, 'db', 'schema.rb')]
       models_results = parse_util(schema_file, Visitor::Schema)
@@ -77,6 +75,15 @@ module Trains
       result_hash
     end
 
+    def parse_models
+      models = Dir.glob(File.join(@dir, 'app', 'models', '**', '*.rb'))
+      model_results = parse_util(models, Visitor::Model)
+
+      model_results
+        .select { |result| result.error.nil? }
+        .map(&:data)
+    end
+
     def get_helpers; end
 
     def get_gemfile; end

data/lib/trains/utils/migration_tailor.rb

@@ -13,7 +13,7 @@ module Trains
               fields: mig.fields,
               version: mig.version
             )
-          when :add_column, :add_reference
+          when :add_column, :add_column_with_default, :add_reference
             models[mig.table_name].fields.push(*mig.fields)
           when :remove_column
             column =

data/lib/trains/version.rb

@@ -1,3 +1,3 @@
 module Trains
-  VERSION = '0.0.11'.freeze
+  VERSION = '0.0.13'.freeze
 end

data/lib/trains/visitor/migration.rb

@@ -19,6 +19,7 @@ module Trains
       ].freeze
       COLUMN_MODIFIERS = %i[
         add_column
+        add_column_with_default
         change_column
         add_reference
         remove_column

data/lib/trains/visitor/model.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Trains
+  module Visitor
+    # Visit ActiveRecord models and add migrations for reference fields
+    class Model < Base
+      POSSIBLE_ASSOCIATIONS = %i[
+        has_many
+        belongs_to
+        has_one
+        has_and_belongs_to_many
+      ]
+      attr_reader :result
+
+      # skipcq: RB-LI1087
+      def initialize
+        @result = []
+      end
+
+      def on_class(node)
+        return unless node.parent_class
+        return unless node.parent_class.source == 'ApplicationRecord'
+
+        @model_class = node.identifier.source
+        node.each_descendant(:send) { |send_node| parse_model(send_node) }
+      end
+
+      def parse_model(node)
+        return unless POSSIBLE_ASSOCIATIONS.include?(node.method_name)
+
+        @result << DTO::Migration.new(
+          @model_class,
+          :add_column,
+          [DTO::Field.new(node.arguments.first.value, :bigint)],
+          nil
+        )
+
+        return unless node.method_name == :has_and_belongs_to_many
+
+        @result <<
+          DTO::Migration.new(
+            node.arguments.first.value.to_s.singularize.camelize,
+            :add_column,
+            [DTO::Field.new(@model_class.tableize.to_sym, :bigint)],
+            nil
+          )
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trains
 version: !ruby/object:Gem::Version
-  version: 0.0.11
+  version: 0.0.13
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-06-16 00:00:00.000000000 Z
+date: 2023-06-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -103,6 +103,7 @@ files:
 - lib/trains/visitor/controller.rb
 - lib/trains/visitor/helper.rb
 - lib/trains/visitor/migration.rb
+- lib/trains/visitor/model.rb
 - lib/trains/visitor/route.rb
 - lib/trains/visitor/schema.rb
 - trains.gemspec