55 lines
1.2 KiB
TypeScript
55 lines
1.2 KiB
TypeScript
import EventEmitter from 'eventemitter3';
|
|
import { Connect } from '.';
|
|
|
|
export class ExtServer {
|
|
private EE: EventEmitter;
|
|
|
|
constructor() {
|
|
this.EE = new EventEmitter();
|
|
chrome.runtime.onConnect.addListener((port) => {
|
|
this.EE.emit('connect', new ExtConnect(port));
|
|
});
|
|
}
|
|
|
|
onConnect(callback: (con: Connect) => void) {
|
|
this.EE.on('connect', callback);
|
|
}
|
|
}
|
|
|
|
export function connect() {
|
|
return new ExtConnect(chrome.runtime.connect());
|
|
}
|
|
|
|
export class ExtConnect implements Connect {
|
|
private EE: EventEmitter;
|
|
private port: chrome.runtime.Port;
|
|
|
|
constructor(port: chrome.runtime.Port) {
|
|
this.EE = new EventEmitter();
|
|
this.port = port;
|
|
port.onMessage.addListener((message) => {
|
|
this.EE.emit('message', message);
|
|
});
|
|
port.onDisconnect.addListener(() => {
|
|
this.EE.emit('disconnect');
|
|
this.EE.removeAllListeners();
|
|
});
|
|
}
|
|
|
|
postMessage(message: unknown) {
|
|
this.port.postMessage(message);
|
|
}
|
|
|
|
onMessage(callback: (message: unknown) => void) {
|
|
this.EE.on('message', callback);
|
|
}
|
|
|
|
onDisconnect(callback: () => void) {
|
|
this.EE.on('disconnect', callback);
|
|
}
|
|
|
|
disconnect() {
|
|
this.port.disconnect();
|
|
}
|
|
}
|