Error callback on dispatcher disconnect

ts/packages/agentServer/client/src/agentServerClient.ts

@@ -22,6 +22,7 @@ export async function connectDispatcher(
     clientIO: ClientIO,
     url: string | URL,
     options?: DispatcherConnectOptions,
+    onDisconnect?: () => void,
 ): Promise<Dispatcher> {
     return new Promise((resolve, reject: (e: Error) => void) => {
         const ws = new WebSocket(url); // Replace with the actual WebSocket server URL
@@ -74,7 +75,9 @@ export async function connectDispatcher(
         ws.onclose = (event: WebSocket.CloseEvent) => {
             debug("WebSocket connection closed", event.code, event.reason);
             channel.notifyDisconnected();
-            if (!resolved) {
+            if (resolved) {
+                onDisconnect?.();
+            } else {
                 reject(new Error(`Failed to connect to dispatcher at ${url}`));
             }
         };

ts/packages/cli/src/commands/connect.ts

@@ -43,6 +43,11 @@ export default class Connect extends Command {
             const dispatcher = await connectDispatcher(
                 clientIO,
                 `ws://localhost:${flags.port}`,
+                undefined,
+                () => {
+                    console.error("Disconnected from dispatcher");
+                    process.exit(1);
+                },
             );
             try {
                 let processed = false;

ts/packages/shell/src/main/instance.ts

@@ -132,6 +132,14 @@ async function initializeDispatcher(
             newDispatcher = await connectDispatcher(
                 clientIO,
                 `ws://localhost:${connect}`,
+                undefined,
+                () => {
+                    dialog.showErrorBox(
+                        "Disconnected",
+                        "The connection to the dispatcher was lost.",
+                    );
+                    app.quit();
+                },
             );
             debugShellInit(
                 "Connected to remote dispatcher",