class Dry::Types::Enum

def call_safe(input, &) = type.call_safe(map_value(input), &)

def call_unsafe(input) = type.call_unsafe(map_value(input))

def default(*)
  raise ".enum(*values).default(value) is not supported. Call " \
        ".default(value).enum(*values) instead"
end

def each_value(&)
  values.each(&)
end

def initialize(type, **options)
  super
  @mapping = options.fetch(:mapping).freeze
  @values = @mapping.keys.freeze
  @inverted_mapping = @mapping.invert.freeze
  freeze
end

def joined_values
  mapping.keys.map { |value|
    if value.is_a?(::String)
      value
    else
      value.inspect
    end
  }.join("|")
end

def map_value(input)
  if input.equal?(Undefined)
    type.call
  elsif mapping.key?(input)
    input
  else
    inverted_mapping.fetch(input, input)
  end
end

def name = "#{super}(#{joined_values})"

def to_ast(meta: true)
  [:enum, [type.to_ast(meta: meta), mapping]]
end

def to_s = PRINTER.(self)

def try(input, &) = super(map_value(input))