Something went wrong. Try again.
source dump of claude code forked from oppi.li/claude-code
Something went wrong. Try again.
1.9 kB · 76 lines
TypeScript
at main
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677export class Stream<T> implements AsyncIterator<T> { private readonly queue: T[] = [] private readResolve?: (value: IteratorResult<T>) => void private readReject?: (error: unknown) => void private isDone: boolean = false private hasError: unknown | undefined private started = false
constructor(private readonly returned?: () => void) {}
[Symbol.asyncIterator](): AsyncIterableIterator<T> { if (this.started) { throw new Error('Stream can only be iterated once') } this.started = true return this }
next(): Promise<IteratorResult<T, unknown>> { if (this.queue.length > 0) { return Promise.resolve({ done: false, value: this.queue.shift()!, }) } if (this.isDone) { return Promise.resolve({ done: true, value: undefined }) } if (this.hasError) { return Promise.reject(this.hasError) } return new Promise<IteratorResult<T>>((resolve, reject) => { this.readResolve = resolve this.readReject = reject }) }
enqueue(value: T): void { if (this.readResolve) { const resolve = this.readResolve this.readResolve = undefined this.readReject = undefined resolve({ done: false, value }) } else { this.queue.push(value) } }
done() { this.isDone = true if (this.readResolve) { const resolve = this.readResolve this.readResolve = undefined this.readReject = undefined resolve({ done: true, value: undefined }) } }
error(error: unknown) { this.hasError = error if (this.readReject) { const reject = this.readReject this.readResolve = undefined this.readReject = undefined reject(error) } }
return(): Promise<IteratorResult<T, unknown>> { this.isDone = true if (this.returned) { this.returned() } return Promise.resolve({ done: true, value: undefined }) }}