feat(stream): pion-based WebRTC byte streamer for browser playback
Replaces the broken anacrolix WebTorrent path with a custom WebRTC peer that the browser drives directly. Architecture matches plan/clever- weaving-dove.md (Fase 2 + 3 + 6 of the streaming pivot). - engine/wire: shared 12-byte binary frame format (Hello / RangeReq / RangeData / RangeEnd / Cancel / Ping / Pong / SeekHint). Roundtrip + oversized-frame rejection tests. - agent/signal_client: SSE consumer + POST sender for SDP/ICE relay through /api/internal/stream/signal/<id>; auto-reconnects. - engine/webrtc_stream: pion v4 PeerConnection + DataChannel pump. Reads file via os.ReadAt, chunks RangeData at 16 KiB, honours app- level backpressure with SetBufferedAmountLowThreshold. - cmd/daemon dispatcher learns mode webrtc_stream + new webrtcSessionRegistry tracks per-session cancel funcs for clean shutdown. - engine/probe + hwaccel + transcoder: foundation for Fase 2.5 (codec detection, NVENC/QSV/VAAPI/VideoToolbox autodetection, ffmpeg pipe wrapper to fragmented MP4). Integration into webrtc_stream still pending. - pion/webrtc/v4 promoted from indirect to direct dep. End-to-end against unarr-dev confirms a 122 MB 1080p H.264 / AAC MP4 plays in Chrome with the new pipeline.
This commit is contained in:
parent
4c52d9b039
commit
4314c06c5c
17 changed files with 2308 additions and 1 deletions
153
internal/agent/signal_client_test.go
Normal file
153
internal/agent/signal_client_test.go
Normal file
|
|
@ -0,0 +1,153 @@
|
|||
package agent
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"sync"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
// fakeSSEServer streams a fixed set of SSE events then closes the connection.
|
||||
func fakeSSEServer(t *testing.T, msgs []SignalMessage, holdOpenAfter bool) *httptest.Server {
|
||||
t.Helper()
|
||||
return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Header.Get("Authorization") != "Bearer test-key" {
|
||||
http.Error(w, "auth", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "text/event-stream")
|
||||
w.Header().Set("Cache-Control", "no-cache")
|
||||
flusher, ok := w.(http.Flusher)
|
||||
if !ok {
|
||||
t.Fatal("server: ResponseWriter is not a Flusher")
|
||||
}
|
||||
fmt.Fprint(w, "retry: 1500\n\n")
|
||||
flusher.Flush()
|
||||
for _, m := range msgs {
|
||||
data, _ := json.Marshal(m)
|
||||
fmt.Fprintf(w, "id: %d\nevent: signal\ndata: %s\n\n", m.TS, data)
|
||||
flusher.Flush()
|
||||
}
|
||||
// Send a heartbeat comment to verify it's ignored.
|
||||
fmt.Fprint(w, ": heartbeat\n\n")
|
||||
flusher.Flush()
|
||||
if holdOpenAfter {
|
||||
// Hold the connection until the client disconnects so the test can
|
||||
// exercise stream.Close().
|
||||
<-r.Context().Done()
|
||||
}
|
||||
}))
|
||||
}
|
||||
|
||||
func TestSignalStreamReadsMessages(t *testing.T) {
|
||||
want := []SignalMessage{
|
||||
{From: SignalRoleBrowser, Type: SignalMsgOffer, Payload: "{sdp:1}", TS: 1},
|
||||
{From: SignalRoleBrowser, Type: SignalMsgCandidate, Payload: "{cand:1}", TS: 2},
|
||||
}
|
||||
srv := fakeSSEServer(t, want, false)
|
||||
defer srv.Close()
|
||||
|
||||
c := NewClient(srv.URL, "test-key", "test-ua")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
||||
defer cancel()
|
||||
|
||||
stream, err := c.OpenSignalStream(ctx, "session-1")
|
||||
if err != nil {
|
||||
t.Fatalf("open: %v", err)
|
||||
}
|
||||
defer stream.Close()
|
||||
|
||||
var got []SignalMessage
|
||||
for m := range stream.Events() {
|
||||
got = append(got, m)
|
||||
if len(got) == len(want) {
|
||||
break
|
||||
}
|
||||
}
|
||||
if len(got) != len(want) {
|
||||
t.Fatalf("got %d messages, want %d", len(got), len(want))
|
||||
}
|
||||
for i, m := range got {
|
||||
if m.From != want[i].From || m.Type != want[i].Type || m.Payload != want[i].Payload {
|
||||
t.Errorf("[%d] mismatch: %+v want %+v", i, m, want[i])
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestSignalStreamPropagatesAuthError(t *testing.T) {
|
||||
srv := fakeSSEServer(t, nil, false)
|
||||
defer srv.Close()
|
||||
|
||||
c := NewClient(srv.URL, "wrong-key", "test-ua")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
||||
defer cancel()
|
||||
|
||||
_, err := c.OpenSignalStream(ctx, "session-1")
|
||||
if err == nil {
|
||||
t.Fatal("expected auth error, got nil")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSignalStreamCloseCancelsRead(t *testing.T) {
|
||||
srv := fakeSSEServer(t, nil, true)
|
||||
defer srv.Close()
|
||||
|
||||
c := NewClient(srv.URL, "test-key", "test-ua")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
stream, err := c.OpenSignalStream(ctx, "session-1")
|
||||
if err != nil {
|
||||
t.Fatalf("open: %v", err)
|
||||
}
|
||||
|
||||
// Close on a separate goroutine then make sure the events channel drains.
|
||||
var wg sync.WaitGroup
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
time.Sleep(50 * time.Millisecond)
|
||||
stream.Close()
|
||||
}()
|
||||
|
||||
for range stream.Events() {
|
||||
// drain
|
||||
}
|
||||
wg.Wait()
|
||||
}
|
||||
|
||||
func TestPostSignalSendsCorrectBody(t *testing.T) {
|
||||
var bodySeen map[string]any
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Header.Get("Authorization") != "Bearer test-key" {
|
||||
http.Error(w, "auth", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
_ = json.NewDecoder(r.Body).Decode(&bodySeen)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
fmt.Fprint(w, `{"ok":true}`)
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
c := NewClient(srv.URL, "test-key", "test-ua")
|
||||
err := c.PostSignal(context.Background(), "sess-x", SignalMessage{
|
||||
Type: SignalMsgAnswer,
|
||||
Payload: "{sdp:answer}",
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("post: %v", err)
|
||||
}
|
||||
if bodySeen["from"] != string(SignalRoleAgent) {
|
||||
t.Errorf("expected from=agent, got %v", bodySeen["from"])
|
||||
}
|
||||
if bodySeen["type"] != string(SignalMsgAnswer) {
|
||||
t.Errorf("expected type=answer, got %v", bodySeen["type"])
|
||||
}
|
||||
if bodySeen["payload"] != "{sdp:answer}" {
|
||||
t.Errorf("expected payload mismatch, got %v", bodySeen["payload"])
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue