Improve sandbox deploy script

Author: nhinsch

scripts/publish_sandbox.sh

@@ -5,22 +5,52 @@
 # This product includes software developed at Datadog (https://www.datadoghq.com/).
 # Copyright 2021 Datadog, Inc.
 
-# Usage: VERSION=5 ARCHITECTURE=amd64 ./scripts/publish_sandbox.sh
+# Usage: VERSION=5 ARCHITECTURE=[amd64|arm64] ./scripts/publish_sandbox.sh
+
+# VERSION is optional. By default, increment the version by 1.
+# ARCHITECTURE is optional. The default is amd64.
 
 set -e
 
-# Move into the root directory
-SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
-cd $SCRIPTS_DIR/..
+if [ -z $ARCHITECTURE ]; then
+    echo "No architecture specified, defaulting to amd64"
+    ARCHITECTURE="amd64"
+fi
 
 if [ "$ARCHITECTURE" != "amd64" -a "$ARCHITECTURE" != "arm64" ]; then
-    echo "ERROR: For sandbox deploys, you must specify an architecture (amd64 or arm64)"
+    echo "ERROR: Invalid architecture (must be amd64 OR arm64 for sandbox deploys)"
     exit 1
 fi
 
-./scripts/build_binary_and_layer_dockerized.sh
-REGIONS=sa-east-1 aws-vault exec sandbox-account-admin -- ./scripts/publish_layers.sh
+if [ "$ARCHITECTURE" == "amd64" ]; then
+    LAYER_NAME="Datadog-Extension"
+fi
+if [ "$ARCHITECTURE" == "arm64" ]; then
+    LAYER_NAME="Datadog-Extension-ARM"
+fi
+
+if [ -z $VERSION ]; then
+    echo "No version specified, automatically incrementing version number"
+
+    LAST_LAYER_VERSION=$(
+        aws-vault exec sandbox-account-admin -- \
+        aws lambda list-layer-versions \
+            --layer-name $LAYER_NAME \
+            --region sa-east-1 \
+        | jq -r ".LayerVersions | .[0] |  .Version" \
+    )
+    if [ "$LAST_LAYER_VERSION" == "null" ]; then
+        echo "Error auto-detecting the last layer version"
+        exit 1
+    else
+        VERSION=$(($LAST_LAYER_VERSION+1))
+        echo "Will publish new layer version as $VERSION"
+    fi
+fi
+
+# Move into the root directory
+SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
+cd $SCRIPTS_DIR/..
 
-# Automatically create PR against github.com/DataDog/documentation
-# If you'd like to test, please uncomment the below line
-# VERSION=$VERSION LAYER=datadog-lambda-extension ./scripts/create_documentation_pr.sh
+VERSION=$VERSION ARCHITECTURE=$ARCHITECTURE ./scripts/build_binary_and_layer_dockerized.sh
+VERSION=$VERSION ARCHITECTURE=$ARCHITECTURE REGIONS=sa-east-1 aws-vault exec sandbox-account-admin -- ./scripts/publish_layers.sh