//go:generate $WIT_BINDGEN_WRPC go --world async-server --out-dir bindings/async_server --package github.com/wrpc/wit-bindgen-wrpc/tests/go/bindings/async_server ../wit

package integration

import (
	"bytes"
	"context"
	"log/slog"

	wrpc "github.com/wrpc/wrpc/go"
)

type AsyncHandler struct{}

func (AsyncHandler) WithStreams(ctx context.Context, complete bool) (wrpc.ReadCompleter, wrpc.ReceiveCompleter[[][]string], error) {
	slog.DebugContext(ctx, "handling `with-streams`", "complete", complete)
	buf := bytes.NewBuffer([]byte("test"))
	str := wrpc.NewCompleteReceiver([][]string{{"foo", "bar"}, {"baz"}})
	if complete {
		return wrpc.NewCompleteReader(buf), str, nil
	} else {
		return wrpc.NewPendingByteReader(buf), wrpc.NewPendingReceiver(str), nil
	}
}