feat: Allow combining -p and stdin for prompt input (#4406)

Co-authored-by: Allen Hutchison <adh@google.com>
This commit is contained in:
Wietse Venema 2025-08-14 20:08:59 +02:00 committed by GitHub
parent 798c4d1311
commit ec7b84191f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 2 deletions

View File

@ -297,8 +297,11 @@ export async function main() {
}
// If not a TTY, read from stdin
// This is for cases where the user pipes input directly into the command
if (!process.stdin.isTTY && !input) {
input += await readStdin();
if (!process.stdin.isTTY) {
const stdinData = await readStdin();
if (stdinData) {
input = `${stdinData}\n\n${input}`;
}
}
if (!input) {
console.error('No input provided via stdin.');

View File

@ -5,14 +5,26 @@
*/
export async function readStdin(): Promise<string> {
const MAX_STDIN_SIZE = 8 * 1024 * 1024; // 8MB
return new Promise((resolve, reject) => {
let data = '';
let totalSize = 0;
process.stdin.setEncoding('utf8');
const onReadable = () => {
let chunk;
while ((chunk = process.stdin.read()) !== null) {
if (totalSize + chunk.length > MAX_STDIN_SIZE) {
const remainingSize = MAX_STDIN_SIZE - totalSize;
data += chunk.slice(0, remainingSize);
console.warn(
`Warning: stdin input truncated to ${MAX_STDIN_SIZE} bytes.`,
);
process.stdin.destroy(); // Stop reading further
break;
}
data += chunk;
totalSize += chunk.length;
}
};