defmodule Ash.Resource.Transformers.ValidateRelationshipAttributes do
@moduledoc """
Validates that all relationships point to valid fields
"""
use Spark.Dsl.Transformer
alias Spark.Dsl.Transformer
@impl true
def after_compile?, do: true
@impl true
def transform(dsl) do
attribute_names =
dsl
|> Transformer.get_entities([:attributes])
|> Enum.map(& &1.name)
dsl
|> Transformer.get_entities([:relationships])
|> Enum.reject(fn relationship ->
Map.get(relationship, :manual) || Map.get(relationship, :no_attributes?)
end)
|> Enum.filter(& &1.validate_destination_attribute?)
|> Enum.each(&validate_relationship(&1, attribute_names))
{:ok, dsl}
end
defp validate_relationship(relationship, attribute_names) do
unless relationship.source_attribute in attribute_names do
raise Spark.Error.DslError,
path: [:relationships, relationship.name],
message:
"Relationship `#{relationship.name}` expects source field `#{relationship.source_attribute}` to be defined"
end
if Code.ensure_loaded?(relationship.destination) do
if relationship.type == :many_to_many do
if Code.ensure_loaded?(relationship.through) do
through_attributes =
relationship.through
|> Ash.Resource.Info.attributes()
|> Enum.map(& &1.name)
unless relationship.source_attribute_on_join_resource in through_attributes do
raise Spark.Error.DslError,
path: [:relationships, relationship.name],
message:
"Relationship `#{relationship.name}` expects source attribute on resource `#{relationship.source_attribute_on_join_resource}` to be defined on #{inspect(relationship.through)}"
end
unless relationship.destination_attribute_on_join_resource in through_attributes do
raise Spark.Error.DslError,
path: [:relationships, relationship.name],
message:
"Relationship `#{relationship.name}` expects destination attribute on join resource `#{relationship.destination_attribute_on_join_resource}` to be defined on #{inspect(relationship.through)}"
end
end
end
destination_attributes =
relationship.destination
|> Ash.Resource.Info.attributes()
|> Enum.map(& &1.name)
unless relationship.destination_attribute in destination_attributes do
raise Spark.Error.DslError,
path: [:relationships, relationship.name],
message:
"Relationship `#{relationship.name}` expects destination field `#{relationship.destination_attribute}` to be defined on #{inspect(relationship.destination)}"
end
end
end
end