mirror of
https://github.com/philomena-dev/philomena.git
synced 2024-11-27 13:47:58 +01:00
49 lines
1.3 KiB
Elixir
49 lines
1.3 KiB
Elixir
|
defmodule PhilomenaWeb.AdminCountersPlug do
|
||
|
|
@moduledoc """
|
||
|
|
This plug stores the counts used by the admin bar.
|
||
|
|
## Example
|
||
|
|
plug PhilomenaWeb.AdminCountersPlug
|
||
|
|
"""
|
||
|
|
|
||
|
|
alias Plug.Conn
|
||
|
|
alias Philomena.DuplicateReports
|
||
|
|
alias Philomena.Reports
|
||
|
|
alias Philomena.UserLinks
|
||
|
|
alias Philomena.DnpEntries
|
||
|
|
|
||
|
|
import Plug.Conn, only: [assign: 3]
|
||
|
|
|
||
|
|
@doc false
|
||
|
|
@spec init(any()) :: any()
|
||
|
|
def init(opts), do: opts
|
||
|
|
|
||
|
|
@doc false
|
||
|
|
@spec call(Plug.Conn.t()) :: Plug.Conn.t()
|
||
|
|
def call(conn), do: call(conn, nil)
|
||
|
|
|
||
|
|
@doc false
|
||
|
|
@spec call(Plug.Conn.t(), any()) :: Plug.Conn.t()
|
||
|
|
def call(conn, _opts) do
|
||
|
|
user = conn.assigns.current_user
|
||
|
|
|
||
|
|
maybe_assign_admin_metrics(conn, staff?(user))
|
||
|
|
end
|
||
|
|
|
||
|
|
defp maybe_assign_admin_metrics(conn, false), do: conn
|
||
|
|
defp maybe_assign_admin_metrics(conn, true) do
|
||
|
|
duplicate_reports = DuplicateReports.count_duplicate_reports()
|
||
|
|
reports = Reports.count_reports()
|
||
|
|
user_links = UserLinks.count_user_links()
|
||
|
|
dnps = DnpEntries.count_dnp_entries()
|
||
|
|
|
||
|
|
conn
|
||
|
|
|> assign(:duplicate_report_count, duplicate_reports)
|
||
|
|
|> assign(:report_count, reports)
|
||
|
|
|> assign(:user_link_count, user_links)
|
||
|
|
|> assign(:dnp_entry_count, dnps)
|
||
|
|
end
|
||
|
|
|
||
|
|
defp staff?(%{role: role}) when role in ["assistant", "moderator", "admin"], do: true
|
||
|
|
defp staff?(_user), do: false
|
||
|
|
end
|