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

feat: add STRM tree recognize rename option

Browse Source
This commit is contained in:
ShukeBta
2026-07-03 01:46:13 +08:00
parent 9833c60e38
commit 470a639091
7 changed files with 118 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/service/strm_tree_import_test.go
+30
View File
@@ -44,6 +44,36 @@ func TestGenerateSTRMFromTreePaths(t *testing.T) {
}
}
func TestGenerateSTRMFromTreeRecognizeRenameOutputPaths(t *testing.T) {
outDir := filepath.Join(t.TempDir(), "strm")
svc := NewSTRMService(zap.NewNop(), nil, nil)
res, err := svc.GenerateFromTree(t.Context(), GenerateSTRMTreeOptions{
Provider: "openlist",
Paths: []string{"/电视剧/国产剧/南部档案/Season 01/Archives.The.Nanyang.Mystery.S01E02.2160p.WEB-DL.mkv", "/电影/Dune.Part.Two.2024.2160p.WEB-DL.mkv"},
SourceRoot: "/电视剧",
OutputDir: outDir,
RecognizeRename: true,
})
if err != nil {
t.Fatal(err)
}
if res.Generated != 2 || len(res.Errors) != 0 {
t.Fatalf("result = %#v, want two generated renamed STRM files", res)
}
episode := readSTRM(t, filepath.Join(outDir, "南部档案", "Season 01", "南部档案 S01E02.strm"))
if !strings.Contains(episode, "ref=%2F%E7%94%B5%E8%A7%86%E5%89%A7%2F%E5%9B%BD%E4%BA%A7%E5%89%A7%2F%E5%8D%97%E9%83%A8%E6%A1%A3%E6%A1%88%2FSeason+01%2FArchives.The.Nanyang.Mystery.S01E02.2160p.WEB-DL.mkv") {
t.Fatalf("episode strm URL = %q, want original cloud ref preserved", episode)
}
movie := readSTRM(t, filepath.Join(outDir, "Dune Part Two (2024)", "Dune Part Two (2024).strm"))
if !strings.Contains(movie, "ref=%2F%E7%94%B5%E5%BD%B1%2FDune.Part.Two.2024.2160p.WEB-DL.mkv") {
t.Fatalf("movie strm URL = %q, want original cloud ref preserved", movie)
}
if _, err := os.Stat(filepath.Join(outDir, "国产剧", "南部档案", "Season 01", "Archives.The.Nanyang.Mystery.S01E02.2160p.WEB-DL.strm")); !os.IsNotExist(err) {
t.Fatalf("recognize rename should not leave original episode output path, stat err=%v", err)
}
}
func TestGenerateSTRMFromTreeSupportsCommonVideoExtensions(t *testing.T) {
outDir := filepath.Join(t.TempDir(), "strm")
svc := NewSTRMService(zap.NewNop(), nil, nil)