Roo-Code/apps/cli/scripts/test-stdin-stream.ts

import path from "path"
import { fileURLToPath } from "url"
import readline from "readline"

import { execa } from "execa"

const __dirname = path.dirname(fileURLToPath(import.meta.url))
const cliRoot = path.resolve(__dirname, "..")

async function main() {
	const child = execa(
		"pnpm",
		["dev", "--print", "--stdin-prompt-stream", "--provider", "roo", "--output-format", "stream-json"],
		{
			cwd: cliRoot,
			stdin: "pipe",
			stdout: "pipe",
			stderr: "pipe",
			reject: false,
			forceKillAfterDelay: 2_000,
		},
	)

	child.stdout?.on("data", (chunk) => process.stdout.write(chunk))
	child.stderr?.on("data", (chunk) => process.stderr.write(chunk))

	console.log("[wrapper] Type a message and press Enter to send it.")
	console.log("[wrapper] Type /exit to close stdin and let the CLI finish.")

	const rl = readline.createInterface({
		input: process.stdin,
		output: process.stdout,
		terminal: true,
	})

	rl.on("line", (line) => {
		if (line.trim() === "/exit") {
			console.log("[wrapper] Closing stdin...")
			child.stdin?.end()
			rl.close()
			return
		}

		if (!child.stdin?.destroyed) {
			child.stdin?.write(`${line}\n`)
		}
	})

	const onSignal = (signal: NodeJS.Signals) => {
		console.log(`[wrapper] Received ${signal}, forwarding to CLI...`)
		rl.close()
		child.kill(signal)
	}

	process.on("SIGINT", () => onSignal("SIGINT"))
	process.on("SIGTERM", () => onSignal("SIGTERM"))

	const result = await child
	rl.close()
	console.log(`[wrapper] CLI exited with code ${result.exitCode}`)
	process.exit(result.exitCode ?? 1)
}

main().catch((error) => {
	console.error("[wrapper] Fatal error:", error)
	process.exit(1)
})