lib/ree_lib/packages/ree_enum/package/ree_enum/base_enum_mapper.rb



# frozen_string_literal: true
require_relative "value"

class ReeEnum::BaseEnumMapper < ReeMapper::AbstractType
  attr_reader :enum

  def initialize(enum)
    @enum = enum
  end

  contract(ReeEnum::Value => Or[Integer, String])
  def db_dump(value)
    value.mapped_value
  end

  contract(Or[Integer, String] => ReeEnum::Value).throws(ReeMapper::CoercionError)
  def db_load(value)
    enum_val = @enum.get_values.by_mapped_value(value)

    if !enum_val
      raise ReeMapper::CoercionError.new("should be one of #{enum_inspection}, got `#{truncate(value.inspect)}`")
    end

    enum_val
  end

  private

  def enum_inspection
    @enum_inspection ||= truncate(@enum.get_values.each.map(&:to_s).inspect)
  end
end