app/controllers/settings/two_factor_auths_controller.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

# frozen_string_literal: true

class Settings::TwoFactorAuthsController < ApplicationController
  layout 'admin'

  before_action :authenticate_user!

  def show; end

  def new
    redirect_to settings_two_factor_auth_path if current_user.otp_required_for_login

    @confirmation = Form::TwoFactorConfirmation.new
    current_user.otp_secret = User.generate_otp_secret(32)
    current_user.save!
    set_qr_code
  end

  def create
    if current_user.validate_and_consume_otp!(confirmation_params[:code])
      current_user.otp_required_for_login = true
      @codes = current_user.generate_otp_backup_codes!
      current_user.save!
      flash[:notice] = I18n.t('two_factor_auth.enabled_success')
    else
      @confirmation = Form::TwoFactorConfirmation.new
      set_qr_code
      flash.now[:alert] = I18n.t('two_factor_auth.wrong_code')
      render action: :new
    end
  end

  def recovery_codes
    @codes = current_user.generate_otp_backup_codes!
    current_user.save!
    flash[:notice] = I18n.t('two_factor_auth.recovery_codes_regenerated')
  end

  def disable
    current_user.otp_required_for_login = false
    current_user.save!

    redirect_to settings_two_factor_auth_path
  end

  private

  def set_qr_code
    @provision_url = current_user.otp_provisioning_uri(current_user.email, issuer: Rails.configuration.x.local_domain)
    @qrcode        = RQRCode::QRCode.new(@provision_url)
  end

  def confirmation_params
    params.require(:form_two_factor_confirmation).permit(:code)
  end
end