memEx/lib/memex/accounts/invites.ex

defmodule Memex.Accounts.Invites do
  @moduledoc """
  The Invites context.
  """

  import Ecto.Query, warn: false
  alias Ecto.Multi
  alias Memex.Accounts.{Invite, User}
  alias Memex.Repo

  @invite_token_length 20

  @doc """
  Returns the list of invites.

  ## Examples

      iex> list_invites(%User{id: 123, role: :admin})
      [%Invite{}, ...]

  """
  @spec list_invites(User.t()) :: [Invite.t()]
  def list_invites(%User{role: :admin}) do
    Repo.all(from i in Invite, order_by: i.name)
  end

  @doc """
  Gets a single invite for a user

  Raises `Ecto.NoResultsError` if the Invite does not exist.

  ## Examples

      iex> get_invite!(123, %User{id: 123, role: :admin})
      %Invite{}

      > get_invite!(456, %User{id: 123, role: :admin})
      ** (Ecto.NoResultsError)

  """
  @spec get_invite!(Invite.id(), User.t()) :: Invite.t()
  def get_invite!(id, %User{role: :admin}) do
    Repo.get!(Invite, id)
  end

  @doc """
  Returns if an invite token is still valid

  ## Examples

      iex> valid_invite_token?("valid_token")
      %Invite{}

      iex> valid_invite_token?("invalid_token")
      nil
  """
  @spec valid_invite_token?(Invite.token() | nil) :: boolean()
  def valid_invite_token?(token) when token in [nil, ""], do: false

  def valid_invite_token?(token) do
    Repo.exists?(
      from i in Invite,
        where: i.token == ^token,
        where: i.disabled_at |> is_nil()
    )
  end

  @doc """
  Uses invite by decrementing uses_left, or marks invite invalid if it's been
  completely used.
  """
  @spec use_invite(Invite.token()) :: {:ok, Invite.t()} | {:error, :invalid_token}
  def use_invite(invite_token) do
    Multi.new()
    |> Multi.run(:invite, fn _changes_so_far, _repo ->
      invite_token |> get_invite_by_token()
    end)
    |> Multi.update(:decrement_invite, fn %{invite: invite} ->
      decrement_invite_changeset(invite)
    end)
    |> Repo.transaction()
    |> case do
      {:ok, %{decrement_invite: invite}} -> {:ok, invite}
      {:error, :invite, :invalid_token, _changes_so_far} -> {:error, :invalid_token}
    end
  end

  @spec get_invite_by_token(Invite.token() | nil) :: {:ok, Invite.t()} | {:error, :invalid_token}
  defp get_invite_by_token(token) when token in [nil, ""], do: {:error, :invalid_token}

  defp get_invite_by_token(token) do
    Repo.one(
      from i in Invite,
        where: i.token == ^token,
        where: i.disabled_at |> is_nil()
    )
    |> case do
      nil -> {:error, :invalid_token}
      invite -> {:ok, invite}
    end
  end

  @spec get_use_count(Invite.t(), User.t()) :: non_neg_integer() | nil
  def get_use_count(%Invite{id: invite_id} = invite, user) do
    [invite] |> get_use_counts(user) |> Map.get(invite_id)
  end

  @spec get_use_counts([Invite.t()], User.t()) :: %{Invite.id() => non_neg_integer()}
  def get_use_counts(invites, %User{role: :admin}) do
    invite_ids = invites |> Enum.map(fn %{id: invite_id} -> invite_id end)

    Repo.all(
      from u in User,
        where: u.invite_id in ^invite_ids,
        group_by: u.invite_id,
        select: {u.invite_id, count(u.id)}
    )
    |> Map.new()
  end

  @spec decrement_invite_changeset(Invite.t()) :: Invite.changeset()
  defp decrement_invite_changeset(%Invite{uses_left: nil} = invite) do
    invite |> Invite.update_changeset(%{})
  end

  defp decrement_invite_changeset(%Invite{uses_left: 1} = invite) do
    now = NaiveDateTime.utc_now() |> NaiveDateTime.truncate(:second)
    invite |> Invite.update_changeset(%{uses_left: 0, disabled_at: now})
  end

  defp decrement_invite_changeset(%Invite{uses_left: uses_left} = invite) do
    invite |> Invite.update_changeset(%{uses_left: uses_left - 1})
  end

  @doc """
  Creates a invite.

  ## Examples

      iex> create_invite(%User{id: 123, role: :admin}, %{field: value})
      {:ok, %Invite{}}

      iex> create_invite(%User{id: 123, role: :admin}, %{field: bad_value})
      {:error, %Changeset{}}

  """
  @spec create_invite(User.t(), attrs :: map()) ::
          {:ok, Invite.t()} | {:error, Invite.changeset()}
  def create_invite(%User{role: :admin} = user, attrs) do
    token =
      :crypto.strong_rand_bytes(@invite_token_length)
      |> Base.url_encode64()
      |> binary_part(0, @invite_token_length)

    Invite.create_changeset(user, token, attrs) |> Repo.insert()
  end

  @doc """
  Updates a invite.

  ## Examples

      iex> update_invite(invite, %{field: new_value}, %User{id: 123, role: :admin})
      {:ok, %Invite{}}

      iex> update_invite(invite, %{field: bad_value}, %User{id: 123, role: :admin})
      {:error, %Changeset{}}

  """
  @spec update_invite(Invite.t(), attrs :: map(), User.t()) ::
          {:ok, Invite.t()} | {:error, Invite.changeset()}
  def update_invite(invite, attrs, %User{role: :admin}) do
    invite |> Invite.update_changeset(attrs) |> Repo.update()
  end

  @doc """
  Deletes a invite.

  ## Examples

      iex> delete_invite(invite, %User{id: 123, role: :admin})
      {:ok, %Invite{}}

      iex> delete_invite(invite, %User{id: 123, role: :admin})
      {:error, %Changeset{}}

  """
  @spec delete_invite(Invite.t(), User.t()) ::
          {:ok, Invite.t()} | {:error, Invite.changeset()}
  def delete_invite(invite, %User{role: :admin}) do
    invite |> Repo.delete()
  end

  @doc """
  Deletes a invite.

  ## Examples

      iex> delete_invite(invite, %User{id: 123, role: :admin})
      %Invite{}

  """
  @spec delete_invite!(Invite.t(), User.t()) :: Invite.t()
  def delete_invite!(invite, %User{role: :admin}) do
    invite |> Repo.delete!()
  end
end