mirror of
https://github.com/truewhile/MeBox.git
synced 2026-11-12 19:34:24 +08:00
truewhile
4f19e2210e
外挂字幕原来直接拿文件名的语言后缀当名称（chs、chs[BilibiliID_danmu]）， 内嵌字幕没有 title 标签时也是裸语言码（chi）。改成统一映射成中文名： chi/zh -> 中文，chs -> 简体中文，jpn -> 日本語，并把弹幕转存的 [BilibiliID_danmu] 文件标注为「弹幕」，避免和对白字幕混淆。 认不出来的标记保持原样（含 und），Emby 客户端的「字幕 (ASS)」兜底不变。
305 lines
9.3 KiB
Go
305 lines
9.3 KiB
Go
package service
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/truewhile/MeBox/internal/config"
|
|
"github.com/truewhile/MeBox/internal/model"
|
|
"github.com/truewhile/MeBox/internal/repository"
|
|
)
|
|
|
|
func TestSubtitleDiscoverNoTracksReturnsEmptySlice(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.Library{}, &model.Media{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
dir := t.TempDir()
|
|
media := model.Media{
|
|
Title: "No Subtitles",
|
|
Path: filepath.Join(dir, "No Subtitles.mkv"),
|
|
}
|
|
if err := db.Create(&media).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
svc := NewSubtitleService(&config.Config{}, zap.NewNop(), repository.New(db))
|
|
tracks, err := svc.Discover(t.Context(), media.ID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if tracks == nil {
|
|
t.Fatal("tracks is nil, want empty slice")
|
|
}
|
|
if len(tracks) != 0 {
|
|
t.Fatalf("len(tracks) = %d, want 0", len(tracks))
|
|
}
|
|
}
|
|
|
|
func TestDiscoverExternalOnlyDoesNotResolveOrProbeCloudMedia(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.Library{}, &model.Media{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
media := model.Media{
|
|
Title: "Cloud Media",
|
|
Path: "cloud://115/example/video.mkv",
|
|
STRMURL: "cloud://115/example/video.mkv",
|
|
}
|
|
if err := db.Create(&media).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
svc := NewSubtitleService(&config.Config{}, zap.NewNop(), repository.New(db))
|
|
resolveCalls := 0
|
|
svc.SetStrmPlayTargetResolver(func(context.Context, string) (*StrmPlayResult, error) {
|
|
resolveCalls++
|
|
return nil, errors.New("resolver must not be called")
|
|
})
|
|
|
|
tracks, err := svc.DiscoverExternalOnly(t.Context(), media.ID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if resolveCalls != 0 {
|
|
t.Fatalf("STRM resolver called %d times, want 0", resolveCalls)
|
|
}
|
|
if len(tracks) != 0 {
|
|
t.Fatalf("len(tracks) = %d, want 0", len(tracks))
|
|
}
|
|
}
|
|
|
|
func TestEmbeddedSubtitleProbeClassifiesTextAndBitmapTracks(t *testing.T) {
|
|
var probe embeddedSubtitleProbe
|
|
raw := []byte(`{"streams":[
|
|
{"index":2,"codec_name":"ass","tags":{"language":"chi","title":"中文"},"disposition":{"default":1}},
|
|
{"index":4,"codec_name":"hdmv_pgs_subtitle","tags":{"language":"eng"},"disposition":{"forced":1}}
|
|
]}`)
|
|
if err := json.Unmarshal(raw, &probe); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tracks := subtitleTracksFromProbe(probe)
|
|
if len(tracks) != 2 {
|
|
t.Fatalf("len(tracks) = %d, want 2", len(tracks))
|
|
}
|
|
if tracks[0].Delivery != "ass" || tracks[0].Path != "embedded:2" {
|
|
t.Fatalf("text track = %#v", tracks[0])
|
|
}
|
|
if tracks[1].Delivery != "burn" || tracks[1].StreamIndex != 4 {
|
|
t.Fatalf("bitmap track = %#v", tracks[1])
|
|
}
|
|
}
|
|
|
|
func TestSubtitleDeliveryClassifiesASSForLibass(t *testing.T) {
|
|
if got := subtitleDeliveryForCodec("ass"); got != "ass" {
|
|
t.Fatalf("ass delivery = %q, want ass", got)
|
|
}
|
|
if got := subtitleDeliveryForCodec("ssa"); got != "ass" {
|
|
t.Fatalf("ssa delivery = %q, want ass", got)
|
|
}
|
|
if got := subtitleDeliveryForCodec("subrip"); got != "webvtt" {
|
|
t.Fatalf("subrip delivery = %q, want webvtt", got)
|
|
}
|
|
}
|
|
|
|
func TestNormaliseTimecode(t *testing.T) {
|
|
cases := map[string]string{
|
|
"0:00:01": "00:00:01",
|
|
"0:00:01.51": "00:00:01.510", // ASS centiseconds -> 3-digit ms
|
|
"0:00:04.123": "00:00:04.123", // already 3 digits
|
|
"00:00:12.07": "00:00:12.070", // 2-digit fraction padded
|
|
"00:01:30.1234": "00:01:30.123", // capped at 3 digits
|
|
"00:00:1.5": "00:00:01.500", // single-digit seconds and fraction
|
|
}
|
|
for in, want := range cases {
|
|
if got := normaliseTimecode(in); got != want {
|
|
t.Errorf("normaliseTimecode(%q) = %q, want %q", in, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAssToVTTDeduplicatesDialogueAndNormalisesLineBreaks(t *testing.T) {
|
|
body := strings.Join([]string{
|
|
`Dialogue: 0,0:00:01.00,0:00:02.00,Default,,0,0,0,,{\an2}第一行\N第二行`,
|
|
`Dialogue: 1,0:00:01.00,0:00:02.00,Copy,,0,0,0,,{\bord2}第一行\N第二行`,
|
|
`Dialogue: 0,0:00:03.00,0:00:04.00,Default,,0,0,0,,{\i1}`,
|
|
}, "\n")
|
|
|
|
got := assToVTT(body)
|
|
if strings.Count(got, "第一行\n第二行") != 1 {
|
|
t.Fatalf("duplicate ASS dialogue was not collapsed:\n%s", got)
|
|
}
|
|
if strings.Contains(got, `\N`) || strings.Contains(got, `\an2`) {
|
|
t.Fatalf("ASS control sequences leaked into WebVTT:\n%s", got)
|
|
}
|
|
if strings.Contains(got, "00:00:03.000 --> 00:00:04.000") {
|
|
t.Fatalf("empty styled dialogue should be omitted:\n%s", got)
|
|
}
|
|
}
|
|
|
|
func TestSubtitleServeRawWritesSourceBytes(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.Library{}, &model.Media{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
dir := t.TempDir()
|
|
videoPath := filepath.Join(dir, "MovieName.mkv")
|
|
subPath := filepath.Join(dir, "MovieName.ass")
|
|
raw := "Dialogue: 0,0:00:01.00,0:00:02.00,Default,,0,0,0,,線合なライン\n"
|
|
if err := os.WriteFile(videoPath, []byte("video"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(subPath, []byte(raw), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
media := model.Media{Title: "MovieName", Path: videoPath}
|
|
if err := db.Create(&media).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
svc := NewSubtitleService(&config.Config{}, zap.NewNop(), repository.New(db))
|
|
var buf bytes.Buffer
|
|
if err := svc.ServeRaw(t.Context(), media.ID, subPath, &buf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// ServeRaw must NOT convert ASS->VTT; it returns the exact source bytes.
|
|
if got := buf.String(); got != raw {
|
|
t.Fatalf("ServeRaw returned %q, want raw %q", got, raw)
|
|
}
|
|
|
|
buf.Reset()
|
|
if err := svc.ServeASS(t.Context(), media.ID, subPath, &buf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := buf.String(); got != raw {
|
|
t.Fatalf("ServeASS returned %q, want raw ASS %q", got, raw)
|
|
}
|
|
|
|
buf.Reset()
|
|
if err := svc.ServeRaw(t.Context(), media.ID, videoPath, &buf); err == nil {
|
|
t.Fatal("ServeRaw accepted a non-subtitle file")
|
|
}
|
|
}
|
|
|
|
func TestSubtitleLanguageNameMapsCommonTags(t *testing.T) {
|
|
cases := map[string]string{
|
|
"chs": "简体中文",
|
|
"CHS": "简体中文",
|
|
"zh-Hans": "简体中文",
|
|
"cht": "繁体中文",
|
|
"chi": "中文",
|
|
"zh": "中文",
|
|
"jpn": "日本語",
|
|
"eng": "English",
|
|
"kor": "한국어",
|
|
"chs[bilibiliid_danmu]": "简体中文",
|
|
"und": "",
|
|
"": "",
|
|
}
|
|
for raw, want := range cases {
|
|
if got := subtitleLanguageName(raw); got != want {
|
|
t.Errorf("subtitleLanguageName(%q) = %q, want %q", raw, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSubtitleDisplayLabelKeepsUnknownTags(t *testing.T) {
|
|
// 认不出来的标记原样返回：Emby 的 DisplayTitle 依赖 `und` 走「字幕 (ASS)」兜底。
|
|
if got := subtitleDisplayLabel("und"); got != "und" {
|
|
t.Fatalf("subtitleDisplayLabel(und) = %q, want und", got)
|
|
}
|
|
if got := subtitleDisplayLabel("weird"); got != "weird" {
|
|
t.Fatalf("subtitleDisplayLabel(weird) = %q, want weird", got)
|
|
}
|
|
}
|
|
|
|
func TestIsDanmakuSubtitle(t *testing.T) {
|
|
if !isDanmakuSubtitle("S01E01.chs[BilibiliID_danmu].ass") {
|
|
t.Fatal("danmaku sidecar was not detected")
|
|
}
|
|
if !isDanmakuSubtitle("弹幕.ass") {
|
|
t.Fatal("Chinese danmaku name was not detected")
|
|
}
|
|
if isDanmakuSubtitle("S01E01.chs.ass") {
|
|
t.Fatal("normal subtitle was misdetected as danmaku")
|
|
}
|
|
}
|
|
|
|
func TestDiscoverExternalSubtitleLabelsAreReadable(t *testing.T) {
|
|
dir := t.TempDir()
|
|
videoPath := filepath.Join(dir, "S01E01.mkv")
|
|
files := map[string]string{
|
|
"S01E01.chs.ass": "[Script Info]\n",
|
|
"S01E01.chs[BilibiliID_danmu].ass": "[Script Info]\n",
|
|
}
|
|
for name, body := range files {
|
|
if err := os.WriteFile(filepath.Join(dir, name), []byte(body), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
m := &model.Media{Path: videoPath}
|
|
tracks := discoverExternalSubtitleTracks(m)
|
|
if len(tracks) != 2 {
|
|
t.Fatalf("len(tracks) = %d, want 2: %#v", len(tracks), tracks)
|
|
}
|
|
labels := map[string]string{}
|
|
for _, track := range tracks {
|
|
labels[filepath.Base(track.Path)] = track.Label
|
|
}
|
|
if got := labels["S01E01.chs.ass"]; got != "简体中文" {
|
|
t.Fatalf("chs.ass label = %q, want 简体中文", got)
|
|
}
|
|
if got := labels["S01E01.chs[BilibiliID_danmu].ass"]; got != "简体中文 · 弹幕" {
|
|
t.Fatalf("danmaku label = %q, want 简体中文 · 弹幕", got)
|
|
}
|
|
}
|
|
|
|
func TestEmbeddedSubtitleProbeUsesReadableLanguage(t *testing.T) {
|
|
var probe embeddedSubtitleProbe
|
|
raw := []byte(`{"streams":[
|
|
{"index":2,"codec_name":"hdmv_pgs_subtitle","tags":{"language":"chi"},"disposition":{"default":1}},
|
|
{"index":3,"codec_name":"subrip","tags":{"language":"jpn"}},
|
|
{"index":4,"codec_name":"subrip","tags":{"language":"chi","title":"简日双语"}}
|
|
]}`)
|
|
if err := json.Unmarshal(raw, &probe); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tracks := subtitleTracksFromProbe(probe)
|
|
if len(tracks) != 3 {
|
|
t.Fatalf("len(tracks) = %d, want 3", len(tracks))
|
|
}
|
|
if got := tracks[0].Label; got != "中文（默认）（内嵌·图片）" {
|
|
t.Fatalf("bitmap label = %q", got)
|
|
}
|
|
if got := tracks[1].Label; got != "日本語（内嵌）" {
|
|
t.Fatalf("text label = %q", got)
|
|
}
|
|
// 容器自带 title 时优先用 title。
|
|
if got := tracks[2].Label; got != "简日双语（内嵌）" {
|
|
t.Fatalf("titled label = %q", got)
|
|
}
|
|
}
|