From 2c458053185c128247460e67d63c5d42af0285cb Mon Sep 17 00:00:00 2001
From: AuroraWright <AuroraWright@users.noreply.github.com>
Date: Sun, 28 Jan 2024 13:40:03 +0100
Subject: [PATCH] Add window closed detection and catch SIGINT

---
 owocr/run.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/owocr/run.py b/owocr/run.py
index d9bbe69..154f619 100644
--- a/owocr/run.py
+++ b/owocr/run.py
@@ -1,4 +1,5 @@
 import sys
+import signal
 import time
 import threading
 from pathlib import Path
@@ -196,6 +197,19 @@ def on_key_release(key):
         first_pressed = None
 
 
+def signal_handler(sig, frame):
+    global terminated
+    logger.info('Terminated!')
+    terminated = True
+
+
+def on_window_closed(alive):
+    global terminated
+    if not alive:
+        logger.info('Window closed, terminated!')
+        terminated = True
+
+
 def on_window_activated(active):
     global screencapture_window_active
     screencapture_window_active = active
@@ -423,11 +437,10 @@ def run(read_from=None,
             coord_height = target_window.height
             if screen_capture_only_active_windows:
                 screencapture_window_active = target_window.isActive
-                target_window.watchdog.start(isActiveCB=on_window_activated, resizedCB=on_window_resized, movedCB=on_window_moved)
+                target_window.watchdog.start(isAliveCB=on_window_closed, isActiveCB=on_window_activated, resizedCB=on_window_resized, movedCB=on_window_moved)
             else:
                 screencapture_window_visible = not target_window.isMinimized
-                target_window.watchdog.start(isMinimizedCB=on_window_minimized, resizedCB=on_window_resized, movedCB=on_window_moved)
-            target_window.watchdog.setTryToFind(True)
+                target_window.watchdog.start(isAliveCB=on_window_closed, isMinimizedCB=on_window_minimized, resizedCB=on_window_resized, movedCB=on_window_moved)
 
         global sct_params
         sct_params = {'top': coord_top, 'left': coord_left, 'width': coord_width, 'height': coord_height, 'mon': screen_capture_monitor}
@@ -446,6 +459,7 @@ def run(read_from=None,
             if path.suffix.lower() in allowed_extensions:
                 old_paths.add(get_path_key(path))
 
+    signal.signal(signal.SIGINT, signal_handler)
     while not terminated:
         if read_from == 'websocket':
             while True:
@@ -526,7 +540,6 @@ def run(read_from=None,
         windows_clipboard_thread.join()
     elif read_from == 'screencapture' and screencapture_window_mode:
         target_window.watchdog.stop()
-    user_input_thread.join()
     tmp_paused_listener.stop()
 
 if __name__ == '__main__':