Fix cloud media metadata repair and merging

internal/service/external_id_hints.go

@@ -0,0 +1,96 @@
+package service
+
+import (
+	"regexp"
+	"strings"
+)
+
+type mediaExternalIDHints struct {
+	TMDbID    int
+	BangumiID int
+	DoubanID  string
+	TheTVDBID string
+}
+
+var externalIDHintPatterns = []struct {
+	name string
+	re   *regexp.Regexp
+}{
+	{"tmdb", regexp.MustCompile(`(?i)(?:^|[^a-z0-9])(?:tmdb|tmdbid)[\s_:=#-]*(\d{2,})`)},
+	{"bangumi", regexp.MustCompile(`(?i)(?:^|[^a-z0-9])(?:bangumi|bgm)[\s_:=#-]*(\d{2,})`)},
+	{"douban", regexp.MustCompile(`(?i)(?:^|[^a-z0-9])(?:douban|db)[\s_:=#-]*(\d{2,})`)},
+	{"thetvdb", regexp.MustCompile(`(?i)(?:^|[^a-z0-9])(?:thetvdb|tvdb)[\s_:=#-]*(\d{2,})`)},
+}
+
+func externalIDHintsFromText(raw string) mediaExternalIDHints {
+	var hints mediaExternalIDHints
+	for _, item := range externalIDHintPatterns {
+		m := item.re.FindStringSubmatch(raw)
+		if len(m) < 2 {
+			continue
+		}
+		value := strings.TrimSpace(m[1])
+		switch item.name {
+		case "tmdb":
+			hints.TMDbID = mustAtoi(value)
+		case "bangumi":
+			hints.BangumiID = mustAtoi(value)
+		case "douban":
+			hints.DoubanID = value
+		case "thetvdb":
+			hints.TheTVDBID = value
+		}
+	}
+	return hints
+}
+
+func (h mediaExternalIDHints) useful() bool {
+	return h.TMDbID > 0 || h.BangumiID > 0 || strings.TrimSpace(h.DoubanID) != "" || strings.TrimSpace(h.TheTVDBID) != ""
+}
+
+func (h mediaExternalIDHints) applyToLocalMetadata(meta *LocalMetadata) *LocalMetadata {
+	if !h.useful() {
+		return meta
+	}
+	if meta == nil {
+		meta = &LocalMetadata{}
+	}
+	if h.TMDbID > 0 && meta.TMDbID <= 0 {
+		meta.TMDbID = h.TMDbID
+	}
+	if h.BangumiID > 0 && meta.BangumiID <= 0 {
+		meta.BangumiID = h.BangumiID
+	}
+	if strings.TrimSpace(h.DoubanID) != "" && meta.DoubanID == "" {
+		meta.DoubanID = strings.TrimSpace(h.DoubanID)
+	}
+	if strings.TrimSpace(h.TheTVDBID) != "" && meta.TheTVDBID == "" {
+		meta.TheTVDBID = strings.TrimSpace(h.TheTVDBID)
+	}
+	meta.PathHint = true
+	return meta
+}
+
+func pathHintMetadata(raw string, _ bool) (*LocalMetadata, mediaExternalIDHints) {
+	hints := externalIDHintsFromText(raw)
+	title, year := cloudSeriesTitleFromMediaPath(raw)
+	if title == "" {
+		title, year = CleanQuery(raw)
+	}
+	if !hints.useful() && title == "" && year <= 0 {
+		return nil, hints
+	}
+	meta := (&mediaExternalIDHints{
+		TMDbID:    hints.TMDbID,
+		BangumiID: hints.BangumiID,
+		DoubanID:  hints.DoubanID,
+		TheTVDBID: hints.TheTVDBID,
+	}).applyToLocalMetadata(&LocalMetadata{PathHint: true})
+	if title != "" {
+		meta.Title = title
+	}
+	if year > 0 {
+		meta.Year = year
+	}
+	return meta, hints
+}