nixos/modules/base/ssh.nix

{
  config,
  pkgs,
  username,
  hostname,
  lib,
  ...
}: let
  # Load all public keys from ../../secrets/hosts/*/id_ed25519.pub
  allHosts = lib.attrNames (lib.filterAttrs (_: type: type == "directory") (builtins.readDir ../../secrets/hosts));
  publicKeys = map (x: builtins.readFile (../../secrets/hosts + ("/" + x) + "/id_ed25519.pub")) allHosts;
in {
  age.secrets.ssh-key = {
    rekeyFile = ../../secrets/hosts + ("/" + hostname) + /id_ed25519.age;
    path = "/home/" + username + "/.ssh/id_ed25519";
    mode = "0600";
    owner = username;
  };
  age.secrets.ssh-key-ao = {
    rekeyFile = ../../secrets/ssh-key-ao.age;
    mode = "0600";
    owner = username;
  };

  fireproof.home-manager = {
    home.file.".ssh/id_ed25519.pub".source = ../../secrets/hosts + ("/" + hostname) + "/id_ed25519.pub";
    programs.ssh = {
      enable = true;
      forwardAgent = true;
      serverAliveInterval = 60;
      serverAliveCountMax = 10;
      matchBlocks = {
        "*" = {
          identityFile = "${config.age.secrets.ssh-key.path}";
        };
        homelab = {
          hostname = "x.nickolaj.com";
          user = "nickolaj";
        };
        # Work hostnames definded in ./networking.nix
        "bastion.ao" = {
          user = "nij";
          identityFile = "${config.age.secrets.ssh-key-ao.path}";
        };
        "clickhouse.ao" = {
          user = "ubuntu";
          hostname = "51.158.205.48";
          identityFile = "${config.age.secrets.ssh-key-ao.path}";
        };
        "flex.ao" = {
          user = "nij";
          hostname = "192.168.2.5";
          proxyJump = "bastion.ao";
          identityFile = "${config.age.secrets.ssh-key-ao.path}";
        };
        "scw.ao" = {
          user = "nij";
          hostname = "51.15.81.1";
          proxyJump = lib.mkDefault "dev.ao";
          identityFile = "${config.age.secrets.ssh-key-ao.path}";
        };
        "dev.ao" = {
          user = "nij";
          hostname = "192.168.2.28";
          proxyJump = lib.mkDefault "bastion.ao";
          identityFile = "${config.age.secrets.ssh-key-ao.path}";
        };
        "staging.ao" = {
          user = "staging";
          hostname = "172.16.2.102";
          proxyJump = lib.mkDefault "bastion.ao";
          identityFile = "${config.age.secrets.ssh-key-ao.path}";
        };
      };
    };
  };

  programs.ssh.startAgent = true;
  services.openssh = {
    enable = true;
    settings.PasswordAuthentication = false;
    settings.KbdInteractiveAuthentication = false;
  };

  systemd.user.services."add-ssh-keys" = {
    description = "Add SSH keys to ssh-agent";
    after = ["network.target" "ssh-agent.service"];
    requires = ["ssh-agent.service"];
    wantedBy = ["default.target"];
    serviceConfig = {
      Type = "oneshot";
      ExecStartPre = ''
        ${pkgs.coreutils}/bin/sleep 5
      '';
      ExecStart = ''
        ${pkgs.openssh}/bin/ssh-add -q ${config.age.secrets.ssh-key-ao.path}
      '';
    };
  };

  users.users.${username}.openssh.authorizedKeys.keys = publicKeys;
}