refactor scraper service helpers

2026-10-08 14:26:37 +08:00 · 2026-06-26 14:08:37 +08:00
parent c08460baa7
commit 09c4e9252c
4 changed files with 392 additions and 373 deletions
@@ -0,0 +1,118 @@
+package service
+
+import (
+	"context"
+	"strings"
+
+	"go.uber.org/zap"
+
+	"github.com/ShukeBta/MediaStationGo/internal/model"
+)
+
+func (s *ScraperService) matchFromMediaExternalIDs(ctx context.Context, m *model.Media, lib *model.Library) *Match {
+	if s == nil || m == nil {
+		return nil
+	}
+	mediaType := ""
+	if lib != nil {
+		mediaType = lib.Type
+	}
+	if m.TMDbID > 0 {
+		if mediaIsEpisodic(m, lib) {
+			mediaType = "tv"
+		}
+		if match := s.manualTMDbMatchByID(ctx, m.TMDbID, normalizeMediaType(mediaType, m.Title, "")); match != nil {
+			if s.mediaExternalIDMatchTrusted(m, lib, match, "tmdb") {
+				return match
+			}
+		}
+	}
+	if strings.TrimSpace(m.DoubanID) != "" && s.douban != nil && s.douban.Enabled() {
+		if match, err := s.douban.GetMatchByID(ctx, strings.TrimSpace(m.DoubanID)); err == nil && match != nil {
+			if s.mediaExternalIDMatchTrusted(m, lib, match, "douban") {
+				return match
+			}
+		} else if err != nil {
+			s.log.Debug("douban id lookup failed", zap.String("media_id", m.ID), zap.String("douban_id", m.DoubanID), zap.Error(err))
+		}
+	}
+	if m.BangumiID > 0 && s.bangumi != nil && s.bangumi.Enabled() {
+		if match, err := s.bangumi.GetSubject(ctx, m.BangumiID); err == nil && match != nil {
+			if s.mediaExternalIDMatchTrusted(m, lib, match, "bangumi") {
+				return match
+			}
+		} else if err != nil {
+			s.log.Debug("bangumi id lookup failed", zap.String("media_id", m.ID), zap.Int("bangumi_id", m.BangumiID), zap.Error(err))
+		}
+	}
+	if strings.TrimSpace(m.TheTVDBID) != "" && s.thetvdb != nil && s.thetvdb.Enabled() {
+		if match, err := s.thetvdb.GetSeriesMatchByID(ctx, strings.TrimSpace(m.TheTVDBID)); err == nil && match != nil {
+			if s.mediaExternalIDMatchTrusted(m, lib, match, "thetvdb") {
+				return match
+			}
+		} else if err != nil {
+			s.log.Debug("thetvdb id lookup failed", zap.String("media_id", m.ID), zap.String("thetvdb_id", m.TheTVDBID), zap.Error(err))
+		}
+	}
+	return nil
+}
+
+func (s *ScraperService) mediaExternalIDMatchTrusted(m *model.Media, lib *model.Library, match *Match, source string) bool {
+	if match == nil || strings.TrimSpace(match.Title) == "" {
+		return false
+	}
+	if !mediaIsEpisodic(m, lib) {
+		return true
+	}
+	for _, candidate := range scrapeQueryCandidates(m, lib) {
+		if unsafeAutomaticEpisodeQuery(candidate) {
+			continue
+		}
+		if automaticMetadataTitleTrusted(candidate, match) {
+			return true
+		}
+	}
+	if s != nil && s.log != nil && m != nil {
+		s.log.Warn("episode external id match rejected",
+			zap.String("media_id", m.ID),
+			zap.String("source", source),
+			zap.String("title", match.Title),
+			zap.String("original_name", match.OriginalName),
+			zap.Int("tmdb_id", match.TMDbID),
+			zap.Int("bangumi_id", match.BangumiID),
+			zap.String("douban_id", match.DoubanID),
+			zap.String("thetvdb_id", match.TheTVDBID))
+	}
+	return false
+}
+
+func (s *ScraperService) applyFanartArtwork(ctx context.Context, match *Match) {
+	if s == nil || s.fanart == nil || !s.fanart.Enabled() || match == nil {
+		return
+	}
+	apply := func(a *Artwork) {
+		if a == nil {
+			return
+		}
+		if a.Poster != "" {
+			match.PosterURL = a.Poster
+		}
+		if a.Backdrop != "" {
+			match.BackdropURL = a.Backdrop
+		}
+	}
+	if match.TMDbID > 0 {
+		if a, err := s.fanart.MovieArtwork(ctx, match.TMDbID); err == nil {
+			apply(a)
+		} else {
+			s.log.Debug("fanart movie artwork failed", zap.Int("tmdb_id", match.TMDbID), zap.Error(err))
+		}
+	}
+	if strings.TrimSpace(match.TheTVDBID) != "" {
+		if a, err := s.fanart.TVArtwork(ctx, strings.TrimSpace(match.TheTVDBID)); err == nil {
+			apply(a)
+		} else {
+			s.log.Debug("fanart tv artwork failed", zap.String("thetvdb_id", match.TheTVDBID), zap.Error(err))
+		}
+	}
+}