gitlab-org--gitlab-foss/app/controllers/profiles/two_factor_auths_controller.rb

72 lines
1.9 KiB
Ruby
Raw Normal View History

class Profiles::TwoFactorAuthsController < Profiles::ApplicationController
skip_before_action :check_2fa_requirement
2015-12-18 15:29:13 -05:00
def new
2015-03-27 20:53:08 -04:00
unless current_user.otp_secret
2015-05-23 18:47:53 -04:00
current_user.otp_secret = User.generate_otp_secret(32)
2015-03-27 20:53:08 -04:00
end
2015-12-23 21:02:52 -05:00
unless current_user.otp_grace_period_started_at && two_factor_grace_period
current_user.otp_grace_period_started_at = Time.current
end
2015-12-23 21:02:52 -05:00
current_user.save! if current_user.changed?
2015-03-27 20:53:08 -04:00
if two_factor_grace_period_expired?
flash.now[:alert] = 'You must enable Two-factor Authentication for your account.'
else
grace_period_deadline = current_user.otp_grace_period_started_at + two_factor_grace_period.hours
flash.now[:alert] = "You must enable Two-factor Authentication for your account before #{l(grace_period_deadline)}."
end
2015-03-27 20:53:08 -04:00
@qr_code = build_qr_code
end
def create
if current_user.validate_and_consume_otp!(params[:pin_code])
current_user.two_factor_enabled = true
@codes = current_user.generate_otp_backup_codes!
2015-03-27 20:53:08 -04:00
current_user.save!
render 'create'
2015-03-27 20:53:08 -04:00
else
@error = 'Invalid pin code'
@qr_code = build_qr_code
2015-03-27 20:53:08 -04:00
render 'new'
end
end
2015-03-31 15:42:17 -04:00
def codes
@codes = current_user.generate_otp_backup_codes!
2015-03-31 15:42:17 -04:00
current_user.save!
end
def destroy
current_user.disable_two_factor!
redirect_to profile_account_path
end
2015-03-27 20:53:08 -04:00
2015-12-23 21:02:52 -05:00
def skip
if two_factor_grace_period_expired?
2015-12-23 21:02:52 -05:00
redirect_to new_profile_two_factor_auth_path, alert: 'Cannot skip two factor authentication setup'
else
session[:skip_tfa] = current_user.otp_grace_period_started_at + two_factor_grace_period.hours
redirect_to root_path
end
end
2015-03-27 20:53:08 -04:00
private
def build_qr_code
issuer = "#{issuer_host} | #{current_user.email}"
2015-03-27 20:53:08 -04:00
uri = current_user.otp_provisioning_uri(current_user.email, issuer: issuer)
RQRCode::render_qrcode(uri, :svg, level: :m, unit: 3)
end
def issuer_host
Gitlab.config.gitlab.host
end
end