MeBox/internal/service/external_id_hints.go

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
}