From b28532043649ed7882f7e132f631094af9fe5f3e Mon Sep 17 00:00:00 2001
From: Manuel Thalmann <m@nuth.ch>
Date: Thu, 16 May 2024 18:17:18 +0200
Subject: [PATCH] Allow configuring `vfs`

---
 lib/modules/rclone.nix | 66 ++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 64 insertions(+), 2 deletions(-)

diff --git a/lib/modules/rclone.nix b/lib/modules/rclone.nix
index 574079d..7faa2a5 100644
--- a/lib/modules/rclone.nix
+++ b/lib/modules/rclone.nix
@@ -89,6 +89,44 @@ let
           default = "";
         };
 
+        vfs = {
+          enable = lib.mkEnableOption "Virtual File System";
+
+          mode = lib.mkOption {
+            type = lib.types.nullOr (lib.types.enum [
+              "minimal"
+              "writes"
+              "full"
+            ]);
+            description = "The cache mode to use.";
+            default = "full";
+          };
+
+          dirCacheTime = lib.mkOption {
+            type = lib.types.nullOr lib.types.str;
+            description = "The time to cache directory entries for.";
+            default = null;
+          };
+
+          pollInterval = lib.mkOption {
+            type = lib.types.nullOr lib.types.str;
+            description = "The time to wait between polling for changes.";
+            default = null;
+          };
+
+          bufferSize = lib.mkOption {
+            type = lib.types.nullOr lib.types.str;
+            description = "The maximum size of the buffer per size to allocate.";
+            default = null;
+          };
+
+          maxAge = lib.mkOption {
+            type = lib.types.nullOr lib.types.str;
+            description = "The maximum age of cached files to keep.";
+            default = null;
+          };
+        };
+
         extraArgs = lib.mkOption {
           type = lib.types.listOf lib.types.str;
           description = "A set of additional arguments to pass to `rclone mount`.";
@@ -115,7 +153,23 @@ let
               (name: path: "export ${name}=\"$(cat ${lib.escapeShellArg path})\"")
               config.secrets));
 
-        args = config.extraArgs;
+        args = config.extraArgs ++ (
+          let
+            vfs = config.vfs;
+          in (
+            lib.optionals vfs.enable
+              (builtins.attrValues (
+                lib.attrsets.concatMapAttrs
+                  (name: value:
+                    lib.optionalAttrs
+                      (value != null)
+                      { name = "--${name}=${lib.escapeShellArg value}"; })
+                  {
+                    vfs-cache-mode = vfs.mode;
+                    vfs-cache-poll-interval = vfs.pollInterval;
+                    vfs-cache-max-size = vfs.bufferSize;
+                    vfs-cache-max-age = vfs.maxAge;
+                  }))));
       };
     });
 
@@ -320,6 +374,12 @@ in {
         global = true;
       };
 
+      globalArgs = lib.mkOption {
+        type = lib.types.listOf lib.types.str;
+        description = "The arguments to pass to `rclone mount` for each configuration.";
+        default = [ ];
+      };
+
       configs = (builtins.mapAttrs
         (name: provider: lib.mkOption {
           type = lib.types.attrsOf (lib.types.submodule provider.module);
@@ -394,7 +454,9 @@ in {
                         ${sync.secretsScript}
                         mkdir -p ${sync.path}
                         mkdir -p /tmp/rclone
-                        ${lib.getExe pkgs.rclone} mount --config ${configFile} ${name}: ${sync.path}
+                        ${lib.getExe pkgs.rclone} mount ${
+                          builtins.concatStringsSep " " (cfg.globalArgs ++ sync.args)
+                        } --config ${configFile} ${name}: ${sync.path}
                       '';
                     in
                       (lib.getExe script);