From ac8aac0e349beebbf3082db385f547fc58efab52 Mon Sep 17 00:00:00 2001
From: Malte Poll <1780588+malt3@users.noreply.github.com>
Date: Thu, 16 Nov 2023 14:41:34 +0100
Subject: [PATCH] ci: allow jobs to install tools from pinned nixpkgs (#2605)

---
 .github/actions/setup_bazel_nix/action.yml | 48 ++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/.github/actions/setup_bazel_nix/action.yml b/.github/actions/setup_bazel_nix/action.yml
index 286ccf2e9..7ce0202fe 100644
--- a/.github/actions/setup_bazel_nix/action.yml
+++ b/.github/actions/setup_bazel_nix/action.yml
@@ -12,6 +12,10 @@ inputs:
   rbePlatform:
     description: "RBE platform to use. If empty, RBE will not be used."
     required: false
+  nixTools:
+    description: "Nix tools to install as list of strings separated by newlines. If empty, no tools will be installed."
+    default: ""
+    required: false
 
 runs:
   using: "composite"
@@ -267,3 +271,47 @@ runs:
         echo "common --disk_cache=" >> "${WORKSPACE}/.bazeloverwriterc"
         echo "common --repository_cache=" >> "${WORKSPACE}/.bazeloverwriterc"
         echo "::endgroup::"
+
+    - name: Install nix tools
+      if: inputs.nixTools != ''
+      shell: bash
+      env:
+        tools: ${{ inputs.nixTools }}
+        repository: ${{ github.repository }}
+        gitSha: ${{ github.sha }}
+      run: |
+        echo "::group::Install nix tools"
+        toolsNixList=$(printf ' "%s"' ${tools[@]})
+        toolsNixList="[ ${toolsNixList} ]"
+        expressionFile=$(mktemp)
+        cat << "EOF" > "${expressionFile}"
+        { tools, repository, rev }:
+        let
+          repoFlake = builtins.getFlake ("github:" + repository + "/" + rev);
+          nixpkgs = repoFlake.inputs.nixpkgsUnstable;
+          pkgs = import nixpkgs { system = builtins.currentSystem; };
+          toolPkgs = map (p: pkgs.${p}) tools;
+        in
+        {
+          tools = pkgs.symlinkJoin { name = "tools"; paths = [ toolPkgs ]; };
+          pathVar = pkgs.lib.makeBinPath toolPkgs;
+        }
+        EOF
+        # ensure the store paths are created
+        nix-build \
+          --no-out-link \
+          --arg tools "${toolsNixList}" \
+          --argstr repository "${repository}" \
+          --argstr rev "${gitSha}" \
+          --attr tools \
+          "${expressionFile}"
+        # evaluate the path expression
+        # EXTRA_PATH=/nix/store/...:/nix/store/...:/nix/store/...
+        EXTRA_PATH=$(nix eval --raw --file "${expressionFile}" \
+          --arg tools "${toolsNixList}" \
+          --argstr repository "${repository}" \
+          --argstr rev "${gitSha}" \
+          pathVar)
+        echo "EXTRA_PATH=${EXTRA_PATH}"
+        echo "${EXTRA_PATH}" >> "${GITHUB_PATH}"
+        echo "::endgroup::"