33 lines
690 B
Ruby
33 lines
690 B
Ruby
|
namespace :gitlab do
|
||
|
namespace :shell do
|
||
|
desc "GITLAB | Setup gitlab-shell"
|
||
|
task :setup => :environment do
|
||
|
setup
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def setup
|
||
|
warn_user_is_not_gitlab
|
||
|
|
||
|
puts "This will rebuild an authorized_keys file."
|
||
|
puts "You will lose any data stored in /home/git/.ssh/authorized_keys."
|
||
|
ask_to_continue
|
||
|
puts ""
|
||
|
|
||
|
system("echo '# Managed by gitlab-shell' > /home/git/.ssh/authorized_keys")
|
||
|
|
||
|
Key.find_each(:batch_size => 1000) do |key|
|
||
|
if Gitlab::Shell.new.add_key(key.shell_id, key.key)
|
||
|
print '.'
|
||
|
else
|
||
|
print 'F'
|
||
|
end
|
||
|
end
|
||
|
|
||
|
rescue Gitlab::TaskAbortedByUserError
|
||
|
puts "Quitting...".red
|
||
|
exit 1
|
||
|
end
|
||
|
end
|
||
|
|