From e940c13019011f9881e19e98a2a2d2810128fc0e Mon Sep 17 00:00:00 2001
From: timoxa0 <tima012007@gmail.com>
Date: Mon, 13 May 2024 19:40:05 +0500
Subject: [PATCH] Add exit handler

---
 lon_deployer/main.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/lon_deployer/main.py b/lon_deployer/main.py
index d97b720..ecf9066 100644
--- a/lon_deployer/main.py
+++ b/lon_deployer/main.py
@@ -1,4 +1,5 @@
 import argparse
+import atexit
 import logging
 import re
 import signal
@@ -37,10 +38,18 @@ def handle_sigint(*_) -> None:
         exit_counter += 1
 
 
+def exit_handler(*_) -> None:
+    global adb
+    if adb is not None:
+        with console.status("[cyan]Stopping adb server", spinner="line", spinner_style="white"):
+            adb.server_kill()
+
+
 def main() -> int:
     global adb
 
     signal.signal(signal.SIGINT, handle_sigint)
+    atexit.register(exit_handler)
 
     parser = argparse.ArgumentParser(
         description="Linux on Nabu deployer",