brick 1.0.23 → 1.0.26

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 24731ac9bd56261179224610406247cc4438c61e411aa3fe96205b2e97fc8cb8
-  data.tar.gz: 785d12e36fcf788c05cff3a02ca2ec17fc418b0ea15b54cb6923c83c6f362d3b
+  metadata.gz: 922a55ec392e2ea9cef11c7297e4c2e4883e918130a23e49102b5a7e2442140e
+  data.tar.gz: 7cda213ec2e7cdccae096ef4ad8091a53a91359a1e6fabeb753cc2b579ac5c16
 SHA512:
-  metadata.gz: '090a3c4d1b3cb4761799e05adf5caa0c83f60cf985f97ed6856beb05620014e9e7401689719209488092c6c4a7605e53781cdb71ad9e7ec204ccaddca0d92cca'
-  data.tar.gz: 8d1f9512c948359e7d73cd7cc32302f226df7e3e744fab562e01e16170830be28d0e2e15bcd7b437f66ed79e366b0ff33b7f578e4c76ffde1dd62cdcec059069
+  metadata.gz: 7b817257f9c2aedd47bc2471b12e13edc8b370b71920258755f0f9a7594060e065d022e8d4b2b360cdc41f2a60def02cfc406889d526d80088670bd64acd58be
+  data.tar.gz: 1ea923301103dcc53853940770861ab5c2635f1280026e0740287274c0e9bb7d566ddcf96db0d3858467b788bc9ee4b73e7bcfc61145be1bbb43775cfb859b80

data/lib/brick/config.rb

@@ -97,6 +97,15 @@ module Brick
       @mutex.synchronize { @has_ones = hos }
     end
 
+    # Polymorphic associations
+    def polymorphics
+      @mutex.synchronize { @polymorphics }
+    end
+
+    def polymorphics=(polys)
+      @mutex.synchronize { @polymorphics = polys }
+    end
+
     def model_descrips
       @mutex.synchronize { @model_descrips ||= {} }
     end
@@ -113,6 +122,14 @@ module Brick
       @mutex.synchronize { @sti_namespace_prefixes = prefixes }
     end
 
+    def schema_to_analyse
+      @mutex.synchronize { @schema_to_analyse }
+    end
+
+    def schema_to_analyse=(schema)
+      @mutex.synchronize { @schema_to_analyse = schema }
+    end
+
     def skip_database_views
       @mutex.synchronize { @skip_database_views }
     end

data/lib/brick/extensions.rb

@@ -74,8 +74,7 @@ module ActiveRecord
       dsl
     end
 
-    # Pass in true or a JoinArray
-    def self.brick_parse_dsl(build_array = nil, prefix = [], translations = {})
+    def self.brick_parse_dsl(build_array = nil, prefix = [], translations = {}, is_polymorphic = false)
       build_array = ::Brick::JoinArray.new.tap { |ary| ary.replace([build_array]) } if build_array.is_a?(::Brick::JoinHash)
       build_array = ::Brick::JoinArray.new unless build_array.nil? || build_array.is_a?(Array)
       members = []
@@ -87,12 +86,17 @@ module ActiveRecord
           if bracket_name
             if ch == ']' # Time to process a bracketed thing?
               parts = bracket_name.split('.')
-              first_parts = parts[0..-2].map { |part| klass = klass.reflect_on_association(part_sym = part.to_sym).klass; part_sym }
+              first_parts = parts[0..-2].map do |part|
+                klass = klass.reflect_on_association(part_sym = part.to_sym).klass
+                part_sym
+              end
               parts = prefix + first_parts + [parts[-1]]
               if parts.length > 1
-                s = build_array
-                parts[0..-3].each { |v| s = s[v.to_sym] }
-                s[parts[-2]] = nil # unless parts[-2].empty? # Using []= will "hydrate" any missing part(s) in our whole series
+                unless is_polymorphic
+                  s = build_array
+                  parts[0..-3].each { |v| s = s[v.to_sym] }
+                  s[parts[-2]] = nil # unless parts[-2].empty? # Using []= will "hydrate" any missing part(s) in our whole series
+                end
                 translations[parts[0..-2].join('.')] = klass
               end
               members << parts
@@ -115,8 +119,8 @@ module ActiveRecord
     # If available, parse simple DSL attached to a model in order to provide a friendlier name.
     # Object property names can be referenced in square brackets like this:
     # { 'User' => '[profile.firstname] [profile.lastname]' }
-    def brick_descrip
-      self.class.brick_descrip(self)
+    def brick_descrip(data = nil, pk_alias = nil)
+      self.class.brick_descrip(self, data, pk_alias)
     end
 
     def self.brick_descrip(obj, data = nil, pk_alias = nil)
@@ -136,11 +140,7 @@ module ActiveRecord
                         this_obj = obj
                         bracket_name.split('.').each do |part|
                           obj_name += ".#{part}"
-                          this_obj = if caches.key?(obj_name)
-                                       caches[obj_name]
-                                     else
-                                       (caches[obj_name] = this_obj&.send(part.to_sym))
-                                     end
+                          this_obj = caches.fetch(obj_name) { caches[obj_name] = this_obj&.send(part.to_sym) }
                         end
                         this_obj&.to_s || ''
                       end
@@ -160,7 +160,8 @@ module ActiveRecord
       end
       if is_brackets_have_content
         output
-      elsif pk_alias
+      elsif (pk_alias ||= primary_key)
+        pk_alias = [pk_alias] unless pk_alias.is_a?(Array)
         id = []
         pk_alias.each do |pk_alias_part|
           if (pk_part = obj.send(pk_alias_part))
@@ -194,7 +195,10 @@ module ActiveRecord
   private
 
     def self._brick_get_fks
-      @_brick_get_fks ||= reflect_on_all_associations.select { |a2| a2.macro == :belongs_to }.map(&:foreign_key)
+      @_brick_get_fks ||= reflect_on_all_associations.select { |a2| a2.macro == :belongs_to }.each_with_object([]) do |bt, s|
+        s << bt.foreign_key
+        s << bt.foreign_type if bt.polymorphic?
+      end
     end
   end
 
@@ -295,8 +299,14 @@ module ActiveRecord
       if is_add_bts || is_add_hms
         bts, hms, associatives = ::Brick.get_bts_and_hms(klass)
         bts.each do |_k, bt|
+          next if bt[2] # Polymorphic?
+
           # join_array will receive this relation name when calling #brick_parse_dsl
-          bt_descrip[bt.first] = [bt.last, bt.last.brick_parse_dsl(join_array, bt.first, translations)]
+          bt_descrip[bt.first] = if bt[1].is_a?(Array)
+                                   bt[1].each_with_object({}) { |bt_class, s| s[bt_class] = bt_class.brick_parse_dsl(join_array, bt.first, translations, true) }
+                                 else
+                                   { bt.last => bt[1].brick_parse_dsl(join_array, bt.first, translations) }
+                                 end
         end
         skip_klass_hms = ::Brick.config.skip_index_hms[klass.name] || {}
         hms.each do |k, hm|
@@ -314,7 +324,7 @@ module ActiveRecord
         when 2
           assoc_name = ks.first.to_sym
           # Make sure it's a good association name and that the model has that column name
-          next unless klass.reflect_on_association(assoc_name)&.klass&.columns&.any? { |col| col.name == ks.last }
+          next unless klass.reflect_on_association(assoc_name)&.klass&.column_names&.any?(ks.last)
 
           join_array[assoc_name] = nil # Store this relation name in our special collection for .joins()
         end
@@ -330,22 +340,31 @@ module ActiveRecord
         id_for_tables = Hash.new { |h, k| h[k] = [] }
         field_tbl_names = Hash.new { |h, k| h[k] = {} }
         bt_columns = bt_descrip.each_with_object([]) do |v, s|
