mirror of https://github.com/laurent22/joplin.git
55 lines
1.0 KiB
TypeScript
55 lines
1.0 KiB
TypeScript
import { createInterface } from 'readline/promises';
|
|
|
|
const iterateStdin = async function*(prompt: string) {
|
|
let nextLineListeners: (()=> void)[] = [];
|
|
const dispatchAllListeners = () => {
|
|
const listeners = nextLineListeners;
|
|
nextLineListeners = [];
|
|
for (const listener of listeners) {
|
|
listener();
|
|
}
|
|
};
|
|
|
|
const rl = createInterface({
|
|
input: process.stdin,
|
|
output: process.stdout,
|
|
});
|
|
rl.setPrompt(prompt);
|
|
|
|
let buffer: string[] = [];
|
|
rl.on('line', (line) => {
|
|
buffer.push(line);
|
|
dispatchAllListeners();
|
|
});
|
|
|
|
let done = false;
|
|
rl.on('close', () => {
|
|
done = true;
|
|
dispatchAllListeners();
|
|
});
|
|
|
|
const readNextLines = () => {
|
|
return new Promise<string|null>(resolve => {
|
|
if (done) {
|
|
resolve(null);
|
|
} else if (buffer.length > 0) {
|
|
resolve(buffer.join('\n'));
|
|
buffer = [];
|
|
} else {
|
|
nextLineListeners.push(() => {
|
|
resolve(buffer.join('\n'));
|
|
buffer = [];
|
|
});
|
|
}
|
|
});
|
|
};
|
|
|
|
while (!done) {
|
|
rl.prompt();
|
|
const lines = await readNextLines();
|
|
yield lines;
|
|
}
|
|
};
|
|
|
|
export default iterateStdin;
|