# 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.User
  alias Pleroma.Repo

  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