Add output and confirmation to gitlab:two_factor:disable_for_all_users

This commit is contained in:
Robert Speicher 2015-10-08 23:19:56 -04:00
parent 5a28a5b726
commit cd46f4c367
1 changed files with 15 additions and 2 deletions

View File

@ -2,8 +2,21 @@ namespace :gitlab do
namespace :two_factor do
desc "GitLab | Disable Two-factor authentication (2FA) for all users"
task disable_for_all_users: :environment do
User.with_two_factor.find_each do |user|
user.disable_two_factor!
scope = User.with_two_factor
count = scope.count
if count > 0
puts "This will disable 2FA for #{count.to_s.red} users..."
begin
ask_to_continue
scope.find_each(&:disable_two_factor!)
puts "Successfully disabled 2FA for #{count} users.".green
rescue Gitlab::TaskAbortedByUserError
puts "Quitting...".red
end
else
puts "There are currently no users with 2FA enabled.".yellow
end
end
end