eae5d8c807
These conflicts will be resolved in the following commits. We do it this way so that PR review is possible.
58 lines
2.3 KiB
TypeScript
58 lines
2.3 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import { ipcMain, WebContents } from 'electron';
|
|
import { Event, Emitter } from 'vs/base/common/event';
|
|
import { IPCServer, ClientConnectionEvent } from 'vs/base/parts/ipc/common/ipc';
|
|
import { Protocol as ElectronProtocol } from 'vs/base/parts/ipc/common/ipc.electron';
|
|
import { IDisposable, toDisposable } from 'vs/base/common/lifecycle';
|
|
import { VSBuffer } from 'vs/base/common/buffer';
|
|
|
|
interface IIPCEvent {
|
|
event: { sender: WebContents; };
|
|
message: Buffer | null;
|
|
}
|
|
|
|
function createScopedOnMessageEvent(senderId: number, eventName: string): Event<VSBuffer | null> {
|
|
const onMessage = Event.fromNodeEventEmitter<IIPCEvent>(ipcMain, eventName, (event, message) => ({ event, message }));
|
|
const onMessageFromSender = Event.filter(onMessage, ({ event }) => event.sender.id === senderId);
|
|
|
|
return Event.map(onMessageFromSender, ({ message }) => message ? VSBuffer.wrap(message) : message);
|
|
}
|
|
|
|
/**
|
|
* An implemention of `IPCServer` on top of Electron `ipcMain` API.
|
|
*/
|
|
export class Server extends IPCServer {
|
|
|
|
private static readonly Clients = new Map<number, IDisposable>();
|
|
|
|
private static getOnDidClientConnect(): Event<ClientConnectionEvent> {
|
|
const onHello = Event.fromNodeEventEmitter<WebContents>(ipcMain, 'vscode:hello', ({ sender }) => sender);
|
|
|
|
return Event.map(onHello, webContents => {
|
|
const id = webContents.id;
|
|
const client = Server.Clients.get(id);
|
|
|
|
if (client) {
|
|
client.dispose();
|
|
}
|
|
|
|
const onDidClientReconnect = new Emitter<void>();
|
|
Server.Clients.set(id, toDisposable(() => onDidClientReconnect.fire()));
|
|
|
|
const onMessage = createScopedOnMessageEvent(id, 'vscode:message') as Event<VSBuffer>;
|
|
const onDidClientDisconnect = Event.any(Event.signal(createScopedOnMessageEvent(id, 'vscode:disconnect')), onDidClientReconnect.event);
|
|
const protocol = new ElectronProtocol(webContents, onMessage);
|
|
|
|
return { protocol, onDidClientDisconnect };
|
|
});
|
|
}
|
|
|
|
constructor() {
|
|
super(Server.getOnDidClientConnect());
|
|
}
|
|
}
|