40 lines
919 B
Elixir
40 lines
919 B
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.Plugs.UserFetcherPlug do
|
|
alias Pleroma.Repo
|
|
alias Pleroma.User
|
|
|
|
import Plug.Conn
|
|
|
|
def init(options) do
|
|
options
|
|
end
|
|
|
|
def call(conn, _options) do
|
|
with %{auth_credentials: %{username: username}} <- conn.assigns,
|
|
{:ok, %User{} = user} <- user_fetcher(username) do
|
|
conn
|
|
|> assign(:auth_user, user)
|
|
else
|
|
_ -> conn
|
|
end
|
|
end
|
|
|
|
defp user_fetcher(username_or_email) do
|
|
{
|
|
:ok,
|
|
cond do
|
|
# First, try logging in as if it was a name
|
|
user = Repo.get_by(User, %{nickname: username_or_email}) ->
|
|
user
|
|
|
|
# If we get nil, we try using it as an email
|
|
user = Repo.get_by(User, %{email: username_or_email}) ->
|
|
user
|
|
end
|
|
}
|
|
end
|
|
end
|