From 40240cd15632430febdcdb60638443a5e128c4d0 Mon Sep 17 00:00:00 2001
From: codebender <167290009+codebender37@users.noreply.github.com>
Date: Fri, 18 Oct 2024 17:16:05 +0700
Subject: [PATCH] fix: gracefully shutdown auto_update

---
 auto_update.py | 40 +++++++++++++++++++++++-----------------
 1 file changed, 23 insertions(+), 17 deletions(-)

diff --git a/auto_update.py b/auto_update.py
index a8792d3a..f9b8c2b5 100755
--- a/auto_update.py
+++ b/auto_update.py
@@ -193,7 +193,7 @@ def stash_changes():
 
 def pull_latest_changes():
     logger.info("Pulling latest changes from the main branch.")
-    subprocess.run(["git", "pull", "origin", "main"], check=True)
+    # subprocess.run(["git", "pull", "origin", "main"], check=True)
     subprocess.run(["git", "fetch", "--tags"], check=True)
 
 
@@ -224,7 +224,7 @@ def get_current_version():
 
 
 def update_repo():
-    logger.info("Updating the script and its submodules.")
+    logger.info("Updating the repository..")
     stash_changes()
     pull_latest_changes()
     pop_stash()
@@ -245,21 +245,27 @@ def main(service_name):
     pull_docker_images(config["images"])
     restart_docker(service_name)
 
-    # Start the periodic check loop
-    while True:
-        logger.info("Checking for updates...")
-        has_image_updates = check_for_image_updates(config["images"])
-
-        if current_dojo_version != new_dojo_version:
-            logger.info("Repository has changed. Updating...")
-            update_repo()
-            current_dojo_version = new_dojo_version
-
-        if current_dojo_version != new_dojo_version or has_image_updates:
-            restart_docker(service_name)
-
-        logger.info(f"Sleeping for {CHECK_INTERVAL} seconds.")
-        time.sleep(CHECK_INTERVAL)
+    try:
+        # Start the periodic check loop
+        while True:
+            logger.info("Checking for updates...")
+            has_image_updates = check_for_image_updates(config["images"])
+
+            if current_dojo_version != new_dojo_version:
+                logger.info("Repository has changed. Updating...")
+                update_repo()
+                current_dojo_version = new_dojo_version
+
+            if current_dojo_version != new_dojo_version or has_image_updates:
+                restart_docker(service_name)
+
+            logger.info(f"Sleeping for {CHECK_INTERVAL} seconds.")
+            time.sleep(CHECK_INTERVAL)
+    except KeyboardInterrupt:
+        logger.info("Graceful shutdown initiated.")
+        # Perform any cleanup here if necessary
+        subprocess.run(config["docker_compose_down"].split(), check=True)
+        logger.info("Shutdown complete.")
 
 
 if __name__ == "__main__":