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;