@effect-ts/system/_src/Experimental/Stream/_internal/api/takeUntil.ts

Effect-TS is a zero dependency set of libraries to write highly productive, purely functional TypeScript at scale.

// ets_tracing: off

import * as CK from "../../../../Collections/Immutable/Chunk/index.js"
import type { Predicate } from "../../../../Function/index.js"
import * as CH from "../../Channel/index.js"
import * as C from "../core.js"

/**
 * Takes all elements of the stream until the specified predicate evaluates
 * to `true`.
 */
export function takeUntil_<R, E, A>(
  self: C.Stream<R, E, A>,
  f: Predicate<A>
): C.Stream<R, E, A> {
  const loop: CH.Channel<R, E, CK.Chunk<A>, unknown, E, CK.Chunk<A>, any> = CH.readWith(
    (chunk) => {
      const taken = CK.takeWhile_(chunk, (_) => !f(_))
      const last = CK.take_(CK.drop_(chunk, CK.size(taken)), 1)

      if (CK.isEmpty(last)) {
        return CH.zipRight_(CH.write(taken), loop)
      } else {
        return CH.write(CK.concat_(taken, last))
      }
    },
    (_) => CH.fail(_),
    (_) => CH.succeed(_)
  )

  return new C.Stream(self.channel[">>>"](loop))
}

/**
 * Takes all elements of the stream until the specified predicate evaluates
 * to `true`.
 *
 * @ets_data_first takeUntil_
 */
export function takeUntil<A>(f: Predicate<A>) {
  return <R, E>(self: C.Stream<R, E, A>) => takeUntil_(self, f)
}