106 lines
2.9 KiB
Nix
106 lines
2.9 KiB
Nix
{
|
|
inputs,
|
|
pkgs,
|
|
lib,
|
|
config,
|
|
...
|
|
}:
|
|
let
|
|
cfg = config.services.snapshot_manager;
|
|
snapshotManagerPackages =
|
|
ps: with ps; [
|
|
apprise
|
|
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";
|
|
};
|
|
};
|
|
};
|
|
};
|
|
}
|