Add projects

backend/internal/pluginhost/http_stream_bridge.go

@@ -0,0 +1,83 @@
+package pluginhost
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+	"sync"
+	"sync/atomic"
+
+	"github.com/router-for-me/CLIProxyAPI/v7/sdk/pluginapi"
+)
+
+type hostHTTPStreamBridge struct {
+	next    atomic.Uint64
+	mu      sync.Mutex
+	streams map[string]hostHTTPStreamEntry
+}
+
+type hostHTTPStreamEntry struct {
+	chunks <-chan pluginapi.HTTPStreamChunk
+	cancel context.CancelFunc
+}
+
+func newHostHTTPStreamBridge() *hostHTTPStreamBridge {
+	return &hostHTTPStreamBridge{streams: make(map[string]hostHTTPStreamEntry)}
+}
+
+func (b *hostHTTPStreamBridge) open(chunks <-chan pluginapi.HTTPStreamChunk, cancel context.CancelFunc) string {
+	if b == nil || chunks == nil {
+		if cancel != nil {
+			cancel()
+		}
+		return ""
+	}
+	id := strconv.FormatUint(b.next.Add(1), 10)
+	b.mu.Lock()
+	b.streams[id] = hostHTTPStreamEntry{chunks: chunks, cancel: cancel}
+	b.mu.Unlock()
+	return id
+}
+
+func (b *hostHTTPStreamBridge) read(ctx context.Context, id string) (pluginapi.HTTPStreamChunk, bool, error) {
+	if b == nil || id == "" {
+		return pluginapi.HTTPStreamChunk{}, true, fmt.Errorf("http stream id is required")
+	}
+	b.mu.Lock()
+	entry := b.streams[id]
+	b.mu.Unlock()
+	if entry.chunks == nil {
+		return pluginapi.HTTPStreamChunk{}, true, fmt.Errorf("http stream %s is not open", id)
+	}
+	if ctx == nil {
+		ctx = context.Background()
+	}
+	select {
+	case <-ctx.Done():
+		b.close(id)
+		return pluginapi.HTTPStreamChunk{}, true, ctx.Err()
+	case chunk, ok := <-entry.chunks:
+		if !ok {
+			b.close(id)
+			return pluginapi.HTTPStreamChunk{}, true, nil
+		}
+		if chunk.Err != nil {
+			b.close(id)
+			return chunk, true, nil
+		}
+		return chunk, false, nil
+	}
+}
+
+func (b *hostHTTPStreamBridge) close(id string) {
+	if b == nil || id == "" {
+		return
+	}
+	b.mu.Lock()
+	entry := b.streams[id]
+	delete(b.streams, id)
+	b.mu.Unlock()
+	if entry.cancel != nil {
+		entry.cancel()
+	}
+}