1 files changed, 17 insertions, 6 deletions

diff --git a/custom/borgbackup.nix b/custom/borgbackup.nix
index 222b1999..c08168ed 100644
--- a/custom/borgbackup.nix
+++ b/custom/borgbackup.nix
@@ -16,6 +16,11 @@ let
         default = [];
       };
 
+      prune = mkOption {
+        type = types.attrsOf (types.listOf types.str);
+        default = {};
+      };
+
       interval = mkOption {
         type = types.str;
         default = "6h";
@@ -54,7 +59,7 @@ in {
   config = mkIf (any (t: t.paths != []) (attrValues cfg.targets)) {
     services.btrfs-snapshots.enable = mkIf (cfg.snapshots == "btrfs") true;
   
-    systemd.timers = listToAttrs (map ({ target, path, tCfg }: nameValuePair "borgbackup-${target}@${path}" {
+    systemd.timers = listToAttrs (map ({ target, path, tCfg }: nameValuePair "borgbackup-${target}-${path}" {
       wantedBy = [ "timers.target" ];
 
       timerConfig = {
@@ -64,9 +69,9 @@ in {
       };
     }) (flatten (mapAttrsToList (target: tCfg: map (path: { inherit target path tCfg; }) tCfg.paths) cfg.targets)));
 
-    systemd.services = mapAttrs' (target: tCfg: nameValuePair "borgbackup-${target}@" (let
+    systemd.services = listToAttrs (map ({ target, path,  tCfg }: nameValuePair "borgbackup-${target}@${path}" (let
       deps = flatten [
-        (optional (cfg.snapshots == "btrfs") "btrfs-snapshot@%i.service")
+        (optional (cfg.snapshots == "btrfs") "btrfs-snapshot@%p.service")
         (optional tCfg.network "network-online.target")
       ];
     in {
@@ -87,10 +92,16 @@ in {
             --patterns-from .backup \
             --one-file-system \
             --compression auto,lzma \
-            ${tCfg.repo}::${cfg.prefix}$1-{utcnow}
+            ${tCfg.repo}::${cfg.prefix}${path}-{utcnow}
         '';
       in if tCfg.lock == null then borgCmd else "flock -xo /var/lock/${tCfg.lock} ${borgCmd}";
-      scriptArgs = "%i";
+
+      preStop = mkIf (hasAttr path tCfg.prune) ''
+        borg prune \
+          --prefix "${tcfg.prefix}${path}" \
+          ${concatStringsSep " " tCfg.prune."${path}"} \
+          ${tCfg.repo}
+      '';
 
       unitConfig = {
         AssertPathIsDirectory = mkIf (tCfg.lock != null) "/var/lock";
@@ -104,6 +115,6 @@ in {
         IOSchedulingPriority = 7;
         SuccessExitStatus = [1 2];
       };
-    })) cfg.targets;
+    }))) (flatten (mapAttrsToList (target: tCfg: map (path: { inherit target path tCfg; }) tCfg.paths) cfg.targets)));
   };
 }