0.2.56 - Reliable trailer playback

2026-08-12 11:05:07 +12:00
parent 9777eb0952
commit f2d052dbf6
22 changed files with 1438 additions and 55 deletions
@@ -0,0 +1,116 @@
+package trailer
+
+import (
+	"context"
+	"io"
+	"net/http"
+	"strings"
+	"testing"
+)
+
+type roundTripFunc func(*http.Request) (*http.Response, error)
+
+func (fn roundTripFunc) RoundTrip(request *http.Request) (*http.Response, error) {
+	return fn(request)
+}
+
+func TestYouTubeVideoID(t *testing.T) {
+	for _, raw := range []string{
+		"https://www.youtube.com/watch?v=dQw4w9WgXcQ",
+		"https://youtu.be/dQw4w9WgXcQ",
+		"https://www.youtube.com/embed/dQw4w9WgXcQ",
+		"https://youtube.com/shorts/dQw4w9WgXcQ",
+	} {
+		if got := youtubeVideoID(raw); got != "dQw4w9WgXcQ" {
+			t.Fatalf("youtubeVideoID(%q) = %q", raw, got)
+		}
+	}
+	if got := youtubeVideoID("https://example.com/watch?v=dQw4w9WgXcQ"); got != "" {
+		t.Fatalf("accepted a non-YouTube host: %q", got)
+	}
+}
+
+func TestProviderHostMatchingRejectsLookalikeDomains(t *testing.T) {
+	if newAppleProvider(http.DefaultClient).Supports("https://notapple.com/trailer.mov") {
+		t.Fatal("lookalike Apple host was accepted")
+	}
+	if youtubeVideoID("https://notyoutube.com/watch?v=dQw4w9WgXcQ") != "" {
+		t.Fatal("lookalike YouTube host was accepted")
+	}
+}
+
+func TestYouTubeResolverReturnsValidatedProgressiveStream(t *testing.T) {
+	client := &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) {
+		body := ""
+		status := http.StatusOK
+		headers := http.Header{}
+		switch request.URL.Host {
+		case "www.youtube.com":
+			body = `{"playabilityStatus":{"status":"OK"},"streamingData":{"formats":[` +
+				`{"url":"https://media.example/trailer.mp4","mimeType":"video/mp4; codecs=avc1,mp4a","height":720,"bitrate":1000}]}}`
+			headers.Set("Content-Type", "application/json")
+		case "media.example":
+			status = http.StatusPartialContent
+			headers.Set("Content-Type", "video/mp4")
+		default:
+			t.Fatalf("unexpected request to %s", request.URL)
+		}
+		return &http.Response{
+			StatusCode: status,
+			Header:     headers,
+			Body:       io.NopCloser(strings.NewReader(body)),
+			Request:    request,
+		}, nil
+	})}
+
+	resolver := New(client)
+	result, err := resolver.Resolve(context.Background(), Source{
+		Provider: "youtube",
+		URL:      "https://youtu.be/dQw4w9WgXcQ",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if result.URL != "https://media.example/trailer.mp4" || result.MimeType != "video/mp4" {
+		t.Fatalf("unexpected result: %+v", result)
+	}
+}
+
+func TestApplePageChoosesBestValidatedStream(t *testing.T) {
+	client := &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) {
+		body := ""
+		headers := http.Header{}
+		status := http.StatusOK
+		if request.URL.Path == "/page" {
+			body = `<a href="https://trailers.apple.com/film_h720p.mov">720</a>` +
+				`<a href="https://trailers.apple.com/film_h1080p.mov">1080</a>`
+			headers.Set("Content-Type", "text/html")
+		} else {
+			status = http.StatusPartialContent
+			headers.Set("Content-Type", "video/quicktime")
+		}
+		return &http.Response{
+			StatusCode: status,
+			Header:     headers,
+			Body:       io.NopCloser(strings.NewReader(body)),
+			Request:    request,
+		}, nil
+	})}
+	resolver := New(client)
+	result, err := resolver.Resolve(context.Background(), Source{
+		Provider: "apple", URL: "https://trailers.apple.com/page",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !strings.Contains(result.URL, "1080") {
+		t.Fatalf("did not choose the best stream: %+v", result)
+	}
+}
+
+func TestBalancedJSONObjectIgnoresBracesInsideStrings(t *testing.T) {
+	body := `before marker = {"value":"}" ,"nested":{"ok":true}} after`
+	if got := balancedJSONObject(body, "marker = "); got != `{"value":"}" ,"nested":{"ok":true}}` {
+		t.Fatalf("balanced object = %q", got)
+	}
+}