activerecord-exclusive-arc 0.2.3 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 315a60d914bac0fcce5129cddb165afa90604355e2c4bc9c46cad1b2d5aac79a
-  data.tar.gz: 5b93a4be4ddba10b47eaaaddbd5ba7cb5df072b2462114e01945e2c2f9b051d6
+  metadata.gz: 2fa6d4449542465ef3cc8420e9428a6faee1b4a91c547f5df660aeb615c15638
+  data.tar.gz: f945d9a76c78658e9c3a7a357f9c46202377ccd11488ec1cd286dfb467dd8eb2
 SHA512:
-  metadata.gz: 502088f01538ad5c31fbb4052f2a03c7f4dbe5723533d452b07abb331dae7d20c6dc4b87f3c5e019731c2b6cfce43834726f4cc89febf32e971593d2ecb39e44
-  data.tar.gz: 74c62d1c094c033ff0224af98128fe0d65a114e1f36099845cf1d3312cbd39d34a73db71f65b275eda92746be51d063e62ca65feb31849578833951af8115976
+  metadata.gz: 8c829330cc1049ffddf223122cab10681e454f13184ea0ae7218bb55ed8f3b90e3ede9e140907917b2402f7a5740b0857ffe5c71dd0ed798a9a82cc364c30438
+  data.tar.gz: 140aea768b0e93d5e7cf22b7c0c6517a7ce19ecc64e48bce405501ed06adcc47072a8f789bccabb20fb9b0fb366b2928d5b14573bff9bed91245a4262fb360af

data/Gemfile

@@ -1,11 +1,14 @@
 source "https://rubygems.org"
 
 gem "debug"
-gem "minitest"
-gem "rails", "~> #{ENV.fetch("RAILS_VERSION", "7.0")}"
-gem "rake"
+if (rails_version = ENV["RAILS_VERSION"])
+  gem "rails", "~> #{rails_version}.0"
+else
+  gem "rails"
+end
 gem "pg"
 gem "sqlite3"
 gem "standard", "~> 1.26"
+gem "mysql2"
 
 gemspec

data/README.md

@@ -5,16 +5,7 @@ types of ActiveRecord models.
 
 ### Doesn’t Rails already provide a way to do this?
 
