effect/src/internal/singleShotGen.ts

The missing standard library for TypeScript, for writing production-grade software.

/** @internal */
export class SingleShotGen<T, A> implements Generator<T, A> {
  called = false

  constructor(readonly self: T) {
  }

  next(a: A): IteratorResult<T, A> {
    return this.called ?
      ({
        value: a,
        done: true
      }) :
      (this.called = true,
        ({
          value: this.self,
          done: false
        }))
  }

  return(a: A): IteratorResult<T, A> {
    return ({
      value: a,
      done: true
    })
  }

  throw(e: unknown): IteratorResult<T, A> {
    throw e
  }

  [Symbol.iterator](): Generator<T, A> {
    return new SingleShotGen<T, A>(this.self)
  }
}