lib/disko: add more filesystem functions

lib/disko.nix

@@ -1,4 +1,8 @@
 { lib }:
+let
+  inherit (builtins) mapAttrs removeAttrs;
+  inherit (lib) optionals;
+in
 rec {
   # Constants
 
@@ -17,17 +21,52 @@ rec {
     } // content;
   };
 
+  mkESP' = mountOptions: size: mountpoint: {
+    inherit size;
+    type = "EF00";
+    content = {
+      type = "filesystem";
+      format = "vfat";
+      inherit mountpoint mountOptions;
+    };
+  };
+  mkESP = mkESP' defaultMountOptions;
+
+  mkTmpfs' = mountOptions: size: mode: {
+    fsType = "tmpfs";
+    mountOptions = mountOptions ++ [ "size=${size}" "mode=${mode}" ];
+  };
+  mkTmpfs = size: mkTmpfs' defaultMountOptions size "755";
+
+  # btrfs
+
+  mkBtrfsPart = size: mountpoint: content: {
+    inherit size;
+    content = {
+      inherit mountpoint;
+      type = "btrfs";
+    } // content;
+  };
+
+  mkBtrfsSubvols' = mountOptions: mapAttrs (n: v: {
+    mountpoint = n;
+    mountOptions = mountOptions ++ (optionals (v ? mountOptions) v.mountOptions);
+  } // (removeAttrs v ["mountOptions"]));
+  mkBtrfsSubvols = mkBtrfsSubvols' defaultMountOptions;
+
   # ZFS
 
-  mkZDisk = id: pool: mkDisk id {
+  mkZPart = size: pool: {
-    partitions = {
+    inherit size;
-      zfs = {
-        size = "100%";
     content = {
       type = "zfs";
       inherit pool;
     };
   };
+
+  mkZDisk = id: pool: mkDisk id {
+    partitions = {
+      zfs = mkZPart "100%" pool;
     };
   };