# frozen_string_literal: true

module Anyway
  # Contains a mapping between type IDs/names and deserializers
  class TypeRegistry
    class << self
      def default
        @default ||= TypeRegistry.new
      end
    end

    def initialize
      @registry = {}
    end

    def accept(name_or_object, &block)
      if !block && !name_or_object.respond_to?(:call)
        raise ArgumentError, "Please, provide a type casting block or an object implementing #call(val) method"
      end

      registry[name_or_object] = block || name_or_object
    end

    def deserialize(raw, type_id, array: false)
      return if raw.nil?

      caster =
        if type_id.is_a?(Symbol) || type_id.nil?
          registry.fetch(type_id) { raise ArgumentError, "Unknown type: #{type_id}" }
        else
          raise ArgumentError, "Type must implement #call(val): #{type_id}" unless type_id.respond_to?(:call)
          type_id
        end

      if array
        raw_arr = raw.is_a?(String) ? raw.split(/\s*,\s*/) : Array(raw)
        raw_arr.map { caster.call(_1) }
      else
        caster.call(raw)
      end
    end

    def dup
      new_obj = self.class.allocate
      new_obj.instance_variable_set(:@registry, registry.dup)
      new_obj
    end

    private

    attr_reader :registry
  end

  TypeRegistry.default.tap do |obj|
    obj.accept(nil, &:itself)
    obj.accept(:string, &:to_s)
    obj.accept(:integer, &:to_i)
    obj.accept(:float, &:to_f)

    obj.accept(:date) do
      require "date" unless defined?(::Date)

      next _1 if _1.is_a?(::Date)

      next _1.to_date if _1.respond_to?(:to_date)

      ::Date.parse(_1)
    end

    obj.accept(:datetime) do
      require "date" unless defined?(::Date)

      next _1 if _1.is_a?(::DateTime)

      next _1.to_datetime if _1.respond_to?(:to_datetime)

      ::DateTime.parse(_1)
    end

    obj.accept(:uri) do
      require "uri" unless defined?(::URI)

      next _1 if _1.is_a?(::URI)

      ::URI.parse(_1)
    end

    obj.accept(:boolean) do
      _1.to_s.match?(/\A(true|t|yes|y|1)\z/i)
    end
  end

  unless "".respond_to?(:safe_constantize)
    require "anyway/ext/string_constantize"
    using Anyway::Ext::StringConstantize
  end

  # TypeCaster is an object responsible for type-casting.
  # It uses a provided types registry and mapping, and also
  # accepts a fallback typecaster.
  class TypeCaster
    using Ext::DeepDup
    using Ext::Hash

    def initialize(mapping, registry: TypeRegistry.default, fallback: ::Anyway::AutoCast)
      @mapping = mapping.deep_dup
      @registry = registry
      @fallback = fallback
    end

    def coerce(key, val, config: mapping)
      caster_config = config[key.to_sym]

      return fallback.coerce(key, val) unless caster_config

      case caster_config
      in Hash[array:, type:, **nil]
        registry.deserialize(val, type, array: array)
      in Hash[config: subconfig]
        subconfig = subconfig.safe_constantize if subconfig.is_a?(::String)
        raise ArgumentError, "Config is not found: #{subconfig}" unless subconfig

        subconfig.new(val)
      in Hash
        return val unless val.is_a?(Hash)

        caster_config.each do |k, v|
          ks = k.to_s
          next unless val.key?(ks)

          val[ks] = coerce(k, val[ks], config: caster_config)
        end

        val
      else
        registry.deserialize(val, caster_config)
      end
    end

    private

    attr_reader :mapping, :registry, :fallback
  end
end