ci: look for arch-specific snapshots when necessary

Author: yaauie

.ci/docker-setup.sh

@@ -6,6 +6,46 @@ set -e
 
 VERSION_URL="https://raw.githubusercontent.com/elastic/logstash/master/ci/logstash_releases.json"
 
+download_and_load_docker_snapshot_artifact() {
+  project="${1}"
+  stack_version="${ELASTIC_STACK_VERSION}"
+  case "${DISTRIBUTION}" in
+    default) distribution_suffix="" ;; # default has no suffix
+          *) distribution_suffix="${DISTRIBUTION/*/-}${DISTRIBUTION}" ;;
+  esac
+
+  artifact_type="docker-image"
+  artifact_name_base="${project}${distribution_suffix}-${stack_version}-${artifact_type}"
+  echo "Downloading snapshot docker artifact: ${artifact_name_base}"
+
+  artifact_name_noarch="${artifact_name_base}.tar.gz"
+  artifact_name_arch="${artifact_name_base}-x86_64.tar.gz"
+
+  jq_extract_artifact_url=".build.projects.\"${project}\".packages | (.\"${artifact_name_noarch}\" // .\"${artifact_name_arch}\") | .url"
+
+  artifact_list=$(curl --silent "https://artifacts-api.elastic.co/v1/versions/${stack_version}/builds/latest")
+  artifact_url=$(echo "${artifact_list}" | jq --raw-output "${jq_extract_artifact_url}")
+
+  if [[ "${artifact_url}" == "null" ]]; then
+    echo "Failed to find '${artifact_name_noarch}'"
+    echo "Failed to find '${artifact_name_arch}'"
+    echo "Listing:"
+    echo "${artifact_list}" | jq --raw-output ".build.projects.\"${project}\".packages | keys | map(select(contains(\"${artifact_type}\")))"
+    return 1
+  fi
+
+  echo "${artifact_url}"
+
+  cd /tmp
+  curl "${artifact_url}" > "${project}-docker-image.tar.gz"
+  tar xfvz "${project}-docker-image.tar.gz" repositories
+  echo "Loading ${project} docker image: "
+  cat repositories
+  docker load < "${project}-docker-image.tar.gz"
+  rm "${project}-docker-image.tar.gz"
+  cd -
+}
+
 if [ "$ELASTIC_STACK_VERSION" ]; then
     echo "Fetching versions from $VERSION_URL"
     VERSIONS=$(curl --silent $VERSION_URL)
@@ -23,43 +63,12 @@ if [ "$ELASTIC_STACK_VERSION" ]; then
       export ELASTIC_STACK_VERSION=$ELASTIC_STACK_RETRIEVED_VERSION
     fi
 
-    if [[ "$DISTRIBUTION" = "oss" ]]; then
-      export DISTRIBUTION_SUFFIX="-oss"
-    else
-      export DISTRIBUTION_SUFFIX=""
-    fi
-
     echo "Testing against version: $ELASTIC_STACK_VERSION"
 
     if [[ "$ELASTIC_STACK_VERSION" = *"-SNAPSHOT" ]]; then
-        cd /tmp
-
-        jq=".build.projects.\"logstash\".packages.\"logstash$DISTRIBUTION_SUFFIX-$ELASTIC_STACK_VERSION-docker-image.tar.gz\".url"
-        result=$(curl --silent https://artifacts-api.elastic.co/v1/versions/$ELASTIC_STACK_VERSION/builds/latest | jq -r $jq)
-        echo $result
-        curl $result > logstash-docker-image.tar.gz
-        tar xfvz logstash-docker-image.tar.gz  repositories
-        echo "Loading docker image: "
-        cat repositories
-        docker load < logstash-docker-image.tar.gz
-        rm logstash-docker-image.tar.gz
-        cd -
-
+        download_and_load_docker_snapshot_artifact "logstash" || exit 1
         if [ "$INTEGRATION" == "true" ]; then
-
-          cd /tmp
-
-          jq=".build.projects.\"elasticsearch\".packages.\"elasticsearch$DISTRIBUTION_SUFFIX-$ELASTIC_STACK_VERSION-docker-image.tar.gz\".url"
-          result=$(curl --silent https://artifacts-api.elastic.co/v1/versions/$ELASTIC_STACK_VERSION/builds/latest | jq -r $jq)
-          echo $result
-          curl $result > elasticsearch-docker-image.tar.gz
-          tar xfvz elasticsearch-docker-image.tar.gz  repositories
-          echo "Loading docker image: "
-          cat repositories
-          docker load < elasticsearch-docker-image.tar.gz
-          rm elasticsearch-docker-image.tar.gz
-          cd -
-
+          download_and_load_docker_snapshot_artifact "elasticsearch" || exit 1
         fi
     fi