@effect-ts/system/_src/Effect/withChildren.ts

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

// ets_tracing: off

import * as SS from "../Collections/Immutable/SortedSet/index.js"
import type { Runtime } from "../Fiber/index.js"
import { track } from "../Supervisor/index.js"
import { chain_, supervised } from "./core.js"
import { descriptor } from "./descriptor.js"
import type { Effect, UIO } from "./effect.js"
import { map_ } from "./map.js"

/**
 * Locally installs a supervisor and an effect that succeeds with all the
 * children that have been forked in the returned effect.
 */
export function withChildren<R, E, A>(
  get: (_: UIO<SS.SortedSet<Runtime<any, any>>>) => Effect<R, E, A>,
  __trace?: string
) {
  return chain_(
    track,
    (supervisor) =>
      supervised(supervisor)(
        get(
          chain_(supervisor.value, (children) =>
            map_(descriptor, (d) => SS.filter_(children, (_) => _.id !== d.id))
          )
        )
      ),
    __trace
  )
}