package cmd

import (
	"context"
	"log"
	"sync"
)

// webrtcRegistry tracks per-session cancel funcs for active custom WebRTC
// streams (engine.RunWebRTCStream goroutines). Each session lives only as
// long as its DataChannel; the registry exists so duplicate sync responses
// don't double-spawn the same session and so daemon shutdown can drain.
var webrtcRegistry = &webrtcSessionRegistry{
	cancels: make(map[string]context.CancelFunc),
}

type webrtcSessionRegistry struct {
	mu      sync.Mutex
	cancels map[string]context.CancelFunc
}

func (r *webrtcSessionRegistry) has(sessionID string) bool {
	r.mu.Lock()
	defer r.mu.Unlock()
	_, ok := r.cancels[sessionID]
	return ok
}

func (r *webrtcSessionRegistry) add(sessionID string, cancel context.CancelFunc) {
	r.mu.Lock()
	defer r.mu.Unlock()
	r.cancels[sessionID] = cancel
}

func (r *webrtcSessionRegistry) remove(sessionID string) {
	r.mu.Lock()
	defer r.mu.Unlock()
	delete(r.cancels, sessionID)
}

// cancelAllWebRTCSessions cancels every running session. Called on daemon
// shutdown so pion peers and SSE consumers exit cleanly.
func cancelAllWebRTCSessions() {
	webrtcRegistry.mu.Lock()
	cancels := make([]context.CancelFunc, 0, len(webrtcRegistry.cancels))
	for _, c := range webrtcRegistry.cancels {
		cancels = append(cancels, c)
	}
	webrtcRegistry.cancels = make(map[string]context.CancelFunc)
	webrtcRegistry.mu.Unlock()
	for _, c := range cancels {
		c()
	}
}

// stdLogger is a tiny adapter so engine.RunWebRTCStream can log through the
// standard library logger without pulling in a logging dependency.
type stdLogger struct{}

func (stdLogger) Infof(format string, args ...any)  { log.Printf(format, args...) }
func (stdLogger) Warnf(format string, args ...any)  { log.Printf("WARN: "+format, args...) }
func (stdLogger) Errorf(format string, args ...any) { log.Printf("ERROR: "+format, args...) }