services/yt-dlp: refactor

This commit is contained in:
Gabriel Arazas 2022-08-11 09:43:17 +08:00
parent 0927743845
commit 36909a281f
2 changed files with 13 additions and 9 deletions

View File

@ -3,6 +3,8 @@
let
cfg = config.services.yt-dlp;
serviceLevelArgs = lib.escapeShellArgs cfg.extraArgs;
jobType = { name, config, options, ... }: {
options = {
urls = lib.mkOption {
@ -148,12 +150,11 @@ in {
ExecStart = let
scriptName =
"yt-dlp-archive-service-${config.home.username}-${name}";
jobLevelArgs = lib.escapeShellArgs value.extraArgs;
urls = lib.escapeShellArgs urls;
archiveScript = pkgs.writeShellScriptBin scriptName ''
${cfg.package}/bin/yt-dlp ${lib.escapeShellArgs cfg.extraArgs} ${
lib.escapeShellArgs value.extraArgs
} ${lib.escapeShellArgs value.urls} --paths ${
lib.escapeShellArg cfg.archivePath
}
${cfg.package}/bin/yt-dlp ${serviceLevelArgs} ${jobLevelArgs} \
${urls} --paths ${lib.escapeShellArg cfg.archivePath}
'';
in "${archiveScript}/bin/${scriptName}";
};

View File

@ -3,6 +3,8 @@
let
cfg = config.services.yt-dlp;
serviceLevelArgs = lib.escapeShellArgs cfg.extraArgs;
jobType = { name, config, options, ... }: {
options = {
urls = lib.mkOption {
@ -125,7 +127,9 @@ in {
# There's no need to go to the working directory since yt-dlp has the
# `--paths` flag.
config = lib.mkIf cfg.enable {
systemd.services = lib.mapAttrs' (name: value:
systemd.services = lib.mapAttrs' (name: value: let
jobLevelArgs = lib.escapeShellArgs value.extraArgs;
in
lib.nameValuePair "yt-dlp-archive-service-${name}" {
wantedBy = [ "multi-user.target" ];
description = "yt-dlp archive job for group '${name}'";
@ -136,9 +140,8 @@ in {
mkdir -p ${lib.escapeShellArg cfg.archivePath}
'';
script = ''
yt-dlp ${lib.escapeShellArgs cfg.extraArgs} ${
lib.escapeShellArgs value.extraArgs
} ${lib.escapeShellArgs value.urls} --paths ${cfg.archivePath}
yt-dlp ${serviceLevelArgs} ${jobLevelArgs} \
${lib.escapeShellArgs value.urls} --paths ${lib.escapeShellArg cfg.archivePath}
'';
startAt = value.startAt;
serviceConfig = {