52 lines
1.2 KiB
Ruby
52 lines
1.2 KiB
Ruby
|
require_relative 'collab'
|
||
|
|
require 'tty-prompt'
|
||
|
|
|
||
|
|
$prompt = TTY::Prompt.new
|
||
|
|
|
||
|
|
|
||
|
|
def init
|
||
|
|
choices = %w(Keys Users Quit)
|
||
|
|
case $prompt.select("What do you want to do?", choices)
|
||
|
|
when "Users"
|
||
|
|
users
|
||
|
|
when "Keys"
|
||
|
|
keys
|
||
|
|
when "Quit"
|
||
|
|
exit
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def keys
|
||
|
|
choices = $users
|
||
|
|
user = $prompt.select("Pick a user", choices)
|
||
|
|
keys_user(user)
|
||
|
|
end
|
||
|
|
|
||
|
|
def keys_user(user)
|
||
|
|
choices = {'List keys' => :list ,'Add key' => :add, 'Remove key' => :remove, 'Reset authorized_keys' => :reset}
|
||
|
|
case $prompt.select("What do you want to do for user #{user}?", choices)
|
||
|
|
when :list
|
||
|
|
get_ssh_pubkeys(user).each{|k| puts k.pretty}
|
||
|
|
keys_user user
|
||
|
|
when :add
|
||
|
|
key = $prompt.ask("Please enter a valid public SSH key", echo: false)
|
||
|
|
add_ssh_pubkey(user,key)
|
||
|
|
when :remove
|
||
|
|
keys = get_ssh_pubkeys(user)#.map{|k| puts k.pretty}
|
||
|
|
key = $prompt.select("Select the key to delete", keys)
|
||
|
|
unless $prompt.no?("Are you sure you want to delete this key?")
|
||
|
|
remove_ssh_pubkey(user,key)
|
||
|
|
end
|
||
|
|
when :reset
|
||
|
|
unless $prompt.no?("Are you sure you want to reset the keys for this user?")
|
||
|
|
unless $prompt.no?("Really sure?")
|
||
|
|
reset_pubkeys(user)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
while true
|
||
|
|
init
|
||
|
|
end
|