enum_fields 0.0.5 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4d2ba45a5964fee807097e941cd817f7e1df72ace9094c8bf2620c7462bd06d2
-  data.tar.gz: 4b9c320af51de403c10631881531efe3455d2ac5946df3b5798d03e825e06ce3
+  metadata.gz: 9a282e3cb07038a6db32bfed63217bc85ec6901541b26e373e1c5a78e4ee4106
+  data.tar.gz: 26cc2fccb3171185ac16e0edeab889b235c64b7d7b289b9958234c5da7064e73
 SHA512:
-  metadata.gz: 6d16f74b5eb526d1cd7ea287a75b5a261679d432be001736b13e0ce52f442bc5806176e8e5020ae605b475eea608f83af25cd487eed1d181d50af30ad1a24dbf
-  data.tar.gz: e5ea072511fa65e0ea6f5221f9c39406af917533aecca0a6e962be8bf743fc17740033b32064d59579fac25f5cebfd337e2eb002b96546e14bd830275d8bc34e
+  metadata.gz: b989a5b41c4e07609bf00dd08db9555ee7f544cda58f91ca4db114ba3a66fbeecfd892077ae22f96721bfd40478b609fe77bbea123853e2e5ba4aa10354a1bb4
+  data.tar.gz: 2e1c2fbdeee50ee23a328cd160b25ee1d86b52d0e4a42198841c167d1fff231045340171adfbacf0fe5d406ce4bcd83d5a4d151aff63dedeeaf54de9e1dc7832

data/lib/enum_fields/enum_field.rb

@@ -11,6 +11,7 @@ module EnumFields
       @accessor = accessor.to_sym
       @column_name = options.fetch(:column, @accessor).to_sym
       @definition = Definition.new(definition)
+      @options = options
     end
 
     def define!
@@ -143,13 +144,56 @@ module EnumFields
     end
 
     def define_validation!
-      return if @definition.blank?
+      return unless column_validatable?
 
+      if column_polymorphic_association_name.present?
+        define_polymorphic_validation!
+      else
+        define_standard_validation!
+      end
+    end
+
+    def define_standard_validation!
       valid_values = @definition.values.pluck(:value)
+
       @model_class.validates(@column_name, inclusion: {
         in: valid_values,
         allow_nil: true,
       })
     end
+
+    def define_polymorphic_validation!
+      association_name = column_polymorphic_association_name
+      column_name = @column_name
+      accessor = @accessor
+
+      @model_class.validate do
+        association_obj = respond_to?(association_name) ? public_send(association_name) : nil
+        valid_values = self.class.public_send("#{accessor}_values")
+
+        if association_obj
+          errors.add(association_name, "must be one of: #{valid_values.join(', ')}") unless valid_values.include?(association_obj.class.name)
+        else
+          column_value = attributes[column_name.to_s]
+          next if column_value.nil?
+
+          errors.add(column_name, "must be one of: #{valid_values.join(', ')}") unless valid_values.include?(column_value)
+        end
+      end
+    end
+
+    def column_validatable?
+      @definition.present? && @options.fetch(:validate, true)
+    end
+
+    def column_polymorphic_association_name
+      return nil unless @model_class.respond_to?(:reflect_on_all_associations)
+
+      reflection = @model_class.reflect_on_all_associations(:belongs_to).find do |r|
+        r.options[:polymorphic] && "#{r.name}_type" == @column_name.to_s
+      end
+
+      reflection&.name
+    end
   end
 end

data/lib/enum_fields/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module EnumFields
-  VERSION = '0.0.5'
+  VERSION = '0.1.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: enum_fields
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.1.0
 platform: ruby
 authors:
 - Kinnell Shah