platform/clusters/stokes/default.nix

let
  stokes = {
    node.address = "10.1.62.2";
    node.myvnc = true;

    features = {
      os = {
        externalInterface = "eno1";
        nfs.enable = true;
        nfs.exports = ''
          /exports 10.1.61.0/24(insecure,rw,sync,no_subtree_check,crossmnt,fsid=0,no_root_squash)
        '';
      };

      hpc = {
        slurm.server = true;
        frontend = true;
      };

      monitoring = {
        server = {
          enable = true;
          scrapeHosts = [
            "frontend" "mds0-0"
            "c0-1" "c0-2" "c0-3" "c0-4" "c0-5" "c0-6" "c0-7" "c0-8"
          ];
          defaultAlertReceiver = {
            email_configs = [
              { to = "jonas.juselius@tromso.serit.no"; }
            ];
          };
          pageAlertReceiver = {
            webhook_configs = [
              {
                url = "https://prometheus-msteams.k2.itpartner.no/stokes";
                http_config = {
                  tls_config = { insecure_skip_verify = true; };
                };
              }
            ];
          };
        };
        webUI.enable = true;
        webUI.acmeEmail = "innovasjon@itpartner.no";
        webUI.allow = [
            "10.1.2.0/24"
            "172.19.254.0/24"
        ];
        infiniband-exporter = {
          enable = true;
          nameMap = ''
            0x0c42a10300ddc4bc "frontend"
            0x1c34da0300787798 "mds0-0"
            0x0c42a10300dbe7f4 "c0-1"
            0x0c42a10300dbe7d8 "c0-2"
            0x0c42a10300dbe800 "c0-3"
            0x0c42a10300dbec80 "c0-4"
            0x0c42a10300dbea50 "c0-5"
            0x0c42a10300dbeb2c "c0-6"
            0x0c42a10300dbe7fc "c0-7"
            0x0c42a10300dbe5a0 "c0-8"
          '';
        };
        slurm-exporter = {
          enable = true;
          port = 6080;
        };
      };
    };

    networking = {
      interfaces.eno1 = {
        useDHCP = false;
        ipv4.addresses = [ {
          address = "10.1.62.2";
          prefixLength = 24;
        } ];
      };
      interfaces.enp175s0f0 = {
        useDHCP = false;
        ipv4.addresses = [ {
          address = "10.1.61.100";
          prefixLength = 24;
        } ];
      };
      interfaces.ibp59s0 = {
        useDHCP = false;
        ipv4.addresses = [ {
          address = "10.1.63.100";
          prefixLength = 24;
        } ];
      };
      defaultGateway = "10.1.62.1";
      firewall.extraCommands = ''
        iptables -I INPUT -s 10.1.61.0/24 -j ACCEPT
        iptables -t nat -A POSTROUTING -s 10.1.63.0/24 -j MASQUERADE
      '';
    };

    fileSystems ={
      "/exports/home" = {
        device = "/home";
        options = [ "bind" ];
      };
      "/exports/opt" = {
        device = "/opt";
        options = [ "bind" ];
      };
    };

    security.pam.services.sshd.googleAuthenticator.enable = true;

    nix.extraOptions = ''
      secret-key-files = /etc/nix/stokes.private
    '';

    services.xserver = {
      enable = true;
      enableCtrlAltBackspace = true;
      layout = "us";
      xkbVariant = "altgr-intl";
      xkbOptions = "eurosign:e";
      displayManager = {
        gdm.enable = true;
        job.logToFile = true;
      };
      desktopManager.xfce.enable = true;
    };

    services.prometheus.alertmanager.configuration.global = {
        smtp_smarthost = "smtpgw.itpartner.no:465";
        smtp_auth_username = "utvikling";
        smtp_auth_password = "S0m3rp0m@de#21!";
        smtp_hello = "stokes.regnekraft.io";
        smtp_from = "noreply@stokes.regnekraft.io";
    };

    imports = [ ./cluster.nix ./hw/frontend.nix ];
  };

  compute = {
    features = {
      os.externalInterface = "eno33";
      hpc.compute = true;
    };

    fileSystems = {
      "/home" = {
        device = "10.1.61.100:/home";
        fsType = "nfs";
      };
      "/opt" = {
        device = "10.1.61.100:/opt";
        fsType = "nfs";
      };
    };
  };

  genComputeNodes = idx: nNodes:
    let
      nodeList = builtins.genList (x: x + 1) nNodes;
      mkCompute = n:
        let
           ip    = "10.1.61.${toString (n + 100)}";
           ipoib = "10.1.63.${toString (n + 100)}";
           name = "c${toString idx}-${toString n}";
           hw = ./hw + "/${name}.nix";
        in {
          "${name}" = {
            node = {
              address = ip;
              i40efix = true;
            };
            networking = {
              interfaces.eno33 = {
                useDHCP = false;
                ipv4.addresses = [ {
                  address = ip;
                  prefixLength = 24;
                } ];
                ipv4.routes = [ {
                  address = "10.1.62.2";
                  prefixLength = 32;
                  via = "10.1.61.100";
                } ];

              };
              interfaces.ibp65s0 = {
                useDHCP = false;
                ipv4.addresses = [ {
                  address = ipoib;
                  prefixLength = 24;
                } ];
              };
            };
            imports = [ ./cluster.nix hw ];
          } // compute;
        };
    in
      builtins.foldl' (a: n: a // mkCompute n) {} nodeList;
in
  { inherit stokes; } // genComputeNodes 0 8