Current section

Files

Jump to
ash_admin lib ash_admin actor_plug plug.ex
Raw

lib/ash_admin/actor_plug/plug.ex

defmodule AshAdmin.ActorPlug.Plug do
@moduledoc false
@behaviour AshAdmin.ActorPlug
import AshAdmin.Helpers
require Ash.Query
@impl true
def actor_assigns(socket, session) do
otp_app = socket.endpoint.config(:otp_app)
apis = apis(otp_app)
session =
Phoenix.LiveView.get_connect_params(socket) || session
actor_paused =
if is_nil(session["actor_paused"]) do
true
else
session_bool(session["actor_paused"])
end
[
actor: actor_from_session(socket.endpoint, session),
actor_api: actor_api_from_session(socket.endpoint, session),
actor_resources: actor_resources(apis),
authorizing: session_bool(session["actor_authorizing"]),
actor_paused: actor_paused
]
end
@impl true
def set_actor_session(conn) do
case conn.cookies do
%{"actor_resource" => "undefined"} ->
conn
session ->
case session do
%{
"actor_resource" => resource,
"actor_api" => api,
"actor_action" => action,
"actor_primary_key" => primary_key
} ->
authorizing = session["actor_authorizing"] || false
actor_paused =
if is_nil(session["actor_paused"]) do
true
else
session["actor_paused"]
end
actor = actor_from_session(conn.private.phoenix_endpoint, session)
authorizing = session_bool(authorizing)
actor_paused = session_bool(actor_paused)
conn
|> Plug.Conn.put_session(:actor_resource, resource)
|> Plug.Conn.put_session(:actor_api, api)
|> Plug.Conn.put_session(:actor_action, action)
|> Plug.Conn.put_session(:actor_primary_key, primary_key)
|> Plug.Conn.put_session(:actor_authorizing, authorizing)
|> Plug.Conn.put_session(:actor_paused, actor_paused)
|> Plug.Conn.assign(:actor, actor)
|> Plug.Conn.assign(:authorizing, authorizing || false)
|> Plug.Conn.assign(:actor_paused, actor_paused)
|> Plug.Conn.assign(:authorizing, authorizing)
_ ->
conn
end
end
end
defp session_bool(value) do
case value do
"true" ->
true
"false" ->
false
"undefined" ->
false
boolean when is_boolean(boolean) ->
boolean
nil ->
false
end
end
defp actor_resources(apis) do
apis
|> Enum.flat_map(fn api ->
api
|> Ash.Api.Info.resources()
|> Enum.filter(fn resource ->
AshAdmin.Helpers.primary_action(resource, :read) && AshAdmin.Resource.actor?(resource)
end)
|> Enum.map(fn resource -> {api, resource} end)
end)
end
defp apis(otp_app) do
otp_app
|> Application.get_env(:ash_apis)
|> Enum.filter(&AshAdmin.Api.show?/1)
end
defp actor_api_from_session(endpoint, %{"actor_api" => api}) do
otp_app = endpoint.config(:otp_app)
apis = Application.get_env(otp_app, :ash_apis)
Enum.find(apis, fn allowed_api ->
AshAdmin.Api.show?(allowed_api) && AshAdmin.Api.name(allowed_api) == api
end)
end
defp actor_api_from_session(_, _), do: nil
defp actor_from_session(endpoint, %{
"actor_resource" => resource,
"actor_api" => api,
"actor_primary_key" => primary_key,
"actor_action" => action
})
when not is_nil(resource) and not is_nil(api) do
otp_app = endpoint.config(:otp_app)
apis = Application.get_env(otp_app, :ash_apis)
api =
Enum.find(apis, fn allowed_api ->
AshAdmin.Api.show?(allowed_api) && AshAdmin.Api.name(allowed_api) == api
end)
resource =
if api do
api
|> Ash.Api.Info.resources()
|> Enum.find(fn api_resource ->
AshAdmin.Resource.name(api_resource) == resource
end)
end
if api && resource do
action =
if action do
Ash.Resource.Info.action(resource, String.to_existing_atom(action), :read)
end
case decode_primary_key(resource, primary_key) do
:error ->
nil
{:ok, filter} ->
resource
|> Ash.Query.filter(^filter)
|> api.read_one!(action: action, authorize?: false)
end
end
end
defp actor_from_session(_, _), do: nil
end