81 lines
2.2 KiB
Elixir
81 lines
2.2 KiB
Elixir
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
|