enum_fields 0.0.5 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4d2ba45a5964fee807097e941cd817f7e1df72ace9094c8bf2620c7462bd06d2
-  data.tar.gz: 4b9c320af51de403c10631881531efe3455d2ac5946df3b5798d03e825e06ce3
+  metadata.gz: 6f02f9ddd27cf9f20da714f32817b9d5da178834ff87d41ac465f0250b0fd024
+  data.tar.gz: 5c8e63ac0d57bea298c7dce4298596cf60d3e2d66791e2d336ef4730ce40a188
 SHA512:
-  metadata.gz: 6d16f74b5eb526d1cd7ea287a75b5a261679d432be001736b13e0ce52f442bc5806176e8e5020ae605b475eea608f83af25cd487eed1d181d50af30ad1a24dbf
-  data.tar.gz: e5ea072511fa65e0ea6f5221f9c39406af917533aecca0a6e962be8bf743fc17740033b32064d59579fac25f5cebfd337e2eb002b96546e14bd830275d8bc34e
+  metadata.gz: 48a6e303906852608449e28ac134dcab3fa5c1e0bc39b39d54075c8311800ef427ce771e3f5cecf18391d079e383616f2ec281331d3e5a6eeb86cc392d1e516a
+  data.tar.gz: 618ad4bc993ec5adf45aa9d7df42b514750b34787b07fa776331673729f27e6068da1fa51acd0dbb94ebf589b5a62d9cec6f2cc89d39971481aee8a465eae206

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!
@@ -100,7 +101,7 @@ module EnumFields
         definitions = self.class.enum_field_for(accessor)
         return nil if definitions.blank?
 
-        definitions[column_value]
+        definitions[column_value] || definitions.values.find { |d| d[:value] == column_value }
       end
     end
 
@@ -116,7 +117,8 @@ module EnumFields
           definitions = self.class.enum_field_for(accessor)
           return nil if definitions.blank?
 
-          definitions.dig(column_value, property)
+          metadata = definitions[column_value] || definitions.values.find { |d| d[:value] == column_value }
+          metadata&.dig(property)
         end
       end
     end
@@ -143,13 +145,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.1'
 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.1
 platform: ruby
 authors:
 - Kinnell Shah