2020-10-16 14:09:04 -04:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
module Users
|
|
|
|
class ValidateOtpService < BaseService
|
2020-12-10 07:09:43 -05:00
|
|
|
include ::Gitlab::Auth::Otp::Fortinet
|
|
|
|
|
2020-10-16 14:09:04 -04:00
|
|
|
def initialize(current_user)
|
|
|
|
@current_user = current_user
|
2020-12-10 07:09:43 -05:00
|
|
|
@strategy = if forti_authenticator_enabled?(current_user)
|
2020-10-16 14:09:04 -04:00
|
|
|
::Gitlab::Auth::Otp::Strategies::FortiAuthenticator.new(current_user)
|
2020-12-10 07:09:43 -05:00
|
|
|
elsif forti_token_cloud_enabled?(current_user)
|
|
|
|
::Gitlab::Auth::Otp::Strategies::FortiTokenCloud.new(current_user)
|
2020-10-16 14:09:04 -04:00
|
|
|
else
|
|
|
|
::Gitlab::Auth::Otp::Strategies::Devise.new(current_user)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def execute(otp_code)
|
|
|
|
strategy.validate(otp_code)
|
|
|
|
rescue StandardError => ex
|
|
|
|
Gitlab::ErrorTracking.log_exception(ex)
|
|
|
|
error(message: ex.message)
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
attr_reader :strategy
|
|
|
|
end
|
|
|
|
end
|