97 lines
2.8 KiB
Elixir
97 lines
2.8 KiB
Elixir
defmodule MatrixServer.Account do
|
|
use Ecto.Schema
|
|
|
|
import Ecto.{Changeset, Query}
|
|
|
|
alias MatrixServer.{Repo, Account, Device}
|
|
alias Ecto.Multi
|
|
|
|
@max_mxid_length 255
|
|
|
|
@primary_key {:localpart, :string, []}
|
|
schema "accounts" do
|
|
field :password_hash, :string, redact: true
|
|
has_many :devices, Device, foreign_key: :localpart
|
|
timestamps(updated_at: false)
|
|
end
|
|
|
|
def available?(localpart) when is_binary(localpart) do
|
|
if Regex.match?(MatrixServer.localpart_regex(), localpart) and
|
|
String.length(localpart) <= localpart_length() do
|
|
if Repo.one!(
|
|
Account
|
|
|> where([a], a.localpart == ^localpart)
|
|
|> select([a], count(a))
|
|
) == 0 do
|
|
:ok
|
|
else
|
|
{:error, :user_in_use}
|
|
end
|
|
else
|
|
{:error, :invalid_username}
|
|
end
|
|
end
|
|
|
|
def register(params) do
|
|
Multi.new()
|
|
|> Multi.insert(:account, changeset(%Account{}, params))
|
|
|> Multi.insert(:device, fn %{account: account} ->
|
|
device_id = Device.generate_device_id(account.localpart)
|
|
|
|
params =
|
|
Map.update(params, :device_id, device_id, fn
|
|
nil -> device_id
|
|
x -> x
|
|
end)
|
|
|
|
Ecto.build_assoc(account, :devices)
|
|
|> Device.changeset(params)
|
|
end)
|
|
|> Multi.run(:device_with_access_token, &Device.insert_new_access_token/2)
|
|
end
|
|
|
|
def login(%{localpart: localpart, password: password} = params) do
|
|
fn repo ->
|
|
case repo.one(from a in Account, where: a.localpart == ^localpart) do
|
|
%Account{password_hash: hash} = account ->
|
|
if Bcrypt.verify_pass(password, hash) do
|
|
device_id = Map.get(params, :device_id, Device.generate_device_id(localpart))
|
|
access_token = Device.generate_access_token(localpart, device_id)
|
|
|
|
case Device.login(account, device_id, access_token, params) do
|
|
{:ok, device} -> device
|
|
{:error, _cs} -> repo.rollback(:forbidden)
|
|
end
|
|
else
|
|
repo.rollback(:forbidden)
|
|
end
|
|
|
|
nil ->
|
|
repo.rollback(:forbidden)
|
|
end
|
|
end
|
|
end
|
|
|
|
def by_access_token(access_token) do
|
|
Device
|
|
|> where([d], d.access_token == ^access_token)
|
|
|> join(:inner, [d], a in assoc(d, :account))
|
|
|> select([d, a], {a, d})
|
|
|> Repo.one()
|
|
end
|
|
|
|
def changeset(account, params \\ %{}) do
|
|
account
|
|
|> cast(params, [:localpart, :password_hash])
|
|
|> validate_required([:localpart, :password_hash])
|
|
|> validate_length(:password_hash, max: 60)
|
|
|> validate_format(:localpart, MatrixServer.localpart_regex())
|
|
|> validate_length(:localpart, max: localpart_length())
|
|
|> unique_constraint(:localpart, name: :accounts_pkey)
|
|
end
|
|
|
|
defp localpart_length do
|
|
# Subtract the "@" and ":" in the MXID.
|
|
@max_mxid_length - 2 - String.length(MatrixServer.server_name())
|
|
end
|
|
end
|