From d9b0104f8841c8b3cdc4018da453fbbdcf8718c3 Mon Sep 17 00:00:00 2001
From: CanbiZ <47820557+MickLesk@users.noreply.github.com>
Date: Mon, 18 Aug 2025 15:55:43 +0200
Subject: [PATCH] Update paperless-ngx.sh

---
 ct/paperless-ngx.sh | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/ct/paperless-ngx.sh b/ct/paperless-ngx.sh
index 0d51673d..4ba8a9f0 100644
--- a/ct/paperless-ngx.sh
+++ b/ct/paperless-ngx.sh
@@ -49,10 +49,21 @@ function update_script() {
       msg_info "Migrating old Paperless-ngx installation to uv"
       rm -rf /opt/paperless/venv
       find /opt/paperless -name "__pycache__" -type d -exec rm -rf {} +
-      sed -i 's|ExecStart=.*manage.py document_consumer|ExecStart=uv run -- python manage.py document_consumer|' /etc/systemd/system/paperless-consumer.service
-      sed -i 's|ExecStart=celery|ExecStart=uv run -- celery|' /etc/systemd/system/paperless-scheduler.service
-      sed -i 's|ExecStart=celery|ExecStart=uv run -- celery|' /etc/systemd/system/paperless-task-queue.service
-      sed -i 's|ExecStart=granian|ExecStart=uv run -- granian|' /etc/systemd/system/paperless-webserver.service
+      declare -A PATCHES=(
+        ["paperless-consumer.service"]="ExecStart=.*manage.py document_consumer|ExecStart=uv run -- python manage.py document_consumer"
+        ["paperless-scheduler.service"]="ExecStart=celery|ExecStart=uv run -- celery"
+        ["paperless-task-queue.service"]="ExecStart=celery|ExecStart=uv run -- celery"
+        ["paperless-webserver.service"]="ExecStart=.*|ExecStart=uv run -- granian --interface asginl --ws \"paperless.asgi:application\""
+      )
+      for svc in "${!PATCHES[@]}"; do
+        path=$(systemctl show -p FragmentPath "$svc" | cut -d= -f2)
+        if [[ -n "$path" && -f "$path" ]]; then
+          sed -i "s|${PATCHES[$svc]%|*}|${PATCHES[$svc]#*|}|" "$path"
+          msg_ok "Patched $svc"
+        else
+          msg_error "Service file for $svc not found!"
+        fi
+      done
       $STD systemctl daemon-reexec
       $STD systemctl daemon-reload
       cd /opt/paperless