fix(cli): improve stdin handling and add initial state check (#6747)

packages/cli/src/utils/readStdin.ts

@@ -9,40 +9,86 @@ export async function readStdin(): Promise<string> {
   return new Promise((resolve, reject) => {
     let data = '';
     let totalSize = 0;
+    let hasReceivedData = false;
+
     process.stdin.setEncoding('utf8');
 
-    const onReadable = () => {
+    function cleanup() {
-      let chunk;
+      clearTimeout(timeout);
-      while ((chunk = process.stdin.read()) !== null) {
+      process.stdin.removeListener('readable', onReadable);
+      process.stdin.removeListener('end', onEnd);
+      process.stdin.removeListener('error', onError);
+    }
+
+    function processChunk(chunk: string): boolean {
+      hasReceivedData = true;
       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
+        process.stdin.destroy();
-          break;
+        return true; // Indicates truncation occurred
-        }
+      } else {
         data += chunk;
         totalSize += chunk.length;
+        return false;
+      }
     }
-    };
 
-    const onEnd = () => {
+    function checkInitialState(): boolean {
+      if (process.stdin.destroyed || process.stdin.readableEnded) {
+        cleanup();
+        resolve('');
+        return true;
+      }
+
+      const chunk = process.stdin.read();
+      if (chunk !== null) {
+        processChunk(chunk);
+        return false;
+      }
+
+      if (!process.stdin.readable) {
+        cleanup();
+        resolve('');
+        return true;
+      }
+
+      return false;
+    }
+
+    if (checkInitialState()) {
+      return;
+    }
+
+    function onReadable() {
+      let chunk;
+      while ((chunk = process.stdin.read()) !== null) {
+        const truncated = processChunk(chunk);
+        if (truncated) {
+          break;
+        }
+      }
+    }
+
+    function onEnd() {
       cleanup();
       resolve(data);
-    };
+    }
 
-    const onError = (err: Error) => {
+    function onError(err: Error) {
       cleanup();
       reject(err);
-    };
+    }
 
-    const cleanup = () => {
+    const timeout = setTimeout(() => {
-      process.stdin.removeListener('readable', onReadable);
+      if (!hasReceivedData) {
-      process.stdin.removeListener('end', onEnd);
+        cleanup();
-      process.stdin.removeListener('error', onError);
+        resolve('');
-    };
+      }
+    }, 50);
 
     process.stdin.on('readable', onReadable);
     process.stdin.on('end', onEnd);