forked from shibao/cannery
450 lines
12 KiB
Elixir
450 lines
12 KiB
Elixir
defmodule Lokal.Accounts do
|
|
@moduledoc """
|
|
The Accounts context.
|
|
"""
|
|
|
|
import Ecto.Query, warn: false
|
|
alias Lokal.{Mailer, Repo}
|
|
alias Lokal.Accounts.{User, UserToken}
|
|
alias Ecto.{Changeset, Multi}
|
|
|
|
## Database getters
|
|
|
|
@doc """
|
|
Gets a user by email.
|
|
|
|
## Examples
|
|
|
|
iex> get_user_by_email("foo@example.com")
|
|
%User{}
|
|
|
|
iex> get_user_by_email("unknown@example.com")
|
|
nil
|
|
|
|
"""
|
|
@spec get_user_by_email(String.t()) :: User.t() | nil
|
|
def get_user_by_email(email) when is_binary(email), do: Repo.get_by(User, email: email)
|
|
|
|
@doc """
|
|
Gets a user by email and password.
|
|
|
|
## Examples
|
|
|
|
iex> get_user_by_email_and_password("foo@example.com", "correct_password")
|
|
%User{}
|
|
|
|
iex> get_user_by_email_and_password("foo@example.com", "invalid_password")
|
|
nil
|
|
|
|
"""
|
|
@spec get_user_by_email_and_password(String.t(), String.t()) ::
|
|
User.t() | nil
|
|
def get_user_by_email_and_password(email, password)
|
|
when is_binary(email) and is_binary(password) do
|
|
user = Repo.get_by(User, email: email)
|
|
if User.valid_password?(user, password), do: user
|
|
end
|
|
|
|
@doc """
|
|
Gets a single user.
|
|
|
|
Raises `Ecto.NoResultsError` if the User does not exist.
|
|
|
|
## Examples
|
|
|
|
iex> get_user!(123)
|
|
%User{}
|
|
|
|
iex> get_user!(456)
|
|
** (Ecto.NoResultsError)
|
|
|
|
"""
|
|
@spec get_user!(User.t()) :: User.t()
|
|
def get_user!(id), do: Repo.get!(User, id)
|
|
|
|
@doc """
|
|
Returns all users grouped by role.
|
|
|
|
## Examples
|
|
|
|
iex> list_users_by_role(%User{id: 123, role: :admin})
|
|
[admin: [%User{}], user: [%User{}, %User{}]]
|
|
|
|
"""
|
|
@spec list_all_users_by_role(User.t()) :: %{String.t() => [User.t()]}
|
|
def list_all_users_by_role(%User{role: :admin}) do
|
|
Repo.all(from u in User, order_by: u.email) |> Enum.group_by(fn user -> user.role end)
|
|
end
|
|
|
|
@doc """
|
|
Returns all users for a certain role.
|
|
|
|
## Examples
|
|
|
|
iex> list_users_by_role(%User{id: 123, role: :admin})
|
|
[%User{}]
|
|
|
|
"""
|
|
@spec list_users_by_role(:admin | :user) :: [User.t()]
|
|
def list_users_by_role(role) do
|
|
role = role |> to_string()
|
|
Repo.all(from u in User, where: u.role == ^role, order_by: u.email)
|
|
end
|
|
|
|
## User registration
|
|
|
|
@doc """
|
|
Registers a user.
|
|
|
|
## Examples
|
|
|
|
iex> register_user(%{field: value})
|
|
{:ok, %User{}}
|
|
|
|
iex> register_user(%{field: bad_value})
|
|
{:error, %Changeset{}}
|
|
|
|
"""
|
|
@spec register_user(map()) :: {:ok, User.t()} | {:error, Changeset.t(User.new_user())}
|
|
def register_user(attrs) do
|
|
# if no registered users, make first user an admin
|
|
role =
|
|
if Repo.one!(from u in User, select: count(u.id), distinct: true) == 0,
|
|
do: "admin",
|
|
else: "user"
|
|
|
|
%User{} |> User.registration_changeset(attrs |> Map.put("role", role)) |> Repo.insert()
|
|
end
|
|
|
|
@doc """
|
|
Returns an `%Changeset{}` for tracking user changes.
|
|
|
|
## Examples
|
|
|
|
iex> change_user_registration(user)
|
|
%Changeset{data: %User{}}
|
|
|
|
"""
|
|
@spec change_user_registration(User.t() | User.new_user()) ::
|
|
Changeset.t(User.t() | User.new_user())
|
|
@spec change_user_registration(User.t() | User.new_user(), map()) ::
|
|
Changeset.t(User.t() | User.new_user())
|
|
def change_user_registration(user, attrs \\ %{}),
|
|
do: User.registration_changeset(user, attrs, hash_password: false)
|
|
|
|
## Settings
|
|
|
|
@doc """
|
|
Returns an `%Changeset{}` for changing the user email.
|
|
|
|
## Examples
|
|
|
|
iex> change_user_email(user)
|
|
%Changeset{data: %User{}}
|
|
|
|
"""
|
|
@spec change_user_email(User.t(), map()) :: Changeset.t(User.t())
|
|
def change_user_email(user, attrs \\ %{}), do: User.email_changeset(user, attrs)
|
|
|
|
@doc """
|
|
Returns an `%Changeset{}` for changing the user role.
|
|
|
|
## Examples
|
|
|
|
iex> change_user_role(user)
|
|
%Changeset{data: %User{}}
|
|
|
|
"""
|
|
@spec change_user_role(User.t(), atom()) :: Changeset.t(User.t())
|
|
def change_user_role(user, role), do: User.role_changeset(user, role)
|
|
|
|
@doc """
|
|
Emulates that the email will change without actually changing
|
|
it in the database.
|
|
|
|
## Examples
|
|
|
|
iex> apply_user_email(user, "valid password", %{email: ...})
|
|
{:ok, %User{}}
|
|
|
|
iex> apply_user_email(user, "invalid password", %{email: ...})
|
|
{:error, %Changeset{}}
|
|
|
|
"""
|
|
@spec apply_user_email(User.t(), String.t(), map()) ::
|
|
{:ok, User.t()} | {:error, Changeset.t(User.t())}
|
|
def apply_user_email(user, password, attrs) do
|
|
user
|
|
|> User.email_changeset(attrs)
|
|
|> User.validate_current_password(password)
|
|
|> Changeset.apply_action(:update)
|
|
end
|
|
|
|
@doc """
|
|
Updates the user email using the given token.
|
|
|
|
If the token matches, the user email is updated and the token is deleted.
|
|
The confirmed_at date is also updated to the current time.
|
|
"""
|
|
@spec update_user_email(User.t(), String.t()) :: :ok | :error
|
|
def update_user_email(user, token) do
|
|
context = "change:#{user.email}"
|
|
|
|
with {:ok, query} <- UserToken.verify_change_email_token_query(token, context),
|
|
%UserToken{sent_to: email} <- Repo.one(query),
|
|
{:ok, _} <- Repo.transaction(user_email_multi(user, email, context)) do
|
|
:ok
|
|
else
|
|
_ -> :error
|
|
end
|
|
end
|
|
|
|
@spec user_email_multi(User.t(), String.t(), String.t()) :: Multi.t()
|
|
defp user_email_multi(user, email, context) do
|
|
changeset = user |> User.email_changeset(%{email: email}) |> User.confirm_changeset()
|
|
|
|
Multi.new()
|
|
|> Multi.update(:user, changeset)
|
|
|> Multi.delete_all(:tokens, UserToken.user_and_contexts_query(user, [context]))
|
|
end
|
|
|
|
@doc """
|
|
Delivers the update email instructions to the given user.
|
|
|
|
## Examples
|
|
|
|
iex> deliver_update_email_instructions(user, current_email, &Routes.user_update_email_url(conn, :edit, &1))
|
|
{:ok, %{to: ..., body: ...}}
|
|
|
|
"""
|
|
@spec deliver_update_email_instructions(User.t(), String.t(), function) :: Job.t()
|
|
def deliver_update_email_instructions(user, current_email, update_email_url_fun)
|
|
when is_function(update_email_url_fun, 1) do
|
|
{encoded_token, user_token} = UserToken.build_email_token(user, "change:#{current_email}")
|
|
Repo.insert!(user_token)
|
|
Mailer.deliver_update_email_instructions(user, update_email_url_fun.(encoded_token))
|
|
end
|
|
|
|
@doc """
|
|
Returns an `%Changeset{}` for changing the user password.
|
|
|
|
## Examples
|
|
|
|
iex> change_user_password(user)
|
|
%Changeset{data: %User{}}
|
|
|
|
"""
|
|
@spec change_user_password(User.t(), map()) :: Changeset.t(User.t())
|
|
def change_user_password(user, attrs \\ %{}),
|
|
do: User.password_changeset(user, attrs, hash_password: false)
|
|
|
|
@doc """
|
|
Updates the user password.
|
|
|
|
## Examples
|
|
|
|
iex> update_user_password(user, "valid password", %{password: ...})
|
|
{:ok, %User{}}
|
|
|
|
iex> update_user_password(user, "invalid password", %{password: ...})
|
|
{:error, %Changeset{}}
|
|
|
|
"""
|
|
@spec update_user_password(User.t(), String.t(), map()) ::
|
|
{:ok, User.t()} | {:error, Changeset.t(User.t())}
|
|
def update_user_password(user, password, attrs) do
|
|
changeset =
|
|
user
|
|
|> User.password_changeset(attrs)
|
|
|> User.validate_current_password(password)
|
|
|
|
Multi.new()
|
|
|> Multi.update(:user, changeset)
|
|
|> Multi.delete_all(:tokens, UserToken.user_and_contexts_query(user, :all))
|
|
|> Repo.transaction()
|
|
|> case do
|
|
{:ok, %{user: user}} -> {:ok, user}
|
|
{:error, :user, changeset, _} -> {:error, changeset}
|
|
end
|
|
end
|
|
|
|
@doc """
|
|
Deletes a user. must be performed by an admin or the same user!
|
|
|
|
## Examples
|
|
|
|
iex> delete_user!(user_to_delete, %User{id: 123, role: :admin})
|
|
%User{}
|
|
|
|
iex> delete_user!(%User{id: 123}, %User{id: 123})
|
|
%User{}
|
|
|
|
"""
|
|
@spec delete_user!(User.t(), User.t()) :: User.t()
|
|
def delete_user!(user, %User{role: :admin}), do: user |> Repo.delete!()
|
|
def delete_user!(%User{id: user_id} = user, %User{id: user_id}), do: user |> Repo.delete!()
|
|
|
|
## Session
|
|
|
|
@doc """
|
|
Generates a session token.
|
|
"""
|
|
@spec generate_user_session_token(User.t()) :: String.t()
|
|
def generate_user_session_token(user) do
|
|
{token, user_token} = UserToken.build_session_token(user)
|
|
Repo.insert!(user_token)
|
|
token
|
|
end
|
|
|
|
@doc """
|
|
Gets the user with the given signed token.
|
|
"""
|
|
@spec get_user_by_session_token(String.t()) :: User.t()
|
|
def get_user_by_session_token(token) do
|
|
{:ok, query} = UserToken.verify_session_token_query(token)
|
|
Repo.one(query)
|
|
end
|
|
|
|
@doc """
|
|
Deletes the signed token with the given context.
|
|
"""
|
|
@spec delete_session_token(String.t()) :: :ok
|
|
def delete_session_token(token) do
|
|
Repo.delete_all(UserToken.token_and_context_query(token, "session"))
|
|
:ok
|
|
end
|
|
|
|
@doc """
|
|
Returns a boolean if registration is allowed or not
|
|
"""
|
|
@spec allow_registration?() :: boolean()
|
|
def allow_registration? do
|
|
Application.get_env(:lokal, LokalWeb.Endpoint)[:registration] == "public" or
|
|
list_users_by_role(:admin) |> Enum.empty?()
|
|
end
|
|
|
|
@doc """
|
|
Checks if user is an admin
|
|
"""
|
|
@spec is_admin?(User.t()) :: boolean()
|
|
def is_admin?(%User{id: user_id}) do
|
|
Repo.one(from u in User, where: u.id == ^user_id and u.role == :admin)
|
|
|> is_nil()
|
|
end
|
|
|
|
## Confirmation
|
|
|
|
@doc """
|
|
Delivers the confirmation email instructions to the given user.
|
|
|
|
## Examples
|
|
|
|
iex> deliver_user_confirmation_instructions(user, &Routes.user_confirmation_url(conn, :confirm, &1))
|
|
{:ok, %{to: ..., body: ...}}
|
|
|
|
iex> deliver_user_confirmation_instructions(confirmed_user, &Routes.user_confirmation_url(conn, :confirm, &1))
|
|
{:error, :already_confirmed}
|
|
|
|
"""
|
|
@spec deliver_user_confirmation_instructions(User.t(), function) :: Job.t()
|
|
def deliver_user_confirmation_instructions(user, confirmation_url_fun)
|
|
when is_function(confirmation_url_fun, 1) do
|
|
if user.confirmed_at do
|
|
{:error, :already_confirmed}
|
|
else
|
|
{encoded_token, user_token} = UserToken.build_email_token(user, "confirm")
|
|
Repo.insert!(user_token)
|
|
Mailer.deliver_confirmation_instructions(user, confirmation_url_fun.(encoded_token))
|
|
end
|
|
end
|
|
|
|
@doc """
|
|
Confirms a user by the given token.
|
|
|
|
If the token matches, the user account is marked as confirmed
|
|
and the token is deleted.
|
|
"""
|
|
@spec confirm_user(String.t()) :: {:ok, User.t()} | atom()
|
|
def confirm_user(token) do
|
|
with {:ok, query} <- UserToken.verify_email_token_query(token, "confirm"),
|
|
%User{} = user <- Repo.one(query),
|
|
{:ok, %{user: user}} <- Repo.transaction(confirm_user_multi(user)) do
|
|
{:ok, user}
|
|
else
|
|
_ -> :error
|
|
end
|
|
end
|
|
|
|
@spec confirm_user_multi(User.t()) :: Multi.t()
|
|
def confirm_user_multi(user) do
|
|
Multi.new()
|
|
|> Multi.update(:user, User.confirm_changeset(user))
|
|
|> Multi.delete_all(:tokens, UserToken.user_and_contexts_query(user, ["confirm"]))
|
|
end
|
|
|
|
## Reset password
|
|
|
|
@doc """
|
|
Delivers the reset password email to the given user.
|
|
|
|
## Examples
|
|
|
|
iex> deliver_user_reset_password_instructions(user, &Routes.user_reset_password_url(conn, :edit, &1))
|
|
{:ok, %{to: ..., body: ...}}
|
|
|
|
"""
|
|
@spec deliver_user_reset_password_instructions(User.t(), function()) :: Job.t()
|
|
def deliver_user_reset_password_instructions(user, reset_password_url_fun)
|
|
when is_function(reset_password_url_fun, 1) do
|
|
{encoded_token, user_token} = UserToken.build_email_token(user, "reset_password")
|
|
Repo.insert!(user_token)
|
|
Mailer.deliver_reset_password_instructions(user, reset_password_url_fun.(encoded_token))
|
|
end
|
|
|
|
@doc """
|
|
Gets the user by reset password token.
|
|
|
|
## Examples
|
|
|
|
iex> get_user_by_reset_password_token("validtoken")
|
|
%User{}
|
|
|
|
iex> get_user_by_reset_password_token("invalidtoken")
|
|
nil
|
|
|
|
"""
|
|
@spec get_user_by_reset_password_token(String.t()) :: User.t() | nil
|
|
def get_user_by_reset_password_token(token) do
|
|
with {:ok, query} <- UserToken.verify_email_token_query(token, "reset_password"),
|
|
%User{} = user <- Repo.one(query) do
|
|
user
|
|
else
|
|
_ -> nil
|
|
end
|
|
end
|
|
|
|
@doc """
|
|
Resets the user password.
|
|
|
|
## Examples
|
|
|
|
iex> reset_user_password(user, %{password: "new long password", password_confirmation: "new long password"})
|
|
{:ok, %User{}}
|
|
|
|
iex> reset_user_password(user, %{password: "valid", password_confirmation: "not the same"})
|
|
{:error, %Changeset{}}
|
|
|
|
"""
|
|
@spec reset_user_password(User.t(), map()) :: {:ok, User.t()} | {:error, Changeset.t(User.t())}
|
|
def reset_user_password(user, attrs) do
|
|
Multi.new()
|
|
|> Multi.update(:user, User.password_changeset(user, attrs))
|
|
|> Multi.delete_all(:tokens, UserToken.user_and_contexts_query(user, :all))
|
|
|> Repo.transaction()
|
|
|> case do
|
|
{:ok, %{user: user}} -> {:ok, user}
|
|
{:error, :user, changeset, _} -> {:error, changeset}
|
|
end
|
|
end
|
|
end
|