From 500df0f2f34f797e891ca4257870423ecac2ab19 Mon Sep 17 00:00:00 2001
From: CanbiZ <47820557+MickLesk@users.noreply.github.com>
Date: Mon, 27 Oct 2025 10:51:32 +0100
Subject: [PATCH] Update dispatcharr.sh

---
 ct/dispatcharr.sh | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/ct/dispatcharr.sh b/ct/dispatcharr.sh
index 6b76fec57..91113904b 100644
--- a/ct/dispatcharr.sh
+++ b/ct/dispatcharr.sh
@@ -46,6 +46,20 @@ function update_script() {
     if [[ -f /opt/dispatcharr/.env ]]; then
       cp /opt/dispatcharr/.env /tmp/dispatcharr.env.backup
     fi
+
+    if [[ -f /opt/dispatcharr/start-gunicorn.sh ]]; then
+      cp /opt/dispatcharr/start-gunicorn.sh /tmp/start-gunicorn.sh.backup
+    fi
+    if [[ -f /opt/dispatcharr/start-celery.sh ]]; then
+      cp /opt/dispatcharr/start-celery.sh /tmp/start-celery.sh.backup
+    fi
+    if [[ -f /opt/dispatcharr/start-celerybeat.sh ]]; then
+      cp /opt/dispatcharr/start-celerybeat.sh /tmp/start-celerybeat.sh.backup
+    fi
+    if [[ -f /opt/dispatcharr/start-daphne.sh ]]; then
+      cp /opt/dispatcharr/start-daphne.sh /tmp/start-daphne.sh.backup
+    fi
+
     if [[ -f /opt/dispatcharr/.env ]]; then
       set -o allexport
       source /opt/dispatcharr/.env
@@ -65,6 +79,21 @@ function update_script() {
       mv /tmp/dispatcharr.env.backup /opt/dispatcharr/.env
       msg_info "Restored environment configuration"
     fi
+
+    # Restore service scripts
+    if [[ -f /tmp/start-gunicorn.sh.backup ]]; then
+      mv /tmp/start-gunicorn.sh.backup /opt/dispatcharr/start-gunicorn.sh
+    fi
+    if [[ -f /tmp/start-celery.sh.backup ]]; then
+      mv /tmp/start-celery.sh.backup /opt/dispatcharr/start-celery.sh
+    fi
+    if [[ -f /tmp/start-celerybeat.sh.backup ]]; then
+      mv /tmp/start-celerybeat.sh.backup /opt/dispatcharr/start-celerybeat.sh
+    fi
+    if [[ -f /tmp/start-daphne.sh.backup ]]; then
+      mv /tmp/start-daphne.sh.backup /opt/dispatcharr/start-daphne.sh
+    fi
+
     cd /opt/dispatcharr || exit
     rm -rf .venv
     $STD uv venv