fix shell command execution output markdown block issue

Author: yueny2020

packages/core/src/codewhispererChat/tools/executeBash.ts

@@ -117,6 +117,14 @@ export interface CommandValidation {
     warning?: string
 }
 
+// Interface for timestamped output chunks
+interface TimestampedChunk {
+    timestamp: number
+    isStdout: boolean
+    content: string
+    isFirst: boolean
+}
+
 export class ExecuteBash {
     private readonly command: string
     private readonly workingDirectory?: string
@@ -207,8 +215,33 @@ export class ExecuteBash {
             const stdoutBuffer: string[] = []
             const stderrBuffer: string[] = []
 
-            let firstChunk = true
-            let firstStderrChunk = true
+            // Use a queue to maintain chronological order of chunks
+            // This ensures that the output is processed in the exact order it was generated by the child process.
+            const outputQueue: TimestampedChunk[] = []
+            let processingQueue = false
+            const firstChunk = new AtomicBoolean(true)
+
+            // Process the queue in order
+            const processQueue = () => {
+                if (processingQueue || outputQueue.length === 0) {
+                    return
+                }
+
+                processingQueue = true
+
+                try {
+                    // Sort by timestamp to ensure chronological order
+                    outputQueue.sort((a, b) => a.timestamp - b.timestamp)
+
+                    while (outputQueue.length > 0) {
+                        const chunk = outputQueue.shift()!
+                        ExecuteBash.handleTimestampedChunk(chunk, stdoutBuffer, stderrBuffer, updates)
+                    }
+                } finally {
+                    processingQueue = false
+                }
+            }
+
             const childProcessOptions: ChildProcessOptions = {
                 spawnOptions: {
                     cwd: this.workingDirectory,
@@ -217,12 +250,26 @@ export class ExecuteBash {
                 collect: false,
                 waitForStreams: true,
                 onStdout: (chunk: string) => {
-                    ExecuteBash.handleChunk(firstChunk ? '```console\n' + chunk : chunk, stdoutBuffer, updates)
-                    firstChunk = false
+                    const isFirst = firstChunk.getAndSet(false)
+                    const timestamp = Date.now()
+                    outputQueue.push({
+                        timestamp,
+                        isStdout: true,
+                        content: chunk,
+                        isFirst,
+                    })
+                    processQueue()
                 },
                 onStderr: (chunk: string) => {
-                    ExecuteBash.handleChunk(firstStderrChunk ? '```console\n' + chunk : chunk, stderrBuffer, updates)
-                    firstStderrChunk = false
+                    const isFirst = firstChunk.getAndSet(false)
+                    const timestamp = Date.now()
+                    outputQueue.push({
+                        timestamp,
+                        isStdout: false,
+                        content: chunk,
+                        isFirst,
+                    })
+                    processQueue()
                 },
             }
 
@@ -261,6 +308,17 @@ export class ExecuteBash {
         })
     }
 
+    private static handleTimestampedChunk(
+        chunk: TimestampedChunk,
+        stdoutBuffer: string[],
+        stderrBuffer: string[],
+        updates?: Writable
+    ): void {
+        const buffer = chunk.isStdout ? stdoutBuffer : stderrBuffer
+        const content = chunk.isFirst ? '```console\n' + chunk.content : chunk.content
+        ExecuteBash.handleChunk(content, buffer, updates)
+    }
+
     private static handleChunk(chunk: string, buffer: string[], updates?: Writable) {
         try {
             updates?.write(chunk)
@@ -307,3 +365,17 @@ export class ExecuteBash {
         updates.end()
     }
 }
+
+class AtomicBoolean {
+    private value: boolean
+
+    constructor(initialValue: boolean) {
+        this.value = initialValue
+    }
+
+    public getAndSet(newValue: boolean): boolean {
+        const oldValue = this.value
+        this.value = newValue
+        return oldValue
+    }
+}