src/vs/platform/agentHost/node/messagePortProtocolServer.ts
165 LOC · 149 covered · 16 uncovered · 47 ranges · 179 concepts · 9 introducers · 80 tests
File neighbourhood
The centred file is linked to every concept that introduces one of its ranges, every test that runs code from the file, and the gray connector concepts standing between those tests and the file's own introducer concepts. Undirected links join concepts to every file where they introduce source and concepts to the tests they introduce; arrows show specialization between the displayed concepts and bridge only concepts omitted from this view. Concept colors match the source ranges below; connector concepts have no source color and are shown in gray.
Focused file, its introducer and connector concepts, their introduced files, and tests that run code from the file
In the embedded map, ordinary wheel input scrolls the page; use the visible controls to zoom and drag to pan. Open the full-screen map for canvas navigation: wheel pans, Ctrl/Command plus wheel zooms, and arrow keys pan when this region is focused. On touch screens, open the full-screen map to pan or pinch. If JavaScript or WebGL is unavailable, use the related-file, concept, and source links on this page.
Graph controls are ready.
Interactive rendering requires JavaScript and WebGL. Use the related-file, concept, and source links on this page while the interactive map is unavailable.
/*---------------------------------------------------------------------------------------------
messagePortProtocolServer.ts ×13
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See License.txt in the project root for license information.
*--------------------------------------------------------------------------------------------*/
import { Emitter, Event } from '../../../base/common/event.js';
import { Disposable } from '../../../base/common/lifecycle.js';
import { IServerChannel } from '../../../base/parts/ipc/common/ipc.js';
import { JSON_RPC_PARSE_ERROR, type AhpServerNotification, type JsonRpcNotification, type JsonRpcParseErrorResponse, type JsonRpcRequest, type JsonRpcResponse, type ProtocolMessage } from '../common/state/sessionProtocol.js';
import type { IProtocolServer, IProtocolTransport } from '../common/state/sessionTransport.js';
/**
* Adapts MessagePort IPC clients to Agent Host Protocol transports.
*
* Consumers must call {@link closeClient} when a UtilityProcessServer client
* connection disappears.
*/
export class MessagePortProtocolServer<TContext> extends Disposable implements IProtocolServer, IServerChannel<TContext> {
private readonly _onConnection = this._register(new Emitter<IProtocolTransport>());
readonly onConnection = this._onConnection.event;
readonly address = undefined;
private readonly _transports = new Map<TContext, MessagePortProtocolTransport>();
listen<T>(ctx: TContext, event: string): Event<T> {
case 'frame':
throw new Error(`Invalid listen: ${event}`);
async call<T>(ctx: TContext, command: string, arg?: unknown): Promise<T> {
case 'connect': {
const transport = this._getOrCreateTransport(ctx);
if (transport.connect()) {
this._onConnection.fire(transport);
}
return undefined as T;
}
case 'send': {
if (typeof arg !== 'string') {
throw new Error('send: arg must be a string frame');
}
const transport = this._transports.get(ctx);
if (!transport?.isConnected) {
}
transport.acceptFrame(arg);
return undefined as T;
}
case 'close':
return undefined as T;
throw new Error(`Invalid call: ${command}`);
/**
* Closes a client's transport after its owning IPC connection disappears.
*/
closeClient(ctx: TContext): void {
if (!transport) {
return;
}
this._transports.delete(ctx);
transport.dispose();
}
override dispose(): void {
this._transports.clear();
for (const transport of transports) {
}
}
private _getOrCreateTransport(ctx: TContext): MessagePortProtocolTransport {
throw new Error('MessagePortProtocolServer is disposed');
}
let transport = this._transports.get(ctx);
if (!transport) {
transport = new MessagePortProtocolTransport();
this._transports.set(ctx, transport);
const onClose = transport.onClose(() => {
onClose.dispose();
if (this._transports.get(ctx) === transport) {
this._transports.delete(ctx);
}
}
return transport;
}
class MessagePortProtocolTransport extends Disposable implements IProtocolTransport {
messagePortProtocolServer.ts ×19
private readonly _onFrame = this._register(new Emitter<string>());
readonly onFrame = this._onFrame.event;
private readonly _onMessage = this._register(new Emitter<ProtocolMessage>());
readonly onMessage = this._onMessage.event;
private readonly _onClose = this._register(new Emitter<void>());
readonly onClose = this._onClose.event;
private _isConnected = false;
private _isClosed = false;
get isConnected(): boolean {
}
connect(): boolean {
return false;
}
this._isConnected = true;
return true;
}
acceptFrame(frame: string): void {
this._onMessage.fire(JSON.parse(frame) as ProtocolMessage);
} catch {
this.send({ jsonrpc: '2.0', id: null, error: { code: JSON_RPC_PARSE_ERROR, message: 'Parse error' } });
messagePortProtocolServer.ts ×1
}
send(message: ProtocolMessage | AhpServerNotification | JsonRpcNotification | JsonRpcParseErrorResponse | JsonRpcResponse | JsonRpcRequest): void {
return;
}
this._onFrame.fire(JSON.stringify(message));
}
override dispose(): void {
}
this._isClosed = true;
this._isConnected = false;
this._onClose.fire();
super.dispose();
}