integrate VM definitions

flake.nix

@@ -39,6 +39,7 @@
       lib = pkgs.lib;
       pkgs-unstable = nixpkgs-unstable.legacyPackages.${system};
       machines = import ./nixos/machines;
+      physicalMachines = lib.filterAttrs (n: v: v.type == "physical") machines;
       # TODO: Maybe use mergeAttrLists
       mkNixosSystems = systemDef:
         nixpkgs.lib.foldlAttrs
@@ -47,12 +48,12 @@
               "${name}" = nixpkgs.lib.nixosSystem (systemDef machine);
             })
           { }
-          machines;
+          physicalMachines;
       mkDeployNodes = nodeDef:
         nixpkgs.lib.foldlAttrs
           (acc: name: machine: acc // { "${name}" = nodeDef machine; })
           { }
-          machines;
+          physicalMachines;
     in
     {
       devShells.${system}.default = pkgs.mkShell {
@@ -76,15 +77,8 @@
 
       nixosConfigurations = mkNixosSystems (machine: {
         inherit system;
-        specialArgs = { inherit kubenix dns; };
+        specialArgs = { inherit machines machine kubenix dns microvm disko agenix; };
-        modules = [
+        modules = [ ./nixos ];
-          microvm.nixosModules.host
-          machine.nixosModule
-          disko.nixosModules.disko
-          agenix.nixosModules.default
-          ./nixos
-          { networking.hostName = machine.name; }
-        ];
       });
 
       deploy = {
@@ -92,10 +86,11 @@
         user = "root";
 
         nodes = mkDeployNodes (machine: {
-          hostname = machine.hostName;
+          # TODO: simply get this from nixos configuration?
+          hostname = "${machine.hostName}.${machine.domain}";
           profiles.system = {
             path = deploy-rs.lib.${system}.activate.nixos
-              self.nixosConfigurations.${machine.name};
+              self.nixosConfigurations.${machine.hostName};
           };
         });
       };
@@ -103,56 +98,5 @@
       checks = builtins.mapAttrs
         (system: deployLib: deployLib.deployChecks self.deploy)
         deploy-rs.lib;
-
-      apps.${system} =
-        let
-          hostNames = builtins.concatStringsSep " " (builtins.map (host: "\"${host.config.networking.fqdn}\"") (builtins.attrValues self.nixosConfigurations));
-        in
-        {
-          reboot-all-vms =
-            let
-              reboot-all-vms = pkgs.writeScriptBin "reboot-all-vms" ''
-                hostNames=(${hostNames})
-                for hostName in ''${hostNames[@]}; do
-                  units=$(${pkgs.openssh}/bin/ssh root@$hostName systemctl list-units --all)
-                  microvmUnits=$(${pkgs.coreutils}/bin/echo "$units" | ${pkgs.gnugrep}/bin/grep 'microvm@.*\.service' | ${pkgs.gawkInteractive}/bin/awk '{print $1}')
-                  if [ -n "$microvmUnits" ]; then
-                    for microvmUnit in "$microvmUnits"; do
-                      ${pkgs.coreutils}/bin/echo "Restarting $microvmUnit on $hostName"
-                      ${pkgs.openssh}/bin/ssh root@$hostName systemctl restart "$microvmUnit"
-                    done
-                  fi
-                done
-              '';
-            in
-            {
-              type = "app";
-              program = "${reboot-all-vms}/bin/reboot-all-vms";
-            };
-
-          reboot-vm =
-            let
-              reboot-vm = pkgs.writeScriptBin "reboot-vm" ''
-                if [ -z "$1" ]; then
-                  ${pkgs.coreutils}/bin/echo "Please provide a VM name!"
-                  exit 1
-                fi
-
-                hostNames=(${hostNames})
-                unitName="microvm@$1.service"
-                for hostName in ''${hostNames[@]}; do
-                  units=$(${pkgs.openssh}/bin/ssh root@$hostName systemctl list-units --all)
-                  if [[ "''${units[@]}" =~ "$unitName" ]]; then
-                    ${pkgs.coreutils}/bin/echo "Restarting $unitName on $hostName"
-                    ${pkgs.openssh}/bin/ssh root@$hostName systemctl restart "$unitName"
-                  fi
-                done
-              '';
-            in
-            {
-              type = "app";
-              program = "${reboot-vm}/bin/reboot-vm";
-            };
-        };
     };
 }

nixos/default.nix

@@ -1,10 +1,15 @@
-{ pkgs, config, lib, modulesPath, ... }: {
+{ pkgs, config, lib, modulesPath, machine, microvm, disko, agenix, machines, ... }: {
   imports = [
     (modulesPath + "/installer/scan/not-detected.nix")
     ./modules
     ./lab.nix
+    machine.nixosModule
+    disko.nixosModules.disko
+    agenix.nixosModules.default
+    microvm.nixosModules.host
   ];
 
+  config = {
     boot = {
       kernelModules = [ "kvm-intel" ];
       extraModulePackages = [ ];
@@ -116,8 +121,7 @@
       radvd
     ];
 
-  hardware.cpu.intel.updateMicrocode =
+    hardware.cpu.intel.updateMicrocode = config.hardware.enableRedistributableFirmware;
-    lib.mkDefault config.hardware.enableRedistributableFirmware;
 
     age.identityPaths = [ "/etc/age_ed25519" ];
 
@@ -139,4 +143,26 @@
         fi
       '';
     };
+
+    microvm.vms =
+      let
+        vmsForHypervisor = lib.attrValues (lib.filterAttrs (n: v: v.type == "virtual" && v.hypervisorName == machine.hostName) machines);
+      in
+      lib.attrsets.mergeAttrsList (map
+        (vm:
+          {
+            "${vm.hostName}" = {
+              # TODO Simplify?
+              specialArgs = { inherit agenix disko pkgs lib microvm; machine = vm; hypervisorConfig = config; };
+              config = {
+                imports = [
+                  ./vm.nix
+                ];
+              };
+            };
+          }
+        )
+        vmsForHypervisor
+      );
+  };
 }

nixos/machines/default.nix

@@ -1,7 +1,9 @@
 {
   jefke = {
-    name = "jefke";
+    # TODO: directly set and read this from nixos config.
-    hostName = "jefke.hyp";
+    hostName = "jefke";
+    domain = "hyp";
+    type = "physical";
 
     nixosModule.lab = {
       terraformDatabase.enable = true;
@@ -20,8 +22,9 @@
   };
 
   atlas = {
-    name = "atlas";
+    hostName = "atlas";
-    hostName = "atlas.hyp";
+    domain = "hyp";
+    type = "physical";
 
     nixosModule = { config, ... }:
       let inherit (config.lab.networking) dmzServicesIPv4 dmzServicesIPv6; in
@@ -49,8 +52,9 @@
   };
 
   lewis = {
-    name = "lewis";
+    hostName = "lewis";
-    hostName = "lewis.hyp";
+    domain = "hyp";
+    type = "physical";
 
     nixosModule = { pkgs, ... }: {
       lab = {
@@ -67,35 +71,21 @@
           userCert = builtins.readFile ./lewis_user_ed25519-cert.pub;
         };
       };
+    };
+  };
 
-      microvm.vms.my-microvm.config = {
+  my-microvm = {
-        services.openssh.enable = true;
+    hostName = "my-microvm";
-        networking.firewall.enable = false;
+    domain = "dmz";
-
+    type = "virtual";
-        users.users.root.openssh.authorizedKeys.keys = [
+    hypervisorName = "lewis";
-          "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIOodpLr+FDRyKyHjucHizNLVFHZ5AQmE9GmxMnOsSoaw pimkunis@thinkpadpim"
-          "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINUZp4BCxf7uLa1QWonx/Crf8tYZ5MKIZ+EuaBa82LrV user@user-laptop"
-        ];
 
+    nixosModule = { pkgs, config, ... }: {
       programs.bash.interactiveShellInit = ''
         echo "Hello world from inside a virtual machine!" | ${pkgs.lolcat}/bin/lolcat
       '';
 
-        microvm = {
+      lab.vmMacAddress = "BA:DB:EE:F0:00:00";
-          shares = [{
-            source = "/nix/store";
-            mountPoint = "/nix/.ro-store";
-            tag = "ro-store";
-            proto = "virtiofs";
-          }];
-
-          interfaces = [{
-            type = "tap";
-            id = "vm-my-microvm";
-            mac = "48:2D:63:E1:C5:39";
-          }];
-        };
-      };
     };
   };
 }

nixos/modules/networking/default.nix

@@ -1,4 +1,4 @@
-{ lib, config, ... }:
+{ lib, config, machine, ... }:
 let cfg = config.lab.networking;
 in {
   imports = [ ./dmz ];
@@ -89,17 +89,18 @@ in {
 
   config = {
     networking = {
-      domain = "hyp";
+      hostName = machine.hostName;
+      domain = machine.domain;
+      nftables.enable = true;
+      useDHCP = machine.type == "virtual";
+
       firewall = {
         enable = true;
         checkReversePath = false;
       };
-
-      nftables.enable = true;
-      useDHCP = false;
     };
 
-    systemd.network = {
+    systemd.network = lib.mkIf (machine.type == "physical") {
       enable = true;
 
       netdevs = {

nixos/modules/storage.nix

@@ -1,4 +1,4 @@
-{ lib, config, ... }:
+{ lib, config, machine, ... }:
 let cfg = config.lab.storage;
 in {
   options.lab.storage = {
@@ -25,7 +25,7 @@ in {
     };
   };
 
-  config = {
+  config = lib.mkIf (machine.type == "physical") {
     fileSystems.${cfg.dataMountPoint}.device = cfg.dataPartition;
 
     # TODO: Rename this to 'osDisk'. Unfortunately, we would need to run nixos-anywhere again then.

nixos/vm.nix

@@ -0,0 +1,119 @@
+{ pkgs, lib, config, agenix, disko, machine, hypervisorConfig, ... }: {
+  imports = [
+    ./modules
+    ./lab.nix
+    machine.nixosModule
+    disko.nixosModules.disko
+    agenix.nixosModules.default
+  ];
+
+  options.lab.vmMacAddress = lib.mkOption {
+    type = lib.types.str;
+    description = ''
+      The MAC address of the VM's main NIC.
+    '';
+  };
+
+  # TODO: remove overlap with physical nixos module
+  # Perhaps a sane defaults module?
+  config = {
+    system.stateVersion = hypervisorConfig.system.stateVersion;
+    time.timeZone = "Europe/Amsterdam";
+
+    i18n = {
+      defaultLocale = "en_US.UTF-8";
+
+      extraLocaleSettings = {
+        LC_ADDRESS = "nl_NL.UTF-8";
+        LC_IDENTIFICATION = "nl_NL.UTF-8";
+        LC_MEASUREMENT = "nl_NL.UTF-8";
+        LC_MONETARY = "nl_NL.UTF-8";
+        LC_NAME = "nl_NL.UTF-8";
+        LC_NUMERIC = "nl_NL.UTF-8";
+        LC_PAPER = "nl_NL.UTF-8";
+        LC_TELEPHONE = "nl_NL.UTF-8";
+        LC_TIME = "nl_NL.UTF-8";
+      };
+    };
+
+    services = {
+      openssh = {
+        enable = true;
+        openFirewall = true;
+
+        settings = {
+          PasswordAuthentication = false;
+          KbdInteractiveAuthentication = false;
+        };
+      };
+
+      xserver = {
+        layout = "us";
+        xkbVariant = "";
+      };
+    };
+
+    users.users.root.openssh.authorizedKeys.keys = [
+      "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIOodpLr+FDRyKyHjucHizNLVFHZ5AQmE9GmxMnOsSoaw pimkunis@thinkpadpim"
+      "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINUZp4BCxf7uLa1QWonx/Crf8tYZ5MKIZ+EuaBa82LrV user@user-laptop"
+    ];
+
+    programs = {
+      ssh = {
+        knownHosts = {
+          dmz = {
+            hostNames = [ "*.dmz" ];
+            publicKey =
+              "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIAX2IhgHNxC6JTvLu9cej+iWuG+uJFMXn4AiRro9533x";
+            certAuthority = true;
+          };
+
+          hypervisors = {
+            hostNames = [ "*.hyp" ];
+            publicKey =
+              "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIFzRkH3d/KVJQouswY/DMpenWbDFVOnI3Vut0xR0e1tb";
+            certAuthority = true;
+          };
+        };
+
+      };
+
+      neovim = {
+        enable = true;
+        vimAlias = true;
+        viAlias = true;
+      };
+    };
+
+    environment.systemPackages = with pkgs; [
+      neofetch
+      wget
+      git
+      btop
+      htop
+      ripgrep
+      dig
+      tree
+      file
+      tcpdump
+      lsof
+      parted
+      radvd
+    ];
+
+    microvm = {
+      shares = [{
+        source = "/nix/store";
+        mountPoint = "/nix/.ro-store";
+        tag = "ro-store";
+        proto = "virtiofs";
+      }];
+
+      interfaces = [{
+        type = "tap";
+        id = "vm-${machine.hostName}";
+        mac = config.lab.vmMacAddress;
+      }];
+    };
+  };
+}