|
| 1 | +defmodule Hawk.Authority.Session do |
| 2 | + @moduledoc """ |
| 3 | + Small helpers for carrying `Hawk.Authority` through web sessions and assigns. |
| 4 | +
|
| 5 | + Hawk does not own application authentication. Apps resolve their current user |
| 6 | + however they like, then store the resulting authority with these stable keys so |
| 7 | + controllers and LiveViews can share the same convention. |
| 8 | + """ |
| 9 | + |
| 10 | + alias Hawk.Authority |
| 11 | + |
| 12 | + @default_key :hawk_authority |
| 13 | + |
| 14 | + @doc """ |
| 15 | + Returns Hawk's default assign/session key for authorities. |
| 16 | + """ |
| 17 | + def default_key, do: @default_key |
| 18 | + |
| 19 | + @doc """ |
| 20 | + Converts an authority into a session-safe map. |
| 21 | + """ |
| 22 | + def dump(%Authority{} = authority) do |
| 23 | + %{ |
| 24 | + "role" => authority.role, |
| 25 | + "identity" => authority.identity, |
| 26 | + "readonly?" => authority.readonly?, |
| 27 | + "system?" => authority.system?, |
| 28 | + "public?" => authority.public?, |
| 29 | + "scopes" => authority.scopes, |
| 30 | + "meta" => authority.meta |
| 31 | + } |
| 32 | + end |
| 33 | + |
| 34 | + @doc """ |
| 35 | + Rebuilds an authority from `dump/1` output. |
| 36 | + """ |
| 37 | + def load(%{"role" => role, "identity" => identity} = data) do |
| 38 | + %Authority{ |
| 39 | + role: normalize_atom(role), |
| 40 | + identity: identity, |
| 41 | + readonly?: Map.get(data, "readonly?", false), |
| 42 | + system?: Map.get(data, "system?", false), |
| 43 | + public?: Map.get(data, "public?", false), |
| 44 | + scopes: normalize_atom_keys(Map.get(data, "scopes", %{})), |
| 45 | + meta: Map.get(data, "meta", %{}) |
| 46 | + } |
| 47 | + end |
| 48 | + |
| 49 | + def load(%{role: role, identity: identity} = data) do |
| 50 | + %Authority{ |
| 51 | + role: normalize_atom(role), |
| 52 | + identity: identity, |
| 53 | + readonly?: Map.get(data, :readonly?, false), |
| 54 | + system?: Map.get(data, :system?, false), |
| 55 | + public?: Map.get(data, :public?, false), |
| 56 | + scopes: normalize_atom_keys(Map.get(data, :scopes, %{})), |
| 57 | + meta: Map.get(data, :meta, %{}) |
| 58 | + } |
| 59 | + end |
| 60 | + |
| 61 | + def load(nil), do: nil |
| 62 | + |
| 63 | + @doc """ |
| 64 | + Stores an authority under the configured assign key on a socket/conn/map. |
| 65 | + """ |
| 66 | + def assign_authority(socket_or_conn, %Authority{} = authority, key \\ @default_key) do |
| 67 | + assign_value(socket_or_conn, key, authority) |
| 68 | + end |
| 69 | + |
| 70 | + @doc """ |
| 71 | + Fetches an authority from assigns or session-like maps. |
| 72 | + """ |
| 73 | + def fetch_authority(source, key \\ @default_key) do |
| 74 | + source |
| 75 | + |> fetch_value(key) |
| 76 | + |> case do |
| 77 | + %Authority{} = authority -> {:ok, authority} |
| 78 | + nil -> :error |
| 79 | + dumped -> {:ok, load(dumped)} |
| 80 | + end |
| 81 | + end |
| 82 | + |
| 83 | + @doc """ |
| 84 | + Fetches an authority or returns `Hawk.Authority.public()`. |
| 85 | + """ |
| 86 | + def authority_or_public(source, key \\ @default_key) do |
| 87 | + case fetch_authority(source, key) do |
| 88 | + {:ok, authority} -> authority |
| 89 | + :error -> Authority.public() |
| 90 | + end |
| 91 | + end |
| 92 | + |
| 93 | + defp assign_value(%{assigns: assigns} = value, key, authority) when is_map(assigns) do |
| 94 | + put_in(value.assigns, Map.put(assigns, key, authority)) |
| 95 | + end |
| 96 | + |
| 97 | + defp assign_value(value, key, authority) when is_map(value) do |
| 98 | + Map.put(value, key, authority) |
| 99 | + end |
| 100 | + |
| 101 | + defp fetch_value(%{assigns: assigns}, key) when is_map(assigns), do: Map.get(assigns, key) |
| 102 | + defp fetch_value(%{} = map, key), do: Map.get(map, key) || Map.get(map, to_string(key)) |
| 103 | + |
| 104 | + defp normalize_atom(value) when is_atom(value), do: value |
| 105 | + defp normalize_atom(value) when is_binary(value), do: String.to_existing_atom(value) |
| 106 | + |
| 107 | + defp normalize_atom_keys(map) when is_map(map) do |
| 108 | + Map.new(map, fn {key, value} -> {normalize_atom(key), value} end) |
| 109 | + end |
| 110 | +end |
0 commit comments