import { Effect, Fiber, PubSub } from "effect"

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

  // Publish multiple messages at once
  const messages = ["Hello", "World", "from", "Effect"]
  const allPublished = yield* PubSub.publishAll(pubsub, messages)
  console.log("All messages published:", allPublished) // true

  // With a smaller capacity and an active subscription
  const smallPubsub = yield* PubSub.bounded<string>(2)
  const manyMessages = ["msg1", "msg2", "msg3", "msg4"]

  yield* Effect.scoped(Effect.gen(function*() {
    const subscription = yield* PubSub.subscribe(smallPubsub)

    // Will suspend until space becomes available for all messages
    const fiber = yield* Effect.forkChild(PubSub.publishAll(smallPubsub, manyMessages))

    const firstBatch = yield* PubSub.takeBetween(subscription, 2, 2)
    console.log("First batch:", firstBatch) // ["msg1", "msg2"]

    const result = yield* Fiber.join(fiber)
    console.log("All messages eventually published:", result) // true

    const secondBatch = yield* PubSub.takeAll(subscription)
    console.log("Second batch:", secondBatch) // ["msg3", "msg4"]
  }))
})