初始提交：短剧库源码（含转码优化 ultrafast）

2026-09-17 19:09:08 +08:00
commit 07dc0f1059
247 changed files with 43242 additions and 0 deletions
@@ -0,0 +1,103 @@
+package app
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+	"testing"
+)
+
+func TestHuangdouVIPKnownFreeAndUnknownRemainDistinct(t *testing.T) {
+	for _, test := range []struct {
+		body string
+		want int
+	}{
+		{`{"pay_type":"vip"}`, 1},
+		{`{"pay_type":"free"}`, 0},
+		{`{"is_vip":"y"}`, 1},
+		{`{"is_vip":false}`, 0},
+		{`{"vip_episodes":[3,"4"]}`, 1},
+		{`{"pay_type":"free","vip_episodes":[3]}`, 1},
+		{`{"episodes":[{"type":"free"},{"type":"vip"}]}`, 1},
+		{`{"name":"VIP 是标题文字"}`, -1},
+		{`{"pay_type":"unknown","vip_episodes":[0,-1,"invalid"]}`, -1},
+		{`{"vip_episodes":[]}`, -1},
+	} {
+		var row map[string]any
+		if err := json.Unmarshal([]byte(test.body), &row); err != nil {
+			t.Fatal(err)
+		}
+		row["id"], row["name"] = "fixture", "无图 VIP 样本"
+		drama := huangdouDramaFromMap(row)
+		actual := -1
+		if drama.VIP != nil {
+			actual = 0
+			if *drama.VIP {
+				actual = 1
+			}
+		}
+		if actual != test.want {
+			t.Fatalf("%s: VIP %d, want %d", test.body, actual, test.want)
+		}
+		ranking, err := parseHuangdouRanking(map[string]any{"list": []any{row}}, 1)
+		if err != nil {
+			t.Fatal(err)
+		}
+		ranked := ranking.Items[0].Drama
+		if (ranked.VIP == nil) != (drama.VIP == nil) || ranked.VIP != nil && *ranked.VIP != *drama.VIP {
+			t.Fatal("ranking lost VIP metadata")
+		}
+	}
+	paid, free := true, false
+	previous := Drama{ID: "huangdou:fixture", VIP: &paid}
+	if value := mergeDramaMetadata(Drama{}, previous).VIP; value == nil || !*value {
+		t.Fatal("incomplete metadata erased VIP flag")
+	}
+	if value := mergeDramaMetadata(Drama{VIP: &free}, previous).VIP; value == nil || *value {
+		t.Fatal("fresh free metadata did not replace VIP flag")
+	}
+}
+
+func TestHuangdouVIPEpisodesPreserveFreeEpisodesAndPreviewRefusal(t *testing.T) {
+	d := rankingTestDownloader(t, func(request *http.Request) (*http.Response, error) {
+		if request.URL.Path != "/api/drama/detail" {
+			t.Fatal("unexpected upstream path", request.URL.Path)
+		}
+		return rankingHTTPResponse(request, 200, `{"data":{"id":"fixture","name":"无图分集样本","pay_type":"vip","free_episodes":1,"episode_count":4,"vip_episodes":[3],"episodes":[{"seq":1,"type":"free"},{"seq":2,"type":"vip"},{"seq":3},{"seq":4,"type":"free"}]}}`), nil
+	})
+	_, chapters, err := d.fetchHuangdouChapters(context.Background(), "fixture")
+	if err != nil || len(chapters) != 4 {
+		t.Fatal(err, len(chapters))
+	}
+	for i, want := range []bool{false, true, true, false} {
+		if chapters[i].VIP != want {
+			t.Fatal("incorrect episode restriction", i+1, chapters[i].VIP)
+		}
+	}
+	if !huangdouPreviewOnly(map[string]any{"is_preview": true}, "https://example.invalid/preview.mp4", "https://example.invalid") {
+		t.Fatal("preview mistaken for a complete episode")
+	}
+}
+
+func TestHuangdouVIPBackfillOnlyWhenInformationMissing(t *testing.T) {
+	calls := 0
+	d := rankingTestDownloader(t, func(request *http.Request) (*http.Response, error) {
+		calls++
+		if request.URL.Path != "/api/drama/detail" {
+			t.Fatal("unexpected upstream path", request.URL.Path)
+		}
+		return rankingHTTPResponse(request, 200, `{"data":{"id":"fixture","name":"无图样本","pay_type":"vip","issue_date":"2026-03-16","click":0,"hot_rate":"123"}}`), nil
+	})
+	cached := Drama{ID: "huangdou:fixture", Source: sourceHuangdou, OnlineDate: "2026-03-16", Views: "0", Heat: "123", SortMetadata: &sortMetadataState{Version: sortMetadataVersion}}
+	if !needsSortMetadata(cached) {
+		t.Fatal("manual update cannot backfill old VIP information")
+	}
+	patch, err := d.fetchDramaSortMetadata(context.Background(), cached)
+	if err != nil || patch.VIP == nil || !*patch.VIP || calls != 1 {
+		t.Fatal("VIP backfill failed", err, calls)
+	}
+	cached = mergeDramaMetadata(patch, cached)
+	if _, err = d.fetchDramaSortMetadata(context.Background(), cached); err != nil || calls != 1 {
+		t.Fatal("known VIP caused unnecessary detail request")
+	}
+}