defmodule Kerosene do
  defstruct items: [], per_page: 10, page: 1, total_pages: 0, total_count: 0, params: []
  import Ecto.Query

  @moduledoc """
  Pagination for Ecto and Phoenix.
  """

  defmacro __using__(opts \\ []) do
    quote do
      def paginate(query, params \\ []) do
        Kerosene.paginate(__MODULE__, query, params, unquote(opts))
      end
    end
  end

  def paginate(repo, query, params, opts) do
    paginate(repo, query, merge_options(opts, params))
  end

  def paginate(repo, query, opts) do
    per_page = get_per_page(opts)
    page = get_page(opts)
    total_count = get_total_count(repo, query)

    kerosene = %Kerosene {
      per_page: per_page,
      page: page,
      total_pages: get_total_pages(total_count, per_page),
      total_count: total_count,
      params: opts[:params]
    }

    {get_items(repo, query, per_page, page), kerosene}
  end

  defp get_items(repo, query, per_page, page) do
    offset = per_page * (page - 1)
    query
    |> limit(^per_page)
    |> offset(^offset)
    |> repo.all
  end

  defp get_total_count(repo, query) do
    query
    |> exclude(:preload)
    |> exclude(:order_by)
    |> exclude(:select)
    |> select([i], count(i.id))
    |> repo.one
  end

  def get_total_pages(count, per_page) do
    Float.ceil(count / per_page) |> trunc
  end

  def get_per_page(params) do
    params 
    |> Keyword.get(:per_page, 10) 
    |> to_integer
  end

  def get_page(params) do
    params 
    |> Keyword.get(:page, 1) 
    |> to_integer
  end

  defp merge_options(opts, params) do
    keyword_params = to_keyword_list(params)
    Keyword.merge(opts, (keyword_params ++ [params: keyword_params]))
  end

  def to_keyword_list(params) when is_map(params) do
    for {key, val} <- params, into: [], do: {String.to_atom(key), val}
  end
  def to_keyword_list(params), do: params

  def to_integer(i) when is_integer(i), do: i
  def to_integer(i) when is_binary(i), do: String.to_integer(i)
  def to_integer(_), do: 1
end