fix plugin uninstalls

backend/decky_loader/plugin/sandboxed_plugin.py

@@ -1,12 +1,12 @@
 from os import path, environ
-from signal import SIG_IGN, SIGINT, SIGTERM, signal
+from signal import SIG_IGN, SIGINT, SIGTERM, getsignal, signal
 from importlib.util import module_from_spec, spec_from_file_location
 from json import dumps, loads
 from logging import getLogger
 from sys import exit, path as syspath, modules as sysmodules
 from traceback import format_exc
 from asyncio import (get_event_loop, new_event_loop,
-                     set_event_loop, sleep)
+                     set_event_loop)
 from setproctitle import setproctitle, setthreadtitle
 
 from .messages import SocketResponseDict, SocketMessageType
@@ -19,6 +19,8 @@ from typing import List, TypeVar, Any
 
 DataType = TypeVar("DataType")
 
+original_term_handler = getsignal(SIGTERM)
+
 class SandboxedPlugin:
     def __init__(self,
                  name: str,
@@ -46,6 +48,8 @@ class SandboxedPlugin:
         self._socket = socket
 
         try:
+            # Ignore signals meant for parent Process
+            # TODO SURELY there's a better way to do this.
             signal(SIGINT, SIG_IGN)
             signal(SIGTERM, SIG_IGN)
 
@@ -120,7 +124,14 @@ class SandboxedPlugin:
         except:
             self.log.error("Failed to start " + self.name + "!\n" + format_exc())
             exit(0)
-        get_event_loop().run_forever()
+        try:
+            get_event_loop().run_forever()
+        except SystemExit:
+            pass
+        except:
+            self.log.error("Loop exited for " + self.name + "!\n" + format_exc())
+        finally:
+            get_event_loop().close()
 
     async def _unload(self):
         try:
@@ -156,6 +167,8 @@ class SandboxedPlugin:
         data = loads(message)
 
         if "stop" in data:
+            # Incase the loader needs to terminate our process soon
+            signal(SIGTERM, original_term_handler)
             self.log.info(f"Calling Loader unload function for {self.name}.")
             await self._unload()
 
@@ -163,10 +176,10 @@ class SandboxedPlugin:
                 self.log.info("Calling Loader uninstall function.")
                 await self._uninstall()
 
-            get_event_loop().stop()
-            while get_event_loop().is_running():
-                await sleep(0.1)
-            get_event_loop().close()
+            self.log.debug("Stopping event loop")
+
+            loop = get_event_loop()
+            loop.call_soon_threadsafe(loop.stop)
             exit(0)
 
         d: SocketResponseDict = {"type": SocketMessageType.RESPONSE, "res": None, "success": True, "id": data["id"]}