Fix bridge SIGTERM handling

app/src/bridge/components/mediaServer.ts

@@ -179,16 +179,19 @@ export async function startMediaServer(filePath: string, port: number) {
 
 export function stopMediaServer() {
     return new Promise<void>((resolve, reject) => {
-        if (mediaServer?.listening) {
-            mediaServer.close(err => {
-                if (err) {
-                    reject();
-                } else {
-                    resolve();
-                }
-            });
-
-            mediaServer = undefined;
+        if (!mediaServer?.listening) {
+            resolve();
+            return;
         }
+
+        mediaServer.close(err => {
+            if (err) {
+                reject();
+            } else {
+                resolve();
+            }
+        });
+
+        mediaServer = undefined;
     });
 }

app/src/daemon.ts

@@ -1,12 +1,22 @@
 import http from "http";
 import https from "https";
-import { spawn } from "child_process";
+import { ChildProcess, spawn } from "child_process";
 import { Readable } from "stream";
 
 import WebSocket from "ws";
 
 import { DecodeTransform, EncodeTransform } from "./transforms.js";
 
+const bridgeInstances = new Set<ChildProcess>();
+
+// Ensure child processes are killed on exit
+process.on("SIGTERM", async () => {
+    for (const bridge of bridgeInstances) {
+        bridge.kill();
+    }
+    process.exit(1);
+});
+
 export interface DaemonOpts {
     host: string;
     port: number;
@@ -46,6 +56,7 @@ export function init(opts: DaemonOpts) {
          * version of self in bridge mode.
          */
         const bridge = spawn(process.execPath, [process.argv[1]]);
+        bridgeInstances.add(bridge);
 
         // socket -> bridge.stdin
         messageStream.pipe(new EncodeTransform()).pipe(bridge.stdin);
@@ -61,7 +72,10 @@ export function init(opts: DaemonOpts) {
 
         // Handle termination
         socket.on("close", () => bridge.kill());
-        bridge.on("exit", () => socket.close());
+        bridge.on("exit", () => {
+            socket.close();
+            bridgeInstances.delete(bridge);
+        });
     });
 
     /**