tcjlj/MeBox
1
0
Fork 0
mirror of https://github.com/truewhile/MeBox.git synced 2026-10-07 13:56:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: skip existing media in STRM tree generation

Browse Source
This commit is contained in:
ShukeBta
2026-07-02 23:35:24 +08:00
parent e700962f2a
commit 077bb10d25
7 changed files with 159 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/service/strm_tree_url_test.go
+61
View File
@@ -8,6 +8,9 @@ import (
"testing"
"go.uber.org/zap"
"github.com/ShukeBta/MediaStationGo/internal/model"
"github.com/ShukeBta/MediaStationGo/internal/repository"
)
func TestGenerateSTRMFromTreeUsesMediaPathFromURLQuery(t *testing.T) {
@@ -164,6 +167,64 @@ func TestGenerateSTRMFromTreeDoesNotDedupeDifferentProviders(t *testing.T) {
}
}
func TestGenerateSTRMFromTreeMissingOnlySkipsMediaAlreadyInLibrary(t *testing.T) {
db := newServiceTestDB(t, &model.Media{})
repos := repository.New(db)
rows := []model.Media{
{
Base: model.Base{ID: "existing-strm"},
Title: "Existing STRM",
Path: "cloud://openlist/Movies/Existing.Path.Copy.mkv",
STRMURL: "/api/cloud/play/openlist?ref=%2FMovies%2FExisting.Movie.mkv",
},
{
Base: model.Base{ID: "existing-cloud-path"},
Title: "Existing Cloud Path",
Path: "cloud://cloud115/Shows/Existing.Show.S01E01.mkv",
},
{
Base: model.Base{ID: "different-provider"},
Title: "Different Provider",
Path: "cloud://cloud115/Shows/Same.Provider.mkv",
},
}
for i := range rows {
if err := repos.DB.Create(&rows[i]).Error; err != nil {
t.Fatal(err)
}
}
outDir := filepath.Join(t.TempDir(), "strm")
svc := NewSTRMService(zap.NewNop(), repos, nil)
res, err := svc.GenerateFromTree(t.Context(), GenerateSTRMTreeOptions{
Provider: "openlist",
Paths: []string{
"cloud://openlist/Movies/Existing.Movie.mkv",
"cloud://openlist/Movies/New.Movie.mkv",
"cloud://cloud115/Shows/Existing.Show.S01E01.mkv",
"cloud://openlist/Shows/Same.Provider.mkv",
},
OutputDir: outDir,
MissingOnly: true,
})
if err != nil {
t.Fatal(err)
}
if res.Generated != 2 || res.Skipped != 2 || len(res.Errors) != 0 {
t.Fatalf("result = %#v, want generated=2 skipped=2", res)
}
if _, err := os.Stat(filepath.Join(outDir, "Movies", "Existing.Movie.strm")); !os.IsNotExist(err) {
t.Fatalf("existing media should not generate STRM, stat err=%v", err)
}
assertFileContains(t, filepath.Join(outDir, "Movies", "New.Movie.strm"), "/api/cloud/play/openlist?ref=%2FMovies%2FNew.Movie.mkv")
assertFileContains(t, filepath.Join(outDir, "Shows", "Same.Provider.strm"), "/api/cloud/play/openlist?ref=%2FShows%2FSame.Provider.mkv")
for _, item := range res.Items {
if item.Title == "Existing.Movie" && item.Reason != "already in media library" {
t.Fatalf("existing media skip reason = %q", item.Reason)
}
}
}
func TestGenerateSTRMFromTreeStripsPathListPrefixes(t *testing.T) {
outDir := filepath.Join(t.TempDir(), "strm")
svc := NewSTRMService(zap.NewNop(), nil, nil)