mirror of
https://github.com/ash-project/ash_graphql.git
synced 2024-09-20 21:32:59 +12:00
617 lines
17 KiB
Elixir
617 lines
17 KiB
Elixir
defmodule AshGraphql.Graphql.Resolver do
|
|
@moduledoc false
|
|
|
|
require Ash.Query
|
|
|
|
def resolve(
|
|
%{arguments: arguments, context: context} = resolution,
|
|
{api, resource, %{type: :get, action: action, identity: identity}}
|
|
) do
|
|
opts = [
|
|
actor: Map.get(context, :actor),
|
|
authorize?: AshGraphql.Api.authorize?(api),
|
|
action: action,
|
|
verbose?: AshGraphql.Api.debug?(api)
|
|
]
|
|
|
|
filter =
|
|
if identity do
|
|
{:ok,
|
|
resource
|
|
|> Ash.Resource.Info.identities()
|
|
|> Enum.find(&(&1.name == identity))
|
|
|> Map.get(:keys)
|
|
|> Enum.map(fn key ->
|
|
{key, Map.get(arguments, key)}
|
|
end)}
|
|
else
|
|
case AshGraphql.Resource.decode_primary_key(resource, Map.get(arguments, :id) || "") do
|
|
{:ok, value} -> {:ok, [id: value]}
|
|
{:error, error} -> {:error, error}
|
|
end
|
|
end
|
|
|
|
result =
|
|
case filter do
|
|
{:ok, filter} ->
|
|
resource
|
|
|> Ash.Query.new()
|
|
|> Ash.Query.set_tenant(Map.get(context, :tenant))
|
|
|> Ash.Query.filter(^filter)
|
|
|> set_query_arguments(action, arguments)
|
|
|> select_fields(resource, resolution)
|
|
|> load_fields(resource, api, resolution)
|
|
|> case do
|
|
{:ok, query} ->
|
|
api.read_one(query, opts)
|
|
|
|
{:error, error} ->
|
|
{:error, error}
|
|
end
|
|
|
|
{:error, error} ->
|
|
{:error, error}
|
|
end
|
|
|
|
Absinthe.Resolution.put_result(resolution, to_resolution(result))
|
|
end
|
|
|
|
def resolve(
|
|
%{arguments: args, context: context} = resolution,
|
|
{api, resource, %{type: :read_one, action: action}}
|
|
) do
|
|
opts = [
|
|
actor: Map.get(context, :actor),
|
|
authorize?: AshGraphql.Api.authorize?(api),
|
|
action: action,
|
|
verbose?: AshGraphql.Api.debug?(api)
|
|
]
|
|
|
|
query =
|
|
case Map.fetch(args, :filter) do
|
|
{:ok, filter} ->
|
|
Ash.Query.filter(resource, ^filter)
|
|
|
|
_ ->
|
|
Ash.Query.new(resource)
|
|
end
|
|
|
|
result =
|
|
query
|
|
|> Ash.Query.set_tenant(Map.get(context, :tenant))
|
|
|> set_query_arguments(action, args)
|
|
|> select_fields(resource, resolution)
|
|
|> load_fields(resource, api, resolution)
|
|
|> case do
|
|
{:ok, query} ->
|
|
api.read_one(query, opts)
|
|
|
|
{:error, error} ->
|
|
{:error, error}
|
|
end
|
|
|
|
Absinthe.Resolution.put_result(resolution, to_resolution(result))
|
|
end
|
|
|
|
def resolve(
|
|
%{arguments: args, context: context, definition: %{selections: selections}} = resolution,
|
|
{api, resource, %{type: :list, action: action}}
|
|
) do
|
|
opts = [
|
|
actor: Map.get(context, :actor),
|
|
authorize?: AshGraphql.Api.authorize?(api),
|
|
action: action,
|
|
verbose?: AshGraphql.Api.debug?(api)
|
|
]
|
|
|
|
page_opts =
|
|
args
|
|
|> Map.take([:limit, :offset, :after, :before])
|
|
|> Enum.reject(fn {_, val} -> is_nil(val) end)
|
|
|
|
opts =
|
|
case page_opts do
|
|
[] ->
|
|
opts
|
|
|
|
page_opts ->
|
|
if Enum.any?(selections, &(&1.name == :count)) do
|
|
page_opts = Keyword.put(page_opts, :count, true)
|
|
Keyword.put(opts, :page, page_opts)
|
|
else
|
|
Keyword.put(opts, :page, page_opts)
|
|
end
|
|
end
|
|
|
|
query =
|
|
case Map.fetch(args, :filter) do
|
|
{:ok, filter} ->
|
|
Ash.Query.filter(resource, ^filter)
|
|
|
|
_ ->
|
|
Ash.Query.new(resource)
|
|
end
|
|
|
|
query =
|
|
case Map.fetch(args, :sort) do
|
|
{:ok, sort} ->
|
|
keyword_sort =
|
|
Enum.map(sort, fn %{order: order, field: field} ->
|
|
{field, order}
|
|
end)
|
|
|
|
Ash.Query.sort(query, keyword_sort)
|
|
|
|
_ ->
|
|
query
|
|
end
|
|
|
|
result =
|
|
query
|
|
|> Ash.Query.set_tenant(Map.get(context, :tenant))
|
|
|> set_query_arguments(action, args)
|
|
|> select_fields(resource, resolution)
|
|
|> load_fields(resource, api, resolution)
|
|
|> case do
|
|
{:ok, query} ->
|
|
query
|
|
|> api.read(opts)
|
|
|> case do
|
|
{:ok, %{results: results, count: count}} ->
|
|
{:ok, %{results: results, count: count}}
|
|
|
|
{:ok, results} ->
|
|
if Ash.Resource.Info.action(resource, action, :read).pagination do
|
|
{:ok, %{results: results, count: Enum.count(results)}}
|
|
else
|
|
{:ok, results}
|
|
end
|
|
|
|
error ->
|
|
error
|
|
end
|
|
|
|
{:error, error} ->
|
|
{:error, error}
|
|
end
|
|
|
|
Absinthe.Resolution.put_result(resolution, to_resolution(result))
|
|
end
|
|
|
|
def mutate(
|
|
%{arguments: %{input: input}, context: context} = resolution,
|
|
{api, resource, %{type: :create, action: action}}
|
|
) do
|
|
{attributes, relationships, arguments} = split_attrs_rels_and_args(input, resource)
|
|
|
|
changeset = Ash.Changeset.new(resource, attributes)
|
|
|
|
changeset_with_relationships = changeset_with_relationships(relationships, changeset)
|
|
|
|
opts = [
|
|
actor: Map.get(context, :actor),
|
|
authorize?: AshGraphql.Api.authorize?(api),
|
|
action: action,
|
|
verbose?: AshGraphql.Api.debug?(api)
|
|
]
|
|
|
|
result =
|
|
changeset_with_relationships
|
|
|> Ash.Changeset.set_tenant(Map.get(context, :tenant))
|
|
|> Ash.Changeset.set_arguments(arguments)
|
|
|> select_fields(resource, resolution, true)
|
|
|> api.create(opts)
|
|
|> case do
|
|
{:ok, value} ->
|
|
case load_fields(value, resource, api, resolution, true) do
|
|
{:ok, result} ->
|
|
{:ok, %{result: result, errors: []}}
|
|
|
|
{:error, error} ->
|
|
{:ok, %{result: nil, errors: to_errors(List.wrap(error))}}
|
|
end
|
|
|
|
{:error, %{changeset: changeset}} ->
|
|
{:ok, %{result: nil, errors: to_errors(changeset.errors)}}
|
|
end
|
|
|
|
Absinthe.Resolution.put_result(resolution, to_resolution(result))
|
|
end
|
|
|
|
def mutate(
|
|
%{arguments: %{input: input} = arguments, context: context} = resolution,
|
|
{api, resource, %{type: :update, action: action, identity: identity}}
|
|
) do
|
|
filter =
|
|
if identity do
|
|
{:ok,
|
|
resource
|
|
|> Ash.Resource.Info.identities()
|
|
|> Enum.find(&(&1.name == identity))
|
|
|> Map.get(:keys)
|
|
|> Enum.map(fn key ->
|
|
{key, Map.get(arguments, key)}
|
|
end)}
|
|
else
|
|
case AshGraphql.Resource.decode_primary_key(resource, Map.get(arguments, :id) || "") do
|
|
{:ok, value} -> {:ok, [id: value]}
|
|
{:error, error} -> {:error, error}
|
|
end
|
|
end
|
|
|
|
case filter do
|
|
{:ok, filter} ->
|
|
resource
|
|
|> Ash.Query.filter(^filter)
|
|
|> Ash.Query.set_tenant(Map.get(context, :tenant))
|
|
|> set_query_arguments(action, arguments)
|
|
|> api.read_one!(verbose?: AshGraphql.Api.debug?(api))
|
|
|> case do
|
|
nil ->
|
|
not_found(filter, resource)
|
|
|
|
initial ->
|
|
{attributes, relationships, arguments} = split_attrs_rels_and_args(input, resource)
|
|
changeset = Ash.Changeset.new(initial, attributes)
|
|
|
|
changeset_with_relationships = changeset_with_relationships(relationships, changeset)
|
|
|
|
opts = [
|
|
actor: Map.get(context, :actor),
|
|
authorize?: AshGraphql.Api.authorize?(api),
|
|
action: action,
|
|
verbose?: AshGraphql.Api.debug?(api)
|
|
]
|
|
|
|
result =
|
|
changeset_with_relationships
|
|
|> Ash.Changeset.set_tenant(Map.get(context, :tenant))
|
|
|> Ash.Changeset.set_arguments(arguments)
|
|
|> select_fields(resource, resolution, true)
|
|
|> api.update(opts)
|
|
|> update_result(resource, api, resolution)
|
|
|
|
Absinthe.Resolution.put_result(resolution, to_resolution(result))
|
|
end
|
|
|
|
{:error, error} ->
|
|
Absinthe.Resolution.put_result(resolution, to_resolution({:error, error}))
|
|
end
|
|
end
|
|
|
|
def mutate(
|
|
%{arguments: arguments, context: context} = resolution,
|
|
{api, resource, %{type: :destroy, action: action, identity: identity}}
|
|
) do
|
|
filter =
|
|
if identity do
|
|
{:ok,
|
|
resource
|
|
|> Ash.Resource.Info.identities()
|
|
|> Enum.find(&(&1.name == identity))
|
|
|> Map.get(:keys)
|
|
|> Enum.map(fn key ->
|
|
{key, Map.get(arguments, key)}
|
|
end)}
|
|
else
|
|
case AshGraphql.Resource.decode_primary_key(resource, Map.get(arguments, :id) || "") do
|
|
{:ok, value} -> {:ok, [id: value]}
|
|
{:error, error} -> {:error, error}
|
|
end
|
|
end
|
|
|
|
case filter do
|
|
{:ok, filter} ->
|
|
resource
|
|
|> Ash.Query.filter(^filter)
|
|
|> Ash.Query.set_tenant(Map.get(context, :tenant))
|
|
|> set_query_arguments(action, arguments)
|
|
|> api.read_one!(verbose?: AshGraphql.Api.debug?(api))
|
|
|> case do
|
|
nil ->
|
|
not_found(filter, resource)
|
|
|
|
initial ->
|
|
opts = destroy_opts(api, context, action)
|
|
|
|
result =
|
|
initial
|
|
|> Ash.Changeset.new()
|
|
|> Ash.Changeset.set_tenant(Map.get(context, :tenant))
|
|
|> select_fields(resource, resolution, true)
|
|
|> api.destroy(opts)
|
|
|> destroy_result(initial, resource, resolution)
|
|
|
|
Absinthe.Resolution.put_result(resolution, to_resolution(result))
|
|
end
|
|
|
|
{:error, error} ->
|
|
Absinthe.Resolution.put_result(resolution, to_resolution({:error, error}))
|
|
end
|
|
end
|
|
|
|
defp not_found(filter, resource) do
|
|
{:ok,
|
|
%{
|
|
result: nil,
|
|
errors: [
|
|
to_errors(
|
|
Ash.Error.Query.NotFound.exception(
|
|
primary_key: Map.new(filter),
|
|
resource: resource
|
|
)
|
|
)
|
|
]
|
|
}}
|
|
end
|
|
|
|
defp clear_fields(result, resource, resolution) do
|
|
resolution
|
|
|> fields(true)
|
|
|> Enum.map(fn field ->
|
|
Ash.Resource.Info.aggregate(resource, field.schema_node.identifier) ||
|
|
Ash.Resource.Info.calculation(resource, field.schema_node.identifier) ||
|
|
Ash.Resource.Info.attribute(resource, field.schema_node.identifier)
|
|
end)
|
|
|> Enum.filter(& &1)
|
|
|> Enum.map(& &1.name)
|
|
|> Enum.reduce(result, fn field, result ->
|
|
Map.put(result, field, nil)
|
|
end)
|
|
end
|
|
|
|
defp load_fields(query_or_record, resource, api, resolution, result? \\ false) do
|
|
loading =
|
|
resolution
|
|
|> fields(result?)
|
|
|> Enum.map(fn field ->
|
|
Ash.Resource.Info.aggregate(resource, field.schema_node.identifier) ||
|
|
Ash.Resource.Info.calculation(resource, field.schema_node.identifier)
|
|
end)
|
|
|> Enum.filter(& &1)
|
|
|> Enum.map(& &1.name)
|
|
|
|
case query_or_record do
|
|
%Ash.Query{} = query ->
|
|
{:ok, Ash.Query.load(query, loading)}
|
|
|
|
record ->
|
|
api.load(record, loading)
|
|
end
|
|
end
|
|
|
|
defp select_fields(query_or_changeset, resource, resolution, result? \\ false) do
|
|
subfields =
|
|
resolution
|
|
|> fields(result?)
|
|
|> Enum.map(&Ash.Resource.Info.attribute(resource, &1.schema_node.identifier))
|
|
|> Enum.filter(& &1)
|
|
|> Enum.map(& &1.name)
|
|
|
|
case query_or_changeset do
|
|
%Ash.Query{} = query ->
|
|
Ash.Query.select(query, subfields)
|
|
|
|
%Ash.Changeset{} = changeset ->
|
|
Ash.Changeset.select(changeset, subfields)
|
|
end
|
|
end
|
|
|
|
defp fields(resolution, result?) do
|
|
if result? do
|
|
resolution
|
|
|> Absinthe.Resolution.project()
|
|
|> Enum.find(&(&1.name == "result"))
|
|
|> Map.get(:selections)
|
|
else
|
|
Absinthe.Resolution.project(resolution)
|
|
end
|
|
end
|
|
|
|
defp set_query_arguments(query, action, arg_values) do
|
|
action = Ash.Resource.Info.action(query.resource, action, :read)
|
|
|
|
action.arguments
|
|
|> Enum.reject(& &1.private?)
|
|
|> Enum.reduce(query, fn argument, query ->
|
|
Ash.Query.set_argument(query, argument.name, Map.get(arg_values, argument.name))
|
|
end)
|
|
end
|
|
|
|
defp destroy_opts(api, context, action) do
|
|
if AshGraphql.Api.authorize?(api) do
|
|
[actor: Map.get(context, :actor), action: action, verbose?: AshGraphql.Api.debug?(api)]
|
|
else
|
|
[action: action, verbose?: AshGraphql.Api.debug?(api)]
|
|
end
|
|
end
|
|
|
|
defp update_result(result, resource, api, resolution) do
|
|
case result do
|
|
{:ok, value} ->
|
|
case load_fields(value, resource, api, resolution, true) do
|
|
{:ok, result} ->
|
|
{:ok, %{result: result, errors: []}}
|
|
|
|
{:error, error} ->
|
|
{:ok, %{result: nil, errors: to_errors(List.wrap(error))}}
|
|
end
|
|
|
|
{:error, error} ->
|
|
{:ok, %{result: nil, errors: to_errors(List.wrap(error))}}
|
|
end
|
|
end
|
|
|
|
defp destroy_result(result, initial, resource, resolution) do
|
|
case result do
|
|
:ok ->
|
|
{:ok, %{result: clear_fields(initial, resource, resolution), errors: []}}
|
|
|
|
{:error, %{changeset: changeset}} ->
|
|
{:ok, %{result: nil, errors: to_errors(changeset.error)}}
|
|
end
|
|
end
|
|
|
|
defp changeset_with_relationships(relationships, changeset) do
|
|
if changeset.action_type == :create do
|
|
Enum.reduce(relationships, changeset, fn {relationship, replacement}, changeset ->
|
|
case decode_related_pkeys(changeset, relationship, replacement) do
|
|
{:ok, replacement} ->
|
|
Ash.Changeset.replace_relationship(changeset, relationship, replacement)
|
|
|
|
{:error, _error} ->
|
|
Ash.Changeset.add_error(changeset, "Invalid relationship primary keys")
|
|
end
|
|
end)
|
|
else
|
|
Enum.reduce(relationships, changeset, fn {relationship, changes}, changeset ->
|
|
[:add, :remove, :replace]
|
|
|> Enum.flat_map(fn op ->
|
|
case Map.fetch(changes, op) do
|
|
{:ok, value} ->
|
|
[{op, value}]
|
|
|
|
_ ->
|
|
[]
|
|
end
|
|
end)
|
|
|> Enum.reduce(changeset, fn
|
|
{:add, add}, changeset ->
|
|
Ash.Changeset.append_to_relationship(changeset, relationship, add)
|
|
|
|
{:remove, remove}, changeset ->
|
|
Ash.Changeset.remove_from_relationship(changeset, relationship, remove)
|
|
|
|
{:replace, replace}, changeset ->
|
|
Ash.Changeset.replace_relationship(changeset, relationship, replace)
|
|
end)
|
|
end)
|
|
end
|
|
end
|
|
|
|
defp decode_related_pkeys(changeset, relationship, primary_keys)
|
|
when is_list(primary_keys) do
|
|
primary_keys
|
|
|> Enum.reduce_while({:ok, []}, fn pkey, {:ok, list} ->
|
|
case AshGraphql.Resource.decode_primary_key(
|
|
Ash.Resource.Info.related(changeset.resource, relationship),
|
|
pkey
|
|
) do
|
|
{:ok, value} -> {:cont, {:ok, [value | list]}}
|
|
{:error, error} -> {:halt, {:error, error}}
|
|
end
|
|
end)
|
|
|> case do
|
|
{:ok, values} -> {:ok, Enum.reverse(values)}
|
|
{:error, error} -> {:error, error}
|
|
end
|
|
end
|
|
|
|
defp decode_related_pkeys(changeset, relationship, primary_key) do
|
|
AshGraphql.Resource.decode_primary_key(
|
|
Ash.Resource.Info.related(changeset.resource, relationship),
|
|
primary_key
|
|
)
|
|
end
|
|
|
|
defp split_attrs_rels_and_args(input, resource) do
|
|
Enum.reduce(input, {%{}, %{}, %{}}, fn {key, value}, {attrs, rels, args} ->
|
|
cond do
|
|
Ash.Resource.Info.public_attribute(resource, key) ->
|
|
{Map.put(attrs, key, value), rels, args}
|
|
|
|
Ash.Resource.Info.public_relationship(resource, key) ->
|
|
{attrs, Map.put(rels, key, value), args}
|
|
|
|
true ->
|
|
{attrs, rels, Map.put(args, key, value)}
|
|
end
|
|
end)
|
|
end
|
|
|
|
defp to_errors(errors) do
|
|
errors
|
|
|> List.wrap()
|
|
|> Enum.map(fn error ->
|
|
if AshGraphql.Error.impl_for(error) do
|
|
AshGraphql.Error.to_error(error)
|
|
else
|
|
%{
|
|
message: "something went wrong."
|
|
}
|
|
end
|
|
end)
|
|
end
|
|
|
|
def resolve_assoc(
|
|
%{source: parent, arguments: args, context: %{loader: loader} = context} = resolution,
|
|
{api, relationship}
|
|
) do
|
|
api_opts = [
|
|
actor: Map.get(context, :actor),
|
|
authorize?: AshGraphql.Api.authorize?(api),
|
|
verbose?: AshGraphql.Api.debug?(api)
|
|
]
|
|
|
|
opts = [
|
|
query: apply_load_arguments(args, Ash.Query.new(relationship.destination)),
|
|
api_opts: api_opts,
|
|
args: args,
|
|
tenant: Map.get(context, :tenant)
|
|
]
|
|
|
|
{batch_key, parent} = {{relationship.name, opts}, parent}
|
|
|
|
do_dataloader(resolution, loader, api, batch_key, args, parent)
|
|
end
|
|
|
|
defp do_dataloader(
|
|
resolution,
|
|
loader,
|
|
api,
|
|
batch_key,
|
|
_args,
|
|
parent
|
|
) do
|
|
loader = Dataloader.load(loader, api, batch_key, parent)
|
|
|
|
fun = fn loader ->
|
|
{:ok, Dataloader.get(loader, api, batch_key, parent)}
|
|
end
|
|
|
|
Absinthe.Resolution.put_result(
|
|
resolution,
|
|
{:middleware, Absinthe.Middleware.Dataloader, {loader, fun}}
|
|
)
|
|
end
|
|
|
|
defp apply_load_arguments(arguments, query) do
|
|
Enum.reduce(arguments, query, fn
|
|
{:limit, limit}, query ->
|
|
Ash.Query.limit(query, limit)
|
|
|
|
{:offset, offset}, query ->
|
|
Ash.Query.offset(query, offset)
|
|
|
|
{:filter, value}, query ->
|
|
decode_and_filter(query, value)
|
|
|
|
{:sort, value}, query ->
|
|
keyword_sort =
|
|
Enum.map(value, fn %{order: order, field: field} ->
|
|
{field, order}
|
|
end)
|
|
|
|
Ash.Query.sort(query, keyword_sort)
|
|
end)
|
|
end
|
|
|
|
defp decode_and_filter(query, value) do
|
|
Ash.Query.filter(query, ^value)
|
|
end
|
|
|
|
defp to_resolution({:ok, value}), do: {:ok, value}
|
|
|
|
defp to_resolution({:error, error}),
|
|
do: {:error, error |> List.wrap() |> Enum.map(&Exception.message(&1))}
|
|
end
|