Module: SqlEnum::ClassMethods

Defined in:

lib/sql_enum/class_methods.rb

Instance Method Summary

• #sql_enum(column_name, options = {}) ⇒ Object

Instance Method Details

#sql_enum(column_name, options = {}) ⇒ Object

# File 'lib/sql_enum/class_methods.rb', line 3

def sql_enum(column_name, options = {})
  # skip redefinitions
  return if defined_enums.key?(column_name.to_s)

  # Query values
  enum_column = EnumColumn.new(table_name, column_name)
  values_map = enum_column.values.to_h { |value| [value.to_sym, value.to_s] }

  # Check option defaults
  prefix = options.fetch(:_prefix, !!SqlEnum.configuration&.default_prefix)
  suffix = options.fetch(:_suffix, !!SqlEnum.configuration&.default_suffix)

  # Define enum using Rails enum
  enum(column_name => values_map, _prefix: prefix, _suffix: suffix)

  # Override reader to return symbols
  type_definition = ->(subtype) { EnumType.new(attr, send(column_name.to_s.pluralize), subtype) }
  attribute(column_name, &type_definition)

  prefix_str = format_affix(column_name, prefix, suffix: '_')
  suffix_str = format_affix(column_name, suffix, prefix: '_')

  # Fix query methods to compare symbols to symbols
  values_map.each_value do |value|
    method_name = "#{prefix_str}#{value}#{suffix_str}?"
    define_method(method_name) { self[column_name] == value.to_sym }
  end
end