-          tbl_name = field_tbl_names[v.first][v.last.first] ||= shift_or_first(chains[v.last.first])
-          if (id_col = v.last.first.primary_key) && !id_for_tables.key?(v.first) # was tbl_name
-            # Accommodate composite primary key by allowing id_col to come in as an array
-            (id_col.is_a?(Array) ? id_col : [id_col]).each do |id_part|
-              selects << "#{"#{tbl_name}.#{id_part}"} AS \"#{(id_alias = "_brfk_#{v.first}__#{id_part}")}\""
-              id_for_tables[v.first] << id_alias
-            end
-            v.last << id_for_tables[v.first]
-          end
-          if (col_name = v.last[1].last&.last)
+          v.last.each do |k1, v1| # k1 is class, v1 is array of columns to snag
+            next if chains[k1].nil?
+
+            tbl_name = field_tbl_names[v.first][k1] ||= shift_or_first(chains[k1])
+            # if (col_name = v1[1].last&.last) # col_name is weak when there are multiple, using sel_col.last instead
             field_tbl_name = nil
-            v.last[1].map { |x| [translations[x[0..-2].map(&:to_s).join('.')], x.last] }.each_with_index do |sel_col, idx|
+            v1.map { |x|
+                    [translations[x[0..-2].map(&:to_s).join('.')], x.last]
+                  }.each_with_index do |sel_col, idx|
               field_tbl_name ||= field_tbl_names[v.first][sel_col.first] ||= shift_or_first(chains[sel_col.first])
-              # col_name is weak when there are multiple, using sel_col.last instead
+
               selects << "#{"#{field_tbl_name}.#{sel_col.last}"} AS \"#{(col_alias = "_brfk_#{v.first}__#{sel_col.last}")}\""
-              v.last[1][idx] << col_alias
+              v1[idx] << col_alias
+            end
+            # end
+
+            unless id_for_tables.key?(v.first)
+              # Accommodate composite primary key by allowing id_col to come in as an array
+              ((id_col = k1.primary_key).is_a?(Array) ? id_col : [id_col]).each do |id_part|
+                id_for_tables[v.first] << if id_part
+                                            selects << "#{"#{tbl_name}.#{id_part}"} AS \"#{(id_alias = "_brfk_#{v.first}__#{id_part}")}\""
+                                            id_alias
+                                          end
+              end
+              v1 << id_for_tables[v.first].compact
             end
           end
         end
@@ -365,21 +384,26 @@ module ActiveRecord
                          hm.foreign_key
                        else
                          fk_col = hm.foreign_key
+                         poly_type = hm.inverse_of.foreign_type if hm.options.key?(:as)
                          hm.klass.primary_key || '*'
                        end
         tbl_alias = "_br_#{hm.name}"
         pri_tbl = hm.active_record
+        on_clause = []
         if fk_col.is_a?(Array) # Composite key?
-          on_clause = []
           fk_col.each_with_index { |fk_col_part, idx| on_clause << "#{tbl_alias}.#{fk_col_part} = #{pri_tbl.table_name}.#{pri_tbl.primary_key[idx]}" }