-It does, but there are decent arguments against the default Rails way of doing polymorphism.
-Consider the fact that the Ruby class name is stored in the database as a string. If you want to
-change the name of the Ruby class used for such reasons, you must also update the database strings
-that represent it. The seeping of application-layer definitions into the database may become a
-liability.
-
-Another common argument concerns referential integrity. _Foreign Key Constraints_ are a common
-mechanism to ensure primary keys of database tables can be reliably used as foreign keys on others.
-This becomes harder to enforce in the databse when a string column that represents a Ruby class is
-one of the components required for unique identification.
+Yes but [here’s a post about why this exists](https://waymondo.com/posts/are-exclusive-arcs-evil/).
 
 ### So how does this work?
 
@@ -88,14 +79,14 @@ Continuing with our example, the generator command would also produce a migratio
 this:
 
 ```ruby
-class CommentCommentableExclusiveArc < ActiveRecord::Migration[7.0]
+class CommentCommentableExclusiveArcPostComment < ActiveRecord::Migration[7.0]
   def change
     add_reference :comments, :post, foreign_key: true, index: {where: "post_id IS NOT NULL"}
     add_reference :comments, :comment, foreign_key: true, index: {where: "comment_id IS NOT NULL"}
     add_check_constraint(
       :comments,
       "(CASE WHEN post_id IS NULL THEN 0 ELSE 1 END + CASE WHEN comment_id IS NULL THEN 0 ELSE 1 END) = 1",
-      name: :commentable
+      name: "commentable"
     )
   end
 end
@@ -127,12 +118,49 @@ Notably, if you want to make an Exclusive Arc optional, you can use the `--optio
 adjust the definition in your `ActiveRecord` model and loosen both the validation and database check
 constraint so that there can be 0 or 1 foreign keys set for the polymorphic reference.
 
+### Updating an existing exclusive arc
+
+If you need to add an additional polymorphic option to an existing exclusive arc, you can simply run
+the generator command again with the additional target. Existing references will be skipped and the
+check constraint will be removed and re-added in a reversible manner.
+
+```
+bin/rails g exclusive_arc Comment commentable post comment page
+```
+
+``` ruby
+class CommentCommentableExclusiveArcPostCommentPage < ActiveRecord::Migration[7.0]
+  def change
+    add_reference :comments, :page, foreign_key: true, index: {where: "page_id IS NOT NULL"}
+    remove_check_constraint(
+      :comments,
+      "(CASE WHEN post_id IS NULL THEN 0 ELSE 1 END + CASE WHEN comment_id IS NULL THEN 0 ELSE 1 END) = 1",
+      name: "commentable"
+    )
+    add_check_constraint(
+      :comments,
+      "(CASE WHEN post_id IS NULL THEN 0 ELSE 1 END + CASE WHEN comment_id IS NULL THEN 0 ELSE 1 END + CASE WHEN page_id IS NULL THEN 0 ELSE 1 END) = 1",
+      name: "commentable"
+    )
+  end
+end
+```
+
+The registration in the model will be updated as well.
+
+``` ruby
+class Comment < ApplicationRecord
+  include ExclusiveArc::Model
+  has_exclusive_arc :commentable, [:post, :comment, :page]
+end
+```
+
 ### Compatibility
 
 Currently `activerecord-exclusive-arc` is tested against a matrix of:
-* Ruby 2.7 and 3.2
-* Rails 6.1 and 7.0
-* `postgresql` and `sqlite3` database adapters
+* Ruby 2.7 and 3.3
+* Rails 6.1, 7.0, 7.1
+* `postgresql`, `sqlite3`, and `mysql2` database adapters
 
 ### Contributing
 

data/lib/exclusive_arc/model.rb

@@ -15,7 +15,9 @@ module ExclusiveArc
           next if reflections[option.to_s]
 
           belongs_to(option, optional: true)
+          validate :"validate_#{arc}"
         end
+
         exclusive_arcs[arc] = Definition.new(
           reflections: reflections.slice(*belong_tos.map(&:to_s)),
           options: args[2] || {}
@@ -39,9 +41,11 @@ module ExclusiveArc
             assign_exclusive_arc(:#{arc}, polymorphic)
             @#{arc} = polymorphic
           end
-        RUBY
 
-        validate :validate_exclusive_arcs
+          def validate_#{arc}
+            validate_exclusive_arc(:#{arc})
+          end
+        RUBY
       end
     end
 
@@ -54,12 +58,11 @@ module ExclusiveArc
       assign_attributes attributes
     end
 
-    def validate_exclusive_arcs
-      exclusive_arcs.each do |(arc, definition)|
-        foreign_key_count = definition.reflections.keys.count { |name| !!public_send(name) }
-        valid = definition.options[:optional] ? foreign_key_count.in?([0, 1]) : foreign_key_count == 1
-        errors.add(arc, :arc_not_exclusive) unless valid
-      end
+    def validate_exclusive_arc(arc)
+      definition = exclusive_arcs.fetch(arc)
+      foreign_key_count = definition.reflections.keys.count { |name| !!public_send(name) }
+      valid = definition.options[:optional] ? foreign_key_count.in?([0, 1]) : foreign_key_count == 1
+      errors.add(arc, :arc_not_exclusive) unless valid
     end
   end
 end

data/lib/exclusive_arc/version.rb

@@ -1,3 +1,3 @@
 module ExclusiveArc
-  VERSION = "0.2.3"
+  VERSION = "0.3.1"
 end

data/lib/generators/exclusive_arc_generator.rb

@@ -31,15 +31,20 @@ class ExclusiveArcGenerator < ActiveRecord::Generators::Base
       model_file_path,
       class_name.demodulize,
       <<~RB
-        #{indents}has_exclusive_arc #{model_exclusive_arcs}
+        #{indents}include ExclusiveArc::Model
       RB
     )
-    inject_into_class(
+    gsub_file(
       model_file_path,
-      class_name.demodulize,
-      <<~RB
-        #{indents}include ExclusiveArc::Model
+      /has_exclusive_arc :#{arc}(.*)$/,
+      ""
+    )
+    inject_into_file(
+      model_file_path,
+      <<~RB,
+        #{indents}has_exclusive_arc #{model_exclusive_arcs}
       RB
+      after: "include ExclusiveArc::Model\n"
     )
   end
 
@@ -52,8 +57,8 @@ class ExclusiveArcGenerator < ActiveRecord::Generators::Base
 
     def add_references
       belong_tos.map do |reference|
-        add_reference(reference)
-      end.join("\n")
+        add_reference(reference) unless column_exists?(reference)
+      end.compact.join("\n")
     end
 
     def add_reference(reference)
@@ -78,11 +83,22 @@ class ExclusiveArcGenerator < ActiveRecord::Generators::Base
     end
 
     def migration_file_name
-      "#{class_name.delete(":").underscore}_#{arc}_exclusive_arc.rb"
+      "#{class_name.delete(":").underscore}_#{arc}_exclusive_arc_#{belong_tos.map(&:underscore).join("_")}.rb"
     end
 
     def migration_class_name
-      [class_name.delete(":").singularize, arc.classify, "ExclusiveArc"].join
+      (
+        [class_name.delete(":").singularize, arc.classify, "ExclusiveArc"] |
+        belong_tos.map(&:classify)
+      ).join
+    end
+
+    def existing_check_constraint
+      @existing_check_constraint ||=
+        class_name.constantize.connection.check_constraints(class_name.constantize.table_name)
+          .find { |constraint| constraint.name == arc }
+    rescue
+      nil
     end
 
     def reference_type(reference)
@@ -98,15 +114,37 @@ class ExclusiveArcGenerator < ActiveRecord::Generators::Base
       reference.tableize
     end
 
+    def column_exists?(reference)
+      foreign_key = foreign_key_name(reference)
+      class_name.constantize.column_names.include?(foreign_key)
+    rescue
+      false
+    end
+
     def foreign_key_name(reference)
       class_name.constantize.reflections[reference].foreign_key
     rescue
       "#{reference}_id"
     end
 
+    def remove_check_constraint
+      return unless class_name.constantize.connection.supports_check_constraints?
+      return if options[:skip_check_constraint]
+      return unless existing_check_constraint&.expression
+
+      <<-RUBY.chomp
+    remove_check_constraint(
+      :#{table_name},
+      "#{existing_check_constraint.expression.squish}",
+      name: "#{arc}"
+    )
+      RUBY
+    end
+
     def add_check_constraint
+      return unless class_name.constantize.connection.supports_check_constraints?
       return if options[:skip_check_constraint]
-      <<-RUBY
+      <<-RUBY.chomp
     add_check_constraint(
       :#{table_name},
       "#{check_constraint}",

data/lib/generators/templates/migration.rb.erb

@@ -1,6 +1,7 @@
 class <%= migration_class_name %> < ActiveRecord::Migration[<%= ActiveRecord::Migration.current_version %>]
   def change
 <%= add_references %>
+<%= remove_check_constraint %>
 <%= add_check_constraint %>
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-exclusive-arc
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.3.1
 platform: ruby
 authors:
 - justin talbott
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-19 00:00:00.000000000 Z
+date: 2024-01-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -89,7 +89,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.12
+rubygems_version: 3.5.4
 signing_key:
 specification_version: 4
 summary: An ActiveRecord extension for polymorphic exclusive arc relationships