Use a closure boolean value to check the first chunk

Author: yueny2020

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

@@ -215,11 +215,18 @@ export class ExecuteBash {
             const stdoutBuffer: string[] = []
             const stderrBuffer: string[] = []
 
+            // Use a closure boolean value firstChunk and a function to get and set its value atomically
+            let isFirstChunk = true
+            const getAndSetFirstChunk = (newValue: boolean): boolean => {
+                const oldValue = isFirstChunk
+                isFirstChunk = newValue
+                return oldValue
+            }
+
             // 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 = () => {
@@ -249,8 +256,8 @@ export class ExecuteBash {
                 },
                 collect: false,
                 waitForStreams: true,
-                onStdout: (chunk: string) => {
-                    const isFirst = firstChunk.getAndSet(false)
+                onStdout: async (chunk: string) => {
+                    const isFirst = getAndSetFirstChunk(false)
                     const timestamp = Date.now()
                     outputQueue.push({
                         timestamp,
@@ -260,8 +267,8 @@ export class ExecuteBash {
                     })
                     processQueue()
                 },
-                onStderr: (chunk: string) => {
-                    const isFirst = firstChunk.getAndSet(false)
+                onStderr: async (chunk: string) => {
+                    const isFirst = getAndSetFirstChunk(false)
                     const timestamp = Date.now()
                     outputQueue.push({
                         timestamp,
@@ -365,17 +372,3 @@ 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
-    }
-}