mirror of
https://github.com/philomena-dev/philomena.git
synced 2024-12-18 06:57:59 +01:00
39 lines
930 B
Elixir
39 lines
930 B
Elixir
defmodule PhilomenaWeb.TotpPlug do
|
|
@moduledoc """
|
|
This plug ensures that a user session has a valid TOTP.
|
|
|
|
## Example
|
|
|
|
plug PhilomenaWeb.TotpPlug
|
|
"""
|
|
|
|
alias PhilomenaWeb.Router.Helpers, as: Routes
|
|
|
|
@doc false
|
|
@spec init(any()) :: any()
|
|
def init(opts), do: opts
|
|
|
|
@doc false
|
|
@spec call(Plug.Conn.t(), any()) :: Plug.Conn.t()
|
|
def call(conn, _opts) do
|
|
case conn.assigns.current_user do
|
|
nil -> conn
|
|
user -> maybe_require_totp_phase(user, conn)
|
|
end
|
|
end
|
|
|
|
defp maybe_require_totp_phase(%{otp_required_for_login: nil}, conn), do: conn
|
|
defp maybe_require_totp_phase(%{otp_required_for_login: false}, conn), do: conn
|
|
|
|
defp maybe_require_totp_phase(_user, conn) do
|
|
case conn.assigns.totp_valid? do
|
|
true ->
|
|
conn
|
|
|
|
_falsy ->
|
|
conn
|
|
|> Phoenix.Controller.redirect(to: Routes.session_totp_path(conn, :new))
|
|
|> Plug.Conn.halt()
|
|
end
|
|
end
|
|
end
|