From 8fa6decb890444c87045aa39931929df00f30389 Mon Sep 17 00:00:00 2001
From: Benjamin Toby <devops@tben.me>
Date: Sun, 20 Sep 2026 18:52:59 +0100
Subject: [PATCH] Update installation script: fix systemd infinite loop

---
 src/scripts/install-wg-ui.sh | 4 ++--
 src/server.ts                | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/scripts/install-wg-ui.sh b/src/scripts/install-wg-ui.sh
index 9bbc97f..f80ad01 100755
--- a/src/scripts/install-wg-ui.sh
+++ b/src/scripts/install-wg-ui.sh
@@ -199,7 +199,7 @@ EOF
     systemctl daemon-reload
     log "enabling and starting $SERVICE_NAME ..."
     systemctl enable --now "$SERVICE_NAME.service"
-    if [ "$UPDATE" = true ]; then
+    if [ "$UPDATE" = true ] && [ -z "${SKIP_SERVICE_RESTART:-}" ]; then
         log "restarting $SERVICE_NAME to load updated code ..."
         systemctl restart "$SERVICE_NAME.service"
     fi
@@ -232,7 +232,7 @@ EOF
     chmod +x "$init_script"
     log "adding $SERVICE_NAME to default runlevel and starting ..."
     rc-update add "$SERVICE_NAME" default
-    if [ "$UPDATE" = true ]; then
+    if [ "$UPDATE" = true ] && [ -z "${SKIP_SERVICE_RESTART:-}" ]; then
         rc-service "$SERVICE_NAME" restart
     else
         rc-service "$SERVICE_NAME" start
diff --git a/src/server.ts b/src/server.ts
index 2fbefe2..72e38d9 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -35,6 +35,7 @@ execSync(path.join(__dirname, "scripts", "install-wg-ui.sh"), {
     stdio: "ignore",
     env: {
         ...process.env,
+        SKIP_SERVICE_RESTART: "1",
     },
 });
 bunext.bunextLog.success(`Wireguard setup complete!`);