Code Update

src/commands/ssh/keys.sh

@@ -1,6 +1,5 @@
 wx-ssh-keys(){
-    echo "       >> SSH / Keys <<       "
-    echo "------------------------------"
+    wx-header "SSH / Keys"
 
     case $1 in
         generate)
@@ -28,6 +27,7 @@ wx-ssh-keys(){
 }
 
 wx-ssh-keys-generate(){
+    wx-restricted
     if [[ ! -z $1 ]]
     then
         if [[ ! -f "$HOME/.ssh/keys/$1" ]]
@@ -38,6 +38,7 @@ wx-ssh-keys-generate(){
 }
 
 wx-ssh-keys-retrieve(){
+    wx-restricted
     if [[ ! -z $1 ]]
     then
         VAULT_STATUS=$(curl -s -o /dev/null -w "%{http_code}" https://$VAULT_DOMAIN/v1/cli/data/$USERNAME/settings/ssh/keys/$1 -X GET --header "X-Vault-Token: ${config["login",$ORG]}")
@@ -50,6 +51,7 @@ wx-ssh-keys-retrieve(){
 }
 
 wx-ssh-keys-save(){
+    wx-restricted
     if [[ ! -z $1 ]]
     then
         if [[ -f "$HOME/.ssh/keys/$1" ]]
@@ -60,5 +62,6 @@ wx-ssh-keys-save(){
 }
 
 wx-ssh-keys-sync(){
+    wx-restricted
     echo ""
 }