ash_postgres/lib/transformers/prevent_multidimensional_array_aggregates.ex
2023-11-14 13:59:30 -05:00

40 lines
1.3 KiB
Elixir

defmodule AshPostgres.Transformers.PreventMultidimensionalArrayAggregates do
@moduledoc false
use Spark.Dsl.Transformer
alias Spark.Dsl.Transformer
def after_compile?, do: true
def transform(dsl) do
resource = Transformer.get_persisted(dsl, :module)
dsl
|> Ash.Resource.Info.aggregates()
|> Stream.filter(&(&1.kind in [:list, :first]))
|> Stream.filter(& &1.field)
|> Enum.each(fn aggregate ->
related = Ash.Resource.Info.related(resource, aggregate.relationship_path)
type = Ash.Resource.Info.field(related, aggregate.field).type
case type do
{:array, _} ->
raise Spark.Error.DslError,
module: resource,
path: [:aggregates, aggregate.name],
message: """
Aggregate not supported.
Aggregate #{inspect(resource)}.#{aggregate.name} is not supported, because its type is `#{aggregate.kind}`, and the destination attribute is an array.
Postgres does not support multidimensional arrays with differing lengths internally. In the future we may be able to remove this restriction
for the `:first` type aggregate, but likely never for `:list`. In the meantime, you will have to use a custom calculation to get this data.
"""
_ ->
:ok
end
end)
{:ok, dsl}
end
end