mirror of
https://github.com/philomena-dev/philomena.git
synced 2024-11-24 04:27:59 +01:00
43 lines
1.1 KiB
Elixir
43 lines
1.1 KiB
Elixir
|
defmodule PhilomenaWeb.Registration.TotpController do
|
||
|
|
use PhilomenaWeb, :controller
|
||
|
|
|
||
|
|
alias Philomena.Users.User
|
||
|
|
alias Philomena.Repo
|
||
|
|
|
||
|
|
def edit(conn, _params) do
|
||
|
|
user = conn.assigns.current_user
|
||
|
|
|
||
|
|
case user.encrypted_otp_secret do
|
||
|
|
nil ->
|
||
|
|
user
|
||
|
|
|> User.create_totp_secret_changeset()
|
||
|
|
|> Repo.update()
|
||
|
|
|
||
|
|
# Redirect to have Pow pick up the changes
|
||
|
|
redirect(conn, to: Routes.registration_totp_path(conn, :edit))
|
||
|
|
|
||
|
|
_ ->
|
||
|
|
changeset = Pow.Plug.change_user(conn)
|
||
|
|
render(conn, "edit.html", changeset: changeset)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def update(conn, params) do
|
||
|
|
backup_codes = User.random_backup_codes()
|
||
|
|
|
||
|
|
conn
|
||
|
|
|> Pow.Plug.current_user()
|
||
|
|
|> User.totp_changeset(params, backup_codes)
|
||
|
|
|> Repo.update()
|
||
|
|
|> case do
|
||
|
|
{:error, changeset} ->
|
||
|
|
render(conn, "edit.html", changeset: changeset)
|
||
|
|
|
||
|
|
{:ok, user} ->
|
||
|
|
conn
|
||
|
|
|> PhilomenaWeb.Plugs.TotpPlug.update_valid_totp_at_for_session(user)
|
||
|
|
|> put_flash(:totp_backup_codes, backup_codes)
|
||
|
|
|> redirect(to: Routes.registration_totp_path(conn, :edit))
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|