From 4d31fbbe2af3b2c5b64bc07d3a09de7b35b562ef Mon Sep 17 00:00:00 2001
From: 514fpv <koishi@514fpv.one>
Date: Tue, 9 Jan 2024 13:17:47 +0800
Subject: [PATCH] feat(fs): add f2fs and zfs

---
 global/fs/default.nix |  5 ++++-
 global/fs/f2fs.nix    | 10 ++++++++++
 global/fs/zfs.nix     | 41 +++++++++++++++++++++++++++++++++++++++++
 3 files changed, 55 insertions(+), 1 deletion(-)
 create mode 100644 global/fs/f2fs.nix
 create mode 100644 global/fs/zfs.nix

diff --git a/global/fs/default.nix b/global/fs/default.nix
index f5b718cf..f5c2569b 100644
--- a/global/fs/default.nix
+++ b/global/fs/default.nix
@@ -6,19 +6,22 @@
 in {
   imports = [
     ./ext4.nix
+    ./f2fs.nix
     ./xfs.nix
+    ./zfs.nix
     #./bcachefs.nix
     ./btrfs.nix
   ];
 
   options.global.fs = {
     type = mkOption {
-      type = with types; enum [ "ext4" "xfs" "bcachefs" "btrfs" ];
+      type = with types; enum [ "ext4" "f2fs" "xfs" "zfs" "bcachefs" "btrfs" ];
       default = "bcachefs";
       description = "filesystem type to use for persistent state storage";
     };
     store = mkOption {
       type = with types; str;
+      default = config.networking.hostName;
       description = "UUID/dataset of nix store backing device";
     };
     esp = {
diff --git a/global/fs/f2fs.nix b/global/fs/f2fs.nix
new file mode 100644
index 00000000..f9043a39
--- /dev/null
+++ b/global/fs/f2fs.nix
@@ -0,0 +1,10 @@
+{ lib
+, config
+, ... }: with lib; let
+  cfg = config.global.fs;
+in mkIf (cfg.type == "f2fs") {
+  fileSystems."/nix" =
+  { device = "/dev/disk/by-uuid/${cfg.store}";
+    fsType = "f2fs";
+  };
+}
diff --git a/global/fs/zfs.nix b/global/fs/zfs.nix
new file mode 100644
index 00000000..beec8db5
--- /dev/null
+++ b/global/fs/zfs.nix
@@ -0,0 +1,41 @@
+{ pkgs
+, lib
+, config
+, ... }: with lib; let
+  cfg = config.global.fs;
+in {
+  options.global.fs.zfs = {
+    persist = mkOption {
+      type = with types; str;
+      default = cfg.store;
+      description = ''
+        pool for persist dataset
+        defaults to nix store dataset
+      '';
+    };
+  };
+
+  config = mkIf (cfg.type == "zfs") {
+    fileSystems = {
+      "/nix" =
+        { device = "${cfg.store}/nix";
+          fsType = "zfs";
+        };
+      "/nix/persist" =
+        { device = "${cfg.zfs.persist}/persist";
+          fsType = "zfs";
+          neededForBoot = true;
+        };
+    } // (mapAttrs' (name: opts: nameValuePair
+    "/nix/persist/home/${name}" {
+      device = "${cfg.zfs.persist}/home/${name}";
+      fsType = "zfs";
+      neededForBoot = true;
+    }) (filterAttrs (n: _: n != "root") config.users.profiles));
+
+    services.zfs.trim.enable = true;
+    services.zfs.autoSnapshot.enable = true;
+    services.zfs.autoScrub.enable = true;
+    boot.zfs.devNodes = mkDefault "/dev/disk/by-partuuid";
+  };
+}