summaryrefslogtreecommitdiff
path: root/lib/pleroma/plugs/user_fetcher_plug.ex
blob: 9cbaaf40a983be5d4c76788ae25bd71d3bc5c64a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
defmodule Pleroma.Plugs.UserFetcherPlug do
  import Plug.Conn
  alias Pleroma.Repo
  alias Pleroma.User

  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