-          joins!("LEFT OUTER
-JOIN (SELECT #{fk_col.join(', ')}, COUNT(#{count_column}) AS _ct_ FROM #{associative&.table_name || hm.klass.table_name} GROUP BY #{(1..fk_col.length).to_a.join(', ')}) AS #{tbl_alias}
-  ON #{on_clause.join(' AND ')}")
+          selects = fk_col.dup
         else
-          joins!("LEFT OUTER
-JOIN (SELECT #{fk_col}, COUNT(#{count_column}) AS _ct_ FROM #{associative&.table_name || hm.klass.table_name} GROUP BY 1) AS #{tbl_alias}
-  ON #{tbl_alias}.#{fk_col} = #{pri_tbl.table_name}.#{pri_tbl.primary_key}")
+          selects = [fk_col]
+          on_clause << "#{tbl_alias}.#{fk_col} = #{pri_tbl.table_name}.#{pri_tbl.primary_key}"
+        end
+        if poly_type
+          selects << poly_type
+          on_clause << "#{tbl_alias}.#{poly_type} = '#{name}'"
         end
+        join_clause = "LEFT OUTER
+JOIN (SELECT #{selects.join(', ')}, COUNT(#{count_column}) AS _ct_ FROM #{associative&.table_name || hm.klass.table_name} GROUP BY #{(1..selects.length).to_a.join(', ')}) AS #{tbl_alias}"
+        joins!("#{join_clause} ON #{on_clause.join(' AND ')}")
       end
       where!(wheres) unless wheres.empty?
       wheres unless wheres.empty? # Return the specific parameters that we did use
@@ -424,8 +448,7 @@ JOIN (SELECT #{fk_col}, COUNT(#{count_column}) AS _ct_ FROM #{associative&.table
               this_module.const_get(class_name)
             else
               # Build STI subclass and place it into the namespace module
-              # %%% Does this ever get used???
-              puts [this_module.const_set(class_name, klass = Class.new(self)).name, class_name].inspect
+              this_module.const_set(class_name, klass = Class.new(self))
               klass
             end
           end
@@ -476,7 +499,7 @@ class Object
       # path_suffix = ActiveSupport::Dependencies.qualified_name_for(Object, args.first).underscore
       # return self._brick_const_missing(*args) if ActiveSupport::Dependencies.search_for_file(path_suffix)
       # If the file really exists, go and snag it:
-      if !(is_found = ActiveSupport::Dependencies.search_for_file(class_name.underscore)) && (filepath = self.name&.split('::'))
+      if !(is_found = ActiveSupport::Dependencies.search_for_file(class_name.underscore)) && (filepath = (self.name || class_name)&.split('::'))
         filepath = (filepath[0..-2] + [class_name]).join('/').underscore + '.rb'
       end
       if is_found
@@ -500,7 +523,7 @@ class Object
                  singular_table_name = ActiveSupport::Inflector.underscore(model_name)
  
                  # Adjust for STI if we know of a base model for the requested model name
-                 table_name = if (base_model = ::Brick.sti_models[model_name]&.fetch(:base, nil))
+                 table_name = if (base_model = ::Brick.sti_models[model_name]&.fetch(:base, ::Brick.existing_stis[model_name]&.constantize))
                                 base_model.table_name
                               else
                                 ActiveSupport::Inflector.pluralize(singular_table_name)
@@ -541,7 +564,7 @@ class Object
         return
       end
 
-      if (base_model = ::Brick.sti_models[model_name]&.fetch(:base, nil))
+      if (base_model = ::Brick.sti_models[model_name]&.fetch(:base, ::Brick.existing_stis[model_name]&.constantize))
         is_sti = true
       else
         base_model = ::Brick.config.models_inherit_from || ActiveRecord::Base
@@ -589,77 +612,11 @@ class Object
           hmts = fks.each_with_object(Hash.new { |h, k| h[k] = [] }) do |fk, hmts|
             # The key in each hash entry (fk.first) is the constraint name
             inverse_assoc_name = (assoc = fk.last)[:inverse]&.fetch(:assoc_name, nil)
-            options = {}
-            singular_table_name = ActiveSupport::Inflector.singularize(assoc[:inverse_table])
-            macro = if assoc[:is_bt]
-                      # Try to take care of screwy names if this is a belongs_to going to an STI subclass
-                      assoc_name = if (primary_class = assoc.fetch(:primary_class, nil)) &&
-                                      sti_inverse_assoc = primary_class.reflect_on_all_associations.find do |a|
-                                        a.macro == :has_many && a.options[:class_name] == self.name && assoc[:fk] = a.foreign_key
-                                      end
-                                     sti_inverse_assoc.options[:inverse_of]&.to_s || assoc_name
-                                   else
-                                     assoc[:assoc_name]
-                                   end
-                      need_class_name = singular_table_name.underscore != assoc_name
-                      need_fk = "#{assoc_name}_id" != assoc[:fk]
-                      if (inverse = assoc[:inverse])
-                        inverse_assoc_name, _x = _brick_get_hm_assoc_name(relations[assoc[:inverse_table]], inverse)
-                        if (has_ones = ::Brick.config.has_ones&.fetch(inverse[:alternate_name].camelize, nil))&.key?(singular_inv_assoc_name = ActiveSupport::Inflector.singularize(inverse_assoc_name))
-                          inverse_assoc_name = if has_ones[singular_inv_assoc_name]
-                                                 need_inverse_of = true
-                                                 has_ones[singular_inv_assoc_name]
-                                               else
-                                                 singular_inv_assoc_name
-                                               end
-                        end
-                      end
-                      :belongs_to
-                    else
-                      # need_class_name = ActiveSupport::Inflector.singularize(assoc_name) == ActiveSupport::Inflector.singularize(table_name.underscore)
-                      # Are there multiple foreign keys out to the same table?
-                      assoc_name, need_class_name = _brick_get_hm_assoc_name(relation, assoc)
-                      need_fk = "#{ActiveSupport::Inflector.singularize(assoc[:inverse][:inverse_table])}_id" != assoc[:fk]
-                      # fks[table_name].find { |other_assoc| other_assoc.object_id != assoc.object_id && other_assoc[:assoc_name] == assoc[assoc_name] }
-                      if (has_ones = ::Brick.config.has_ones&.fetch(model_name, nil))&.key?(singular_assoc_name = ActiveSupport::Inflector.singularize(assoc_name))
-                        assoc_name = if (custom_assoc_name = has_ones[singular_assoc_name])
-                                       need_class_name = custom_assoc_name != singular_assoc_name
-                                       custom_assoc_name
-                                     else
-                                       singular_assoc_name
-                                     end
-                        :has_one
-                      else
-                        :has_many
-                      end
-                    end
-            # Figure out if we need to specially call out the class_name and/or foreign key
-            # (and if either of those then definitely also a specific inverse_of)
-            options[:class_name] = assoc[:primary_class]&.name || singular_table_name.camelize if need_class_name
-            # Work around a bug in CPK where self-referencing belongs_to associations double up their foreign keys
-            if need_fk # Funky foreign key?
-              options[:foreign_key] = if assoc[:fk].is_a?(Array)
-                                        assoc_fk = assoc[:fk].uniq
-                                        assoc_fk.length < 2 ? assoc_fk.first : assoc_fk
-                                      else
-                                        assoc[:fk].to_sym
-                                      end
-            end
-            options[:inverse_of] = inverse_assoc_name.to_sym if inverse_assoc_name && (need_class_name || need_fk || need_inverse_of)
-
-            # Prepare a list of entries for "has_many :through"
-            if macro == :has_many
-              relations[assoc[:inverse_table]][:hmt_fks].each do |k, hmt_fk|
-                next if k == assoc[:fk]
-
-                hmts[ActiveSupport::Inflector.pluralize(hmt_fk.last)] << [assoc, hmt_fk.first]
-              end
+            if (invs = assoc[:inverse_table]).is_a?(Array)
+              invs.each { |inv| build_bt_or_hm(relations, model_name, relation, hmts, assoc, inverse_assoc_name, inv, code) }
+            else
+              build_bt_or_hm(relations, model_name, relation, hmts, assoc, inverse_assoc_name, invs, code)
             end
-
-            # And finally create a has_one, has_many, or belongs_to for this association
-            assoc_name = assoc_name.to_sym
-            code << "  #{macro} #{assoc_name.inspect}#{options.map { |k, v| ", #{k}: #{v.inspect}" }.join}\n"
-            self.send(macro, assoc_name, **options)
             hmts
           end
           hmts.each do |hmt_fk, fks|
@@ -698,6 +655,88 @@ class Object
       [built_model, code]
     end
 
+    def build_bt_or_hm(relations, model_name, relation, hmts, assoc, inverse_assoc_name, inverse_table, code)
+      singular_table_name = inverse_table&.singularize
+      options = {}
+      macro = if assoc[:is_bt]
+                # Try to take care of screwy names if this is a belongs_to going to an STI subclass
+                assoc_name = if (primary_class = assoc.fetch(:primary_class, nil)) &&
+                               sti_inverse_assoc = primary_class.reflect_on_all_associations.find do |a|
+                                 a.macro == :has_many && a.options[:class_name] == self.name && assoc[:fk] = a.foreign_key
+                               end
+                               sti_inverse_assoc.options[:inverse_of]&.to_s || assoc_name
+                             else
+                               assoc[:assoc_name]
+                             end
+                if assoc.key?(:polymorphic)
+                  options[:polymorphic] = true
+                else
+                  need_class_name = singular_table_name.underscore != assoc_name
+                  need_fk = "#{assoc_name}_id" != assoc[:fk]
+                end
+                if (inverse = assoc[:inverse])
+                  inverse_assoc_name, _x = _brick_get_hm_assoc_name(relations[inverse_table], inverse)
+                  has_ones = ::Brick.config.has_ones&.fetch(inverse[:alternate_name].camelize, nil)
+                  if has_ones&.key?(singular_inv_assoc_name = ActiveSupport::Inflector.singularize(inverse_assoc_name))
+                    inverse_assoc_name = if has_ones[singular_inv_assoc_name]
+                                           need_inverse_of = true
+                                           has_ones[singular_inv_assoc_name]
+                                         else
+                                           singular_inv_assoc_name
+                                         end
+                  end
+                end
+                :belongs_to
+              else
+                # need_class_name = ActiveSupport::Inflector.singularize(assoc_name) == ActiveSupport::Inflector.singularize(table_name.underscore)
+                # Are there multiple foreign keys out to the same table?
+                assoc_name, need_class_name = _brick_get_hm_assoc_name(relation, assoc)
+                if assoc.key?(:polymorphic)
+                  options[:as] = assoc[:fk].to_sym
+                else
+                  need_fk = "#{ActiveSupport::Inflector.singularize(assoc[:inverse][:inverse_table])}_id" != assoc[:fk]
+                end
+                # fks[table_name].find { |other_assoc| other_assoc.object_id != assoc.object_id && other_assoc[:assoc_name] == assoc[assoc_name] }
+                if (has_ones = ::Brick.config.has_ones&.fetch(model_name, nil))&.key?(singular_assoc_name = ActiveSupport::Inflector.singularize(assoc_name))
+                  assoc_name = if (custom_assoc_name = has_ones[singular_assoc_name])
+                                 need_class_name = custom_assoc_name != singular_assoc_name
+                                 custom_assoc_name
+                               else
+                                 singular_assoc_name
+                               end
+                  :has_one
+                else
+                  :has_many
+                end
+              end
+      # Figure out if we need to specially call out the class_name and/or foreign key
+      # (and if either of those then definitely also a specific inverse_of)
+      options[:class_name] = assoc[:primary_class]&.name || singular_table_name.camelize if need_class_name
+      # Work around a bug in CPK where self-referencing belongs_to associations double up their foreign keys
+      if need_fk # Funky foreign key?
+        options[:foreign_key] = if assoc[:fk].is_a?(Array)
+                                  assoc_fk = assoc[:fk].uniq
+                                  assoc_fk.length < 2 ? assoc_fk.first : assoc_fk
+                                else
+                                  assoc[:fk].to_sym
+                                end
+      end
+      options[:inverse_of] = inverse_assoc_name.to_sym if inverse_assoc_name && (need_class_name || need_fk || need_inverse_of)
+
+      # Prepare a list of entries for "has_many :through"
+      if macro == :has_many
+        relations[inverse_table][:hmt_fks].each do |k, hmt_fk|
+          next if k == assoc[:fk]
+
+          hmts[ActiveSupport::Inflector.pluralize(hmt_fk.last)] << [assoc, hmt_fk.first]
+        end
+      end
+      # And finally create a has_one, has_many, or belongs_to for this association
+      assoc_name = assoc_name.to_sym
+      code << "  #{macro} #{assoc_name.inspect}#{options.map { |k, v| ", #{k}: #{v.inspect}" }.join}\n"
+      self.send(macro, assoc_name, **options)
+    end
+
     def build_controller(class_name, plural_class_name, model, relations)
       table_name = ActiveSupport::Inflector.underscore(plural_class_name)
       singular_table_name = ActiveSupport::Inflector.singularize(table_name)
@@ -1010,12 +1049,13 @@ module Brick
   # rubocop:enable Style/CommentedKeyword
 
   class << self
-    def _add_bt_and_hm(fk, relations = nil)
-      relations ||= ::Brick.relations
+    def _add_bt_and_hm(fk, relations, is_polymorphic = false)
       bt_assoc_name = fk[1].underscore
       bt_assoc_name = bt_assoc_name[0..-4] if bt_assoc_name.end_with?('_id')
 
       bts = (relation = relations.fetch(fk[0], nil))&.fetch(:fks) { relation[:fks] = {} }
+      # %%% Do we miss out on has_many :through or even HM based on constantizing this model early?
+      # Maybe it's already gotten this info because we got as far as to say there was a unique class
       primary_table = (is_class = fk[2].is_a?(Hash) && fk[2].key?(:class)) ? (primary_class = fk[2][:class].constantize).table_name : fk[2]
       hms = (relation = relations.fetch(primary_table, nil))&.fetch(:fks) { relation[:fks] = {} } unless is_class
 
@@ -1032,7 +1072,7 @@ module Brick
           puts "Brick: Additional reference #{fk.inspect} refers to non-existent #{'table'.pluralize(missing.length)} #{missing.join(' and ')}. (Available tables include #{tables.join(', ')}.)"
           return
         end
-        unless (cols = relations[fk[0]][:cols]).key?(fk[1])
+        unless (cols = relations[fk[0]][:cols]).key?(fk[1]) || (is_polymorphic && cols.key?("#{fk[1]}_id") && cols.key?("#{fk[1]}_type"))
           columns = cols.map { |k, v| "#{k} (#{v.first.split(' ').first})" }
           puts "Brick: Additional reference #{fk.inspect} refers to non-existent column #{fk[1]}. (Columns present in #{fk[0]} are #{columns.join(', ')}.)"
           return
@@ -1047,10 +1087,21 @@ module Brick
         end
       end
       if (assoc_bt = bts[cnstr_name])
-        assoc_bt[:fk] = assoc_bt[:fk].is_a?(String) ? [assoc_bt[:fk], fk[1]] : assoc_bt[:fk].concat(fk[1])
-        assoc_bt[:assoc_name] = "#{assoc_bt[:assoc_name]}_#{fk[1]}"
+        if is_polymorphic
+          # Assuming same fk (don't yet support composite keys for polymorphics)
+          assoc_bt[:inverse_table] << fk[2]
+        else # Expect we could have a composite key going
+          if assoc_bt[:fk].is_a?(String)
+            assoc_bt[:fk] = [assoc_bt[:fk], fk[1]] unless fk[1] == assoc_bt[:fk]
+          elsif assoc_bt[:fk].exclude?(fk[1])
+            assoc_bt[:fk] << fk[1]
+          end
+          assoc_bt[:assoc_name] = "#{assoc_bt[:assoc_name]}_#{fk[1]}"
+        end
       else
-        assoc_bt = bts[cnstr_name] = { is_bt: true, fk: fk[1], assoc_name: bt_assoc_name, inverse_table: primary_table }
+        inverse_table = [primary_table] if is_polymorphic
+        assoc_bt = bts[cnstr_name] = { is_bt: true, fk: fk[1], assoc_name: bt_assoc_name, inverse_table: inverse_table || primary_table }
+        assoc_bt[:polymorphic] = true if is_polymorphic
       end
       if is_class
         # For use in finding the proper :source for a HMT association that references an STI subclass
@@ -1062,11 +1113,16 @@ module Brick
       return if is_class || ::Brick.config.exclude_hms&.any? { |exclusion| fk[0] == exclusion[0] && fk[1] == exclusion[1] && primary_table == exclusion[2] }
 
       if (assoc_hm = hms.fetch((hm_cnstr_name = "hm_#{cnstr_name}"), nil))
-        assoc_hm[:fk] = assoc_hm[:fk].is_a?(String) ? [assoc_hm[:fk], fk[1]] : assoc_hm[:fk].concat(fk[1])
+        if assoc_bt[:fk].is_a?(String)
+          assoc_bt[:fk] = [assoc_bt[:fk], fk[1]] unless fk[1] == assoc_bt[:fk]
+        elsif assoc_bt[:fk].exclude?(fk[1])
+          assoc_bt[:fk] << fk[1]
+        end
         assoc_hm[:alternate_name] = "#{assoc_hm[:alternate_name]}_#{bt_assoc_name}" unless assoc_hm[:alternate_name] == bt_assoc_name
         assoc_hm[:inverse] = assoc_bt
       else
         assoc_hm = hms[hm_cnstr_name] = { is_bt: false, fk: fk[1], assoc_name: fk[0], alternate_name: bt_assoc_name, inverse_table: fk[0], inverse: assoc_bt }
+        assoc_hm[:polymorphic] = true if is_polymorphic
         hm_counts = relation.fetch(:hm_counts) { relation[:hm_counts] = {} }
         hm_counts[fk[0]] = hm_counts.fetch(fk[0]) { 0 } + 1
       end

data/lib/brick/frameworks/rails/engine.rb

@@ -38,6 +38,9 @@ module Brick
 
         # Has one relationships
         ::Brick.has_ones = app.config.brick.fetch(:has_ones, nil)
+
+        # Polymorphic associations
+        ::Brick.polymorphics = app.config.brick.fetch(:polymorphics, nil)
       end
 
       # After we're initialized and before running the rest of stuff, put our configuration in place
@@ -64,12 +67,14 @@ module Brick
               is_template_exists
             end
 
-            def path_keys(fk_name, obj_name, pk)
-              if fk_name.is_a?(Array) && pk.is_a?(Array) # Composite keys?
-                fk_name.zip(pk.map { |pk_part| "#{obj_name}.#{pk_part}" })
-              else
-                [[fk_name, "#{obj_name}.#{pk}"]]
-              end.map { |x| "#{x.first}: #{x.last}"}.join(', ')
+            def path_keys(hm_assoc, fk_name, obj_name, pk)
+              keys = if fk_name.is_a?(Array) && pk.is_a?(Array) # Composite keys?
+                       fk_name.zip(pk.map { |pk_part| "#{obj_name}.#{pk_part}" })
+                     else
+                       [[fk_name, "#{obj_name}.#{pk}"]]
+                     end
+              keys << [hm_assoc.inverse_of.foreign_type, "#{hm_assoc.active_record.name}"] if hm_assoc.options.key?(:as)
+              keys.map { |x| "#{x.first}: #{x.last}"}.join(', ')
             end
 
             alias :_brick_find_template :find_template
@@ -102,12 +107,12 @@ module Brick
                                               "#{obj_name}.#{attrib_name} || 0"
                                             end
 "<%= ct = #{set_ct}
-     link_to \"#\{ct || 'View'\} #{assoc_name}\", #{hm_assoc.klass.name.underscore.pluralize}_path({ #{path_keys(hm_fk_name, obj_name, pk)} }) unless ct&.zero? %>\n"
+     link_to \"#\{ct || 'View'\} #{assoc_name}\", #{hm_assoc.klass.name.underscore.pluralize}_path({ #{path_keys(hm_assoc, hm_fk_name, obj_name, pk)} }) unless ct&.zero? %>\n"
                                  else # has_one
 "<%= obj = #{obj_name}.#{hm.first}; link_to(obj.brick_descrip, obj) if obj %>\n"
                                  end
                 elsif args.first == 'show'
-                  hm_stuff << "<%= link_to '#{assoc_name}', #{hm_assoc.klass.name.underscore.pluralize}_path({ #{path_keys(hm_fk_name, "@#{obj_name}&.first&", pk)} }) %>\n"
+                  hm_stuff << "<%= link_to '#{assoc_name}', #{hm_assoc.klass.name.underscore.pluralize}_path({ #{path_keys(hm_assoc, hm_fk_name, "@#{obj_name}&.first&", pk)} }) %>\n"
                 end
                 s << hm_stuff
               end
@@ -205,7 +210,19 @@ def hide_bcrypt(val)
 end %>"
 
               if ['index', 'show', 'update'].include?(args.first)
-                css << "<% bts = { #{bts.each_with_object([]) { |v, s| s << "#{v.first.inspect} => [#{v.last.first.inspect}, #{v.last[1].name}, #{v.last[1].primary_key.inspect}]"}.join(', ')} } %>"
+                poly_cols = []
+                css << "<% bts = { #{
+                  bts.each_with_object([]) do |v, s|
+                    foreign_models = if v.last[2] # Polymorphic?
+                                       poly_cols << @_brick_model.reflect_on_association(v[1].first).foreign_type
+                                       v.last[1].each_with_object([]) { |x, s| s << "[#{x.name}, #{x.primary_key.inspect}]" }.join(', ')
+                                     else
+                                       "[#{v.last[1].name}, #{v.last[1].primary_key.inspect}]"
+                                     end
+                    s << "#{v.first.inspect} => [#{v.last.first.inspect}, [#{foreign_models}], #{v.last[2].inspect}]"
+                  end.join(', ')
+                } }
+                poly_cols = #{poly_cols.inspect} %>"
               end
 
               # %%% When doing schema select, if there's an ID then remove it, or if we're on a new page go to index
@@ -274,108 +291,112 @@ function changeout(href, param, value) {
                            template_link = "
   <%= link_to 'CSV', #{table_name}_path(format: :csv) %> &nbsp; <a href=\"#\" id=\"sheetsLink\">Sheets</a>
   <div id=\"dropper\" contenteditable=\"true\"></div>
-  <input type=\"button\" id=\"btnImport\" value=\"Import\">"
-                         end
-"#{css}
-<p style=\"color: green\"><%= notice %></p>#{"
-<select id=\"schema\">#{schema_options}</select>" if ::Brick.db_schemas.length > 1}
-<select id=\"tbl\">#{table_options}</select>
-<h1>#{model_name.pluralize}</h1>#{template_link}
+  <input type=\"button\" id=\"btnImport\" value=\"Import\">
+
 <script>
-var dropperDiv = document.getElementById(\"dropper\");
-var btnImport = document.getElementById(\"btnImport\");
-var droppedTSV;
-if (dropperDiv) { // Other interesting events: blur keyup input
-  dropperDiv.addEventListener(\"paste\", function (evt) {
-    droppedTSV = evt.clipboardData.getData('text/plain');
-    var html = evt.clipboardData.getData('text/html');
-    var tbl = html.substring(html.indexOf(\"<tbody>\") + 7, html.lastIndexOf(\"</tbody>\"));
-    console.log(tbl);
-    btnImport.style.display = droppedTSV.length > 0 ? \"block\" : \"none\";
-  });
-  btnImport.addEventListener(\"click\", function () {
-    fetch(changeout(<%= #{obj_name}_path(-1, format: :csv).inspect.html_safe %>, \"_brick_schema\", brickSchema), {
-      method: 'PATCH',
-      headers: { 'Content-Type': 'text/tab-separated-values' },
-      body: droppedTSV
-    }).then(function (tsvResponse) {
-      btnImport.style.display = \"none\";
-      console.log(\"toaster\", tsvResponse);
+  var dropperDiv = document.getElementById(\"dropper\");
+  var btnImport = document.getElementById(\"btnImport\");
+  var droppedTSV;
+  if (dropperDiv) { // Other interesting events: blur keyup input
+    dropperDiv.addEventListener(\"paste\", function (evt) {
+      droppedTSV = evt.clipboardData.getData('text/plain');
+      var html = evt.clipboardData.getData('text/html');
+      var tbl = html.substring(html.indexOf(\"<tbody>\") + 7, html.lastIndexOf(\"</tbody>\"));
+      console.log(tbl);
+      btnImport.style.display = droppedTSV.length > 0 ? \"block\" : \"none\";
     });
-  });
-}
-var sheetUrl;
-var spreadsheetId;
-var sheetsLink = document.getElementById(\"sheetsLink\");
-function gapiLoaded() {
-  // Have a click on the sheets link to bring up the sign-in window.  (Must happen from some kind of user click.)
-  sheetsLink.addEventListener(\"click\", async function (evt) {
-    evt.preventDefault();
-    await gapi.load(\"client\", function () {
-      gapi.client.init({ // Load the discovery doc to initialize the API
-        clientId: \"487319557829-fgj4u660igrpptdji7ev0r5hb6kh05dh.apps.googleusercontent.com\",
-        scope: \"https://www.googleapis.com/auth/spreadsheets https://www.googleapis.com/auth/drive.file\",
-        discoveryDocs: [\"https://sheets.googleapis.com/$discovery/rest?version=v4\"]
-      }).then(function () {
-        gapi.auth2.getAuthInstance().isSignedIn.listen(updateSignInStatus);
-        updateSignInStatus(gapi.auth2.getAuthInstance().isSignedIn.get());
+    btnImport.addEventListener(\"click\", function () {
+      fetch(changeout(<%= #{obj_name}_path(-1, format: :csv).inspect.html_safe %>, \"_brick_schema\", brickSchema), {
+        method: 'PATCH',
+        headers: { 'Content-Type': 'text/tab-separated-values' },
+        body: droppedTSV
+      }).then(function (tsvResponse) {
+        btnImport.style.display = \"none\";
+        console.log(\"toaster\", tsvResponse);
       });
     });
-  });
-}
+  }
+  var sheetUrl;
+  var spreadsheetId;
+  var sheetsLink = document.getElementById(\"sheetsLink\");
+  function gapiLoaded() {
+    // Have a click on the sheets link to bring up the sign-in window.  (Must happen from some kind of user click.)
+    sheetsLink.addEventListener(\"click\", async function (evt) {
+      evt.preventDefault();
+      await gapi.load(\"client\", function () {
+        gapi.client.init({ // Load the discovery doc to initialize the API
+          clientId: \"487319557829-fgj4u660igrpptdji7ev0r5hb6kh05dh.apps.googleusercontent.com\",
+          scope: \"https://www.googleapis.com/auth/spreadsheets https://www.googleapis.com/auth/drive.file\",
+          discoveryDocs: [\"https://sheets.googleapis.com/$discovery/rest?version=v4\"]
+        }).then(function () {
+          gapi.auth2.getAuthInstance().isSignedIn.listen(updateSignInStatus);
+          updateSignInStatus(gapi.auth2.getAuthInstance().isSignedIn.get());
+        });
+      });
+    });
+  }
 
-async function updateSignInStatus(isSignedIn) {
-  if (isSignedIn) {
-    console.log(\"turds!\");
-    await gapi.client.sheets.spreadsheets.create({
-      properties: {
-        title: #{table_name.inspect},
-      },
-      sheets: [
-        // sheet1, sheet2, sheet3
-      ]
-    }).then(function (response) {
-      sheetUrl = response.result.spreadsheetUrl;
-      spreadsheetId = response.result.spreadsheetId;
-      sheetsLink.setAttribute(\"href\", sheetUrl); // response.result.spreadsheetUrl
-      console.log(\"x1\", sheetUrl);
-
-      // Get JSON data
-      fetch(changeout(<%= #{table_name}_path(format: :js).inspect.html_safe %>, \"_brick_schema\", brickSchema)).then(function (response) {
-        response.json().then(function (data) {
-          gapi.client.sheets.spreadsheets.values.append({
-            spreadsheetId: spreadsheetId,
-            range: \"Sheet1\",
-            valueInputOption: \"RAW\",
-            insertDataOption: \"INSERT_ROWS\"
-          }, {
-            range: \"Sheet1\",
-            majorDimension: \"ROWS\",
-            values: data,
-          }).then(function (response2) {
-//            console.log(\"beefcake\", response2);
+  async function updateSignInStatus(isSignedIn) {
+    if (isSignedIn) {
+      console.log(\"turds!\");
+      await gapi.client.sheets.spreadsheets.create({
+        properties: {
+          title: #{table_name.inspect},
+        },
+        sheets: [
+          // sheet1, sheet2, sheet3
+        ]
+      }).then(function (response) {
+        sheetUrl = response.result.spreadsheetUrl;
+        spreadsheetId = response.result.spreadsheetId;
+        sheetsLink.setAttribute(\"href\", sheetUrl); // response.result.spreadsheetUrl
+        console.log(\"x1\", sheetUrl);
+
+        // Get JSON data
+        fetch(changeout(<%= #{table_name}_path(format: :js).inspect.html_safe %>, \"_brick_schema\", brickSchema)).then(function (response) {
+          response.json().then(function (data) {
+            gapi.client.sheets.spreadsheets.values.append({
+              spreadsheetId: spreadsheetId,
+              range: \"Sheet1\",
+              valueInputOption: \"RAW\",
+              insertDataOption: \"INSERT_ROWS\"
+            }, {
+              range: \"Sheet1\",
+              majorDimension: \"ROWS\",
+              values: data,
+            }).then(function (response2) {
+  //            console.log(\"beefcake\", response2);
+            });
           });
         });
       });
-    });
-    window.open(sheetUrl, '_blank');
+      window.open(sheetUrl, '_blank');
+    }
   }
-}
 </script>
 <script async defer src=\"https://apis.google.com/js/api.js\" onload=\"gapiLoaded()\"></script>
-
+"
+                         end
+"#{css}
+<p style=\"color: green\"><%= notice %></p>#{"
+<select id=\"schema\">#{schema_options}</select>" if ::Brick.db_schemas.length > 1}
+<select id=\"tbl\">#{table_options}</select>
+<h1>#{model_name.pluralize}</h1>#{template_link}
 
 <% if @_brick_params&.present? %><h3>where <%= @_brick_params.each_with_object([]) { |v, s| s << \"#\{v.first\} = #\{v.last.inspect\}\" }.join(', ') %></h3><% end %>
 <table id=\"#{table_name}\">
   <thead><tr>#{'<th></th>' if pk}
   <% @#{table_name}.columns.map(&:name).each do |col| %>
-    <% next if col == '#{pk}' || ::Brick.config.metadata_columns.include?(col) %>
+    <% next if col == '#{pk}' || ::Brick.config.metadata_columns.include?(col) || poly_cols.include?(col) %>
     <th>
     <% if (bt = bts[col]) %>
-      BT <%= bt[1].bt_link(bt.first) %>
-    <% else %>
-      <%= col %>
-    <% end %>
+         BT <%
+         bt[1].each do |bt_pair| %><%=
+           bt_pair.first.bt_link(bt.first) %> <%
+         end %><%
+       else %><%=
+         col %><%
+       end %>
     </th>
   <% end %>
   <%# Consider getting the name from the association -- h.first.name -- if a more \"friendly\" alias should be used for a screwy table name %>
@@ -387,21 +408,30 @@ async function updateSignInStatus(isSignedIn) {
   <tr>#{"
     <td><%= link_to '⇛', #{obj_name}_path(#{obj_pk}), { class: 'big-arrow' } %></td>" if obj_pk}
     <% #{obj_name}.attributes.each do |k, val| %>
-      <% next if k == '#{pk}' || ::Brick.config.metadata_columns.include?(k) || k.start_with?('_brfk_') || (k.start_with?('_br_') && k.end_with?('_ct')) %>
+      <% next if k == '#{pk}' || ::Brick.config.metadata_columns.include?(k) || poly_cols.include?(k) || k.start_with?('_brfk_') || (k.start_with?('_br_') && (k.length == 63 || k.end_with?('_ct'))) %>
       <td>
       <% if (bt = bts[k]) %>
         <%# binding.pry # Postgres column names are limited to 63 characters %>
-        <% bt_txt = bt[1].brick_descrip(#{obj_name}, @_brick_bt_descrip[bt.first][1].map { |z| #{obj_name}.send(z.last[0..62]) }, @_brick_bt_descrip[bt.first][2]) %>
-        <% bt_id_col = @_brick_bt_descrip[bt.first][2]; bt_id = #{obj_name}.send(*bt_id_col) if bt_id_col&.present? %>
-        <%= bt_id ? link_to(bt_txt, send(\"#\{bt_obj_path_base = bt[1].name.underscore\}_path\".to_sym, bt_id)) : bt_txt %>
-        <%#= Previously was:  bt_obj = bt[1].find_by(bt[2] => val); link_to(bt_obj.brick_descrip, send(\"#\{bt_obj_path_base = bt[1].name.underscore\}_path\".to_sym, bt_obj.send(bt[1].primary_key.to_sym))) if bt_obj %>
+        <% if bt[2] # Polymorphic?
+             bt_class = #{obj_name}.send(\"#\{bt.first\}_type\")
+             base_class = (::Brick.existing_stis[bt_class] || bt_class).constantize.base_class.name.underscore
+             poly_id = #{obj_name}.send(\"#\{bt.first\}_id\")
+             %><%= link_to(\"#\{bt_class\} ##\{poly_id\}\",
+                           send(\"#\{base_class\}_path\".to_sym, poly_id)) if poly_id %><%
+           else
+             bt_txt = (bt_class = bt[1].first.first).brick_descrip(
+               #{obj_name}, (descrips = @_brick_bt_descrip[bt.first][bt_class])[0..-2].map { |z| #{obj_name}.send(z.last[0..62]) }, (bt_id_col = descrips.last)
+             )
+             bt_id = #{obj_name}.send(*bt_id_col) if bt_id_col&.present? %>
+          <%= bt_id ? link_to(bt_txt, send(\"#\{bt_class.base_class.name.underscore\}_path\".to_sym, bt_id)) : bt_txt %>
+          <%#= Previously was:  bt_obj = bt[1].first.first.find_by(bt[2] => val); link_to(bt_obj.brick_descrip, send(\"#\{bt[1].first.first.name.underscore\}_path\".to_sym, bt_obj.send(bt[1].first.first.primary_key.to_sym))) if bt_obj %>
+        <% end %>
       <% else %>
         <%= hide_bcrypt(val) %>
       <% end %>
       </td>
     <% end %>
-    <td>#{hms_columns.join('</td><td>')}</td>
-    <!-- td>X</td -->
+    #{hms_columns.each_with_object(+'') { |hm_col, s| s << "<td>#{hm_col}</td>" }}
   </tr>
   </tbody>
   <% end %>
@@ -418,35 +448,47 @@ async function updateSignInStatus(isSignedIn) {
 <%= link_to '(See all #{obj_name.pluralize})', #{table_name}_path %>
 <% if obj %>
   <%= # path_options = [obj.#{pk}]
-   # path_options << { '_brick_schema':  } if
-   # url = send(:#{model_name.underscore}_path, obj.#{pk})
-   form_for(obj.becomes(#{model_name})) do |f| %>
+    # path_options << { '_brick_schema':  } if
+    # url = send(:#{model_name.underscore}_path, obj.#{pk})
+    form_for(obj.becomes(#{model_name})) do |f| %>
   <table>
-  <% @#{obj_name}.first.attributes.each do |k, val| %>
+  <% has_fields = false
+    @#{obj_name}.first.attributes.each do |k, val| %>
     <tr>
     <%# %%% Accommodate composite keys %>
     <% next if k == '#{pk}' || ::Brick.config.metadata_columns.include?(k) %>
     <th class=\"show-field\">
-    <% if (bt = bts[k])
-      # Add a final member in this array with descriptive options to be used in <select> drop-downs
-      bt_name = bt[1].name
-      # %%% Only do this if the user has permissions to edit this bt field
-      if bt.length < 4
-        bt << (option_detail = [[\"(No #\{bt_name\} chosen)\", '^^^brick_NULL^^^']])
-        # %%% Accommodate composite keys for obj.pk at the end here
-        bt[1].order(bt[1].primary_key).each { |obj| option_detail << [obj.brick_descrip, obj.send(bt[1].primary_key)] }
-      end %>
-      BT <%= bt[1].bt_link(bt.first) %>
+    <% has_fields = true
+      if (bt = bts[k])
+        # Add a final member in this array with descriptive options to be used in <select> drop-downs
+        bt_name = bt[1].map { |x| x.first.name }.join('/')
+        # %%% Only do this if the user has permissions to edit this bt field
+        if bt[2] # Polymorphic?
+          poly_class_name = @#{obj_name}.first.send(\"#\{bt.first\}_type\")
+          bt_pair = bt[1].find { |pair| pair.first.name == poly_class_name }
+          # descrips = @_brick_bt_descrip[bt.first][bt_class]
+          poly_id = @#{obj_name}.first.send(\"#\{bt.first\}_id\")
+          # bt_class.order(obj_pk = bt_class.primary_key).each { |obj| option_detail << [obj.brick_descrip(nil, obj_pk), obj.send(obj_pk)] }
+        else # No polymorphism, so just get the first one
+          bt_pair = bt[1].first
+        end
+        bt_class = bt_pair.first
+        if bt.length < 4
+          bt << (option_detail = [[\"(No #\{bt_name\} chosen)\", '^^^brick_NULL^^^']])
+          # %%% Accommodate composite keys for obj.pk at the end here
+          bt_class.order(obj_pk = bt_class.primary_key).each { |obj| option_detail << [obj.brick_descrip(nil, obj_pk), obj.send(obj_pk)] }
+        end %>
+        BT <%= bt_class.bt_link(bt.first) %>
     <% else %>
       <%= k %>
     <% end %>
     </th>
     <td>
-    <% if (bt = bts[k]) # bt_obj.brick_descrip
+    <% if bt
       html_options = { prompt: \"Select #\{bt_name\}\" }
       html_options[:class] = 'dimmed' unless val %>
       <%= f.select k.to_sym, bt[3], { value: val || '^^^brick_NULL^^^' }, html_options %>
-      <%= bt_obj = bt[1].find_by(bt[2] => val); link_to('⇛', send(\"#\{bt_obj_path_base = bt_name.underscore\}_path\".to_sym, bt_obj.send(bt[1].primary_key.to_sym)), { class: 'show-arrow' }) if bt_obj %>
+      <%= bt_obj = bt_class.find_by(bt_pair[1] => val); link_to('⇛', send(\"#\{bt_class.base_class.name.underscore\}_path\".to_sym, bt_obj.send(bt_class.primary_key.to_sym)), { class: 'show-arrow' }) if bt_obj %>
     <% else case #{model_name}.column_for_attribute(k).type
       when :string, :text %>
         <% if is_bcrypt?(val) # || .readonly? %>
@@ -467,8 +509,12 @@ async function updateSignInStatus(isSignedIn) {
     <% end %>
     </td>
     </tr>
-    <% end %>
+  <% end
+  if has_fields %>
     <tr><td colspan=\"2\" class=\"right\"><%= f.submit %></td></tr>
+  <% else %>
+    <tr><td colspan=\"2\">(No displayable fields)</td></tr>
+  <% end %>
   </table>
   <% end %>
 
@@ -482,6 +528,7 @@ async function updateSignInStatus(isSignedIn) {
           <tr><td>(none)</td></tr>
         <% else %>
           <% collection.uniq.each do |#{hm_singular_name = hm_name.singularize.underscore}| %>
+            <%# %%% accommodate composite primary key %>
             <tr><td><%= link_to(#{hm_singular_name}.brick_descrip, #{hm.first.klass.name.underscore}_path(#{hm_singular_name}.#{pk})) %></td></tr>
           <% end %>
         <% end %>
@@ -510,7 +557,7 @@ async function updateSignInStatus(isSignedIn) {
         end
 
         # Just in case it hadn't been done previously when we tried to load the brick initialiser,
-        # go make sure we've loaded additional references (virtual foreign keys).
+        # go make sure we've loaded additional references (virtual foreign keys and polymorphic associations).
         ::Brick.load_additional_references
       end
     end

data/lib/brick/version_number.rb

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

data/lib/brick.rb

@@ -85,12 +85,16 @@ module Brick
     @sti_models ||= {}
   end
 
+  def self.existing_stis
+    @existing_stis ||= Brick.config.sti_namespace_prefixes.each_with_object({}) { |snp, s| s[snp.first[2..-1]] = snp.last unless snp.first.end_with?('::') }
+  end
+
   class << self
     attr_accessor :db_schemas
 
     def set_db_schema(params)
       schema = params['_brick_schema'] || 'public'
-      ActiveRecord::Base.connection.execute("SET SEARCH_PATH='#{schema}';") if schema && ::Brick.db_schemas&.include?(schema)
+      ActiveRecord::Base.execute_sql("SET SEARCH_PATH = ?;", schema) if schema && ::Brick.db_schemas&.include?(schema)
     end
 
     # All tables and views (what Postgres calls "relations" including column and foreign key info)
@@ -107,7 +111,13 @@ module Brick
 
         case a.macro
         when :belongs_to
-          s.first[a.foreign_key] = [a.name, a.klass]
+          s.first[a.foreign_key] = if a.polymorphic?
+                                     primary_tables = relations[model.table_name][:fks].find { |_k, fk| fk[:assoc_name] == a.name.to_s }&.last&.fetch(:inverse_table, [])
+                                     models = primary_tables&.map { |table| table.singularize.camelize.constantize }
+                                     [a.name, models, true]
+                                   else
+                                     [a.name, a.klass]
+                                   end
         when :has_many, :has_one # This gets has_many as well as has_many :through
           # %%% weed out ones that don't have an available model to reference
           s.last[a.name] = a
@@ -256,6 +266,12 @@ module Brick
       end
     end
 
+    # Polymorphic associations
+    def polymorphics=(polys)
+      polys = polys.each_with_object({}) { |poly, s| s[poly] = nil } if polys.is_a?(Array)
+      Brick.config.polymorphics = polys || {}
+    end
+
     # DSL templates for individual models to provide prettier descriptions of objects
     # @api public
     def model_descrips=(descrips)
@@ -268,14 +284,48 @@ module Brick
       Brick.config.sti_namespace_prefixes = snp
     end
 
+    # Database schema to use when analysing existing data, such as deriving a list of polymorphic classes
+    # for polymorphics in which it wasn't originally specified.
+    # @api public
+    def schema_to_analyse=(schema)
+      Brick.config.schema_to_analyse = schema
+    end
+
     # Load additional references (virtual foreign keys)
     # This is attempted early if a brick initialiser file is found, and then again as a failsafe at the end of our engine's initialisation
     # %%% Maybe look for differences the second time 'round and just add new stuff instead of entirely deferring
     def load_additional_references
       return if @_additional_references_loaded
 
-      if (ars = ::Brick.config.additional_references)
-        ars.each { |fk| ::Brick._add_bt_and_hm(fk[0..2]) }
+      relations = ::Brick.relations
+      if (ars = ::Brick.config.additional_references) || ::Brick.config.polymorphics
+        ars.each { |fk| ::Brick._add_bt_and_hm(fk[0..2], relations) } if ars
+        if (polys = ::Brick.config.polymorphics)
+          if (schema = ::Brick.config.schema_to_analyse) && ::Brick.db_schemas&.include?(schema)
+            ActiveRecord::Base.execute_sql("SET SEARCH_PATH = ?;", schema)
+          end
+          missing_stis = {}
+          polys.each do |k, v|
+            table_name, poly = k.split('.')
+            v ||= ActiveRecord::Base.execute_sql("SELECT DISTINCT #{poly}_type AS typ FROM #{table_name}").map { |result| result['typ'] }
+            v.each do |type|
+              if relations.key?(primary_table = type.underscore.pluralize)
+                ::Brick._add_bt_and_hm([table_name, poly, primary_table, "(brick) #{table_name}_#{poly}"], relations, true)
+              else
+                missing_stis[primary_table] = type unless ::Brick.existing_stis.key?(type)
+              end
+            end
+          end
+          unless missing_stis.empty?
+            print "
+You might be missing an STI namespace prefix entry for these tables:  #{missing_stis.keys.join(', ')}.
+In config/initializers/brick.rb appropriate entries would look something like:
+  Brick.sti_namespace_prefixes = {"
+            puts missing_stis.map { |_k, missing_sti| "\n    '::#{missing_sti}' => 'YourParentModel'" }.join(',')
+            puts "  }
+(Just trade out YourParentModel with some more appropriate one.)"
+          end
+        end
         @_additional_references_loaded = true
       end
 

data/lib/generators/brick/install_generator.rb

@@ -19,11 +19,31 @@ module Brick
 
     def create_initializer_file
       unless File.exist?(filename = 'config/initializers/brick.rb')
-        # See if we can make suggestions for additional_references
-        resembles_fks = []
-        possible_additional_references = (relations = ::Brick.relations).each_with_object([]) do |v, s|
-          v.last[:cols].each do |col, _type|
+        # See if we can make suggestions for additional_references and polymorphic associations
+        resembles_fks = Hash.new { |h, k| h[k] = [] }
+        possible_polymorphics = {}
+        possible_additional_references = (relations = ::Brick.relations).each_with_object(Hash.new { |h, k| h[k] = [] }) do |v, s|
+          model_filename = "app/models/#{ActiveSupport::Inflector.singularize(v.first)}.rb"
+          v.last[:cols].each do |col, type|
             col_down = col.downcase
+
+            if (is_possible_poly = ['character varying', 'text'].include?(type.first))
+              if col_down.end_with?('_type') &&
+                poly_type_cut_length = -6
+                col_down = col_down[0..-6]
+              elsif col_down.end_with?('type')
+                poly_type_cut_length = -5
+                col_down = col_down[0..-5]
+              else
+                is_possible_poly = false
+              end
+              is_possible_poly = false if col_down.length < 6 # Was it simply called "type" or something else really short?
+              if is_possible_poly && !File.exist?(model_filename) # Make sure a model file isn't present
+                possible_polymorphics["#{v.first}.#{col_down}"] = "'#{v.first}.#{col[0..poly_type_cut_length]}'"
+                next
+              end
+            end
+
             is_possible = true
             if col_down.end_with?('_id')
               col_down = col_down[0..-4]
@@ -40,30 +60,48 @@ module Brick
             if col_down.start_with?('fk_')
               is_possible = true
               col_down = col_down[3..-1]
+            elsif col_down.start_with?('fk')
+              is_possible = true
+              col_down = col_down[2..-1]
             end
             # This possible key not really a primary key and not yet used as a foreign key?
             if is_possible && !(relation = relations.fetch(v.first, {}))[:pkey].first&.last&.include?(col) &&
                !relations.fetch(v.first, {})[:fks]&.any? { |_k, v| v[:is_bt] && v[:fk] == col }
-              if (relations.fetch(f_table = col_down, nil) ||
-                 relations.fetch(f_table = ActiveSupport::Inflector.pluralize(col_down), nil)) &&
-                 # Looks pretty promising ... just make sure a model file isn't present
-                 !File.exist?("app/models/#{ActiveSupport::Inflector.singularize(v.first)}.rb")
-                s << "['#{v.first}', '#{col}', '#{f_table}']"
-              else
-                resembles_fks << "#{v.first}.#{col}"
+              # Starting to look promising ... make sure a model file isn't present
+              if !File.exist?(model_filename)
+                if (relations.fetch(f_table = col_down, nil) ||
+                  relations.fetch(f_table = ActiveSupport::Inflector.pluralize(col_down), nil)) &&
+                  s["#{v.first}.#{col_down}"] << "['#{v.first}', '#{col}', '#{f_table}']"
+                else
+                  resembles_fks["#{v.first}.#{col_down}"] << "#{v.first}.#{col}"
+                end
               end
             end
           end
-          s
         end
 
-        bar = case possible_additional_references.length
+        possible_polymorphics.each_key do |k|
+          # Also matching one of the FK suggestions means it could be polymorphic,
+          # so delete any suggestions for a FK of the same name and only recommend
+          # the polymorphic association.
+          if resembles_fks.key?(k)
+            resembles_fks.delete(k)
+          elsif possible_additional_references.key?(k)
+            possible_additional_references.delete(k)
+          else
+            # While this one has a type, it's missing a corresponding ID column so it isn't polymorphic
+            possible_polymorphics.delete(k)
+          end
+        end
+        resembles_fks = resembles_fks.values.flatten
+
+        bar = case (possible_additional_references = possible_additional_references.values.flatten).length
               when 0
 +"# Brick.additional_references = [['orders', 'customer_id', 'customer'],
 #                                ['customer', 'region_id', 'regions']]"
               when 1
 +"# # Here is a possible additional reference that has been auto-identified for the #{ActiveRecord::Base.connection.current_database} database:
-# Brick.additional_references = [[#{possible_additional_references.first}]"
+# Brick.additional_references = [#{possible_additional_references.first}]"
               else
 +"# # Here are possible additional references that have been auto-identified for the #{ActiveRecord::Base.connection.current_database} database:
 # Brick.additional_references = [
@@ -75,6 +113,26 @@ module Brick
 # #   #{resembles_fks.join(', ')}"
       end
 
+      poly = case (possible_polymorphics = possible_polymorphics.values.flatten).length
+      when 0
+" like this:
+# Brick.polymorphics = [
+#                        'comments.commentable',
+#                        'images.imageable'
+#                      ]"
+      when 1
+".
+# # Here is a possible polymorphic association that has been auto-identified for the #{ActiveRecord::Base.connection.current_database} database:
+# Brick.polymorphics = [#{possible_additional_references.first}]"
+
+      else
+".
+# # Here are possible polymorphic associations that have been auto-identified for the #{ActiveRecord::Base.connection.current_database} database:
+# Brick.polymorphics = [
+#   #{possible_polymorphics.join(",\n#   ")}
+# ]"
+      end
+
       create_file(filename, "# frozen_string_literal: true
 
 # # Settings for the Brick gem
@@ -119,7 +177,7 @@ module Brick
 # # Skip showing counts for these specific has_many associations when building auto-generated #index views.
 # # When there are related tables with a significant number of records, this can lessen the load on the database
 # # considerably, sometimes fixing what might appear to be an index page that just \"hangs\" for no apparent reason.
-Brick.skip_index_hms = ['User.litany_of_woes']
+# Brick.skip_index_hms = ['User.litany_of_woes']
 
 # # By default primary tables involved in a foreign key relationship will indicate a \"has_many\" relationship pointing
 # # back to the foreign table.  In order to represent a \"has_one\" association instead, an override can be provided
@@ -157,6 +215,12 @@ Brick.skip_index_hms = ['User.litany_of_woes']
 # Brick.sti_namespace_prefixes = { '::Animals::' => 'Animal',
 #                                  '::Snake' => 'Reptile' }
 
+# # Database schema to use when analysing existing data, such as deriving a list of polymorphic classes in the case that
+# # it wasn't originally specified.
+# Brick.schema_to_analyse = 'engineering'
+
+# # Polymorphic associations are set up by providing a model name and polymorphic association name#{poly}
+
 # # If a default route is not supplied, Brick attempts to find the most \"central\" table and wires up the default
 # # route to go to the :index action for what would be a controller for that table.  You can specify any controller
 # # name and action you wish in order to override this and have that be the default route when none other has been

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: brick
 version: !ruby/object:Gem::Version
-  version: 1.0.23
+  version: 1.0.26
 platform: ruby
 authors:
 - Lorin Thwaits
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-05-12 00:00:00.000000000 Z
+date: 2022-05-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord