Fix cloud media metadata repair and merging

2026-10-13 00:26:37 +08:00 · 2026-06-18 02:58:30 +08:00
parent b492d9f867
commit 6c50697a1a
12 changed files with 530 additions and 2 deletions
@@ -228,6 +228,113 @@ func TestDeleteCloudLibraryPurgesMountWithoutRecycleBin(t *testing.T) {
 	}
 }

+func TestGroupMediaVersionsMergesEpisodeByExternalIDAcrossLibraries(t *testing.T) {
+	local := model.Media{
+		LibraryID:  "local-tv",
+		Title:      "折腰",
+		Path:       "/media/国产剧/折腰 (2025)/Season 1/折腰.S01E01.mkv",
+		SeasonNum:  1,
+		EpisodeNum: 1,
+		TMDbID:     220269,
+		SizeBytes:  100,
+		PosterURL:  "https://image.tmdb.org/t/p/w500/poster.jpg",
+	}
+	cloud := model.Media{
+		LibraryID:  "cloud-tv",
+		Title:      "折腰",
+		Path:       "cloud://openlist/国产剧/折腰 (2025) {tmdb-220269}/Season 1/折腰.S01E01.mkv",
+		SeasonNum:  1,
+		EpisodeNum: 1,
+		TMDbID:     220269,
+		SizeBytes:  200,
+		STRMURL:    "/api/cloud/play/openlist?ref=/国产剧/折腰/01.mkv",
+	}
+
+	grouped := groupMediaVersions([]model.Media{local, cloud})
+	if len(grouped) != 1 {
+		t.Fatalf("grouped len = %d, want 1: %#v", len(grouped), grouped)
+	}
+	if len(grouped[0].Versions) != 2 {
+		t.Fatalf("versions len = %d, want 2: %#v", len(grouped[0].Versions), grouped[0].Versions)
+	}
+}
+
+func TestMediaUpsertBackfillsExternalIDsForPendingCloudRows(t *testing.T) {
+	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := db.AutoMigrate(&model.Media{}); err != nil {
+		t.Fatal(err)
+	}
+	repos := repository.New(db)
+	path := "cloud://openlist/国漫/折腰 (2025) {tmdb-220269}/Season 1/折腰.S01E01.mkv"
+	if err := repos.DB.Create(&model.Media{
+		LibraryID:    "cloud-tv",
+		Title:        "折腰",
+		Path:         path,
+		SeasonNum:    1,
+		EpisodeNum:   1,
+		ScrapeStatus: "pending",
+	}).Error; err != nil {
+		t.Fatal(err)
+	}
+	if err := repos.Media.Upsert(t.Context(), &model.Media{
+		LibraryID:    "cloud-tv",
+		Title:        "折腰",
+		Path:         path,
+		SeasonNum:    1,
+		EpisodeNum:   1,
+		TMDbID:       220269,
+		Year:         2025,
+		ScrapeStatus: "pending",
+	}); err != nil {
+		t.Fatal(err)
+	}
+	var got model.Media
+	if err := repos.DB.First(&got, "path = ?", path).Error; err != nil {
+		t.Fatal(err)
+	}
+	if got.TMDbID != 220269 || got.Year != 2025 || got.ScrapeStatus != "pending" {
+		t.Fatalf("pending cloud row was not backfilled correctly: tmdb=%d year=%d status=%q", got.TMDbID, got.Year, got.ScrapeStatus)
+	}
+}
+
+func TestRepairCloudPathMetadataBackfillsExistingPlaceholders(t *testing.T) {
+	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := db.AutoMigrate(&model.Media{}); err != nil {
+		t.Fatal(err)
+	}
+	repos := repository.New(db)
+	path := "cloud://openlist/动画电影/雄狮少年2 (2024) {tmdb-1154478}/雄狮少年2 (2024) - 2160p.WEB-DL.H.265.DDP 5.1-ADWeb.mp4"
+	if err := repos.DB.Create(&model.Media{
+		LibraryID:    "cloud-movie",
+		Title:        "雄狮少年2 adweb",
+		Path:         path,
+		ScrapeStatus: "no_match",
+	}).Error; err != nil {
+		t.Fatal(err)
+	}
+	container := &Container{Repo: repos, Log: zap.NewNop()}
+	repaired, err := container.RepairCloudPathMetadata(t.Context())
+	if err != nil {
+		t.Fatal(err)
+	}
+	if repaired != 1 {
+		t.Fatalf("repaired = %d, want 1", repaired)
+	}
+	var got model.Media
+	if err := repos.DB.First(&got, "path = ?", path).Error; err != nil {
+		t.Fatal(err)
+	}
+	if got.TMDbID != 1154478 || got.Year != 2024 || got.Title != "雄狮少年2" || got.ScrapeStatus != "pending" {
+		t.Fatalf("placeholder was not repaired: title=%q tmdb=%d year=%d status=%q", got.Title, got.TMDbID, got.Year, got.ScrapeStatus)
+	}
+}
+
 func TestSoftDeleteCloudMediaPurgesRecordWithoutRecycleBin(t *testing.T) {
 	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
 	if err != nil {