defmodule Defnamed.Check do
  require Defnamed.Exception, as: E

  @type t ::
          :ok
          | E.NotKeyword.t()
          | E.InvalidArgNames.t()
          | E.ArgNamesDuplication.t()
          | E.MissingRequiredArgs.t()

  @doc """
  Shortcut for "no error" state

  ## Examples

  ```
  iex> require #{__MODULE__}
  #{__MODULE__}
  iex> #{__MODULE__}.ok
  :ok
  ```
  """
  defmacro ok do
    :ok
  end

  E.error_exception_list()
  |> Enum.each(fn {error, exception} ->
    @doc """
    Shortcut for #{exception} exception

    ## Examples

    ```
    iex> require #{__MODULE__}
    #{__MODULE__}
    iex> #{__MODULE__}.#{error}("my error message")
    %#{exception}{message: "my error message"}
    ```
    """
    defmacro unquote(error)(message) do
      exception = unquote(exception)

      quote do
        %unquote(exception){message: unquote(message)}
      end
    end
  end)

  @doc """
  Validate keyword list type and keys uniqueness

  ## Examples

  ```
  iex> #{__MODULE__}.validate_kv!([a: 1, b: 2], "Example keyword")
  :ok

  iex> #{__MODULE__}.validate_kv!(123, "Example keyword")
  ** (Defnamed.Exception.NotKeyword) Example keyword should be keyword list without keys duplication, but argument is not a keyword: 123

  iex> #{__MODULE__}.validate_kv!([a: 1, b: 2, b: 2], "Example keyword")
  ** (Defnamed.Exception.ArgNamesDuplication) Example keyword should be keyword list without keys duplication, but keys [:b] are duplicated
  ```
  """
  @spec validate_kv!(term, String.t()) :: :ok | no_return
  def validate_kv!(kv, message)
      when is_binary(message) do
    kv
    |> validate_kv(message)
    |> maybe_raise
  end

  @doc """
  Validate keyword list according given mapset of valid keys and list of required keys

  ## Examples

  ```
  iex> #{__MODULE__}.validate_kv!([a: 1, b: 2], MapSet.new([:a, :b]), [:a], "Example keyword")
  :ok

  iex> #{__MODULE__}.validate_kv!(123, MapSet.new([:a, :b]), [:a], "Example keyword")
  ** (Defnamed.Exception.NotKeyword) Example keyword should be keyword list which can contain only [:a, :b] keys without duplication, and mandatory [:a] keys, but argument is not a keyword: 123

  iex> #{__MODULE__}.validate_kv!([a: 1, b: 2, b: 2], MapSet.new([:a, :b]), [:a], "Example keyword")
  ** (Defnamed.Exception.ArgNamesDuplication) Example keyword should be keyword list which can contain only [:a, :b] keys without duplication, and mandatory [:a] keys, but keys [:b] are duplicated

  iex> #{__MODULE__}.validate_kv!([a: 1, b: 2], MapSet.new([:a]), [:a], "Example keyword")
  ** (Defnamed.Exception.InvalidArgNames) Example keyword should be keyword list which can contain only [:a] keys without duplication, and mandatory [:a] keys, but got invalid :b key

  iex> #{__MODULE__}.validate_kv!([b: 2], MapSet.new([:a, :b]), [:a], "Example keyword")
  ** (Defnamed.Exception.MissingRequiredArgs) Example keyword should be keyword list which can contain only [:a, :b] keys without duplication, and mandatory [:a] keys, but required :a key is not presented
  ```
  """
  @spec validate_kv!(term, MapSet.t(atom), list(atom), String.t()) :: :ok | no_return
  def validate_kv!(kv, valid_keys, required_keys, message)
      when is_list(required_keys) and is_binary(message) do
    kv
    |> validate_kv(valid_keys, required_keys, message)
    |> maybe_raise
  end

  @spec maybe_raise(t) :: :ok | no_return
  defp maybe_raise(ok() = result) do
    result
  end

  E.error_exception_list()
  |> Enum.each(fn {_, exception} ->
    defp maybe_raise(%unquote(exception){} = e) do
      raise(e)
    end
  end)

  @spec validate_kv(term, String.t()) :: t
  defp validate_kv(kv, raw_message) when is_binary(raw_message) do
    message = "#{raw_message} should be keyword list without keys duplication"

    with ok() <- validate_kv_type(kv, message),
         ok() <- validate_kv_uniqueness(kv, message) do
      ok()
    else
      error -> error
    end
  end

  @spec validate_kv(term, MapSet.t(atom), list(atom), String.t()) :: t
  defp validate_kv(kv, valid_keys, required_keys, raw_message)
       when is_list(required_keys) and is_binary(raw_message) do
    validity_message =
      "#{raw_message} should be keyword list which can contain only #{valid_keys |> MapSet.to_list() |> inspect} keys without duplication"

    mandatory_message =
      required_keys
      |> case do
        [] -> ""
        [_ | _] -> "and mandatory #{inspect(required_keys)} keys"
      end

    message =
      [
        validity_message,
        mandatory_message
      ]
      |> Enum.filter(&(&1 != ""))
      |> Enum.join(", ")

    with ok() <- validate_kv_type(kv, message),
         ok() <- validate_kv_uniqueness(kv, message),
         ok() <- validate_kv_required_keys(kv, required_keys, message),
         ok() <- validate_kv_keys(kv, valid_keys, message) do
      ok()
    else
      error -> error
    end
  end

  @spec validate_kv_type(term, String.t()) :: :ok | E.NotKeyword.t()
  defp validate_kv_type(kv, message) when is_binary(message) do
    kv
    |> Keyword.keyword?()
    |> case do
      true -> ok()
      false -> not_keyword("#{message}, but argument is not a keyword: #{inspect(kv)}")
    end
  end

  @spec validate_kv_uniqueness(Keyword.t(), String.t()) :: :ok | E.ArgNamesDuplication.t()
  defp validate_kv_uniqueness(kv, message) when is_list(kv) and is_binary(message) do
    kv_uniq =
      kv
      |> Enum.uniq_by(fn {k, _} -> k end)

    kv_uniq
    |> length
    |> case do
      l when l == length(kv) ->
        ok()

      _ ->
        arg_names_duplication("#{message}, but keys #{Enum.uniq(Keyword.keys(kv) -- Keyword.keys(kv_uniq)) |> inspect} are duplicated")
    end
  end

  @spec validate_kv_required_keys(Keyword.t(), list(atom), String.t()) ::
          :ok | E.MissingRequiredArgs.t()
  defp validate_kv_required_keys(kv, required_keys, message)
       when is_list(kv) and is_list(required_keys) and is_binary(message) do
    actual_keys = kv |> Keyword.keys() |> MapSet.new()

    required_keys
    |> Enum.reduce_while(ok(), fn key, ok() ->
      actual_keys
      |> MapSet.member?(key)
      |> case do
        true ->
          {:cont, ok()}

        false ->
          {:halt, missing_required_args("#{message}, but required #{inspect(key)} key is not presented")}
      end
    end)
  end

  @spec validate_kv_keys(Keyword.t(), MapSet.t(atom), String.t()) :: :ok | E.InvalidArgNames.t()
  defp validate_kv_keys(kv, valid_keys, message)
       when is_list(kv) and is_binary(message) do
    kv
    |> Enum.reduce_while(ok(), fn {key, _}, ok() ->
      valid_keys
      |> MapSet.member?(key)
      |> case do
        true -> {:cont, ok()}
        false -> {:halt, invalid_arg_names("#{message}, but got invalid #{inspect(key)} key")}
      end
    end)
  end
end