Added support for args and stdin to provision.sh

scripts/provision.sh

@@ -2,22 +2,32 @@
 
 set -e
 
-# Provision a remote server using a Bash script
-# Usage example: ./provision.sh provisioning/ user@hostname -p2222
+# Provision a remote server using a directory with a Bash script
+# Usage example:
+#   ./provision.sh provisioning/ '--arg01 --arg02' ssh user@hostname -p2222
 
-cd "$1"; shift
+prov_dir="$1"; shift
+remote_args="$1"; shift
 
-if [ ! -f main.sh ]; then
+if [ ! -f "$prov_dir/main.sh" ]; then
     echo 'File main.sh not found' >&2
     exit 1
 fi
 
-tar -cvzf- . | ssh "$@" '
+# Operations are split in two separate connections because we want the
+# "$prov_dir/main.sh" script to be able to read from our stdin
+
+tar -cvzf- -C "$prov_dir" . | "$@" '
     set -ex
     rm -rf /tmp/provisioning
     mkdir /tmp/provisioning
     tar -xvzf- -C/tmp/provisioning
-    bash /tmp/provisioning/main.sh || result="$?"
+'
+
+# shellcheck disable=SC2016
+"$@" '
+    set -ex
+    bash /tmp/provisioning/main.sh '"$remote_args"' || result="$?"
     rm -rf /tmp/provisioning
     exit "${result:-0}"
 '