# frozen_string_literal: true

require "dry/types/any"

module Dry
  module Types
    # Primitives with {Kernel} coercion methods
    KERNEL_COERCIBLE = {
      string: String,
      integer: Integer,
      float: Float,
      decimal: BigDecimal,
      array: ::Array,
      hash: ::Hash
    }.freeze

    # Primitives with coercions through by convention `to_*` methods
    METHOD_COERCIBLE = {
      symbol: Symbol
    }.freeze

    # By convention methods to coerce {METHOD_COERCIBLE} primitives
    METHOD_COERCIBLE_METHODS = {
      symbol: :to_sym
    }.freeze

    # Primitives that are non-coercible
    NON_COERCIBLE = {
      nil: NilClass,
      class: Class,
      true: TrueClass,
      false: FalseClass,
      date: Date,
      date_time: DateTime,
      time: Time,
      range: Range
    }.freeze

    # All built-in primitives
    ALL_PRIMITIVES = [KERNEL_COERCIBLE, METHOD_COERCIBLE, NON_COERCIBLE].reduce(&:merge).freeze

    # All coercible types
    COERCIBLE = KERNEL_COERCIBLE.merge(METHOD_COERCIBLE).freeze

    # All built-in primitives except {NilClass}
    NON_NIL = ALL_PRIMITIVES.reject { |name, _| name == :nil }.freeze

    # Register generic types for {ALL_PRIMITIVES}
    ALL_PRIMITIVES.each do |name, primitive|
      type = Nominal[primitive].new(primitive)
      register("nominal.#{name}", type)
    end

    # Register strict types for {ALL_PRIMITIVES}
    ALL_PRIMITIVES.each do |name, primitive|
      type = self["nominal.#{name}"].constrained(type: primitive)
      register(name.to_s, type)
      register("strict.#{name}", type)
    end

    # Register {KERNEL_COERCIBLE} types
    KERNEL_COERCIBLE.each do |name, primitive|
      register("coercible.#{name}", self["nominal.#{name}"].constructor(Kernel.method(primitive.name)))
    end

    # Register {METHOD_COERCIBLE} types
    METHOD_COERCIBLE.each_key do |name|
      register(
        "coercible.#{name}", self["nominal.#{name}"].constructor(&METHOD_COERCIBLE_METHODS[name])
      )
    end

    # Register optional strict {NON_NIL} types
    NON_NIL.each_key do |name|
      type = self[name.to_s].optional
      register("optional.strict.#{name}", type)
      register("optional.#{name}", type)
    end

    # Register optional {COERCIBLE} types
    COERCIBLE.each_key do |name|
      register("optional.coercible.#{name}", self["coercible.#{name}"].optional)
    end

    # Register `:bool` since it's common and not a built-in Ruby type :(
    register("nominal.bool", self["nominal.true"] | self["nominal.false"])
    bool = self["strict.true"] | self["strict.false"]
    register("strict.bool", bool)
    register("bool", bool)

    register("any", Any)
    register("nominal.any", Any)
    register("strict.any", Any)
  end
end

require "dry/types/coercions"
require "dry/types/params"
require "dry/types/json"