nix-config/hosts/celaeno/services/hydra/machines.nix

{ config, pkgs, ... }:
let
  buildMachinesFile = (import ./lib/mk-build-machines-file.nix) [
    {
      uri = "ssh://nix-ssh@atlas";
      systems = [ "x86_64-linux" "i686-linux" ];
      sshKey = config.sops.secrets.nix-ssh-key.path;
      maxJobs = 12;
      speedFactor = 150;
      supportedFeatures = [ "kvm" "big-parallel" "nixos-test" ];
    }
    {
      uri = "ssh://nix-ssh@maia";
      systems = [ "x86_64-linux" "i686-linux" ];
      sshKey = config.sops.secrets.nix-ssh-key.path;
      maxJobs = 8;
      speedFactor = 100;
      supportedFeatures = [ "kvm" "big-parallel" "nixos-test" ];
    }
    {
      uri = "localhost";
      systems = [ "aarch64-linux" "x86_64-linux" "i686-linux" ];
      maxJobs = 4;
      # This is the only builder marked as aarch64, so these builds will always
      # run here (regardless of speedFactor).
      # As for x86_64, this machine's factor is much lower than the others, so
      # these builds will only be picked up if the others are offline.
      speedFactor = 1;
      supportedFeatures = [ "kvm" "big-parallel" "nixos-test" ];
    }
  ];
in {
  services.hydra.buildMachinesFiles = [ "/etc/nix/hydra-machines" ];


  systemd = {
    timers.builder-pinger = {
      description = "Build machine pinger timer";
      partOf = [ "builder-pinger.service" ];
      wantedBy = [ "multi-user.target" ];
      timerConfig = {
        OnBootSec = "0";
        OnUnitActiveSec = "30s";
      };
    };
    services.builder-pinger = {
      description = "Build machine pinger";
      enable = true;
      wantedBy = [ "multi-user.target" "post-resume.target" ];
      serviceConfig = {
        Type = "oneshot";
        Restart = "no";
      };
      path = [
        config.nix.package
        config.programs.ssh.package
        pkgs.diffutils
        pkgs.coreutils
      ];
      script = /* bash */ ''
        set -euo pipefail

        final_file="/etc/nix/hydra-machines"
        temp_file="$(mktemp)"

        check_host() {
          line="$1"
          host="$(echo "$line" | cut -d ' ' -f1)"
          key="$(echo "$line" | cut -d ' ' -f3)"

          if [ "$key" == "-" ]; then
              args=""
          else
              args="ssh-key=$key"
          fi
          if [ "$host" == "localhost" ]; then
              host="local"
          fi

          if timeout 20 nix store ping  --store "$host?$args"; then
              echo "$line" >> $temp_file
          fi
        }

        while read -r host_line; do
          check_host "$host_line" &
        done < "${buildMachinesFile}"

        wait

        touch "$final_file"
        if ! diff <(sort "$temp_file") <(sort "$final_file"); then
          mv "$temp_file" "$final_file"
          chmod 755 "$final_file"
          touch "$final_file" # So that hydra-queue-runner refreshes
        fi
      '';
    };
  };
}