1 files changed, 41 insertions, 0 deletions
diff --git a/modules/memory.nix b/modules/memory.nix
new file mode 100644
index 0000000..b20ec8a
--- /dev/null
+++ b/modules/memory.nix
@@ -0,0 +1,41 @@
+{ config, lib, ... }:
+let
+  cfg = config.os.core.memory;
+in
+{
+  options.os.core.memory = {
+    zram = {
+      enable = lib.mkEnableOption "enables zram compression";
+      percent = lib.mkOption {
+        type = lib.types.int;
+        default = 25;
+      };
+    };
+
+    swapfile = {
+      enable = lib.mkEnableOption "enables a swapfile";
+      size = lib.mkOption {
+        type = lib.types.int;
+        default = 8;
+      };
+    };
+  };
+  config = lib.mkMerge [
+    (lib.mkIf cfg.zram.enable {
+      zramSwap = {
+        enable = true;
+        algorithm = "zstd";
+        memoryPercent = cfg.zram.percent;
+        priority = 100;
+      };
+    })
+
+    (lib.mkIf cfg.swapfile.enable {
+      swapDevices = [ {
+        device = "/.swapvol/swapfile";
+        size = cfg.swapfile.size * 1024; 
+        priority = 0;
+      } ];
+    })
+  ];
+}