dotfiles/common/optional/zfs/snapshot.nix

{
  inputs,
  pkgs,
  lib,
  config,
  ...
}:
let
  cfg = config.services.snapshot_manager;
  snapshotManagerPackages =
    ps: with ps; [
      httpx
      typer
    ];
in
{
  options = {
    services.snapshot_manager = {
      enable = lib.mkEnableOption "ZFS snapshot manager";
      path = lib.mkOption {
        type = lib.types.path;
        default = ./snapshot_config.toml;
        description = "Path to the snapshot_manager TOML config.";
      };
      PYTHONPATH = lib.mkOption {
        type = lib.types.str;
        description = ''
          the PYTHONPATH to use for the snapshot_manager service.
        '';
      };
      EnvironmentFile = lib.mkOption {
        type = lib.types.nullOr (lib.types.coercedTo lib.types.path toString lib.types.str);
        default = null;
        description = ''
          Single environment file for the service (e.g. /etc/snapshot-manager/env).
          Use a leading "-" to ignore if missing (systemd feature).
        '';
      };
    };
  };

  config = lib.mkIf cfg.enable {
    nixpkgs.overlays = [
      (final: _prev: {
        snapshot_manager_python = final.python314.withPackages snapshotManagerPackages;
        snapshot_manager_test_python = final.python314.withPackages (
          ps:
          snapshotManagerPackages ps
          ++ (with ps; [
            pyfakefs
            pytest
            pytest-asyncio
            pytest-mock
            pytest-xdist
          ])
        );
        snapshot_manager_tests =
          final.runCommand "snapshot-manager-tests"
            {
              nativeBuildInputs = [ final.snapshot_manager_test_python ];
            }
            ''
              export HOME="$TMPDIR"
              cd ${inputs.self}
              pytest \
                -o cache_dir="$TMPDIR/pytest-cache" \
                tests/test_common.py \
                tests/test_signal_alert.py \
                tests/test_snapshot_manager.py \
                tests/test_zfs.py
              touch "$out"
            '';
      })
    ];

    system.checks = [ pkgs.snapshot_manager_tests ];

    systemd = {
      services.snapshot_manager = {
        description = "ZFS Snapshot Manager";
        requires = [ "zfs-import.target" ];
        after = [ "zfs-import.target" ];
        path = [ pkgs.zfs ];
        environment = {
          PYTHONPATH = cfg.PYTHONPATH;
        };
        serviceConfig = {
          Type = "oneshot";
          ExecStart = "${pkgs.snapshot_manager_python}/bin/python -m python.tools.snapshot_manager ${lib.escapeShellArg cfg.path}";
        }
        // lib.optionalAttrs (cfg.EnvironmentFile != null) {
          EnvironmentFile = cfg.EnvironmentFile;
        };
      };
      timers.snapshot_manager = {
        wantedBy = [ "timers.target" ];
        timerConfig = {
          OnBootSec = "15m";
          OnUnitActiveSec = "15m";
          Unit = "snapshot_manager.service";
        };
      };
    };
  };
}