niveum/packages/niveum-ssh.nix

{
  symlinkJoin,
  writers,
  lib,
  netcat,
  openssh,
  try-connect,
}:
let
  inherit (lib.niveum) machines;
  sshableMachines = lib.filterAttrs (name: value: value ? "sshPort") machines;
in
symlinkJoin {
  name = "niveum-ssh";
  paths = lib.mapAttrsToList (
    hostname: _:
    writers.writeBashBin "niveum-ssh-${hostname}" ''
      reachable=$(${try-connect.${hostname}}/bin/try-connect)

      if [ -z "$reachable" ]; then
        exit 1
      fi

      if [[ "$reachable" == *.onion ]]; then
        exec ${openssh}/bin/ssh -p ${toString machines.${hostname}.sshPort} \
          -o ProxyCommand="${netcat}/bin/nc -x localhost:9050 %h %p" \
          "root@$reachable" "$@"
      else
        exec ${openssh}/bin/ssh -p ${toString machines.${hostname}.sshPort} \
          "root@$reachable" "$@"
      fi
    ''
  ) sshableMachines;
}