defmodule Ash.Policy.Check.ContextEquals do
  @moduledoc false
  use Ash.Policy.SimpleCheck

  @impl true
  def describe(opts) do
    case opts[:key] do
      key when is_atom(key) ->
        "query_or_changeset.context[#{inspect(key)}] == #{inspect(opts[:value])}"

      key when is_list(key) ->
        key =
          Enum.map_join(key, fn key ->
            "[#{inspect(key)}]"
          end)

        "query_or_changeset#{key} == #{inspect(opts[:value])}"
    end
  end

  @impl true
  def match?(_, context, opts) do
    changeset_or_query =
      Map.get(context, :changeset) || Map.get(context, :query) || Map.get(context, :action_input)

    if is_nil(changeset_or_query) do
      false
    else
      get_in(changeset_or_query.context || %{}, List.wrap(opts[:key])) == opts[:value]
    end
  end
end