module Dry::Types::Constructor::Wrapper

def __new__(type) = self.class.new(type, *@__args__.drop(1), **@options)

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

def call_unsafe(input) = fn.(input, type)

def lax
  # return self back because wrapping function
  # can handle failed type check
  self
end

def prepend(new_fn = nil, **options, &block)
  prep_fn = Function[new_fn || block]
  decorated =
    if prep_fn.wrapper?
      type.constructor(prep_fn, **options)
    else
      type.prepend(prep_fn, **options)
    end
  __new__(decorated)
end

def try(input, &)
  value = fn.(input, type)
rescue CoercionError => e
  failure = failure(input, e)
  block_given? ? yield(failure) : failure
else
  type.try(value, &)
end