* Fix #795, fix #704, fix #835 - 2FA requires confirmation to be enabled TOTP secret is not shown again after 2FA is enabled * Clean up
		
			
				
	
	
		
			50 lines
		
	
	
	
		
			1.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			50 lines
		
	
	
	
		
			1.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # 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
 | |
|       current_user.save!
 | |
| 
 | |
|       redirect_to settings_two_factor_auth_path, 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 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
 |