disko test

flake.nix

@@ -33,6 +33,7 @@
 
     stylix.url = "github:danth/stylix";
     base16.url = "github:SenchoPens/base16.nix";
+    disko.url = "github:nix-community/disko/latest";
 
     anyrun.url = "github:Kirottu/anyrun";
     oxicalc.url = "github:DashieTM/OxiCalc";
@@ -80,7 +81,8 @@
       };
       overlays = [
         inputs.nur.overlays.default
-        inputs.chaoticNyx.overlays.default];
+        inputs.chaoticNyx.overlays.default
+      ];
     };
     pkgs = import inputs.nixpkgs {
       system = currentSystem;

lib/default.nix

@@ -44,6 +44,7 @@
       nixos = [
         inputs.home-manager.nixosModules.home-manager
         inputs.stylix.nixosModules.stylix
+        inputs.disko.nixosModules.disko
         ../base
         ../home
         ../modules

modules/conf.nix

@@ -28,6 +28,14 @@
       '';
     };
 
+    defualtDiskId = lib.mkOption {
+      default = "TODO";
+      example = "/dev/disk/by-id/nvme-Force_MP510_19498249000129196385";
+      description = ''
+        The id of the disk to format
+      '';
+    };
+
     cpu = lib.mkOption {
       # TODO: how to enable arm?
       default = "amd";

modules/programs/drives.nix

@@ -3,35 +3,7 @@
   config,
   options,
   ...
-}: let
-  driveModule = lib.types.submodule {
-    options = {
-      name = lib.mkOption {
-        type = lib.types.str;
-        description = ''
-          The path of the drive.
-          Note that a / is already added at the beginning.
-        '';
-        default = "";
-        example = "drive2";
-      };
-      drive = lib.mkOption {
-        type = lib.types.attrsOf lib.types.anything;
-        description = "The attrs of the drive";
-        default = {};
-        example = {
-          device = "/dev/disk/by-label/DRIVE2";
-          fsType = "ext4";
-          options = [
-            "noatime"
-            "nodiratime"
-            "discard"
-          ];
-        };
-      };
-    };
-  };
-in {
+}: {
   options.mods = {
     drives = {
       useSwap = {
@@ -83,8 +55,72 @@ in {
 
   config = (
     lib.optionalAttrs (options ? fileSystems) {
-      fileSystems =
-        builtins.listToAttrs (
+      disko.devices = {
+        disk =
+          {
+            main = (lib.optionalAttrs config.mods.drives.defaultDrives.enable) {
+              device = "${config.conf.defaultDiskId}";
+              type = "disk";
+              content = {
+                type = "gpt";
+                partitions = {
+                  root = {
+                    start = "33G";
+                    end = "30%";
+                    content = {
+                      type = "filesystem";
+                      format = "btrfs";
+                      mountpoint = "/";
+                      mountOptions = [
+                        "noatime"
+                        "nodiratime"
+                        "discard"
+                      ];
+                    };
+                  };
+                  plainSwap = {
+                    start = "1G";
+                    end = "33G";
+                    content = {
+                      type = "swap";
+                      discardPolicy = "both";
+                      resumeDevice = true;
+                    };
+                  };
+                  boot = {
+                    start = "0G";
+                    end = "1G";
+                    content = {
+                      type = "filesystem";
+                      format = "vfat";
+                      mountpoint = "/boot";
+                      mountOptions = [
+                        "rw"
+                        "fmask=0022"
+                        "dmask=0022"
+                        "noatime"
+                      ];
+                    };
+                  };
+                  home = {
+                    start = "30%";
+                    end = "100%";
+                    content = {
+                      type = "filesystem";
+                      format = "btrfs";
+                      mountpoint = "/home";
+                      mountOptions = [
+                        "noatime"
+                        "nodiratime"
+                        "discard"
+                      ];
+                    };
+                  };
+                };
+              };
+            };
+          }
+          // builtins.listToAttrs (
             map (
               {
                 name,
@@ -95,43 +131,8 @@ in {
               }
             )
             config.mods.drives.extraDrives
-        )
-        // (lib.optionalAttrs config.mods.drives.defaultDrives.enable) {
-          "/" = {
-            device = "/dev/disk/by-label/ROOT";
-            fsType = "btrfs";
-            options = [
-              "noatime"
-              "nodiratime"
-              "discard"
-            ];
+          );
       };
-
-          "/boot" = {
-            device = "/dev/disk/by-label/BOOT";
-            fsType = "vfat";
-            options = [
-              "rw"
-              "fmask=0022"
-              "dmask=0022"
-              "noatime"
-            ];
-          };
-
-          "/home" = {
-            device = "/dev/disk/by-label/HOME";
-            fsType = "btrfs";
-            options = [
-              "noatime"
-              "nodiratime"
-              "discard"
-            ];
-          };
-        };
-      # TODO make this convert to choice of drives -> thanks to funny types this doesn't work...
-      swapDevices = lib.mkIf config.mods.drives.useSwap.enable [
-        {device = "/dev/disk/by-label/SWAP";}
-      ];
     }
   );
 }