corex

defmodule Corex.DataTable.Selection do
  import Phoenix.Component, only: [assign: 3]

  @moduledoc """
  Helpers for selectable [`data_table/1`](Corex.DataTable.html#data_table/1) usage in LiveViews.

  Use with [`Corex.DataTable`](Corex.DataTable.html): set `selectable`, `selected`, `on_select`,
  `on_select_all`, and `row_id` on the component (see the **Selectable** pattern in
  [`data_table/1`](Corex.DataTable.html#data_table/1) docs). In the LiveView, call
  `assign_for_selection/3` in `mount/3`, then `handle_select/3` and `handle_select_all/3`
  from the matching `handle_event/3` callbacks.

  ## Example

      def mount(_params, _session, socket) do
        socket =
          socket
          |> assign(:users, fetch_users())
          |> Corex.DataTable.Selection.assign_for_selection(:users, table_id: "users-table", row_id: &"user-\#{&1.id}")

        {:ok, socket}
      end

      def handle_event("select", params, socket) do
        {:noreply, Corex.DataTable.Selection.handle_select(socket, params, :users)}
      end

      def handle_event("select_all", params, socket) do
        {:noreply, Corex.DataTable.Selection.handle_select_all(socket, params, :users)}
      end

      def render(assigns) do
        ~H\"\"\"
        <.data_table
          id="users-table"
          rows={@users}
          row_id={&"user-\#{&1.id}"}
          selectable={true}
          selected={@selected}
          on_select="select"
          on_select_all="select_all"
        >
          <:col :let={user} label="ID">{user.id}</:col>
          <:col :let={user} label="Name">{user.name}</:col>
        </.data_table>
        \"\"\"
      end
  """

  @doc """
  Assigns selection state for a [`data_table/1`](Corex.DataTable.html#data_table/1) instance.

  Use in `mount/3` after assigning the rows list. Options:

  - `:table_id` – required; must match the `id` on [`data_table/1`](Corex.DataTable.html#data_table/1) (e.g. `"users-table"`)
  - `:row_id` – required; must match `row_id` on the same [`data_table/1`](Corex.DataTable.html#data_table/1) component (e.g. `&"user-\#{&1.id}"`). Do not rely on the component `inspect/1` fallback when using these helpers.

  Adds `:selected` (empty list), `:selection_table_id`, and `:selection_row_id`
  for use by [`handle_select/3`](#handle_select/3) and [`handle_select_all/3`](#handle_select_all/3).
  """
  def assign_for_selection(socket, _rows_assign, opts) do
    table_id = Keyword.fetch!(opts, :table_id)
    row_id = Keyword.fetch!(opts, :row_id)

    socket
    |> assign(:selected, [])
    |> assign(:selection_table_id, table_id)
    |> assign(:selection_row_id, row_id)
  end

  @doc """
  Handles the row `on_select` event from [`data_table/1`](Corex.DataTable.html#data_table/1) and returns the updated socket.

  Use in `handle_event("select", params, socket)` (same name as `on_select`) and return
  `{:noreply, Corex.DataTable.Selection.handle_select(socket, params, :users)}`.

  `params` must contain `"id"` (checkbox DOM id) and `"checked"`. Only row ids derived from
  current `rows` via `selection_row_id` are added to `:selected`; forged ids are ignored when
  checking a row. `rows_assign` is the assign key passed to [`data_table/1`](Corex.DataTable.html#data_table/1) as `rows` (e.g. `:users`).
  """
  def handle_select(socket, %{"id" => checkbox_id, "checked" => checked}, rows_assign) do
    table_id = socket.assigns.selection_table_id
    row_id_fn = socket.assigns.selection_row_id
    row_id = String.replace(checkbox_id, "#{table_id}-select-", "")
    rows = socket.assigns[rows_assign] || []
    valid_ids = valid_row_ids(rows, row_id_fn)

    selected =
      cond do
        checked and MapSet.member?(valid_ids, row_id) ->
          [row_id | socket.assigns.selected] |> Enum.uniq()

        checked ->
          socket.assigns.selected

        true ->
          List.delete(socket.assigns.selected, row_id)
      end

    selected = Enum.filter(selected, &MapSet.member?(valid_ids, &1))

    all_selected =
      MapSet.subset?(MapSet.new(selected), valid_ids) and
        length(selected) == MapSet.size(valid_ids)

    socket
    |> assign(:selected, selected)
    |> Corex.Checkbox.set_checked("#{table_id}-select-all", all_selected)
  end

  @doc """
  Handles the `on_select_all` event from [`data_table/1`](Corex.DataTable.html#data_table/1) and returns the updated socket.

  Use in `handle_event("select_all", params, socket)` (same name as `on_select_all`) and return
  `{:noreply, Corex.DataTable.Selection.handle_select_all(socket, params, :users)}`.

  `params` must contain `"checked"`. Syncs all row checkboxes via
  [`Corex.Checkbox.set_checked/3`](Corex.Checkbox.html#set_checked/3). `rows_assign` is the
  assign key passed to [`data_table/1`](Corex.DataTable.html#data_table/1) as `rows`.
  """
  def handle_select_all(socket, %{"checked" => checked}, rows_assign) do
    table_id = socket.assigns.selection_table_id
    row_id_fn = socket.assigns.selection_row_id
    rows = socket.assigns[rows_assign] || []

    selected =
      if checked do
        Enum.map(rows, row_id_fn)
      else
        []
      end

    socket =
      socket
      |> assign(:selected, selected)

    socket =
      Enum.reduce(rows, socket, fn row, acc ->
        Corex.Checkbox.set_checked(acc, "#{table_id}-select-#{row_id_fn.(row)}", checked)
      end)

    socket
  end

  defp valid_row_ids(rows, row_id_fn) when is_function(row_id_fn, 1) do
    rows
    |> Enum.map(row_id_fn)
    |> MapSet.new()
  end
end