#!/usr/bin/env bash

# Copyright Materialize, Inc. and contributors. All rights reserved.
#
# Use of this software is governed by the Business Source License
# included in the LICENSE file at the root of this repository.
#
# As of the Change Date specified in that file, in accordance with
# the Business Source License, use of this software will be governed
# by the Apache License, Version 2.0.
#
# mkpipeline.sh — dynamically renders a pipeline.yml for Buildkite.

# This script's path is hardcoded into the Buildkite UI. It bootstraps the CI
# process by building the CI builder image, in which all other dependencies
# are installed. Unfortunately that means this script needs to be written in
# Bash, since our Python tools are only available once the CI builder image has
# been built.

set -euo pipefail

pipeline=${1:-test}
bootstrap_steps=
tmpfile=$(mktemp)

for arch in x86_64 aarch64; do
  for flavor in stable nightly min; do
    (
      if ! MZ_DEV_CI_BUILDER_ARCH=$arch bin/ci-builder exists $flavor; then
        echo "$arch:$flavor" >> "$tmpfile"
      fi
    ) &
  done
done
wait

while IFS=: read -r arch flavor; do
    queue=builder-linux-x86_64
    if [[ $arch == aarch64 ]]; then
        queue=builder-linux-aarch64-mem
    fi
    tag=$(MZ_DEV_CI_BUILDER_ARCH=$arch bin/ci-builder tag "$flavor")
    bootstrap_steps+="
  - label: bootstrap $flavor $arch
    command: bin/ci-builder exists $flavor || bin/ci-builder push $flavor
    concurrency: 1
    concurrency_group: \"ci-builder:$tag\"
    agents:
      queue: $queue
"
done < "$tmpfile"
rm "$tmpfile"

if [ -n "$bootstrap_steps" ]; then
  exec buildkite-agent pipeline upload <<EOF
steps:
  $bootstrap_steps
  - wait
  - label: mkpipeline
    env:
      CI_BAZEL_BUILD: 1
      CI_BAZEL_REMOTE_CACHE: "https://bazel-remote.dev.materialize.com"
    command: bin/ci-builder run min bin/pyactivate -m ci.mkpipeline $pipeline $@
    priority: 200
    agents:
      queue: hetzner-x86-64-4cpu-8gb-mkpipeline
    retry:
      automatic:
        - exit_status: -1
          signal_reason: none
          limit: 2
        - signal_reason: agent_stop
          limit: 2
EOF
else
  exec bin/ci-builder run min bin/pyactivate -m ci.mkpipeline "$pipeline" "$@"
fi