Merge pull request #28 from balena-os/ab77/patch

adds support for Azure VMs

balena.sh

@@ -11,9 +11,14 @@ for device in $devices; do
     fi
 done
 
+# 1. AWS
+# 2. DigitalOcean
+# 3. Azure
+# 4. Equinix
 metadata_urls=(
   'http://169.254.169.254/latest/user-data'
   'http://169.254.169.254/metadata/v1/user-data'
+  'http://169.254.169.254/metadata/instance/compute/userData?api-version=2023-07-01&format=text;-H Metadata:true'
   'https://metadata.platformequinix.com/userdata'
 )
 
@@ -36,10 +41,15 @@ function curl_with_opts() {
 
 function config_from_metadata() {
     #shellcheck disable=SC2034,SC2039 # /bin/sh is a symbolic link to bash on balenaOS
-    for url in "${metadata_urls[@]}"; do
-        user_data="$(curl_with_opts "${url}")"
-
-        if [ -n "${user_data}" ]; then
+    for metadata_url in "${metadata_urls[@]}"; do
+        url="$(echo "${metadata_url}" | awk -F';' '{print $1}')"
+        headers="$(echo "${metadata_url}" | awk -F';' '{print $2}')"
+        response="$(curl_with_opts ${headers} "${url}")"
+        user_data="$(echo "${response}" | base64 -d)"
+        if [ -z "${user_data}" ]; then
+            user_data="${response}"
+        fi
+        if [ -n "${user_data}" ] && echo "${user_data}" | jq -e . > /dev/null; then
             echo "${user_data}" | jq -r '.cloudConfig="done"'
             break
         fi