From 7ee037fc29ca3f375f6d3543e071f503d79081a3 Mon Sep 17 00:00:00 2001
From: 514fpv <koishi@514fpv.one>
Date: Thu, 22 Feb 2024 11:24:59 +0800
Subject: [PATCH] package(tubesync): add worker option

---
 package/tubesync/nixos.nix | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/package/tubesync/nixos.nix b/package/tubesync/nixos.nix
index e4552f4b..53628cf2 100644
--- a/package/tubesync/nixos.nix
+++ b/package/tubesync/nixos.nix
@@ -13,6 +13,12 @@ in {
       description = "tubesync launcher package";
     };
 
+    workers = mkOption {
+      type = with types; int;
+      default = 1;
+      description = "maximum amount of concurrent workers";
+    };
+
     user = mkOption {
       type = with types; str;
       default = "tubesync";
@@ -59,7 +65,6 @@ in {
   config = mkIf cfg.enable {
     systemd.services = let
       env = {
-        PYTHONPATH = cfg.package.pythonPath;
         GUNICORN_PID_FILE = "${cfg.stateDir}/run/gunicorn.pid";
         GUNICORN_USER = cfg.user;
         GUNICORN_GROUP = cfg.group;
@@ -67,7 +72,9 @@ in {
         CONFIG_BASE_DIR = cfg.stateDir;
         DOWNLOADS_BASE_DIR = cfg.dataDir;
         TUBESYNC_DEBUG = mkIf cfg.debug "True";
+        TUBESYNC_WORKERS = toString cfg.workers;
 
+        PYTHONPATH = cfg.package.pythonPath;
         REDIS_CONNECTION = "redis+socket://"
         + "${cfg.stateDir}/run/redis.sock";
       };