trains 0.0.11 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 227d3fc953f2317d57846387bf0c358b0a1bfd6f70f89ddb760b70410616270e
-  data.tar.gz: edb13b82f62463858971853cd62bb21a7f6223df35625c2a88e651c9da279bbe
+  metadata.gz: c13daf893e01409b469a93f5f786390e671de242b5ac2f4565f3d5c4bcb23d93
+  data.tar.gz: e6b5cb947bcab9841e852895e3a23c4d9966c936ea48e9f5ec5a11deb8f7a7f1
 SHA512:
-  metadata.gz: 9550034596435821878df66664809d21e01a0eced82a0f7579bf1f7d84d501d9043b93248527ab52d5c6b77d13d274449132f7f377ac285294361b4472662470
-  data.tar.gz: ddb3f8f34f5e7940cd98a8a05fae0ecdd600c18bea6a3206d9ef2c4910365f3b673946d8085340cddf4c96f782abe061f8cbfd967a3a4dbb7386a3737a0107ad
+  metadata.gz: 05eebe5b58eff0181d4d9d1c7d6f3cd59a3a70d3c62bd4ac29ca7bdc3ca7f6ca2bdd0227140e1bb2ff1031536d2ebff87fd308d805463492ad89be72e1cc57ff
+  data.tar.gz: 0ddd9c546ee1364226336301a2ff7933533b0219c92da788c21889f1de64ac22472a8b93853d1ce70d9467294fb5d62db1544a3a2ddbbb4d0865668f2d816b5c

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/version.rb

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

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,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: trains
 version: !ruby/object:Gem::Version
-  version: 0.0.11
+  version: 0.0.12
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
@@ -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