From c80dad9c3db35220c4fc1184144c6f03cbcd8747 Mon Sep 17 00:00:00 2001
From: Sakooooo <78461130+Sakooooo@users.noreply.github.com>
Date: Wed, 1 Jan 2025 16:19:36 +0400
Subject: [PATCH] qbuittornet

---
 hosts/sakoserver/configuration.nix   |   8 ++
 modules/server/media/qbittorrent.nix | 114 ++++++++++++++++++++++++++-
 2 files changed, 120 insertions(+), 2 deletions(-)

diff --git a/hosts/sakoserver/configuration.nix b/hosts/sakoserver/configuration.nix
index 0cd70a07..2b20b83f 100644
--- a/hosts/sakoserver/configuration.nix
+++ b/hosts/sakoserver/configuration.nix
@@ -70,6 +70,14 @@
         forgejo.enable = true;
         headscale.enable = true;
       };
+      media = {
+        qbittorrent = {
+          enable = true;
+          torrentPort = 55907;
+          openFirewall = true;
+          group = "media";
+        };
+      };
     };
   };
 
diff --git a/modules/server/media/qbittorrent.nix b/modules/server/media/qbittorrent.nix
index 88792d01..8f7db789 100644
--- a/modules/server/media/qbittorrent.nix
+++ b/modules/server/media/qbittorrent.nix
@@ -2,7 +2,117 @@
 with lib;
 let cfg = config.void.server.media.qbittorrent;
 in {
-  options.void.server.media.qbittorrent = { enable = mkEnableOption false; };
+  options.void.server.media.qbittorrent = {
+    enable = mkEnableOption (lib.mdDoc "qBittorrent headless");
 
-  config = mkIf cfg.enable { };
+    dataDir = mkOption {
+      type = types.path;
+      default = "/var/lib/qbittorrent";
+      description = lib.mdDoc ''
+        The directory where qBittorrent stores its data files.
+      '';
+    };
+
+    user = mkOption {
+      type = types.str;
+      default = "qbittorrent";
+      description = lib.mdDoc ''
+        User account under which qBittorrent runs.
+      '';
+    };
+
+    group = mkOption {
+      type = types.str;
+      default = "media";
+      description = lib.mdDoc ''
+        Group under which qBittorrent runs.
+      '';
+    };
+
+    port = mkOption {
+      type = types.port;
+      default = 8080;
+      description = lib.mdDoc ''
+        qBittorrent web UI port.
+      '';
+    };
+
+    torrentPort = mkOption {
+      type = types.port;
+      default = 8661;
+      description = lib.mdDoc ''
+        Port for incoming connections
+      '';
+    };
+
+    openFirewall = mkOption {
+      type = types.bool;
+      default = false;
+      description = lib.mdDoc ''
+        Open services.qBittorrent.torrentPort to the outside network.
+      '';
+    };
+
+    openWebUIFirewall = mkOption {
+      type = types.bool;
+      default = false;
+      description = lib.mdDoc ''
+        Open services.qBittorrent.port to the outside network.
+      '';
+    };
+
+    package = mkOption {
+      type = types.package;
+      default = pkgs.qbittorrent-nox;
+      defaultText = literalExpression "pkgs.qbittorrent-nox";
+      description = lib.mdDoc ''
+        The qbittorrent package to use.
+      '';
+    };
+  };
+
+  config = mkIf cfg.enable {
+    networking.firewall =
+      mkIf cfg.openFirewall { allowedTCPPorts = [ cfg.torrentPort cfg.port ]; };
+
+    users.users.qbittorrent = { };
+    users.groups.media = { };
+
+    systemd.services.qbittorrent = {
+      # based on the plex.nix service module and
+      # https://github.com/qbittorrent/qBittorrent/blob/master/dist/unix/systemd/qbittorrent-nox%40.service.in
+      description = "qBittorrent-nox service";
+      documentation = [ "man:qbittorrent-nox(1)" ];
+      after = [ "network.target" ];
+      wantedBy = [ "multi-user.target" ];
+
+      serviceConfig = {
+        Type = "simple";
+        User = cfg.user;
+        Group = cfg.group;
+
+        # Run the pre-start script with full permissions (the "!" prefix) so it
+        # can create the data directory if necessary.
+        ExecStartPre = let
+          preStartScript = pkgs.writeScript "qbittorrent-run-prestart" ''
+            #!${pkgs.bash}/bin/bash
+
+            # Create data directory if it doesn't exist
+            if ! test -d "$QBT_PROFILE"; then
+              echo "Creating initial qBittorrent data directory in: $QBT_PROFILE"
+              install -d -m 0755 -o "${cfg.user}" -g "${cfg.group}" "$QBT_PROFILE"
+            fi
+          '';
+        in "!${preStartScript}";
+
+        ExecStart = "${cfg.package}/bin/qbittorrent-nox";
+        UMask = "0002";
+      };
+
+      environment = {
+        QBT_PROFILE = cfg.dataDir;
+        QBT_WEBUI_PORT = toString cfg.port;
+      };
+    };
+  };
 }