@akala/core
Version:
175 lines • 6.71 kB
JavaScript
import { EventEmitter } from "../events/event-emitter.js";
import { IsomorphicBuffer } from "../helpers.js";
import { StatefulSubscription } from "../teardown-manager.js";
import { Deferred } from "../promiseHelpers.js";
export class ProcessStdioAdapter extends EventEmitter {
process;
messageEvent;
openEvent;
closeEvent;
errorEvent;
get open() { return !!this.process.pid; }
pipe(socket) {
this.on('message', (message) => socket.send(message));
this.on('close', () => socket.close());
}
async close() {
this.off('message');
this.off('error');
this.closeEvent(null);
}
send(data) {
if (typeof data === 'string')
return new Promise((resolve, reject) => this.process.stdout.write(data, err => err ? reject(err) : resolve()));
return new Promise((resolve, reject) => this.process.stdout.write(data.toArray(), err => err ? reject(err) : resolve()));
}
messageListeners = [];
off(event, handler) {
switch (event) {
case 'message':
{
let listeners = this.messageListeners;
if (handler)
listeners = listeners.filter(f => f[0] == handler);
var result = false;
for (const listener of listeners) {
this.process.stdin.off('data', listener[1]);
result = !!this.messageListeners.splice(this.messageListeners.indexOf(listener), 1)?.length || result;
}
}
break;
case 'open':
return false;
case 'close':
case 'error':
//eslint-disable-next-line @typescript-eslint/no-explicit-any
this.process.stdout.off(event, handler);
break;
default:
throw new Error(`Unsupported event ${String(event)}`);
}
return true;
}
on(event, handler, options) {
switch (event) {
case 'message':
{
const x = function (data) { return handler.call(this, typeof data === 'string' ? data : IsomorphicBuffer.fromBuffer(data)); };
this.messageListeners.push([handler, x]);
if (options?.once)
this.process.stdin.once('data', x);
else
this.process.stdin.on('data', x);
return new StatefulSubscription(() => {
this.messageListeners.splice(this.messageListeners.findIndex(x => x[0] === handler), 1);
this.process.stdin.off('message', x);
}).unsubscribe;
}
case 'open':
handler(null);
break;
case 'close':
case 'error':
if (options?.once)
this.process.stdin.once(event, handler);
else
this.process.stdin.on(event, handler);
return new StatefulSubscription(() => {
this.process.stdin.off(event, handler);
}).unsubscribe;
case Symbol.dispose:
return super.on(event, handler, options);
default:
throw new Error(`Unsupported event ${String(event)}`);
}
}
once(event, handler) {
switch (event) {
case 'message':
return this.on(event, handler, { once: true });
case 'close':
case 'error':
case 'open':
case Symbol.dispose:
return this.on(event, handler, { once: true });
default:
let x = event;
throw new Error(`Unsupported event ${x}`);
}
}
constructor(process, abort) {
super();
this.process = process;
abort.addEventListener('abort', () => {
this.close();
});
this.messageEvent = this.getOrCreate('message').emit.bind(this.get('message'));
this.closeEvent = this.getOrCreate('close').emit.bind(this.get('close'));
this.errorEvent = this.getOrCreate('error').emit.bind(this.get('error'));
}
}
export class ChildProcessStdioAdapter extends EventEmitter {
process;
messageEvent;
openEvent;
closeEvent;
get open() { return !!this.process.pid; }
pipe(socket) {
this.on('message', (message) => socket.send(message));
this.on('close', () => socket.close());
}
close() {
const deferred = new Deferred();
this.process.on('disconnect', () => deferred.resolve());
this.process.disconnect();
return deferred;
}
send(data) {
return new Promise((resolve, reject) => this.process.stdin.write(data + '\n', err => err ? reject(err) : resolve()));
}
off(event, handler) {
const result = super.off(event, handler);
if (!super.hasListener(event))
switch (event) {
case 'message':
this.process.stdout.off('data', this.messageEvent);
break;
case 'close':
this.process.stdout.off('end', this.closeEvent);
break;
}
return result;
}
on(event, handler, options) {
if (!this.events[event].hasListeners)
switch (event) {
case 'message':
if (options?.once)
this.process.stdout.once('data', this.messageEvent);
else
this.process.stdout.on('data', this.messageEvent);
break;
case 'open':
handler(null);
return () => false;
case 'close':
if (options?.once)
this.process.stdout.once('close', this.closeEvent);
else
this.process.stdout.on('close', this.closeEvent);
break;
}
return super.on(event, handler, options);
}
once(event, handler, options) {
return this.on(event, handler, { ...options, once: true });
}
constructor(process, abort) {
super();
this.process = process;
abort.addEventListener('abort', () => this[Symbol.dispose]());
this.messageEvent = this.getOrCreate('message').emit.bind(this.get('message'));
this.closeEvent = this.getOrCreate('close').emit.bind(this.get('close'));
}
}
//# sourceMappingURL=stdio.js.map