@celljs/ta-web 4.6.1 → 4.8.0

package/src/browser/workbench.ts

@@ -6,6 +6,8 @@ import {
     type TaAuthServer,
     type TaPackageServer,
     type TaRelayServer,
+    type TaRemoteEvent,
+    type TaRemoteServer,
     type TaServer,
     type TaSettingsServer,
     type TaTerminalServer,
@@ -83,6 +85,7 @@ export class TaWorkbenchImpl implements TaWorkbench {
     protected readonly terminalListeners = new Set<(terminalId: string, data: string, exitCode?: number) => void>();
     protected readonly extensionDataListeners = new Set<(name: string, keys: string[], cwd?: string) => void>();
     protected readonly authListeners = new Set<(loginId: string, event: TaAuthEvent) => void>();
+    protected readonly remoteListeners = new Set<(event: TaRemoteEvent) => void>();
 
     get state(): WorkbenchState {
         return this.current;
@@ -116,6 +119,10 @@ export class TaWorkbenchImpl implements TaWorkbench {
         return this.signedIn().relay;
     }
 
+    get remote(): TaRemoteServer {
+        return this.signedIn().remote;
+    }
+
     protected signedIn(): TaConnection {
         if (!this.connection) {
             throw new Error('Not signed in to ta serve.');
@@ -138,6 +145,11 @@ export class TaWorkbenchImpl implements TaWorkbench {
         return () => this.authListeners.delete(listener);
     }
 
+    onRemoteEvent(listener: (event: TaRemoteEvent) => void): () => void {
+        this.remoteListeners.add(listener);
+        return () => this.remoteListeners.delete(listener);
+    }
+
     async setWorkingDirectory(cwd: string): Promise<void> {
         this.update({ server: await this.server.setWorkingDirectory(cwd) });
     }
@@ -239,12 +251,15 @@ export class TaWorkbenchImpl implements TaWorkbench {
                 listener(loginId, event);
             }
         });
+        connection.onRemoteEvent(event => this.emitRemote(event));
         connection.onClose(() => this.update({ status: 'offline' }));
         connection.onReconnecting(delayMs => this.update({ retryAt: Date.now() + delayMs }));
         connection.onOpen(() => {
             this.update({ status: 'online', retryAt: undefined });
             void this.refresh();
+            void this.watchRemote(connection);
         });
+        void this.watchRemote(connection);
         this.update({ status: 'online', error: undefined, retryAt: undefined, server: await connection.server.getServerInfo() });
         await this.refresh();
     }
@@ -280,6 +295,20 @@ export class TaWorkbenchImpl implements TaWorkbench {
         return new SessionStore(this.connection, sessionId, options);
     }
 
+    /** The prompts of ssh that wait for an answer, from now and those already there. */
+    protected async watchRemote(connection: TaConnection): Promise<void> {
+        const prompts = await connection.remote.watch().catch(() => []);
+        for (const prompt of prompts) {
+            this.emitRemote({ type: 'prompt', prompt });
+        }
+    }
+
+    protected emitRemote(event: TaRemoteEvent): void {
+        for (const listener of this.remoteListeners) {
+            listener(event);
+        }
+    }
+
     protected stopRetrying(): void {
         this.generation++;
         clearTimeout(this.retryTimer);