defmodule Shift73kWeb.ShiftLive.Index do use Shift73kWeb, :live_view alias Shift73k.Shifts alias Shift73k.Shifts.Shift alias Shift73kWeb.Roles @impl true def mount(_params, session, socket) do socket |> assign_defaults(session) |> live_okreply() end @impl true def handle_params(params, _url, socket) do current_user = socket.assigns.current_user live_action = socket.assigns.live_action shift = shift_from_params(params) if Roles.can?(current_user, shift, live_action) do socket |> init_today(Date.utc_today()) |> update_agenda() |> assign(:delete_shift, nil) |> apply_action(socket.assigns.live_action, params) |> live_noreply() else socket |> put_flash(:error, "Unauthorised") |> redirect(to: "/") |> live_noreply() end end defp apply_action(socket, :index, _params) do socket |> assign(:page_title, "My Shifts") |> assign(:shift, nil) end defp shift_from_params(params) defp shift_from_params(%{"id" => id}), do: Shifts.get_shift!(id) defp shift_from_params(_params), do: %Shift{} defp init_today(socket, today) do assign(socket, current_date: today, cursor_date: cursor_date(today)) end defp cursor_date(%Date{} = date) do date |> Date.beginning_of_month() |> Date.add(4) end defp assign_date_range(%{assigns: %{cursor_date: cursor_date}} = socket) do assign(socket, date_range: date_range(cursor_date)) end defp date_range(cursor_date) do cursor_date |> Date.beginning_of_month() |> Date.range(Date.end_of_month(cursor_date)) end defp assign_known_shifts(socket) do user = socket.assigns.current_user shifts = Shifts.list_shifts_by_user_in_date_range(user.id, socket.assigns.date_range) assign(socket, :shifts, shifts) end defp update_agenda(socket) do socket |> assign_date_range() |> assign_known_shifts() end @impl true def handle_event("delete", %{"id" => id}, socket) do shift = Shifts.get_shift!(id) {:ok, _} = Shifts.delete_shift(shift) {:noreply, assign_known_shifts(socket)} end @impl true def handle_event("month-nav", %{"month" => nav}, socket) do new_cursor = cond do nav == "now" -> Date.utc_today() true -> socket.assigns.cursor_date |> Date.add((nav == "prev" && -30) || 30) |> cursor_date() end socket |> assign(:cursor_date, new_cursor) |> update_agenda() |> live_noreply() end end