WebRTC

Overlord-Client/cmd/agent/webrtcpub/state.go

@@ -0,0 +1,85 @@
+package webrtcpub
+
+import (
+	"sync"
+	"sync/atomic"
+	"time"
+)
+
+type FrameWriter interface {
+	WriteH264(nalu []byte, dur time.Duration) error
+}
+
+var (
+	writersMu sync.RWMutex
+	writers   = map[string]FrameWriter{}
+)
+
+func registerWriter(id string, w FrameWriter) {
+	if id == "" || w == nil {
+		return
+	}
+	writersMu.Lock()
+	writers[id] = w
+	writersMu.Unlock()
+}
+
+func unregisterWriter(id string) {
+	if id == "" {
+		return
+	}
+	writersMu.Lock()
+	delete(writers, id)
+	writersMu.Unlock()
+}
+
+func IsActive() bool {
+	writersMu.RLock()
+	defer writersMu.RUnlock()
+	return len(writers) > 0
+}
+
+var keyframeWanted atomic.Bool
+
+func RequestKeyframe() {
+	keyframeWanted.Store(true)
+}
+
+func ConsumeKeyframeRequest() bool {
+	return keyframeWanted.Swap(false)
+}
+
+func WriteH264(nalu []byte, dur time.Duration) error {
+	if len(nalu) == 0 {
+		return nil
+	}
+	writersMu.RLock()
+	defer writersMu.RUnlock()
+	for _, w := range writers {
+		_ = w.WriteH264(nalu, dur)
+	}
+	return nil
+}
+
+// Deadass fuck WebRTC bro ts is so confusing
+type Options struct {
+	// (e.g. https://server:5173/api/webrtc/agents/abc/whip).
+	WhipURL string
+	// PublishToken is the bearer token issued by the server.
+	PublishToken string
+	// TLSInsecureSkipVerify mirrors the agent's existing TLS config.
+	TLSInsecureSkipVerify bool
+	// TLSCAPath is an optional custom CA bundle.
+	TLSCAPath string
+}
+
+type ICECandidate struct {
+	Candidate     string `msgpack:"candidate"`
+	SDPMid        string `msgpack:"sdpMid"`
+	SDPMLineIndex uint16 `msgpack:"sdpMLineIndex"`
+}
+
+type P2POfferCallbacks struct {
+	OnICE   func(c ICECandidate)
+	OnClose func()
+}