mobilizon/lib/web/auth/context.ex

defmodule Mobilizon.Web.Auth.Context do
  @moduledoc """
  Guardian context for Mobilizon.Web
  """
  @behaviour Plug

  import Plug.Conn

  alias Mobilizon.Applications.Application, as: AuthApplication
  alias Mobilizon.Applications.ApplicationToken
  alias Mobilizon.Users.User

  @spec init(Plug.opts()) :: Plug.opts()
  def init(opts) do
    opts
  end

  @spec call(Plug.Conn.t(), Plug.opts()) :: Plug.Conn.t()
  def call(%{assigns: %{ip: _}} = conn, _opts), do: conn

  def call(conn, _opts) do
    set_user_information_in_context(conn)
  end

  @spec set_user_information_in_context(Plug.Conn.t()) :: Plug.Conn.t()
  defp set_user_information_in_context(conn) do
    context = %{ip: conn.remote_ip |> :inet.ntoa() |> to_string()}

    user_agent = conn |> Plug.Conn.get_req_header("user-agent") |> List.first()

    {conn, context} =
      case Guardian.Plug.current_resource(conn) do
        %User{} = user ->
          set_user_context({conn, context}, user)

        %ApplicationToken{user: %User{} = user} = app_token ->
          conn
          |> set_app_token_context(context, app_token)
          |> set_user_context(user)

        _ ->
          {conn, context}
      end

    context = if is_nil(user_agent), do: context, else: Map.put(context, :user_agent, user_agent)

    Absinthe.Plug.put_options(conn, context: context)
  end

  defp set_user_context({conn, context}, %User{id: user_id, email: user_email} = user) do
    if Application.get_env(:sentry, :dsn) != nil do
      Sentry.Context.set_user_context(%{
        id: user_id,
        email: user_email,
        ip_address: context.ip
      })
    end

    context = Map.put(context, :current_user, user)
    conn = assign(conn, :user_locale, user.locale)
    {conn, context}
  end

  defp set_app_token_context(
         conn,
         context,
         %ApplicationToken{application: %AuthApplication{client_id: client_id} = app} = app_token
       ) do
    if Application.get_env(:sentry, :dsn) != nil do
      Sentry.Context.set_user_context(%{
        app_token_client_id: client_id
      })
    end

    context =
      context |> Map.put(:current_auth_app_token, app_token) |> Map.put(:current_auth_app, app)

    {conn, context}
  end
end