brick 1.0.154 → 1.0.155

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 647f34b463dc92515c7eb04331726a808b5ea69dd2bbd6ef99c9d6c769868565
-  data.tar.gz: 22ae7fa65845e65a7848f5cfff0fbb52a864b0161783c3f7d28cfb82ba1f88e5
+  metadata.gz: fceffbcf863eb9c03b64a0cb4420a33ded5eb7f3908b40b263a345ae9bf5c164
+  data.tar.gz: 81f9f96bd1463b8b7e7736ed4716bcf78609bae27485af4de15cb23ae12e27ee
 SHA512:
-  metadata.gz: 84034f48010bd159804a19cb865a9cad04ba5efe89bb1e807a7031393d9a64a33d9ca9802f991565f4d73753df087d200099052dde59e4127c6b94370456ec09
-  data.tar.gz: 57b3cca20e566ef0d4ac72a45f24b025feea45421417b23cbe7cb683877ac228463f5f348f931a6024b9cdf750dde6aa6c49293e1ae1a596a70a0d8f3c128d6c
+  metadata.gz: c41a59ca62c80012166225b0cf80505b81fe57ccdfeb92c4add7f227d6d4a9cc0c664aaf5209c31dabd54c3e471fe140fd3dc75bf1441bbed0fca67e9801ffd9
+  data.tar.gz: 83c874515b230b0b607da32e41366284b15e830ebd7f7022fb318db412fef94a0339495de54c9ab0277425af2635cc8ab235675074cbb50f703082979ebd28b5

data/lib/brick/config.rb

@@ -380,6 +380,14 @@ module Brick
       @mutex.synchronize { @always_load_fields = field_set }
     end
 
+    def ignore_migration_fks
+      @mutex.synchronize { @ignore_migration_fks || [] }
+    end
+
+    def ignore_migration_fks=(relations)
+      @mutex.synchronize { @ignore_migration_fks = relations }
+    end
+
     # Add status page showing all resources and what files have been built out for them
     def add_status
       true

data/lib/brick/version_number.rb

@@ -5,7 +5,7 @@ module Brick
   module VERSION
     MAJOR = 1
     MINOR = 0
-    TINY = 154
+    TINY = 155
 
     # PRE is nil unless it's a pre-release (beta, RC, etc.)
     PRE = nil

data/lib/brick.rb

@@ -1164,6 +1164,12 @@ ActiveSupport.on_load(:active_record) do
         mattr_accessor :default_timezone, instance_writer: false
         self.default_timezone = :utc
       end
+
+      unless respond_to?(:primary_abstract_class)
+        def self.primary_abstract_class
+          self.abstract_class = true
+        end
+      end
     end
 
     # Rails < 4.0 cannot do #find_by, #find_or_create_by, or do #pluck on multiple columns, so here are the patches:

data/lib/generators/brick/migrations_generator.rb

@@ -123,21 +123,27 @@ module Brick
       built_schemas = {} # Track all built schemas so we can place an appropriate drop_schema command only in the first
                          # migration in which that schema is referenced, thereby allowing rollbacks to function properly.
       versions_to_create = [] # Resulting versions to be used when updating the schema_migrations table
+      ar_base = Object.const_defined?(:ApplicationRecord) ? ApplicationRecord : Class.new(ActiveRecord::Base)
       # Start by making migrations for fringe tables (those with no foreign keys).
       # Continue layer by layer, creating migrations for tables that reference ones already done, until
       # no more migrations can be created.  (At that point hopefully all tables are accounted for.)
       while (fringe = chosen.reject do |tbl|
+                        snag_fks = []
                         snags = ::Brick.relations.fetch(tbl, nil)&.fetch(:fks, nil)&.select do |_k, v|
                           v[:is_bt] && !v[:polymorphic] &&
                           tbl != v[:inverse_table] && # Ignore self-referencing associations (stuff like "parent_id")
-                          !done.include?(v[:inverse_table])
+                          !done.include?(v[:inverse_table]) &&
+                          ::Brick.config.ignore_migration_fks.exclude?(snag_fk = "#{tbl}.#{v[:fk]}") &&
+                          snag_fks << snag_fk
+                        end
+                        if snags&.present?
+                          # puts snag_fks.inspect
+                          stuck[tbl] = snags
                         end
-                        stuck[tbl] = snags if snags&.present?
                       end).present?
         fringe.each do |tbl|
           next unless (relation = ::Brick.relations.fetch(tbl, nil))&.fetch(:cols, nil)&.present?
 
-          ar_base = Object.const_defined?(:ApplicationRecord) ? ApplicationRecord : Class.new(ActiveRecord::Base)
           pkey_cols = (rpk = relation[:pkey].values.flatten) & (arpk = [ar_base.primary_key].flatten.sort)
           # In case things aren't as standard
           if pkey_cols.empty?
@@ -246,7 +252,8 @@ module Brick
                   suffix << ", index: { name: '#{shorter || idx_name}' }"
                   indexes[shorter || idx_name] = nil
                 end
-                mig << "      t.references :#{fk[:assoc_name]}#{suffix}, foreign_key: { to_table: #{to_table} }\n"
+                primary_key = ::Brick.relations[fk[:inverse_table]][:class_name]&.constantize&.primary_key
+                mig << "      t.references :#{fk[:assoc_name]}#{suffix}, foreign_key: { to_table: #{to_table}#{", primary_key: :#{primary_key}" if primary_key != ar_base.primary_key} }\n"
               end
             else
               next if !id_option&.end_with?('id: false') && pkey_cols&.include?(col)

data/lib/generators/brick/seeds_generator.rb

