import { Effect, PubSub } from "effect"

const program = Effect.gen(function*() {
  const pubsub = yield* PubSub.bounded<string>(10)

  // Subscribe within a scope for automatic cleanup
  yield* Effect.scoped(Effect.gen(function*() {
    const subscription: PubSub.Subscription<string> = yield* PubSub.subscribe(
      pubsub
    )

    yield* PubSub.publishAll(pubsub, ["msg1", "msg2", "msg3"])

    // Take individual messages
    const message = yield* PubSub.take(subscription)
    console.log(message) // "msg1"

    // Take multiple messages
    const messages = yield* PubSub.takeUpTo(subscription, 1)
    console.log(messages) // ["msg2"]
    const allMessages = yield* PubSub.takeAll(subscription)
    console.log(allMessages) // ["msg3"]
  }))
})