vobject

defmodule ICalendar.Util.RRULE do
  @moduledoc """
  This module handles the implementation detail of RRULE properties
  """

  @doc ~S"""
  This function is used by the Value module to serialize %RRULE{} structs
  into RRULE iCalendar strings.

      iex> rrule = %ICalendar.RRULE{frequency: :daily}
      ...> key   = :frequency
      ...> ICalendar.Util.RRULE.serialize(rrule, key)
      "FREQ=DAILY"

  It can handle multiple values in a list format:

      iex> rrule = %ICalendar.RRULE{
      ...>   by_day: [:monday, :tuesday]
      ...> }
      ...> key   = :by_day
      ...> ICalendar.Util.RRULE.serialize(rrule, key)
      "BYDAY=MO,TU"

  """
  def serialize(rrule = %ICalendar.RRULE{}, key) do
    keys = ICalendar.RRULE.string_to_atom_keys(:inverted)
    case Map.fetch(rrule, key) do
      {:ok, value} ->
        serialized_key   = serialize_key(key, keys)
        serialized_value = serialize_value(key, value)
        cond do
          is_bitstring(serialized_key) && is_bitstring(serialized_value) ->
            "#{serialized_key}=#{serialized_value}"
          true -> nil
        end
      :error -> nil
    end
  end

  defp serialize_value(:frequency, value) when is_atom(value) do
    frequencies = ICalendar.RRULE.frequencies(:inverted)
    case Map.fetch(frequencies, value) do
      {:ok, freq} -> freq
      :error      -> nil
    end
  end

  defp serialize_value(:count, value) when not is_nil(value),
  do: Integer.to_string(value)

  defp serialize_value(:until, value),
  do: ICalendar.Value.to_ics(value)
  
  defp serialize_value(:interval, value) when not is_nil(value),
  do: Integer.to_string(value)

  defp serialize_value(:by_second, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ -> Enum.join(value, ",")
    end
  end
  defp serialize_value(:by_minute, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ -> Enum.join(value, ",")
    end
  end
  defp serialize_value(:by_hour, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ -> Enum.join(value, ",")
    end
  end
  defp serialize_value(:by_day, value) when is_list(value) do
    days = ICalendar.RRULE.days(:inverted)
    case Enum.count(value) do
      0 -> nil
      _ ->
        value
        |> Enum.map(fn(val) ->
          case Map.fetch(days, val) do
            {:ok, day} -> day
            :error     -> nil
          end
        end)
        |> Enum.join(",")
    end
  end
  defp serialize_value(:by_month, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ ->
        value
        |> Enum.map(fn(val) ->
          # Retrieve the index of the months array then add one
          Enum.find_index(ICalendar.RRULE.months, &(&1 == val)) + 1
        end)
        |> Enum.join(",")
    end
  end
  defp serialize_value(:by_month_day, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ -> Enum.join(value, ",")
    end
  end
  defp serialize_value(:by_year_day, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ -> Enum.join(value, ",")
    end
  end
  defp serialize_value(:by_week_number, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ -> Enum.join(value, ",")
    end
  end
  defp serialize_value(:by_set_pos, value) when is_list(value) do
    case Enum.count(value) do
      0 -> nil
      _ -> Enum.join(value, ",")
    end
  end
  defp serialize_value(:week_start, value) when is_atom(value) do
    days = ICalendar.RRULE.days(:inverted)
    case Map.fetch(days, value) do
      {:ok, day} -> day
      :error     -> nil
    end
  end
  defp serialize_value(_, value), do: value


  defp serialize_key(key, keys) do
    case Map.fetch(keys, key) do
      {:ok, new_key} -> new_key
      :error -> nil
    end
  end
end