philomena/lib/philomena_web/plugs/current_ban_plug.ex
2020-06-03 21:26:38 -04:00

52 lines
1.2 KiB
Elixir

defmodule PhilomenaWeb.CurrentBanPlug do
@moduledoc """
This plug loads the ban for the current user.
## Example
plug PhilomenaWeb.CurrentBanPlug
"""
alias Philomena.Bans
alias Plug.Conn
alias Pow.Plug
@doc false
@spec init(any()) :: any()
def init(opts), do: opts
@doc false
@spec call(Conn.t(), any()) :: Conn.t()
def call(conn, _opts) do
conn =
conn
|> Conn.fetch_cookies()
fingerprint = conn.cookies["_ses"]
user = Plug.current_user(conn)
ip = conn.remote_ip
ban = Bans.exists_for?(user, ip, fingerprint)
cond do
discourage?(ban) ->
Conn.register_before_send(conn, fn conn ->
:timer.sleep(normal_time())
pass(error?(), conn)
end)
|> Conn.assign(:current_ban, nil)
true ->
Conn.assign(conn, :current_ban, ban)
end
end
defp discourage?(%{note: note}) when is_binary(note), do: String.contains?(note, "discourage")
defp discourage?(_ban), do: false
defp normal_time, do: :rand.normal(5_000, 25_000_000) |> trunc() |> max(0)
defp error?, do: :rand.uniform() < 0.05
defp pass(false, conn), do: conn
defp pass(_true, _conn), do: nil
end