UNPKG

@akala/core

Version:
175 lines 6.71 kB
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