|
| 1 | +defmodule Systemd.Calendar do |
| 2 | + @moduledoc "Helpers for building systemd calendar expressions." |
| 3 | + |
| 4 | + @weekdays %{ |
| 5 | + monday: "Mon", |
| 6 | + mon: "Mon", |
| 7 | + tuesday: "Tue", |
| 8 | + tue: "Tue", |
| 9 | + wednesday: "Wed", |
| 10 | + wed: "Wed", |
| 11 | + thursday: "Thu", |
| 12 | + thu: "Thu", |
| 13 | + friday: "Fri", |
| 14 | + fri: "Fri", |
| 15 | + saturday: "Sat", |
| 16 | + sat: "Sat", |
| 17 | + sunday: "Sun", |
| 18 | + sun: "Sun" |
| 19 | + } |
| 20 | + |
| 21 | + @doc "Builds a daily calendar expression at a time of day." |
| 22 | + @spec daily_at(String.t() | Time.t()) :: String.t() |
| 23 | + def daily_at(time), do: "*-*-* #{time!(time)}" |
| 24 | + |
| 25 | + @doc "Builds a weekly calendar expression for a weekday and time of day." |
| 26 | + @spec weekly_at(atom() | String.t(), String.t() | Time.t()) :: String.t() |
| 27 | + def weekly_at(day, time), do: "#{weekday!(day)} *-*-* #{time!(time)}" |
| 28 | + |
| 29 | + @doc "Builds a monthly calendar expression for a day of month and time of day." |
| 30 | + @spec monthly_at(pos_integer(), String.t() | Time.t()) :: String.t() |
| 31 | + def monthly_at(day, time) when is_integer(day) and day in 1..31 do |
| 32 | + "*-*-#{String.pad_leading(to_string(day), 2, "0")} #{time!(time)}" |
| 33 | + end |
| 34 | + |
| 35 | + def monthly_at(day, _time) do |
| 36 | + raise ArgumentError, "monthly timer day must be an integer from 1 to 31, got: #{inspect(day)}" |
| 37 | + end |
| 38 | + |
| 39 | + defp time!(%Time{} = time) do |
| 40 | + time |
| 41 | + |> Time.truncate(:second) |
| 42 | + |> Time.to_iso8601() |
| 43 | + end |
| 44 | + |
| 45 | + defp time!(value) when is_binary(value) do |
| 46 | + value |
| 47 | + |> normalize_time_string() |
| 48 | + |> Time.from_iso8601() |
| 49 | + |> case do |
| 50 | + {:ok, time} -> |
| 51 | + time!(time) |
| 52 | + |
| 53 | + {:error, _reason} -> |
| 54 | + raise ArgumentError, |
| 55 | + "expected timer time as HH:MM or HH:MM:SS, got: #{inspect(value)}" |
| 56 | + end |
| 57 | + end |
| 58 | + |
| 59 | + defp time!(value) do |
| 60 | + raise ArgumentError, "expected timer time as HH:MM, HH:MM:SS, or Time, got: #{inspect(value)}" |
| 61 | + end |
| 62 | + |
| 63 | + defp normalize_time_string(<<_h1, _h2, ?:, _m1, _m2>> = time), do: time <> ":00" |
| 64 | + defp normalize_time_string(time), do: time |
| 65 | + |
| 66 | + defp weekday!(day) when is_atom(day) do |
| 67 | + case Map.fetch(@weekdays, day) do |
| 68 | + {:ok, weekday} -> weekday |
| 69 | + :error -> raise ArgumentError, "unknown weekday #{inspect(day)}" |
| 70 | + end |
| 71 | + end |
| 72 | + |
| 73 | + defp weekday!(day) when is_binary(day) do |
| 74 | + weekday = String.downcase(day) |
| 75 | + |
| 76 | + @weekdays |
| 77 | + |> Enum.find_value(fn {key, value} -> if Atom.to_string(key) == weekday, do: value end) |
| 78 | + |> case do |
| 79 | + nil -> raise ArgumentError, "unknown weekday #{inspect(day)}" |
| 80 | + value -> value |
| 81 | + end |
| 82 | + end |
| 83 | +end |
0 commit comments