@@ -0,0 +1,116 @@
+# frozen_string_literal: true
+
+require 'rails/generators'
+require 'fancy_gets'
+
+module Brick
+  class SeedsGenerator < ::Rails::Generators::Base
+    include FancyGets
+
+    desc 'Auto-generates a seeds file from existing data.'
+
+    def brick_seeds
+      # %%% If Apartment is active and there's no schema_to_analyse, ask which schema they want
+
+      ::Brick.mode = :on
+      ActiveRecord::Base.establish_connection
+
+      if (tables = ::Brick.relations.reject { |k, v| v.key?(:isView) && v[:isView] == true }.map(&:first).sort).empty?
+        puts "No tables found in database #{ActiveRecord::Base.connection.current_database}."
+        return
+      end
+
+      if File.exist?(seed_file_path = "#{::Rails.root}/db/seeds.rb")
+        puts "WARNING: seeds file #{seed_file_path} appears to already be present."
+      end
+
+      # Generate a list of tables that can be chosen
+      chosen = gets_list(list: tables, chosen: tables.dup)
+      schemas = chosen.each_with_object({}) do |v, s|
+        if (v_parts = v.split('.')).length > 1
+          s[v_parts.first] = nil unless [::Brick.default_schema, 'public'].include?(v_parts.first)
+        end
+      end
+      seeds = +"# Seeds file for #{ActiveRecord::Base.connection.current_database}:"
+      done = []
+      fks = {}
+      stuck = {}
+      indexes = {} # Track index names to make sure things are unique
+      ar_base = Object.const_defined?(:ApplicationRecord) ? ApplicationRecord : Class.new(ActiveRecord::Base)
+      # Start by making entries for fringe models (those with no foreign keys).
+      # Continue layer by layer, creating entries for models that reference ones already done, until
+      # no more entries can be created.  (At that point hopefully all models are accounted for.)
+      while (fringe = chosen.reject do |tbl|
+                        snag_fks = []
+                        snags = ::Brick.relations.fetch(tbl, nil)&.fetch(:fks, nil)&.select do |_k, v|
+                          v[:is_bt] && !v[:polymorphic] &&
+                          tbl != v[:inverse_table] && # Ignore self-referencing associations (stuff like "parent_id")
+                          !done.include?(v[:inverse_table]) &&
+                          ::Brick.config.ignore_migration_fks.exclude?(snag_fk = "#{tbl}.#{v[:fk]}") &&
+                          snag_fks << snag_fk
+                        end
+                        if snags&.present?
+                          # puts snag_fks.inspect
+                          stuck[tbl] = snags
+                        end
+                      end).present?
+        seeds << "\n"
+        fringe.each do |tbl|
+          next unless ::Brick.config.exclude_tables.exclude?(tbl) &&
+                      (relation = ::Brick.relations.fetch(tbl, nil))&.fetch(:cols, nil)&.present? &&
+                      (klass = Object.const_get(class_name = relation[:class_name])).table_exists?
+
+          pkey_cols = (rpk = relation[:pkey].values.flatten) & (arpk = [ar_base.primary_key].flatten.sort)
+          # In case things aren't as standard
+          if pkey_cols.empty?
+            pkey_cols = if rpk.empty? # && relation[:cols][arpk.first]&.first == key_type
+                          arpk
+                        elsif rpk.first
+                          rpk
+                        end
+          end
+          schema = if (tbl_parts = tbl.split('.')).length > 1
+                     if tbl_parts.first == (::Brick.default_schema || 'public')
+                       tbl_parts.shift
+                       nil
+                     else
+                       tbl_parts.first
+                     end
+                   end
+
+          # %%% For the moment we're skipping polymorphics
+          fkeys = relation[:fks].values.select { |assoc| assoc[:is_bt] && !assoc[:polymorphic] }
+          # Refer to this table name as a symbol or dotted string as appropriate
+          # tbl_code = tbl_parts.length == 1 ? ":#{tbl_parts.first}" : "'#{tbl}'"
+          seeds << "  # #{class_name}\n"
+
+          is_empty = true
+          klass.order(*pkey_cols).each do |obj|
+            is_empty = false
+            pk_val = obj.send(pkey_cols.first)
+            fk_vals = []
+            data = []
+            relation[:cols].each do |col, col_type|
+              next if !(fk = fkeys.find { |assoc| col == assoc[:fk] }) &&
+                      pkey_cols.include?(col)
+
+              if (val = obj.send(col)) && (val.is_a?(Time) || val.is_a?(Date))
+                val = val.to_s
+              end
+              if fk
+                fk_vals << "#{fk[:assoc_name]}: #{fk[:inverse_table]}_#{val.inspect}" if val
+              else
+                data << "#{col}: #{val.inspect}"
+              end
+            end
+            seeds << "#{tbl}_#{pk_val} = #{class_name}.create(#{(fk_vals + data).join(', ')})\n"
+          end
+          File.open(seed_file_path, "w") { |f| f.write seeds }
+        end
+        done.concat(fringe)
+        chosen -= done
+      end
+      puts "\n*** Created seeds for #{done.length} models in db/seeds.rb ***"
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: brick
 version: !ruby/object:Gem::Version
-  version: 1.0.154
+  version: 1.0.155
 platform: ruby
 authors:
 - Lorin Thwaits
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-06-29 00:00:00.000000000 Z
+date: 2023-07-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -257,6 +257,7 @@ files:
 - lib/generators/brick/install_generator.rb
 - lib/generators/brick/migrations_generator.rb
 - lib/generators/brick/models_generator.rb
+- lib/generators/brick/seeds_generator.rb
 - lib/generators/brick/templates/add_object_changes_to_versions.rb.erb
 - lib/generators/brick/templates/create_versions.rb.erb
 homepage: https://github.com/lorint/brick