lib/str_enum/model.rb in str_enum-0.1.5 vs lib/str_enum/model.rb in str_enum-0.1.6

@@ -3,20 +3,33 @@
 module StrEnum
   module Model
     extend ActiveSupport::Concern
 
     class_methods do
-      def str_enum(column, values, validate: true, scopes: true, accessor_methods: true, prefix: false, suffix: false, default: true)
+      def str_enum(column, values, validate: true, scopes: true, accessor_methods: true, update_methods: true, prefix: false, suffix: false, default: true, allow_nil: false)
         values = values.map(&:to_s)
-        validates column, presence: true, inclusion: {in: values} if validate
+        if validate
+          validate_options = {inclusion: {in: values}}
+          if allow_nil
+            validate_options[:allow_nil] = true
+          else
+            validate_options[:presence] = true
+          end
+          validates column, validate_options
+        end
         values.each do |value|
           prefix = column if prefix == true
           suffix = column if suffix == true
           method_name = [prefix, value, suffix].select { |v| v }.join("_")
           scope method_name, -> { where(column => value) } if scopes && !respond_to?(method_name)
           if accessor_methods && !method_defined?("#{method_name}?")
             define_method "#{method_name}?" do
               read_attribute(column) == value
+            end
+          end
+          if update_methods && !method_defined?("#{method_name}!")
+            define_method "#{method_name}!" do
+              update!(column => value)
             end
           end
         end
         default_value = default == true ? values.first : default
         after_initialize do