1 files changed, 24 insertions, 13 deletions

diff --git a/system-profiles/tmpfs-root.nix b/system-profiles/tmpfs-root.nix
index 23939c2e..7840dadf 100644
--- a/system-profiles/tmpfs-root.nix
+++ b/system-profiles/tmpfs-root.nix
@@ -1,14 +1,25 @@
-{ ... }: {
-  config = {
-    # system.etc.overlay.enable = true;
-
-    fileSystems."/" = {
-      fsType = "tmpfs";
-      options = [ "mode=0755" ];
-    };
-
-    security.sudo.extraConfig = ''
-      Defaults lecture = never
-    '';
-  };
+{ options, lib, ... }: {
+  config = lib.mkMerge
+    [ (lib.mkIf (options ? disko) {
+        disko.devices.nodev = {
+          "/" = {
+            fsType = "tmpfs";
+            mountOptions = [
+              "mode=0755"
+            ];
+          };
+        };
+      })
+      (lib.mkIf (!(options ? disko)) {
+        fileSystems."/" = {
+          fsType = "tmpfs";
+          options = [ "mode=0755" ];
+        };
+      })
+      {
+        security.sudo.extraConfig = ''
+          Defaults lecture = never
+        '';
+      }
+    ];
 }