defmodule QueryBuilder.Utils do
  @moduledoc false

  def root_schema(query) do
    %{from: %{source: {_, context}}} = Ecto.Queryable.to_query(query)
    context
  end

  def query_to_string(query), do: Inspect.Ecto.Query.to_string(query)

  def inspect_query(query, label \\ nil)

  def inspect_query(query, nil) do
    IO.puts(query_to_string(query))
    query
  end

  def inspect_query(query, label: label) do
    IO.puts("#{label}:\n#{query_to_string(query)}")
    query
  end

  def find_field_and_binding_from_token(query, token, field) do
    split_field = String.split(to_string(field), "@")
    [field, assoc_field] = [Enum.at(split_field, 0), Enum.at(split_field, 1)]

    field = String.to_existing_atom(field)
    assoc_field = String.to_existing_atom(assoc_field || "nil")

    _find_field_and_binding_from_token(query, token, [field, assoc_field])
  end

  defp _find_field_and_binding_from_token(query, _token, [field, nil]) do
    {field, QueryBuilder.Utils.root_schema(query)}
  end

  defp _find_field_and_binding_from_token(_query, token, [field, assoc_field]) do
    {:ok, binding} = find_binding_from_token(token, assoc_field)
    {field, binding}
  end

  defp find_binding_from_token([], _field), do: {:error, :not_found}

  defp find_binding_from_token([assoc_data | tail], field) do
    if field == Map.fetch!(assoc_data, :assoc_field) do
      {:ok, assoc_data.assoc_binding}
    else
      find_binding_from_token(assoc_data.nested_assocs, field) ||
        find_binding_from_token(tail, field)
    end
  end
end