package service import ( "context" "encoding/json" "fmt" "net/http" "net/http/httptest" "strings" "sync" "sync/atomic" "testing" "time" "github.com/glebarez/sqlite" "go.uber.org/zap" "gorm.io/gorm" "github.com/ShukeBta/MediaStationGo/internal/config" "github.com/ShukeBta/MediaStationGo/internal/model" "github.com/ShukeBta/MediaStationGo/internal/repository" ) func TestScanCloudLibraryImportsRecursivePlayableMedia(t *testing.T) { empty := false upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path != "/file/sort" { t.Fatalf("unexpected path %s", r.URL.Path) } w.Header().Set("Content-Type", "application/json") if empty { _, _ = w.Write([]byte(`{"status":200,"code":0,"data":{"list":[]}}`)) return } switch r.URL.Query().Get("pdir_fid") { case "0": _, _ = w.Write([]byte(`{"status":200,"code":0,"data":{"list":[ {"fid":"d1","file_name":"Movies","dir":true,"size":0}, {"fid":"f1","file_name":"Root.Movie.2024.mkv","dir":false,"size":123} ]}}`)) case "d1": _, _ = w.Write([]byte(`{"status":200,"code":0,"data":{"list":[ {"fid":"f2","file_name":"Nested.Show.S01E02.mp4","dir":false,"size":456} ]}}`)) default: t.Fatalf("unexpected pdir_fid %q", r.URL.Query().Get("pdir_fid")) } })) defer upstream.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.Setting{}, &model.StorageConfig{}); err != nil { t.Fatal(err) } repos := repository.New(db) log := zap.NewNop() storage := NewStorageConfigService(log, repos, NewCryptoService("", log)) if _, err := storage.Save(t.Context(), StorageInput{ Type: "quark", Config: map[string]any{ "cookie": "kps=test", "base": upstream.URL, }, }); err != nil { t.Fatal(err) } lib := model.Library{Name: "夸克网盘", Path: "cloud://quark/0", Type: "tv", Enabled: true} if err := repos.Library.Create(t.Context(), &lib); err != nil { t.Fatal(err) } scanner := NewScannerService(&config.Config{}, log, repos, NewHub(log), nil, nil) scanner.SetStorageConfig(storage) res, err := scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("scan cloud: %v", err) } if res.Visited != 2 || res.Added != 2 { t.Fatalf("scan result = %#v, want visited=2 added=2", res) } var rows []model.Media if err := repos.DB.Order("path").Find(&rows).Error; err != nil { t.Fatal(err) } if len(rows) != 2 { t.Fatalf("media rows = %d, want 2: %#v", len(rows), rows) } if rows[0].Path != "cloud://quark/Movies/Nested.Show.S01E02.mp4" || !strings.Contains(rows[0].STRMURL, "ref=f2") { t.Fatalf("nested media path/strm wrong: path=%q strm=%q", rows[0].Path, rows[0].STRMURL) } if rows[0].SeasonNum != 1 || rows[0].EpisodeNum != 2 { t.Fatalf("nested episode metadata wrong: %#v", rows[0]) } if rows[1].Path != "cloud://quark/Root.Movie.2024.mkv" || rows[1].STRMURL != "/api/cloud/play/quark?ref=f1" { t.Fatalf("root media path/strm wrong: path=%q strm=%q", rows[0].Path, rows[0].STRMURL) } res, err = scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("rescan same cloud: %v", err) } if res.Added != 0 || res.Updated != 0 || res.Skipped != 2 { t.Fatalf("same cloud rescan should skip unchanged rows, got %#v", res) } empty = true res, err = scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("rescan cloud: %v", err) } if res.Removed != 2 { t.Fatalf("removed = %d, want 2", res.Removed) } if got := countMedia(t, repos); got != 0 { t.Fatalf("media count after prune = %d, want 0", got) } var allRows int64 if err := repos.DB.Unscoped().Model(&model.Media{}).Count(&allRows).Error; err != nil { t.Fatal(err) } if allRows != 0 { t.Fatalf("unscoped media count after cloud prune = %d, want 0", allRows) } } func TestScanCloudLibraryListsChildDirectoriesConcurrently(t *testing.T) { var active int32 var maxActive int32 var releaseOnce sync.Once release := make(chan struct{}) upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path != "/file/sort" { t.Errorf("unexpected path %s", r.URL.Path) return } w.Header().Set("Content-Type", "application/json") switch r.URL.Query().Get("pdir_fid") { case "0": _, _ = w.Write([]byte(`{"status":200,"code":0,"data":{"list":[ {"fid":"d1","file_name":"A","dir":true,"size":0}, {"fid":"d2","file_name":"B","dir":true,"size":0} ]}}`)) case "d1", "d2": cur := atomic.AddInt32(&active, 1) defer atomic.AddInt32(&active, -1) for { prev := atomic.LoadInt32(&maxActive) if cur <= prev || atomic.CompareAndSwapInt32(&maxActive, prev, cur) { break } } if cur >= 2 { releaseOnce.Do(func() { close(release) }) } select { case <-release: case <-r.Context().Done(): return case <-time.After(1500 * time.Millisecond): t.Errorf("child directory requests were not concurrent") return } id := r.URL.Query().Get("pdir_fid") _, _ = fmt.Fprintf(w, `{"status":200,"code":0,"data":{"list":[{"fid":"f-%s","file_name":"Movie.%s.mkv","dir":false,"size":123}]}}`, id, id) default: t.Errorf("unexpected pdir_fid %q", r.URL.Query().Get("pdir_fid")) } })) defer upstream.Close() db, err := gorm.Open(sqlite.Open("file:cloud_scan_concurrent?mode=memory&cache=shared"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.Setting{}, &model.StorageConfig{}); err != nil { t.Fatal(err) } repos := repository.New(db) log := zap.NewNop() storage := NewStorageConfigService(log, repos, NewCryptoService("", log)) if _, err := storage.Save(t.Context(), StorageInput{ Type: "quark", Config: map[string]any{ "cookie": "kps=test", "base": upstream.URL, }, }); err != nil { t.Fatal(err) } lib := model.Library{Name: "夸克网盘", Path: "cloud://quark/0", Type: "movie", Enabled: true} if err := repos.Library.Create(t.Context(), &lib); err != nil { t.Fatal(err) } cfg := &config.Config{} cfg.App.CloudScanMaxConcurrent = 2 scanner := NewScannerService(cfg, log, repos, NewHub(log), nil, nil) scanner.SetStorageConfig(storage) ctx, cancel := context.WithTimeout(t.Context(), 2*time.Second) defer cancel() res, err := scanner.ScanLibrary(ctx, lib.ID) if err != nil { t.Fatalf("scan cloud: %v", err) } if got := atomic.LoadInt32(&maxActive); got < 2 { t.Fatalf("max concurrent child lists = %d, want >= 2", got) } if res.Visited != 2 || res.Added != 2 { t.Fatalf("scan result = %#v, want visited=2 added=2", res) } } func TestCloudLibraryPathParsing(t *testing.T) { typ, dir, ok := parseCloudLibraryPath("cloud://cloud115/abc%20123?ignored=1") if !ok || typ != "cloud115" || dir != "abc 123" { t.Fatalf("parse path got typ=%q dir=%q ok=%v", typ, dir, ok) } typ, dir, ok = parseCloudLibraryPath("cloud://quark?dir=0") if !ok || typ != "quark" || dir != "" { t.Fatalf("parse query got typ=%q dir=%q ok=%v", typ, dir, ok) } if ref := cloudEntryRef("cloud115", "fid", "pick"); ref != "pick" { t.Fatalf("115 ref = %q, want pick", ref) } } func TestCloudMountConflictDetectsNestedMounts(t *testing.T) { root := model.Library{Base: model.Base{ID: "root"}, Name: "115", Path: "cloud://cloud115", Enabled: true} childPath := BuildCloudLibraryPath("cloud115", "child-id", "parent-id/child-id") info, ok := ParseCloudLibraryMount(childPath) if !ok || info.ScanDir != "child-id" || info.DisplayDir != "parent-id/child-id" { t.Fatalf("parse child mount = %#v ok=%v", info, ok) } conflict := FindCloudMountConflict([]model.Library{root}, "cloud115", "child-id", "parent-id/child-id") if conflict != nil { t.Fatalf("child mount under existing root should be allowed, got conflict %#v", conflict) } sibling := model.Library{Base: model.Base{ID: "sibling"}, Name: "Sibling", Path: BuildCloudLibraryPath("cloud115", "sibling-id", "parent-id/sibling-id"), Enabled: true} conflict = FindCloudMountConflict([]model.Library{sibling}, "cloud115", "child-id", "parent-id/child-id") if conflict != nil { t.Fatalf("sibling conflict = %#v, want nil", conflict) } conflict = FindCloudMountConflict([]model.Library{sibling}, "cloud115", "parent-id", "parent-id") if conflict == nil || !conflict.Nested { t.Fatalf("parent mount over existing child = %#v, want nested conflict", conflict) } oldIDPath := BuildCloudLibraryPath("cloud115", "child-id", "old-parent-id/child-id") conflict = FindCloudMountConflict([]model.Library{{Base: model.Base{ID: "old"}, Name: "Old", Path: oldIDPath, Enabled: true}}, "cloud115", "child-id", "父目录/子目录") if conflict == nil || !conflict.Exact { t.Fatalf("same scan dir with renamed display path = %#v, want exact conflict", conflict) } root.CreatedAt = root.CreatedAt.Add(-1) child := model.Library{Base: model.Base{ID: "child"}, Name: "Child", Path: childPath, Enabled: true} if shadow := CloudLibraryShadowed([]model.Library{root, child}, child); shadow != nil { t.Fatalf("child should not be shadowed by root: %#v", shadow) } if shadow := CloudLibraryShadowed([]model.Library{root, child}, root); shadow == nil || !shadow.Nested { t.Fatalf("root should be shadowed by child, got %#v", shadow) } } func TestCancelCloudScansForProviderSignalsRunningScan(t *testing.T) { scanner := NewScannerService(&config.Config{}, zap.NewNop(), repository.New(nil), NewHub(zap.NewNop()), nil, nil) cancelled := false scanner.cloudScans["lib-1"] = &cloudScanEntry{ status: CloudScanStatus{LibraryID: "lib-1", Provider: "openlist", State: "running"}, cancel: func() { cancelled = true }, } if got := scanner.CancelCloudScansForProvider("openlist"); got != 1 { t.Fatalf("cancelled = %d, want 1", got) } if !cancelled { t.Fatal("cancel func was not called") } if state := scanner.cloudScans["lib-1"].status.State; state != "canceling" { t.Fatalf("state = %q, want canceling", state) } } func TestInferCloudMountMediaType(t *testing.T) { cases := map[string]string{ "/日漫": "anime", "/国漫": "anime", "/欧美动漫": "anime", "/电视剧/国产剧": "tv", "/电视剧/欧美剧": "tv", "/电视剧/日韩剧": "tv", "/电影/动画电影": "movie", "/电影/华语电影": "movie", "/电影/外语电影": "movie", "/综艺": "variety", } for dir, want := range cases { if got := InferCloudMountMediaType(dir, "OpenList · "+dir); got != want { t.Fatalf("%s type = %s, want %s", dir, got, want) } } } func TestScan115CloudLibraryKeepsDisplayHierarchyAndSeasonCounts(t *testing.T) { upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path != "/files" { t.Fatalf("unexpected path %s", r.URL.Path) } w.Header().Set("Content-Type", "application/json") switch r.URL.Query().Get("cid") { case "100": _, _ = w.Write([]byte(`{"state":true,"data":[ {"cid":"s1","n":"Season 1","s":0}, {"cid":"s2","n":"Season 2","s":0} ]}`)) case "s1": _, _ = w.Write([]byte(`{"state":true,"data":[ {"fid":"f101","n":"剑来 - S01E01.mkv","s":1001,"pc":"pick101"}, {"fid":"f125","n":"剑来 - S01E25.mkv","s":1025,"pc":"pick125"} ]}`)) case "s2": _, _ = w.Write([]byte(`{"state":true,"data":[ {"fid":"f201","n":"剑来 - S02E01.mkv","s":2001,"pc":"pick201"}, {"fid":"f204","n":"剑来 - S02E04.mkv","s":2004,"pc":"pick204"} ]}`)) default: t.Fatalf("unexpected cid %q", r.URL.Query().Get("cid")) } })) defer upstream.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.Setting{}, &model.StorageConfig{}); err != nil { t.Fatal(err) } repos := repository.New(db) log := zap.NewNop() storage := NewStorageConfigService(log, repos, NewCryptoService("", log)) if _, err := storage.Save(t.Context(), StorageInput{ Type: "cloud115", Config: map[string]any{ "cookie": "UID=test", "base": upstream.URL, }, }); err != nil { t.Fatal(err) } lib := model.Library{Name: "115 · 国漫 · 剑来", Path: BuildCloudLibraryPath("cloud115", "100", "动漫/国漫/剑来"), Type: "anime", Enabled: true} if err := repos.Library.Create(t.Context(), &lib); err != nil { t.Fatal(err) } scanner := NewScannerService(&config.Config{}, log, repos, NewHub(log), nil, nil) scanner.SetStorageConfig(storage) res, err := scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("scan cloud115: %v", err) } if res.Added != 4 { t.Fatalf("scan result = %#v, want added=4", res) } var rows []model.Media if err := repos.DB.Order("path").Find(&rows).Error; err != nil { t.Fatal(err) } if len(rows) != 4 { t.Fatalf("media rows = %d, want 4", len(rows)) } want := map[string][2]int{ "cloud://cloud115/动漫/国漫/剑来/Season 1/剑来 - S01E01.mkv": {1, 1}, "cloud://cloud115/动漫/国漫/剑来/Season 1/剑来 - S01E25.mkv": {1, 25}, "cloud://cloud115/动漫/国漫/剑来/Season 2/剑来 - S02E01.mkv": {2, 1}, "cloud://cloud115/动漫/国漫/剑来/Season 2/剑来 - S02E04.mkv": {2, 4}, } for _, row := range rows { seasonEpisode, ok := want[row.Path] if !ok { t.Fatalf("unexpected path %q", row.Path) } if row.Title != "剑来" { t.Fatalf("title = %q, want 剑来", row.Title) } if row.SeasonNum != seasonEpisode[0] || row.EpisodeNum != seasonEpisode[1] { t.Fatalf("%s season/episode = %d/%d, want %d/%d", row.Path, row.SeasonNum, row.EpisodeNum, seasonEpisode[0], seasonEpisode[1]) } if !strings.Contains(row.STRMURL, "ref=pick") { t.Fatalf("115 playback should keep pickcode ref, got %q", row.STRMURL) } } } func TestCloudSeriesTitlePrefersShowFolder(t *testing.T) { title, year := cloudSeriesTitleFromMediaPath("cloud://openlist/国产剧/紫川 (2024) {tmdb-247590}/Season 2/紫川.2024.S02E24.第24集.2160p.WEB-DL.H.265-ColorTV.mkv") if title != "紫川" || year != 2024 { t.Fatalf("cloud series title = %q/%d, want 紫川/2024", title, year) } title, year = cloudSeriesTitleFromMediaPath("cloud://openlist/国产剧/紫川.2024.S02E24.mkv") if title != "" || year != 0 { t.Fatalf("single category folder should not override title, got %q/%d", title, year) } } func TestCloudMetadataNeedsRefreshWhenPathHintConflicts(t *testing.T) { existing := existingCloudMedia{ Year: 2025, TMDbID: 220269, } local := &LocalMetadata{ Year: 2025, TMDbID: 296753, PathHint: true, } if !cloudMetadataNeedsRefresh(existing, local) { t.Fatal("conflicting explicit cloud path hint should refresh existing media") } } func TestScanCloudLibraryReadsRemoteSTRMTarget(t *testing.T) { upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.Method { case "PROPFIND": if r.URL.Path != "/dav/Links" { t.Fatalf("unexpected propfind path %s", r.URL.Path) } w.Header().Set("Content-Type", "application/xml") w.WriteHeader(http.StatusMultiStatus) _, _ = w.Write([]byte(` /dav/Links/ /dav/Links/Movie.strm Movie.strm32 `)) case http.MethodGet: if r.URL.Path != "/dav/Links/Movie.strm" { t.Fatalf("unexpected get path %s", r.URL.Path) } _, _ = w.Write([]byte("https://cdn.example.com/Movie.mkv\n")) default: t.Fatalf("unexpected method %s", r.Method) } })) defer upstream.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.Setting{}, &model.StorageConfig{}); err != nil { t.Fatal(err) } repos := repository.New(db) log := zap.NewNop() storage := NewStorageConfigService(log, repos, NewCryptoService("", log)) if _, err := storage.Save(t.Context(), StorageInput{ Type: "openlist", Config: map[string]any{ "url": upstream.URL, }, }); err != nil { t.Fatal(err) } lib := model.Library{Name: "OpenList · Links", Path: "cloud://openlist/Links", Type: "movie", Enabled: true} if err := repos.Library.Create(t.Context(), &lib); err != nil { t.Fatal(err) } scanner := NewScannerService(&config.Config{}, log, repos, NewHub(log), nil, nil) scanner.SetStorageConfig(storage) res, err := scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("scan cloud: %v", err) } if res.Added != 1 { t.Fatalf("scan result = %#v, want added=1", res) } var media model.Media if err := repos.DB.First(&media).Error; err != nil { t.Fatal(err) } if media.Path != "cloud://openlist/Links/Movie.strm" { t.Fatalf("path = %q", media.Path) } if media.STRMURL != "https://cdn.example.com/Movie.mkv" { t.Fatalf("strm target = %q", media.STRMURL) } } func TestScanCloudLibraryReadsRemoteNFOAndArtwork(t *testing.T) { upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.Method { case "PROPFIND": w.Header().Set("Content-Type", "application/xml") w.WriteHeader(http.StatusMultiStatus) switch r.URL.Path { case "/dav/Anime/JianLai": _, _ = w.Write([]byte(` /dav/Anime/JianLai/ /dav/Anime/JianLai/tvshow.nfotvshow.nfo64 /dav/Anime/JianLai/poster.jpgposter.jpg1024 /dav/Anime/JianLai/Season1/Season1 `)) case "/dav/Anime/JianLai/Season1": _, _ = w.Write([]byte(` /dav/Anime/JianLai/Season1/ /dav/Anime/JianLai/Season1/JianLai.S01E01.mkvJianLai.S01E01.mkv2048 /dav/Anime/JianLai/Season1/JianLai.S01E01.nfoJianLai.S01E01.nfo128 `)) default: t.Fatalf("unexpected propfind path %s", r.URL.Path) } case http.MethodGet: switch r.URL.Path { case "/dav/Anime/JianLai/tvshow.nfo": _, _ = w.Write([]byte(`剑来2024天地有剑气`)) case "/dav/Anime/JianLai/Season1/JianLai.S01E01.nfo": _, _ = w.Write([]byte(`剑来第一集11`)) case "/dav/Anime/JianLai/poster.jpg": w.Header().Set("Content-Type", "image/jpeg") _, _ = w.Write([]byte("cloud-poster-bytes")) default: t.Fatalf("unexpected get path %s", r.URL.Path) } default: t.Fatalf("unexpected method %s", r.Method) } })) defer upstream.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.Setting{}, &model.StorageConfig{}); err != nil { t.Fatal(err) } repos := repository.New(db) log := zap.NewNop() storage := NewStorageConfigService(log, repos, NewCryptoService("", log)) if _, err := storage.Save(t.Context(), StorageInput{ Type: "openlist", Config: map[string]any{ "url": upstream.URL, }, }); err != nil { t.Fatal(err) } lib := model.Library{Name: "OpenList · 国漫 · 剑来", Path: "cloud://openlist/Anime/JianLai", Type: "anime", Enabled: true} if err := repos.Library.Create(t.Context(), &lib); err != nil { t.Fatal(err) } scanner := NewScannerService(&config.Config{}, log, repos, NewHub(log), nil, nil) scanner.SetStorageConfig(storage) imageProxy := NewImageProxy(&config.Config{Cache: config.CacheConfig{CacheDir: t.TempDir()}}, log) scanner.SetImageProxy(imageProxy) res, err := scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("scan cloud: %v", err) } if res.Added != 1 || res.LocalMetadata != 1 { t.Fatalf("scan result = %#v, want added=1 local_metadata=1", res) } var media model.Media if err := repos.DB.First(&media).Error; err != nil { t.Fatal(err) } // 单集名(episode 「第一集」)不得写入 OriginalName(整剧原名/分组键)。 // tvshow.nfo 未提供 originaltitle, 故 OriginalName 应为空。 if media.Title != "剑来" || media.OriginalName != "" || media.Year != 2024 { t.Fatalf("metadata not applied: %#v", media) } if media.SeasonNum != 1 || media.EpisodeNum != 1 { t.Fatalf("episode numbers = %d/%d", media.SeasonNum, media.EpisodeNum) } if media.PosterURL != "/api/cloud/play/openlist?ref=%2FAnime%2FJianLai%2Fposter.jpg" { t.Fatalf("poster url = %q", media.PosterURL) } rec := httptest.NewRecorder() if !imageProxy.ServeCloudCached(rec, httptest.NewRequest(http.MethodGet, media.PosterURL, nil), "openlist:/Anime/JianLai/poster.jpg") { t.Fatal("cloud poster should be cached locally during scan before media is exposed") } if got := rec.Body.String(); got != "cloud-poster-bytes" { t.Fatalf("cached poster body = %q", got) } if media.ScrapeStatus != "matched" { t.Fatalf("scrape status = %q", media.ScrapeStatus) } } func TestScanOpenListCloudLibraryUsesAPIPaginationBeyondFirstPage(t *testing.T) { const totalFiles = 125 requestedPages := map[int]bool{} upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path != "/api/fs/list" { t.Fatalf("unexpected path %s", r.URL.Path) } if r.Method != http.MethodPost { t.Fatalf("unexpected method %s", r.Method) } if r.Header.Get("Authorization") != "openlist-token" { t.Fatalf("missing openlist token: %q", r.Header.Get("Authorization")) } var in struct { Path string `json:"path"` Page int `json:"page"` PerPage int `json:"per_page"` } if err := json.NewDecoder(r.Body).Decode(&in); err != nil { t.Fatalf("decode request: %v", err) } if in.Path != "/Movies" { t.Fatalf("path = %q, want /Movies", in.Path) } if in.PerPage <= 100 { t.Fatalf("per_page = %d, want API pagination larger than legacy 100", in.PerPage) } requestedPages[in.Page] = true effectivePageSize := in.PerPage if effectivePageSize > 100 { effectivePageSize = 100 } start := (in.Page - 1) * effectivePageSize content := []map[string]any{} for idx := start; idx < totalFiles && idx < start+effectivePageSize; idx++ { content = append(content, map[string]any{ "name": fmt.Sprintf("Movie.%03d.mkv", idx+1), "size": int64(1024 + idx), "is_dir": false, }) } w.Header().Set("Content-Type", "application/json") _ = json.NewEncoder(w).Encode(map[string]any{ "code": 200, "message": "success", "data": map[string]any{ "content": content, "total": totalFiles, }, }) })) defer upstream.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.Setting{}, &model.StorageConfig{}); err != nil { t.Fatal(err) } repos := repository.New(db) log := zap.NewNop() storage := NewStorageConfigService(log, repos, NewCryptoService("", log)) if _, err := storage.Save(t.Context(), StorageInput{ Type: "openlist", Config: map[string]any{ "server": upstream.URL, "token": "openlist-token", }, }); err != nil { t.Fatal(err) } lib := model.Library{Name: "OpenList · Movies", Path: BuildCloudLibraryPath("openlist", "/Movies", "/Movies"), Type: "movie", Enabled: true} if err := repos.Library.Create(t.Context(), &lib); err != nil { t.Fatal(err) } scanner := NewScannerService(&config.Config{}, log, repos, NewHub(log), nil, nil) scanner.SetStorageConfig(storage) res, err := scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("scan openlist: %v", err) } if res.Added != totalFiles { t.Fatalf("scan result = %#v, want added=%d", res, totalFiles) } if got := countMedia(t, repos); got != totalFiles { t.Fatalf("media count = %d, want %d", got, totalFiles) } if !requestedPages[1] || !requestedPages[2] { t.Fatalf("expected pagination beyond the first 100 entries, got pages %#v", requestedPages) } } func TestScanCloudLibraryQueuesMissingExistingTrackMetadataBeforeNewFiles(t *testing.T) { const newFiles = maxCloudMediaProbeQueuePerScan + 5 upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path != "/file/sort" || r.URL.Query().Get("pdir_fid") != "0" { t.Fatalf("unexpected cloud list request %s?%s", r.URL.Path, r.URL.RawQuery) } var b strings.Builder b.WriteString(`{"status":200,"code":0,"data":{"list":[`) for i := 0; i < newFiles; i++ { if i > 0 { b.WriteByte(',') } _, _ = fmt.Fprintf(&b, `{"fid":"new-%02d","file_name":"New.Movie.%02d.mkv","dir":false,"size":%d}`, i, i, 1000+i) } _, _ = fmt.Fprintf(&b, `,{"fid":"existing","file_name":"Existing.Show.S01E01.mkv","dir":false,"size":2048}]}}`) _, _ = w.Write([]byte(b.String())) })) defer upstream.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.Setting{}, &model.StorageConfig{}); err != nil { t.Fatal(err) } repos := repository.New(db) log := zap.NewNop() storage := NewStorageConfigService(log, repos, NewCryptoService("", log)) if _, err := storage.Save(t.Context(), StorageInput{ Type: "quark", Config: map[string]any{ "cookie": "kps=test", "base": upstream.URL, }, }); err != nil { t.Fatal(err) } lib := model.Library{Name: "夸克网盘", Path: "cloud://quark/0", Type: "tv", Enabled: true} if err := repos.Library.Create(t.Context(), &lib); err != nil { t.Fatal(err) } existingPath := "cloud://quark/Existing.Show.S01E01.mkv" if err := repos.DB.Create(&model.Media{ LibraryID: lib.ID, Title: "Existing Show", Path: existingPath, SizeBytes: 2048, Container: "mkv", STRMURL: "/api/cloud/play/quark?ref=existing", SeasonNum: 1, EpisodeNum: 1, }).Error; err != nil { t.Fatal(err) } scanner := NewScannerService(&config.Config{}, log, repos, NewHub(log), NewFFprobeService(&config.Config{}, log), nil) scanner.storage = storage res, err := scanner.ScanLibrary(t.Context(), lib.ID) if err != nil { t.Fatalf("scan cloud: %v", err) } if res.Added != newFiles || res.Skipped != 1 { t.Fatalf("scan result = %#v, want new files added and existing skipped", res) } foundExistingProbe := false for { select { case task := <-scanner.cloudMediaProbeQueue: if task.path == existingPath { foundExistingProbe = true } default: if !foundExistingProbe { t.Fatal("existing media missing track metadata did not receive probe budget before new files") } return } } } func TestParseCloudImagePlaybackURL(t *testing.T) { typ, ref, ok := parseCloudImagePlaybackURL("http://nas.local/api/cloud/play/openlist?ref=%2FAnime%2FJianLai%2Fposter.jpg") if !ok || typ != "openlist" || ref != "/Anime/JianLai/poster.jpg" { t.Fatalf("parse cloud image url = typ=%q ref=%q ok=%v", typ, ref, ok) } if _, _, ok := parseCloudImagePlaybackURL("/api/cloud/play/openlist?ref=%2FAnime%2FJianLai%2Fmovie.mkv"); ok { t.Fatal("video cloud url should not be treated as artwork") } if _, _, ok := parseCloudImagePlaybackURL("https://image.tmdb.org/t/p/w500/poster.jpg"); ok { t.Fatal("remote HTTP poster should not be treated as cloud artwork") } }