[nixos/disk] experiment with autosizing zfs partitions in initrd

2 files changed, 58 insertions, 0 deletions

diff --git a/machines/feneri.nix b/machines/feneri.nix
new file mode 100644
index 0000000..d731a36
--- /dev/null
+++ b/machines/feneri.nix
@@ -0,0 +1,19 @@
+{ flake, modulesPath, ... }:
+
+{
+  imports = [
+    flake.outputs.nixosModules.disk
+    (modulesPath + "/profiles/qemu-guest.nix")
+  ];
+
+  nixpkgs.hostPlatform.system = "x86_64-linux";
+
+  depot.disk = {
+    enable = true;
+    # device = "/dev/vda";
+  };
+
+  services.getty.autologinUser = "root";
+
+  networking.hostName = "feneri";
+}
diff --git a/modules/disk.nix b/modules/disk.nix
index 071c9ea..757d436 100644
--- a/modules/disk.nix
+++ b/modules/disk.nix
@@ -61,6 +61,7 @@ in
       kernelParams = [
         "nohibernate"
         "elevator=none"
+        "rd.systemd.unit=rescue.target"
       ];
       supportedFilesystems = [
         "vfat"
@@ -115,6 +116,44 @@ in
       };
     };
 
+    boot.initrd.systemd.services."growroot" = {
+      description = "Grow main partition into any unused disk space.";
+      serviceConfig.Type = "oneshot";
+
+      # If dry-run fails, there is nothing to grow
+      serviceConfig.ExecCondition = "${pkgs.cloud-utils}/bin/growpart /dev/vda 2 --dry-run";
+
+      wantedBy = [ "zfs.target" ];
+      before = [ "zfs-import-zroot.service" ];
+
+      script = "growpart /dev/vda 2 --verbose";
+    };
+
+    boot.initrd.systemd.services."expand-zroot" = {
+      description = "Expand zroot to take up entire partition.";
+      serviceConfig.Type = "oneshot";
+
+      wantedBy = ["zfs.target"];
+      after = ["zfs-import-zroot.service"];
+
+      script = ''
+        # pull partuuid from zpool
+        GUID=$(zpool list -v --json | jq --raw-output '.pools.zroot.vdevs.[].name')
+
+        # re-mark pool as online with '-e' to expand
+        zpool online -e zroot $GUID
+      '';
+    };
+
+    boot.initrd.systemd.emergencyAccess = true;
+    boot.initrd.systemd.initrdBin = [
+      pkgs.cloud-utils
+      pkgs.util-linux
+      pkgs.busybox
+      pkgs.jq
+      pkgs.zfs
+    ];
+
     image.modules.depot =
       { config, modulesPath, ... }:
       {