wire in installer functions automatically

Author: rarescosma

tests/common/install-tools.bash

@@ -6,9 +6,11 @@ trap 'rm -rf $TMPDIR' EXIT
 
 cd "$TMPDIR"
 
-# --------- UNIT ----------
+# --------- Installers for the 'unit' stage ----------
+# NOTE: as long as a function is prefixed with 'unit_install_', it will be automatically
+# picked up by the parallel stage below.
 
-install_bats() {
+unit_install_bats() {
   _parse_version "${BATS_CORE_VERSION}"
   git clone -q --depth 1 https://github.com/bats-core/bats-core.git --branch "v${VERSION}" "${TMPDIR}/bats-core"
   "${TMPDIR}/bats-core/install.sh" /usr/local
@@ -22,23 +24,23 @@ install_bats() {
   echo "✓ bats installed"
 }
 
-install_docs() {
+unit_install_docs() {
   git clone -q --depth 1 --recurse-submodules --shallow-submodules https://github.com/elastisys/welkin.git "${DOCS_PATH}"
   python3 -m venv "${DOCS_PATH}/.venv"
   pushd "${DOCS_PATH}" >/dev/null
   ./.venv/bin/pip install --no-cache-dir --quiet -r requirements.txt
   echo "✓ docs installed"
 }
 
-install_gomplate() {
+unit_install_gomplate() {
   _parse_version "${GOMPLATE_VERSION}"
   curl -LOs "https://github.com/hairyhenderson/gomplate/releases/download/v${VERSION}/gomplate_linux-amd64"
   echo "${SHA}  gomplate_linux-amd64" | sha256sum -c -
   install -Tm 755 gomplate_linux-amd64 /usr/local/bin/gomplate
   echo "✓ gomplate installed"
 }
 
-install_helm() {
+unit_install_helm() {
   _parse_version "${HELM_VERSION}"
   curl -LOs "https://get.helm.sh/helm-v${VERSION}-linux-amd64.tar.gz"
   tar -zxf "helm-v${VERSION}-linux-amd64.tar.gz" linux-amd64/helm
@@ -55,7 +57,7 @@ install_helm() {
   echo "✓ helm-secrets installed"
 }
 
-install_helmfile() {
+unit_install_helmfile() {
   _parse_version "${HELMFILE_VERSION}"
   curl -LOs "https://github.com/helmfile/helmfile/releases/download/v${VERSION}/helmfile_${VERSION}_linux_amd64.tar.gz"
   tar -zxf "helmfile_${VERSION}_linux_amd64.tar.gz" helmfile
@@ -64,15 +66,15 @@ install_helmfile() {
   echo "✓ helmfile installed"
 }
 
-install_kubectl() {
+unit_install_kubectl() {
   _parse_version "${KUBECTL_VERSION}"
   curl -LOs "https://dl.k8s.io/release/v${VERSION}/bin/linux/amd64/kubectl"
   echo "${SHA}  kubectl" | sha256sum -c -
   install -Tm 755 kubectl /usr/local/bin/kubectl
   echo "✓ kubectl installed"
 }
 
-install_kubeconform() {
+unit_install_kubeconform() {
   _parse_version "${KUBECONFORM_VERSION}"
   curl -LOs "https://github.com/yannh/kubeconform/releases/download/v${VERSION}/kubeconform-linux-amd64.tar.gz"
   tar -zxf kubeconform-linux-amd64.tar.gz kubeconform
@@ -81,7 +83,7 @@ install_kubeconform() {
   echo "✓ kubeconform installed"
 }
 
-install_kubelogin() {
+unit_install_kubelogin() {
   _parse_version "${KUBELOGIN_VERSION}"
   curl -LOs "https://github.com/int128/kubelogin/releases/download/v${VERSION}/kubelogin_linux_amd64.zip"
   unzip -q kubelogin_linux_amd64.zip
@@ -90,15 +92,15 @@ install_kubelogin() {
   echo "✓ kubelogin installed"
 }
 
-install_opa() {
+unit_install_opa() {
   _parse_version "${OPA_VERSION}"
   curl -LOs "https://github.com/open-policy-agent/opa/releases/download/v${VERSION}/opa_linux_amd64"
   echo "${SHA}  opa_linux_amd64" | sha256sum -c -
   install -Tm 755 opa_linux_amd64 /usr/local/bin/opa
   echo "✓ opa installed"
 }
 
-install_promtool() {
+unit_install_promtool() {
   _parse_version "${PROMTOOL_VERSION}"
   curl -fsSL "https://github.com/prometheus/prometheus/releases/download/v${VERSION}/prometheus-${VERSION}.linux-amd64.tar.gz" |
     tar -zxf - "prometheus-${VERSION}.linux-amd64/promtool" --strip-components=1
@@ -107,41 +109,43 @@ install_promtool() {
   echo "✓ promtool installed"
 }
 
-install_sops() {
+unit_install_sops() {
   _parse_version "${SOPS_VERSION}"
   curl -LOs "https://github.com/getsops/sops/releases/download/v${VERSION}/sops-v${VERSION}.linux.amd64"
   echo "${SHA}  sops-v${VERSION}.linux.amd64" | sha256sum -c -
   install -Tm 755 "sops-v${VERSION}.linux.amd64" /usr/local/bin/sops
   echo "✓ sops installed"
 }
 
-install_yajsv() {
+unit_install_yajsv() {
   _parse_version "${YAJSV_VERSION}"
   curl -LOs "https://github.com/neilpa/yajsv/releases/download/v${VERSION}/yajsv.linux.amd64"
   echo "${SHA}  yajsv.linux.amd64" | sha256sum -c -
   install -Tm 755 yajsv.linux.amd64 /usr/local/bin/yajsv
   echo "✓ yajsv installed"
 }
 
-install_yq() {
+unit_install_yq() {
   _parse_version "${YQ_VERSION}"
   curl -LOs "https://github.com/mikefarah/yq/releases/download/v${VERSION}/yq_linux_amd64"
   echo "${SHA}  yq_linux_amd64" | sha256sum -c -
   install -Tm 755 yq_linux_amd64 /usr/local/bin/yq
   echo "✓ yq installed"
 }
 
-# --------- MAIN ----------
+# --------- Installers for the 'main' stage ----------
+# NOTE: as long as a function is prefixed with 'main_install_', it will be automatically
+# picked up by the parallel stage below.
 
-install_kind() {
+main_install_kind() {
   _parse_version "${KIND_VERSION}"
   curl -LOs "https://github.com/kubernetes-sigs/kind/releases/download/v${VERSION}/kind-linux-amd64"
   echo "${SHA}  kind-linux-amd64" | sha256sum -c -
   install -Tm 755 kind-linux-amd64 /usr/local/bin/kind
   echo "✓ kind installed"
 }
 
-install_velero() {
+main_install_velero() {
   _parse_version "${VELERO_VERSION}"
   curl -LOs "https://github.com/vmware-tanzu/velero/releases/download/v${VERSION}/velero-v${VERSION}-linux-amd64.tar.gz"
   tar -zxf "velero-v${VERSION}-linux-amd64.tar.gz" "velero-v${VERSION}-linux-amd64"
@@ -150,7 +154,7 @@ install_velero() {
   echo "✓ velero installed"
 }
 
-install_bun() {
+main_install_bun() {
   _parse_version "${BUN_VERSION}"
   curl -LOs "https://github.com/oven-sh/bun/releases/download/bun-v${VERSION}/bun-linux-x64.zip"
   unzip -q bun-linux-x64.zip
@@ -160,7 +164,7 @@ install_bun() {
   echo "✓ bun installed"
 }
 
-# --------- UTILS ----------
+# --------- Utility functions ----------
 
 _git_clone_revision() {
   if [[ $# -lt 3 ]]; then
@@ -184,47 +188,24 @@ _parse_version() {
   SHA="${1#*@sha256:}"
 }
 
-export -f _git_clone_revision _parse_version \
-  install_bats install_docs install_gomplate \
-  install_helm install_helmfile \
-  install_kubectl install_kubeconform install_kubelogin install_opa install_promtool \
-  install_sops install_yajsv install_yq \
-  install_kind install_velero install_bun
-export SHELLOPTS TMPDIR \
-  BATS_CORE_VERSION BATS_ASSERT_REV BATS_DETIK_REV BATS_FILE_REV BATS_MOCK_REV BATS_SUPPORT_REV \
-  DOCS_PATH GOMPLATE_VERSION \
-  HELM_DATA_HOME HELM_VERSION HELM_DIFF_VERSION HELM_SECRETS_VERSION HELMFILE_VERSION \
-  KUBECTL_VERSION KUBECONFORM_VERSION KUBELOGIN_VERSION OPA_VERSION PROMTOOL_VERSION \
-  SOPS_VERSION YAJSV_VERSION YAJSV_SHA YQ_VERSION YQ_SHA \
-  KIND_VERSION VELERO_VERSION BUN_VERSION
+# Re-export versions and revisions
+mapfile -t _version_and_revision_envs < <(compgen -v | grep -E '(_VERSION|_REV)$')
+export SHELLOPTS TMPDIR "${_version_and_revision_envs[@]}"
+
+# Export all utility + installer functions
+mapfile -t _unit_installer_fns < <(declare -F | grep -oP '.+\Kunit_install_.+')
+mapfile -t _main_installer_fns < <(declare -F | grep -oP '.+\Kmain_install_.+')
+export -f _git_clone_revision _parse_version "${_unit_installer_fns[@]}" "${_main_installer_fns[@]}"
 
 do_unit() {
   echo "Installing unit tools in parallel..."
-
   git config --global advice.detachedHead false
-
-  parallel -j "$(nproc)" --will-cite --halt now,fail=1 ::: \
-    install_bats \
-    install_docs \
-    install_gomplate \
-    install_helm \
-    install_helmfile \
-    install_kubectl \
-    install_kubeconform \
-    install_kubelogin \
-    install_opa \
-    install_promtool \
-    install_sops \
-    install_yajsv \
-    install_yq
+  parallel -j "$(nproc)" --will-cite --halt now,fail=1 ::: "${_unit_installer_fns[@]}"
 }
 
 do_main() {
   echo "Installing main tools in parallel..."
-  parallel -j "$(nproc)" --will-cite --halt now,fail=1 ::: \
-    install_kind \
-    install_velero \
-    install_bun
+  parallel -j "$(nproc)" --will-cite --halt now,fail=1 ::: "${_main_installer_fns[@]}"
 }
 
 "$@"