From 2f33daaaa7bce4ff06c5c96c70d26b1379df6cd5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christer=20War=C3=A9n?= <cwchristerw@gmail.com>
Date: Sun, 16 Jun 2024 04:06:43 +0300
Subject: [PATCH] Code Update

---
 src/commands/ssh/keys.sh | 2 ++
 wx                       | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/commands/ssh/keys.sh b/src/commands/ssh/keys.sh
index c66e08e..47975fa 100644
--- a/src/commands/ssh/keys.sh
+++ b/src/commands/ssh/keys.sh
@@ -94,7 +94,9 @@ wx-ssh-keys-retrieve(){
         if [[ $VAULT_STATUS -eq 200 ]]
         then
             echo $(curl https://$VAULT_DOMAIN/v1/cli/data/$USERNAME/settings/ssh/keys/$1 -X GET --header "X-Vault-Token: ${config["login",$ORG]}" -s | jq -r '.data.data.private') | base64 -d > ~/.ssh/keys/$1 2>&1
+            chmod 700 ~/.ssh/keys/$1
             echo $(curl https://$VAULT_DOMAIN/v1/cli/data/$USERNAME/settings/ssh/keys/$1 -X GET --header "X-Vault-Token: ${config["login",$ORG]}" -s | jq -r '.data.data.public') | base64 -d > ~/.ssh/keys/$1.pub 2>&1
+            chmod 700 ~/.ssh/keys/$1.pub
         fi
     fi
 }
diff --git a/wx b/wx
index 013f60c..a8a315e 100755
--- a/wx
+++ b/wx
@@ -585,7 +585,9 @@ wx-ssh-keys-retrieve(){
         if [[ $VAULT_STATUS -eq 200 ]]
         then
             echo $(curl https://$VAULT_DOMAIN/v1/cli/data/$USERNAME/settings/ssh/keys/$1 -X GET --header "X-Vault-Token: ${config["login",$ORG]}" -s | jq -r '.data.data.private') | base64 -d > ~/.ssh/keys/$1 2>&1
+            chmod 700 ~/.ssh/keys/$1
             echo $(curl https://$VAULT_DOMAIN/v1/cli/data/$USERNAME/settings/ssh/keys/$1 -X GET --header "X-Vault-Token: ${config["login",$ORG]}" -s | jq -r '.data.data.public') | base64 -d > ~/.ssh/keys/$1.pub 2>&1
+            chmod 700 ~/.ssh/keys/$1.pub
         fi
     fi
 }