defmodule AshGraphql.Test.PageOfChannelMessagesCalculation do
  @moduledoc false
  use Ash.Calculation

  def load(_, _, context) do
    limit = context[:limit] || 100
    offset = context[:offset] || 0

    [
      :channel_message_count,
      messages: AshGraphql.Test.Message |> Ash.Query.limit(limit) |> Ash.Query.offset(offset)
    ]
  end

  def calculate([post], _, context) do
    limit = context[:limit] || 100
    offset = context[:offset] || 0

    {:ok,
     [
       %{
         count: post.channel_message_count,
         has_next_page: post.channel_message_count > offset + limit,
         results:
           post.messages
           |> Enum.map(
             &%Ash.Union{type: AshGraphql.Test.MessageUnion.struct_to_name(&1), value: &1}
           )
       }
     ]}
  end
end