debug: add Go Live connection tracing and backend data flow logging

Author: Datanoise

relay/webrtc.go

@@ -13,6 +13,7 @@ import (
 	"github.com/kazzmir/opus-go/ogg"
 	"github.com/pion/webrtc/v4"
 	"github.com/pion/webrtc/v4/pkg/media"
+	"github.com/pion/webrtc/v4/pkg/media/oggwriter"
 	"github.com/sirupsen/logrus"
 )
 
@@ -115,6 +116,93 @@ func (wm *WebRTCManager) HandleOffer(mount string, offer webrtc.SessionDescripti
 	return peerConnection.LocalDescription(), nil
 }
 
+type relayWriter struct {
+	relay  *Relay
+	stream *Stream
+}
+
+func (rw *relayWriter) Write(p []byte) (n int, err error) {
+	logrus.Debugf("relayWriter: Broadcasting %d bytes to %s", len(p), rw.stream.MountName)
+	rw.stream.Broadcast(p, rw.relay)
+	return len(p), nil
+}
+
+func (wm *WebRTCManager) HandleSourceOffer(mount string, offer webrtc.SessionDescription) (*webrtc.SessionDescription, error) {
+	peerConnection, err := wm.api.NewPeerConnection(webrtc.Configuration{
+		ICEServers: []webrtc.ICEServer{
+			{URLs: []string{"stun:stun.l.google.com:19302"}},
+		},
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	peerConnection.OnTrack(func(track *webrtc.TrackRemote, receiver *webrtc.RTPReceiver) {
+		logrus.Infof("WebRTC Source: Received track %s from %s", track.ID(), mount)
+		
+		stream := wm.relay.GetOrCreateStream(mount)
+		stream.mu.Lock()
+		stream.ContentType = "audio/ogg" 
+		stream.IsOggStream = true
+		stream.SourceIP = "webrtc-source"
+		// Reset page offsets so we don't sync to old pages from a previous session
+		stream.PageOffsets = make([]int64, 128)
+		stream.PageIndex = 0
+		stream.OggHeaderOffset = stream.Buffer.Head
+		stream.mu.Unlock()
+
+		// Capture headers written by oggwriter.NewWith
+		var headerBuf bytes.Buffer
+		multi := io.MultiWriter(&headerBuf, &relayWriter{wm.relay, stream})
+
+		writer, err := oggwriter.NewWith(multi, 48000, 2)
+		if err != nil {
+			logrus.WithError(err).Error("Failed to create Ogg writer for WebRTC source")
+			return
+		}
+		defer writer.Close()
+
+		// Save captured headers immediately
+		stream.mu.Lock()
+		h := make([]byte, headerBuf.Len())
+		copy(h, headerBuf.Bytes())
+		stream.OggHead = h
+		logrus.Infof("WebRTC Source: Captured %d bytes of Ogg/Opus headers at offset %d", len(h), stream.OggHeaderOffset)
+		stream.mu.Unlock()
+
+		for {
+			rtpPacket, _, err := track.ReadRTP()
+			if err != nil {
+				if err != io.EOF {
+					logrus.WithError(err).Error("Error reading RTP packet from source")
+				}
+				return
+			}
+			if err := writer.WriteRTP(rtpPacket); err != nil {
+				logrus.WithError(err).Error("Error writing RTP to Ogg muxer")
+				return
+			}
+		}
+	})
+
+	if err = peerConnection.SetRemoteDescription(offer); err != nil {
+		return nil, err
+	}
+
+	answer, err := peerConnection.CreateAnswer(nil)
+	if err != nil {
+		return nil, err
+	}
+
+	gatherComplete := webrtc.GatheringCompletePromise(peerConnection)
+	if err = peerConnection.SetLocalDescription(answer); err != nil {
+		return nil, err
+	}
+	<-gatherComplete
+
+	return peerConnection.LocalDescription(), nil
+}
+
 func (wm *WebRTCManager) streamToTrack(pc *webrtc.PeerConnection, track *webrtc.TrackLocalStaticSample, stream *Stream) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()

server/server.go

@@ -287,6 +287,8 @@ func (s *Server) dispatchWebhook(event string, data map[string]interface{}) {
 func (s *Server) setupRoutes() *http.ServeMux {
 	mux := http.NewServeMux()
 	mux.HandleFunc("/admin", s.handleAdmin)
+	mux.HandleFunc("/admin/golive", s.handleGoLive)
+	mux.HandleFunc("/admin/golive/chunk", s.handleGoLiveChunk)
 	mux.HandleFunc("/admin/add-mount", s.handleAddMount)
 	mux.HandleFunc("/admin/toggle-latency", s.handleToggleLatency)
 	mux.HandleFunc("/admin/stats", s.handleStats)
@@ -344,6 +346,7 @@ func (s *Server) setupRoutes() *http.ServeMux {
 	mux.HandleFunc("/logout", s.handleLogout)
 	mux.HandleFunc("/explore", s.handleExplore)
 	mux.HandleFunc("/webrtc/offer", s.handleWebRTCOffer)
+	mux.HandleFunc("/webrtc/source-offer", s.handleWebRTCSourceOffer)
 	mux.HandleFunc("/player/", s.handlePlayer)
 	mux.HandleFunc("/player-webrtc/", s.handleWebRTCPlayer)
 	mux.HandleFunc("/embed/", s.handleEmbed)
@@ -3573,3 +3576,94 @@ func (s *Server) handleInsights(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set("Content-Type", "application/json")
 	json.NewEncoder(w).Encode(stats)
 }
+
+func (s *Server) handleGoLive(w http.ResponseWriter, r *http.Request) {
+	user, ok := s.checkAuth(r)
+	if !ok {
+		http.Redirect(w, r, "/login", http.StatusSeeOther)
+		return
+	}
+
+	csrf := ""
+	if cookie, err := r.Cookie("sid"); err == nil {
+		s.sessionsMu.RLock()
+		if sess, ok := s.sessions[cookie.Value]; ok {
+			csrf = sess.CSRFToken
+		}
+		s.sessionsMu.RUnlock()
+	}
+
+	data := map[string]interface{}{
+		"Config":    s.Config,
+		"User":      user,
+		"Version":   s.Version,
+		"CSRFToken": csrf,
+	}
+
+	w.Header().Set("Content-Type", "text/html")
+	if err := s.tmpl.ExecuteTemplate(w, "go_live.html", data); err != nil {
+		logrus.WithError(err).Error("Go Live template error")
+	}
+}
+
+func (s *Server) handleGoLiveChunk(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodPost {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	mount := r.URL.Query().Get("mount")
+	if mount == "" {
+		http.Error(w, "mount query param required", http.StatusBadRequest)
+		return
+	}
+
+	// We don't check CSRF here because it's a data stream endpoint
+	// but we SHOULD check the session sid
+	if _, ok := s.checkAuth(r); !ok {
+		http.Error(w, "Unauthorized", http.StatusUnauthorized)
+		return
+	}
+
+	body, err := io.ReadAll(r.Body)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	stream := s.Relay.GetOrCreateStream(mount)
+	stream.SourceIP = "webaudio-http"
+	stream.Broadcast(body, s.Relay)
+
+	w.WriteHeader(http.StatusOK)
+}
+
+func (s *Server) handleWebRTCSourceOffer(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodPost {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	mount := r.URL.Query().Get("mount")
+	if mount == "" {
+		http.Error(w, "mount query param required", http.StatusBadRequest)
+		return
+	}
+
+	var offer webrtc.SessionDescription
+	if err := json.NewDecoder(r.Body).Decode(&offer); err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	answer, err := s.WebRTCM.HandleSourceOffer(mount, offer)
+	if err != nil {
+		w.Header().Set("Content-Type", "application/json")
+		w.WriteHeader(http.StatusBadRequest)
+		json.NewEncoder(w).Encode(map[string]string{"error": err.Error()})
+		return
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(answer)
+}

server/templates/admin.html

@@ -101,6 +101,7 @@
                 <div style="margin: 1rem 0 0.5rem 0.8rem; font-size: 0.65rem; color: var(--text-dim); text-transform: uppercase; letter-spacing: 1px;">Broadcasting</div>
                 <button class="nav-link" onclick="showTab('tab-stations')"><i data-lucide="tower-control"></i> Your Stations</button>
                 <button class="nav-link" onclick="showTab('tab-streamer')"><i data-lucide="play-circle"></i> AutoDJ</button>
+                <a href="/admin/golive" id="debug-golive-link" class="nav-link" style="text-decoration: none; display: none;"><i data-lucide="mic"></i> GO LIVE</a>
 
                 {{if eq .User.Role "superadmin"}}
                 <button class="nav-link" onclick="showTab('tab-relays')"><i data-lucide="refresh-ccw"></i> Edge Relays</button>
@@ -1263,6 +1264,8 @@ <h2>Edit AutoDJ</h2>
             if (isDebug) {
                 const debugBar = document.getElementById('debug-stats-bar');
                 if (debugBar) debugBar.style.display = 'flex';
+                const goLiveLink = document.getElementById('debug-golive-link');
+                if (goLiveLink) goLiveLink.style.display = 'flex';
             }
 
             function resize() { var rect = canvas.getBoundingClientRect(); canvas.width = rect.width * 2; canvas.height = rect.height * 2; ctx.scale(2, 2); }

server/templates/autodj_studio.html

@@ -120,13 +120,6 @@
     </style>
 </head>
 <body>
-    <div id="loading-overlay" style="position:fixed; top:0; left:0; width:100%; height:100%; background:var(--bg); display:flex; align-items:center; justify-content:center; z-index:9999;">
-        <div style="text-align:center;">
-            <i data-lucide="loader-2" class="spinner" style="width:48px; height:48px; color:var(--primary);"></i>
-            <p style="margin-top:1rem; color:var(--text-dim);">Initializing Studio...</p>
-        </div>
-    </div>
-
     <header>
         <div style="display: flex; align-items: center; gap: 1.5rem;">
             <a href="/admin#tab-streamer" class="studio-logo">
@@ -238,7 +231,6 @@ <h2>Library</h2>
         let lastSong = "";
 
         window.addEventListener('load', () => {
-            document.getElementById('loading-overlay').style.display = 'none';
             loadLibrary();
             lucide.createIcons();
         });