ActiveRecord::Enum article on Medium

active_record_enum.rb

def assert_valid_value(value)
  unless value.blank? || mapping.has_key?(value) || mapping.has_value?(value)
    raise ArgumentError, "'#{value}' is not a valid #{name}"
  end
end

 

blog_post.rb

class BlogPost < ApplicationRecord
 enum state: %i[draft published]   # [:draft, :published]
 validates :state, inclusion: { in: states.keys }
end

 

blog_post_after_message_edit.rb

class BlogPost < ApplicationRecord
  include ValidatableEnum

  enum state: %i[draft published]
  validates :state, inclusion: { in: states.keys, 
                                  message: "state must be one of #{states.keys}" }
  validatable_enum :state
end

 

blog_post_after_modification.rb

class BlogPost < ApplicationRecord
  include ValidatableEnum

  enum state: %i[draft published]
  validates :state, inclusion: { in: states.keys }
  validatable_enum :state
end

 

validatable_enum.rb

module ValidatableEnum
  extend ActiveSupport::Concern

  class_methods do 
    def validatable_enum(*enums_to_fix)
      enums_to_fix.each do |element|
        decorate_attribute_type(element) do |subtype|
          ValidatableEnumType.new(element, defined_enums.fetch(element.to_s), subtype)
        end
      end
    end
  end

end

 

validatable_enum_type.rb

class ValidatableEnumType < ActiveRecord::Enum::EnumType
  # override assert_valid_value() to supress <ArgumentError>
  # return a value and depend on our own validation
  def assert_valid_value(value)
    value
  end
end