module Dry::Initializer::Dispatchers::WrapType

def call(type: nil, wrap: 0, **options)
  {type: wrapped_type(type, wrap), **options}
end

def wrap_value(value, count, type)
  if count.zero?
    type ? type.call(value) : value
  else
    return [wrap_value(value, count - 1, type)] unless value.is_a?(Array)
    value.map { |item| wrap_value(item, count - 1, type) }
  end
end

def wrapped_type(type, count)
  return type if count.zero?
  ->(value) { wrap_value(value, count, type) }
end