# 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
