Contents

# frozen_string_literal: true

module StoreModel
  # Allows defining Rails-like enums
  module Enum
    # Defines new enum
    #
    # @param name [String] name of the enum to define
    # @param values [Object]
    # @param kwargs [Object]
    def enum(name, values = nil, **kwargs)
      values ||= kwargs[:in] || kwargs
      options = kwargs.slice(:_prefix, :_suffix, :default)

      ensure_hash(values).tap do |mapping|
        define_attribute(name, mapping, options[:default])
        define_reader(name, mapping)
        define_writer(name, mapping)
        define_method("#{name}_value") { attributes[name.to_s] }
        define_method("#{name}_values") { mapping }
        define_predicate_methods(name, mapping, options)
      end
    end

    private

    def define_attribute(name, mapping, default)
      attribute name, cast_type(mapping), default: default
    end

    def define_reader(name, mapping)
      define_method(name) { mapping.key(send("#{name}_value")).to_s }
    end

    def define_writer(name, mapping)
      type = cast_type(mapping)
      define_method("#{name}=") { |value| super type.cast_value(value) }
    end

    def define_predicate_methods(name, mapping, options)
      mapping.each do |label, value|
        label = affixed_label(label, name, options[:_prefix], options[:_suffix])
        define_method("#{label}?") { send(name) == mapping.key(value).to_s }
      end
    end

    def cast_type(mapping)
      StoreModel::Types::EnumType.new(mapping)
    end

    def ensure_hash(values)
      return values if values.is_a?(Hash)

      values.zip(0...values.size).to_h
    end

    def affixed_label(label, name, prefix = nil, suffix = nil)
      prefix = prefix == true ? "#{name}_" : "#{prefix}_" if prefix
      suffix = suffix == true ? "_#{name}" : "_#{suffix}" if suffix

      "#{prefix}#{label}#{suffix}"
    end
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
store_model-1.5.1	lib/store_model/enum.rb
store_model-1.5.0	lib/store_model/enum.rb
store_model-1.4.0	lib/store_model/enum.rb
store_model-1.3.0	lib/store_model/enum.rb
store_model-1.2.0	lib/store_model/enum.rb
store_model-1.1.0	lib/store_model/enum.rb
store_model-1.0.0	lib/store_model/enum.rb
store_model-0.13.0	lib/store_model/enum.rb
store_model-0.12.0	lib/store_model/enum.rb