app/controllers/settings/two_factor_authentications_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

# frozen_string_literal: true

module Settings
  class TwoFactorAuthenticationsController < ApplicationController
    layout 'admin'

    before_action :authenticate_user!
    before_action :verify_otp_required, only: [:create]
    before_action :set_body_classes

    def show
      @confirmation = Form::TwoFactorConfirmation.new
    end

    def create
      current_user.otp_secret = User.generate_otp_secret(32)
      current_user.save!
      redirect_to new_settings_two_factor_authentication_confirmation_path
    end

    def destroy
      if acceptable_code?
        current_user.otp_required_for_login = false
        current_user.save!
        redirect_to settings_two_factor_authentication_path
      else
        flash.now[:alert] = I18n.t('two_factor_authentication.wrong_code')
        @confirmation = Form::TwoFactorConfirmation.new
        render :show
      end
    end

    private

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

    def verify_otp_required
      redirect_to settings_two_factor_authentication_path if current_user.otp_required_for_login?
    end

    def acceptable_code?
      current_user.validate_and_consume_otp!(confirmation_params[:code]) ||
        current_user.invalidate_otp_backup_code!(confirmation_params[:code])
    end

    def set_body_classes
      @body_classes = 'admin'
    end
  end
end