Add projects

backend/internal/pluginhost/callback_contexts.go

@@ -0,0 +1,139 @@
+package pluginhost
+
+import (
+	"context"
+	"strconv"
+	"strings"
+	"sync"
+	"sync/atomic"
+)
+
+type callbackContextRegistry struct {
+	next     atomic.Uint64
+	mu       sync.RWMutex
+	contexts map[string]callbackContextEntry
+}
+
+type callbackContextEntry struct {
+	ctx      context.Context
+	pluginID string
+	cleanup  []func()
+}
+
+func newCallbackContextRegistry() *callbackContextRegistry {
+	return &callbackContextRegistry{contexts: make(map[string]callbackContextEntry)}
+}
+
+func (r *callbackContextRegistry) open(ctx context.Context, pluginID string) (string, func()) {
+	if r == nil {
+		return "", func() {}
+	}
+	if ctx == nil {
+		ctx = context.Background()
+	}
+	pluginID = strings.TrimSpace(pluginID)
+	ctx = withHostCallbackPluginID(ctx, pluginID)
+	id := strconv.FormatUint(r.next.Add(1), 10)
+	r.mu.Lock()
+	r.contexts[id] = callbackContextEntry{ctx: ctx, pluginID: pluginID}
+	r.mu.Unlock()
+
+	var once sync.Once
+	return id, func() {
+		once.Do(func() {
+			var cleanup []func()
+			r.mu.Lock()
+			entry := r.contexts[id]
+			delete(r.contexts, id)
+			r.mu.Unlock()
+			cleanup = entry.cleanup
+			for _, fn := range cleanup {
+				if fn != nil {
+					fn()
+				}
+			}
+		})
+	}
+}
+
+func (r *callbackContextRegistry) pluginID(id string) string {
+	if r == nil || id == "" {
+		return ""
+	}
+	r.mu.RLock()
+	entry := r.contexts[id]
+	r.mu.RUnlock()
+	return strings.TrimSpace(entry.pluginID)
+}
+
+func (r *callbackContextRegistry) addCleanup(id string, cleanup func()) bool {
+	if r == nil || id == "" || cleanup == nil {
+		return false
+	}
+	r.mu.Lock()
+	entry, ok := r.contexts[id]
+	if ok {
+		entry.cleanup = append(entry.cleanup, cleanup)
+		r.contexts[id] = entry
+	}
+	r.mu.Unlock()
+	if !ok {
+		cleanup()
+		return false
+	}
+	return true
+}
+
+func (r *callbackContextRegistry) resolve(id string, fallback context.Context) context.Context {
+	if fallback == nil {
+		fallback = context.Background()
+	}
+	if r == nil || id == "" {
+		return fallback
+	}
+	r.mu.RLock()
+	ctx := r.contexts[id].ctx
+	r.mu.RUnlock()
+	if ctx == nil {
+		return fallback
+	}
+	return ctx
+}
+
+func (h *Host) openCallbackContext(ctx context.Context) (string, func()) {
+	return h.openCallbackContextForPlugin(ctx, "")
+}
+
+func (h *Host) openCallbackContextForPlugin(ctx context.Context, pluginID string) (string, func()) {
+	if h == nil || h.callbackContexts == nil {
+		return "", func() {}
+	}
+	return h.callbackContexts.open(ctx, pluginID)
+}
+
+func (h *Host) addCallbackCleanup(id string, cleanup func()) bool {
+	if h == nil || h.callbackContexts == nil {
+		if id != "" && cleanup != nil {
+			cleanup()
+		}
+		return false
+	}
+	return h.callbackContexts.addCleanup(id, cleanup)
+}
+
+func (h *Host) resolveCallbackContext(id string, fallback context.Context) context.Context {
+	if h == nil || h.callbackContexts == nil {
+		if fallback == nil {
+			return context.Background()
+		}
+		return fallback
+	}
+	return h.callbackContexts.resolve(id, fallback)
+}
+
+func (h *Host) callbackContextPluginID(id string) string {
+	if h == nil || h.callbackContexts == nil {
+		return ""
+	}
+	return h.callbackContexts.pluginID(id)
+}