defmodule Timex.Ecto.Date do
  @moduledoc """
  Support for using Timex with :date fields
  """
  use Timex

  @behaviour Ecto.Type

  def type, do: :date

  @doc """
  Handle casting to Timex.Ecto.Date
  """
  def cast(%Date{} = date), do: {:ok, date}
  # Support embeds_one/embeds_many
  def cast(%{"calendar" => _,
             "year" => y, "month" => m, "day" => d}) do
    date = Timex.to_date({y,m,d})
    {:ok, date}
  end
  def cast(date) when is_binary(date) do
    case Ecto.Date.cast(date) do
      {:ok, d} -> load({d.year,d.month,d.day})
      :error -> :error
    end
  end
  def cast(datetime) do
    case Timex.to_date(datetime) do
      {:error, _} ->
        case Ecto.Date.cast(datetime) do
          {:ok, date} -> load({date.year, date.month, date.day})
          :error -> :error
        end
      %Date{} = d -> {:ok, d}
    end
  end

  @doc """
  Load from the native Ecto representation
  """
  def load({_year, _month, _day} = date), do: {:ok, Timex.to_date(date)}
  def load(_), do: :error

  @doc """
  Convert to native Ecto representation
  """
  def dump(datetime) do
    case Timex.to_erl(datetime) do
      {:error, _}   -> :error
      {{_,_,_}=d,_} -> {:ok, d}
      {_,_,_} = d   -> {:ok, d}
    end
  end

  def autogenerate(precision \\ :sec)
  def autogenerate(_) do
    {date, {_, _, _}} = :erlang.universaltime
    load(date) |> elem(1)
  end
end