trains 0.0.10 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 706372a84f876f65aec9c638d6ad2c17dd6255f6dbc942bccbe1f0f81b86e069
-  data.tar.gz: 431b9033167ed91a657dd00ca806c16c1c3df454f74ec276eb7e79baf8407adf
+  metadata.gz: c13daf893e01409b469a93f5f786390e671de242b5ac2f4565f3d5c4bcb23d93
+  data.tar.gz: e6b5cb947bcab9841e852895e3a23c4d9966c936ea48e9f5ec5a11deb8f7a7f1
 SHA512:
-  metadata.gz: a9b1de71a3735c43afee1f793ad0af94e9d8d7d0fced82023c3927d52e2821cf730b997ae7eeab53816208c11320ed4168c381d3d5fa18d0c1c8c8cb3f008006
-  data.tar.gz: c3eac1a413ece2f2ff937f642843aefb7188b63cd25c03d6f394a223b63a1ebb8c25c0a05e7d63c6c903d94687b2f59beb2eebed7b196c7f2e2405573a363075
+  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.10'.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

data/lib/trains/visitor/schema.rb

@@ -76,6 +76,11 @@ module Trains
           @table_name = table.send_node.first_argument.value
           @columns = build_columns(table)
 
+          @columns << DTO::Field.new(
+            name: :id,
+            type: :bigint
+          )
+
           @models << DTO::Model.new(
             name: @table_name.singularize.camelize,
             fields: @columns,

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trains
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.0.12
 platform: ruby
 authors:
 - Syed Faraaz Ahmad
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-06-15 00:00:00.000000000 Z
+date: 2023-06-16 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