Parameterize key file

1 files changed, 3 insertions, 1 deletions

diff --git a/refresh_keys.sh b/refresh_keys.sh
index 01dc218..122b68d 100755
--- a/refresh_keys.sh
+++ b/refresh_keys.sh
@@ -2,6 +2,8 @@
 
 set -xeo pipefail
 
+auth_key_file=${1:-~/.ssh/authorized_keys}
+
 function main() {
     cd /tmp
     rm -rf /tmp/keys || true
@@ -16,7 +18,7 @@ function main() {
 
     sed -i '/emprespresso/d' ~/.ssh/authorized_keys && grep -qv "emprespresso" ~/.ssh/authorized_keys  || mkdir -p ~/.ssh && touch ~/.ssh/authorized_keys
     echo "+ Removed emprespresso keys"
-    cat authorized_keys >> ~/.ssh/authorized_keys
+    cat authorized_keys >> $auth_key_file
     echo "+ Appended emprespresso keys"
 
     cd /tmp