50 lines
1.2 KiB
Ruby
50 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# == EnforcesAdminAuthentication
|
|
#
|
|
# Controller concern to enforce that users are authenticated as admins
|
|
#
|
|
# Upon inclusion, adds `authenticate_admin!` as a before_action
|
|
#
|
|
module EnforcesAdminAuthentication
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
before_action :authenticate_admin!
|
|
|
|
def self.authorize!(ability, only:)
|
|
actions = Array(only)
|
|
|
|
skip_before_action :authenticate_admin!, only: actions
|
|
before_action -> { authorize_ability!(ability) }, only: actions
|
|
end
|
|
end
|
|
|
|
def authenticate_admin!
|
|
return render_404 unless user_is_admin?
|
|
return unless Gitlab::CurrentSettings.admin_mode
|
|
|
|
unless current_user_mode.admin_mode?
|
|
current_user_mode.request_admin_mode!
|
|
store_location_for(:redirect, request.fullpath) if storable_location?
|
|
redirect_to(new_admin_session_path, notice: _('Re-authentication required'))
|
|
end
|
|
end
|
|
|
|
def storable_location?
|
|
request.path != new_admin_session_path
|
|
end
|
|
|
|
private
|
|
|
|
def user_is_admin?
|
|
current_user.admin?
|
|
end
|
|
|
|
def authorize_ability!(ability)
|
|
return authenticate_admin! if current_user.admin?
|
|
|
|
render_404 unless current_user.can?(ability)
|
|
end
|
|
end
|