mirror of
https://github.com/truewhile/MeBox.git
synced 2026-10-06 05:16:38 +08:00
refactor manual scrape service by responsibility
This commit is contained in:
@@ -3,8 +3,6 @@ package service
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/ShukeBta/MediaStationGo/internal/model"
|
"github.com/ShukeBta/MediaStationGo/internal/model"
|
||||||
@@ -39,154 +37,6 @@ func (r ManualScrapeRequest) EpisodeArtworkOption() *bool {
|
|||||||
return r.EpisodeArtwork
|
return r.EpisodeArtwork
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *ScraperService) ManualSearch(ctx context.Context, media *model.Media, query, provider, mediaType string) ([]ExternalMediaResult, error) {
|
|
||||||
if s == nil || media == nil {
|
|
||||||
return nil, errors.New("media required")
|
|
||||||
}
|
|
||||||
lib, _ := s.repo.Library.FindByID(ctx, media.LibraryID)
|
|
||||||
queries := manualSearchQueries(media, lib, query)
|
|
||||||
if len(queries) == 0 {
|
|
||||||
return nil, errors.New("search query required")
|
|
||||||
}
|
|
||||||
if mediaType == "" {
|
|
||||||
if mediaIsEpisodic(media, lib) {
|
|
||||||
mediaType = "tv"
|
|
||||||
} else if lib != nil {
|
|
||||||
mediaType = lib.Type
|
|
||||||
}
|
|
||||||
}
|
|
||||||
mediaType = normalizeMediaType(mediaType, queries[0], "")
|
|
||||||
providers := manualSearchProviderSet(provider)
|
|
||||||
year := mediaYearHint(media)
|
|
||||||
if year <= 0 {
|
|
||||||
_, year = CleanQuery(queries[0])
|
|
||||||
}
|
|
||||||
|
|
||||||
out := make([]ExternalMediaResult, 0, 6)
|
|
||||||
add := func(source, typ string, match *Match) {
|
|
||||||
if match == nil || strings.TrimSpace(match.Title) == "" {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
out = append(out, ExternalMediaResult{
|
|
||||||
Source: source,
|
|
||||||
MediaType: typ,
|
|
||||||
Title: match.Title,
|
|
||||||
OriginalName: match.OriginalName,
|
|
||||||
Overview: match.Overview,
|
|
||||||
PosterURL: match.PosterURL,
|
|
||||||
BackdropURL: match.BackdropURL,
|
|
||||||
Year: match.Year,
|
|
||||||
Rating: match.Rating,
|
|
||||||
TMDbID: match.TMDbID,
|
|
||||||
BangumiID: match.BangumiID,
|
|
||||||
DoubanID: match.DoubanID,
|
|
||||||
TheTVDBID: match.TheTVDBID,
|
|
||||||
SubscribeKeyword: buildSubscribeKeyword(match.Title, match.Year),
|
|
||||||
SubscribeAliases: buildSubscribeAliases(match.Title, match.OriginalName, match.Year),
|
|
||||||
Languages: match.Languages,
|
|
||||||
Countries: match.Countries,
|
|
||||||
Genres: match.Genres,
|
|
||||||
NSFW: match.NSFW,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
if providers.want("adult") {
|
|
||||||
for _, candidateQuery := range queries {
|
|
||||||
for _, match := range s.manualAdultMatches(ctx, media, candidateQuery) {
|
|
||||||
add("adult", "adult", match)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if providers.want("tmdb") {
|
|
||||||
for _, candidateQuery := range queries {
|
|
||||||
for _, candidate := range s.manualTMDbCandidates(ctx, candidateQuery, year, mediaType) {
|
|
||||||
add("tmdb", candidate.MediaType, candidate.Match)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if providers.want("douban") {
|
|
||||||
for _, candidateQuery := range queries {
|
|
||||||
if match := s.manualDoubanMatch(ctx, candidateQuery); match != nil {
|
|
||||||
add("douban", normalizeMediaType(mediaType, candidateQuery, ""), match)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if providers.want("bangumi") {
|
|
||||||
for _, candidateQuery := range queries {
|
|
||||||
if match := s.manualBangumiMatch(ctx, candidateQuery); match != nil {
|
|
||||||
add("bangumi", "anime", match)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if providers.want("thetvdb") {
|
|
||||||
for _, candidateQuery := range queries {
|
|
||||||
if match := s.manualTheTVDBMatch(ctx, candidateQuery); match != nil {
|
|
||||||
add("thetvdb", "tv", match)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return dedupeExternalMedia(out), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type manualSearchProviders map[string]struct{}
|
|
||||||
|
|
||||||
func manualSearchProviderSet(provider string) manualSearchProviders {
|
|
||||||
out := manualSearchProviders{}
|
|
||||||
for _, field := range strings.FieldsFunc(provider, func(r rune) bool {
|
|
||||||
return r == ',' || r == ';' || r == '|' || r == ' '
|
|
||||||
}) {
|
|
||||||
field = strings.ToLower(strings.TrimSpace(field))
|
|
||||||
if field == "" || field == "all" {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
out[field] = struct{}{}
|
|
||||||
}
|
|
||||||
if len(out) == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p manualSearchProviders) want(provider string) bool {
|
|
||||||
if len(p) == 0 {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
_, ok := p[provider]
|
|
||||||
return ok
|
|
||||||
}
|
|
||||||
|
|
||||||
func manualSearchQueries(media *model.Media, lib *model.Library, query string) []string {
|
|
||||||
seen := map[string]struct{}{}
|
|
||||||
out := make([]string, 0, 4)
|
|
||||||
add := func(value string) {
|
|
||||||
value = strings.Join(strings.Fields(strings.TrimSpace(value)), " ")
|
|
||||||
if value == "" {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
key := strings.ToLower(value)
|
|
||||||
if _, ok := seen[key]; ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
seen[key] = struct{}{}
|
|
||||||
out = append(out, value)
|
|
||||||
}
|
|
||||||
|
|
||||||
add(query)
|
|
||||||
if strings.TrimSpace(query) == "" && media != nil {
|
|
||||||
add(firstText(media.Title, media.OriginalName))
|
|
||||||
}
|
|
||||||
if media != nil {
|
|
||||||
for _, candidate := range scrapeQueryCandidates(media, lib) {
|
|
||||||
add(candidate)
|
|
||||||
}
|
|
||||||
if len(out) == 0 {
|
|
||||||
title, _ := CleanQuery(media.Path)
|
|
||||||
add(title)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) ApplyManualMatch(ctx context.Context, mediaID string, req ManualScrapeRequest) (*model.Media, error) {
|
func (s *ScraperService) ApplyManualMatch(ctx context.Context, mediaID string, req ManualScrapeRequest) (*model.Media, error) {
|
||||||
return s.ApplyManualMatchWithOptions(ctx, mediaID, req, ScrapeOptions{EpisodeArtwork: req.EpisodeArtworkOption()})
|
return s.ApplyManualMatchWithOptions(ctx, mediaID, req, ScrapeOptions{EpisodeArtwork: req.EpisodeArtworkOption()})
|
||||||
}
|
}
|
||||||
@@ -253,275 +103,3 @@ func (s *ScraperService) manualRequestMatch(ctx context.Context, req ManualScrap
|
|||||||
}
|
}
|
||||||
return fallback()
|
return fallback()
|
||||||
}
|
}
|
||||||
|
|
||||||
type manualTMDbCandidate struct {
|
|
||||||
MediaType string
|
|
||||||
Match *Match
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualTMDbCandidates(ctx context.Context, query string, year int, mediaType string) []manualTMDbCandidate {
|
|
||||||
if s.tmdb == nil || !s.tmdb.Enabled() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if id, ok := parsePositiveInt(query); ok {
|
|
||||||
out := make([]manualTMDbCandidate, 0, 2)
|
|
||||||
for _, typ := range manualTMDbIDSearchTypes(mediaType) {
|
|
||||||
if match := s.manualTMDbMatchByIDForType(ctx, id, typ); match != nil {
|
|
||||||
out = append(out, manualTMDbCandidate{MediaType: typ, Match: match})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
out := make([]manualTMDbCandidate, 0, 4)
|
|
||||||
for _, typ := range manualTMDbSearchTypes(mediaType) {
|
|
||||||
switch typ {
|
|
||||||
case "movie":
|
|
||||||
if matches, err := s.tmdb.SearchMovieCandidates(ctx, query, year); err == nil {
|
|
||||||
for _, match := range matches {
|
|
||||||
out = append(out, manualTMDbCandidate{MediaType: "movie", Match: match})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
case "tv":
|
|
||||||
if matches, err := s.tmdb.SearchTVCandidates(ctx, query, year); err == nil {
|
|
||||||
for _, match := range matches {
|
|
||||||
out = append(out, manualTMDbCandidate{MediaType: "tv", Match: match})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualTMDbMatches(ctx context.Context, query string, year int, mediaType string) []*Match {
|
|
||||||
candidates := s.manualTMDbCandidates(ctx, query, year, mediaType)
|
|
||||||
out := make([]*Match, 0, len(candidates))
|
|
||||||
for _, candidate := range candidates {
|
|
||||||
out = append(out, candidate.Match)
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
func manualTMDbIDSearchTypes(mediaType string) []string {
|
|
||||||
switch normalizeMediaType(mediaType, "", "") {
|
|
||||||
case "tv", "anime", "variety":
|
|
||||||
return []string{"tv", "movie"}
|
|
||||||
case "movie", "adult":
|
|
||||||
return []string{"movie", "tv"}
|
|
||||||
default:
|
|
||||||
return []string{"movie", "tv"}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func manualTMDbSearchTypes(mediaType string) []string {
|
|
||||||
if strings.TrimSpace(mediaType) == "" {
|
|
||||||
return []string{"movie", "tv"}
|
|
||||||
}
|
|
||||||
switch normalizeMediaType(mediaType, "", "") {
|
|
||||||
case "tv", "anime", "variety":
|
|
||||||
return []string{"tv", "movie"}
|
|
||||||
case "movie", "adult":
|
|
||||||
return []string{"movie"}
|
|
||||||
default:
|
|
||||||
return []string{"movie", "tv"}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualAdultMatches(ctx context.Context, media *model.Media, query string) []*Match {
|
|
||||||
candidates := []string{query}
|
|
||||||
if media != nil {
|
|
||||||
candidates = append(candidates, media.Path, media.OriginalName, media.Title)
|
|
||||||
}
|
|
||||||
out := make([]*Match, 0, 1)
|
|
||||||
seen := map[string]struct{}{}
|
|
||||||
for _, candidate := range candidates {
|
|
||||||
code := normalizeAdultCode(candidate)
|
|
||||||
if code == "" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if _, ok := seen[code]; ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
seen[code] = struct{}{}
|
|
||||||
if match := s.manualAdultMatch(ctx, code); match != nil {
|
|
||||||
out = append(out, match)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualAdultMatch(ctx context.Context, code string) *Match {
|
|
||||||
if s.adult == nil || !s.adult.Enabled() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
match, err := s.adult.Search(ctx, code)
|
|
||||||
if err != nil || match == nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualTMDbMatchByID(ctx context.Context, id int, mediaType string) *Match {
|
|
||||||
if s.tmdb == nil || !s.tmdb.Enabled() || id <= 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if mediaType == "tv" || mediaType == "anime" || mediaType == "variety" {
|
|
||||||
if match, err := s.tmdb.GetTVMatch(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if match, err := s.tmdb.GetMovieMatch(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
if match, err := s.tmdb.GetTVMatch(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualTMDbMatchByIDForType(ctx context.Context, id int, mediaType string) *Match {
|
|
||||||
if s.tmdb == nil || !s.tmdb.Enabled() || id <= 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
switch normalizeMediaType(mediaType, "", "") {
|
|
||||||
case "tv", "anime", "variety":
|
|
||||||
if match, err := s.tmdb.GetTVMatch(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
case "movie", "adult":
|
|
||||||
if match, err := s.tmdb.GetMovieMatch(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualDoubanMatch(ctx context.Context, query string) *Match {
|
|
||||||
if s.douban == nil || !s.douban.Enabled() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if id, ok := parsePositiveIDString(query); ok {
|
|
||||||
if match, err := s.douban.GetMatchByID(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
}
|
|
||||||
match, err := s.douban.SearchMatch(ctx, query)
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualBangumiMatch(ctx context.Context, query string) *Match {
|
|
||||||
if s.bangumi == nil || !s.bangumi.Enabled() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if id, ok := parsePositiveInt(query); ok {
|
|
||||||
if match, err := s.bangumi.GetSubject(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
}
|
|
||||||
match, err := s.bangumi.Search(ctx, query)
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *ScraperService) manualTheTVDBMatch(ctx context.Context, query string) *Match {
|
|
||||||
if s.thetvdb == nil || !s.thetvdb.Enabled() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if id, ok := parsePositiveIDString(normalizeTheTVDBSeriesID(query)); ok {
|
|
||||||
if match, err := s.thetvdb.GetSeriesMatchByID(ctx, id); err == nil && match != nil {
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
}
|
|
||||||
match, err := s.thetvdb.SearchSeries(ctx, query)
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
|
|
||||||
func mergeManualRequestIntoMatch(match *Match, req ManualScrapeRequest) *Match {
|
|
||||||
if match == nil {
|
|
||||||
match = &Match{}
|
|
||||||
}
|
|
||||||
if req.Title != "" {
|
|
||||||
match.Title = req.Title
|
|
||||||
}
|
|
||||||
if mediaType := normalizeOrganizeMediaType(req.MediaType); mediaType != "" {
|
|
||||||
match.MediaType = mediaType
|
|
||||||
}
|
|
||||||
if req.OriginalName != "" {
|
|
||||||
match.OriginalName = req.OriginalName
|
|
||||||
}
|
|
||||||
if req.Overview != "" {
|
|
||||||
match.Overview = req.Overview
|
|
||||||
}
|
|
||||||
if req.PosterURL != "" {
|
|
||||||
match.PosterURL = req.PosterURL
|
|
||||||
}
|
|
||||||
if req.BackdropURL != "" {
|
|
||||||
match.BackdropURL = req.BackdropURL
|
|
||||||
}
|
|
||||||
if req.Year > 0 {
|
|
||||||
match.Year = req.Year
|
|
||||||
}
|
|
||||||
if req.Rating > 0 {
|
|
||||||
match.Rating = req.Rating
|
|
||||||
}
|
|
||||||
if req.TMDbID > 0 {
|
|
||||||
match.TMDbID = req.TMDbID
|
|
||||||
}
|
|
||||||
if req.BangumiID > 0 {
|
|
||||||
match.BangumiID = req.BangumiID
|
|
||||||
}
|
|
||||||
if req.DoubanID != "" {
|
|
||||||
match.DoubanID = req.DoubanID
|
|
||||||
}
|
|
||||||
if req.TheTVDBID != "" {
|
|
||||||
match.TheTVDBID = req.TheTVDBID
|
|
||||||
}
|
|
||||||
if len(req.Genres) > 0 {
|
|
||||||
match.Genres = req.Genres
|
|
||||||
}
|
|
||||||
if len(req.Countries) > 0 {
|
|
||||||
match.Countries = req.Countries
|
|
||||||
}
|
|
||||||
if len(req.Languages) > 0 {
|
|
||||||
match.Languages = req.Languages
|
|
||||||
}
|
|
||||||
if req.NSFW {
|
|
||||||
match.NSFW = true
|
|
||||||
}
|
|
||||||
return match
|
|
||||||
}
|
|
||||||
|
|
||||||
func isTVLikeTMDbMatch(match *Match, mediaType string) bool {
|
|
||||||
return mediaType == "tv" || mediaType == "anime" || mediaType == "variety"
|
|
||||||
}
|
|
||||||
|
|
||||||
func parsePositiveInt(value string) (int, bool) {
|
|
||||||
value = strings.TrimSpace(value)
|
|
||||||
if strings.Contains(value, ":") {
|
|
||||||
value = value[strings.LastIndex(value, ":")+1:]
|
|
||||||
}
|
|
||||||
id, err := strconv.Atoi(strings.TrimSpace(value))
|
|
||||||
return id, err == nil && id > 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func parsePositiveIDString(value string) (string, bool) {
|
|
||||||
id, ok := parsePositiveInt(value)
|
|
||||||
if !ok {
|
|
||||||
return "", false
|
|
||||||
}
|
|
||||||
return strconv.Itoa(id), true
|
|
||||||
}
|
|
||||||
|
|
||||||
func manualScrapeBatchName(ids []string) string {
|
|
||||||
if len(ids) == 1 {
|
|
||||||
return ids[0]
|
|
||||||
}
|
|
||||||
return fmt.Sprintf("%d 个媒体", len(ids))
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -0,0 +1,90 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func mergeManualRequestIntoMatch(match *Match, req ManualScrapeRequest) *Match {
|
||||||
|
if match == nil {
|
||||||
|
match = &Match{}
|
||||||
|
}
|
||||||
|
if req.Title != "" {
|
||||||
|
match.Title = req.Title
|
||||||
|
}
|
||||||
|
if mediaType := normalizeOrganizeMediaType(req.MediaType); mediaType != "" {
|
||||||
|
match.MediaType = mediaType
|
||||||
|
}
|
||||||
|
if req.OriginalName != "" {
|
||||||
|
match.OriginalName = req.OriginalName
|
||||||
|
}
|
||||||
|
if req.Overview != "" {
|
||||||
|
match.Overview = req.Overview
|
||||||
|
}
|
||||||
|
if req.PosterURL != "" {
|
||||||
|
match.PosterURL = req.PosterURL
|
||||||
|
}
|
||||||
|
if req.BackdropURL != "" {
|
||||||
|
match.BackdropURL = req.BackdropURL
|
||||||
|
}
|
||||||
|
if req.Year > 0 {
|
||||||
|
match.Year = req.Year
|
||||||
|
}
|
||||||
|
if req.Rating > 0 {
|
||||||
|
match.Rating = req.Rating
|
||||||
|
}
|
||||||
|
if req.TMDbID > 0 {
|
||||||
|
match.TMDbID = req.TMDbID
|
||||||
|
}
|
||||||
|
if req.BangumiID > 0 {
|
||||||
|
match.BangumiID = req.BangumiID
|
||||||
|
}
|
||||||
|
if req.DoubanID != "" {
|
||||||
|
match.DoubanID = req.DoubanID
|
||||||
|
}
|
||||||
|
if req.TheTVDBID != "" {
|
||||||
|
match.TheTVDBID = req.TheTVDBID
|
||||||
|
}
|
||||||
|
if len(req.Genres) > 0 {
|
||||||
|
match.Genres = req.Genres
|
||||||
|
}
|
||||||
|
if len(req.Countries) > 0 {
|
||||||
|
match.Countries = req.Countries
|
||||||
|
}
|
||||||
|
if len(req.Languages) > 0 {
|
||||||
|
match.Languages = req.Languages
|
||||||
|
}
|
||||||
|
if req.NSFW {
|
||||||
|
match.NSFW = true
|
||||||
|
}
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
|
||||||
|
func isTVLikeTMDbMatch(match *Match, mediaType string) bool {
|
||||||
|
return mediaType == "tv" || mediaType == "anime" || mediaType == "variety"
|
||||||
|
}
|
||||||
|
|
||||||
|
func parsePositiveInt(value string) (int, bool) {
|
||||||
|
value = strings.TrimSpace(value)
|
||||||
|
if strings.Contains(value, ":") {
|
||||||
|
value = value[strings.LastIndex(value, ":")+1:]
|
||||||
|
}
|
||||||
|
id, err := strconv.Atoi(strings.TrimSpace(value))
|
||||||
|
return id, err == nil && id > 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func parsePositiveIDString(value string) (string, bool) {
|
||||||
|
id, ok := parsePositiveInt(value)
|
||||||
|
if !ok {
|
||||||
|
return "", false
|
||||||
|
}
|
||||||
|
return strconv.Itoa(id), true
|
||||||
|
}
|
||||||
|
|
||||||
|
func manualScrapeBatchName(ids []string) string {
|
||||||
|
if len(ids) == 1 {
|
||||||
|
return ids[0]
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%d 个媒体", len(ids))
|
||||||
|
}
|
||||||
@@ -0,0 +1,197 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/ShukeBta/MediaStationGo/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type manualTMDbCandidate struct {
|
||||||
|
MediaType string
|
||||||
|
Match *Match
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualTMDbCandidates(ctx context.Context, query string, year int, mediaType string) []manualTMDbCandidate {
|
||||||
|
if s.tmdb == nil || !s.tmdb.Enabled() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if id, ok := parsePositiveInt(query); ok {
|
||||||
|
out := make([]manualTMDbCandidate, 0, 2)
|
||||||
|
for _, typ := range manualTMDbIDSearchTypes(mediaType) {
|
||||||
|
if match := s.manualTMDbMatchByIDForType(ctx, id, typ); match != nil {
|
||||||
|
out = append(out, manualTMDbCandidate{MediaType: typ, Match: match})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
out := make([]manualTMDbCandidate, 0, 4)
|
||||||
|
for _, typ := range manualTMDbSearchTypes(mediaType) {
|
||||||
|
switch typ {
|
||||||
|
case "movie":
|
||||||
|
if matches, err := s.tmdb.SearchMovieCandidates(ctx, query, year); err == nil {
|
||||||
|
for _, match := range matches {
|
||||||
|
out = append(out, manualTMDbCandidate{MediaType: "movie", Match: match})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case "tv":
|
||||||
|
if matches, err := s.tmdb.SearchTVCandidates(ctx, query, year); err == nil {
|
||||||
|
for _, match := range matches {
|
||||||
|
out = append(out, manualTMDbCandidate{MediaType: "tv", Match: match})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualTMDbMatches(ctx context.Context, query string, year int, mediaType string) []*Match {
|
||||||
|
candidates := s.manualTMDbCandidates(ctx, query, year, mediaType)
|
||||||
|
out := make([]*Match, 0, len(candidates))
|
||||||
|
for _, candidate := range candidates {
|
||||||
|
out = append(out, candidate.Match)
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func manualTMDbIDSearchTypes(mediaType string) []string {
|
||||||
|
switch normalizeMediaType(mediaType, "", "") {
|
||||||
|
case "tv", "anime", "variety":
|
||||||
|
return []string{"tv", "movie"}
|
||||||
|
case "movie", "adult":
|
||||||
|
return []string{"movie", "tv"}
|
||||||
|
default:
|
||||||
|
return []string{"movie", "tv"}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func manualTMDbSearchTypes(mediaType string) []string {
|
||||||
|
if strings.TrimSpace(mediaType) == "" {
|
||||||
|
return []string{"movie", "tv"}
|
||||||
|
}
|
||||||
|
switch normalizeMediaType(mediaType, "", "") {
|
||||||
|
case "tv", "anime", "variety":
|
||||||
|
return []string{"tv", "movie"}
|
||||||
|
case "movie", "adult":
|
||||||
|
return []string{"movie"}
|
||||||
|
default:
|
||||||
|
return []string{"movie", "tv"}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualAdultMatches(ctx context.Context, media *model.Media, query string) []*Match {
|
||||||
|
candidates := []string{query}
|
||||||
|
if media != nil {
|
||||||
|
candidates = append(candidates, media.Path, media.OriginalName, media.Title)
|
||||||
|
}
|
||||||
|
out := make([]*Match, 0, 1)
|
||||||
|
seen := map[string]struct{}{}
|
||||||
|
for _, candidate := range candidates {
|
||||||
|
code := normalizeAdultCode(candidate)
|
||||||
|
if code == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if _, ok := seen[code]; ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
seen[code] = struct{}{}
|
||||||
|
if match := s.manualAdultMatch(ctx, code); match != nil {
|
||||||
|
out = append(out, match)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualAdultMatch(ctx context.Context, code string) *Match {
|
||||||
|
if s.adult == nil || !s.adult.Enabled() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
match, err := s.adult.Search(ctx, code)
|
||||||
|
if err != nil || match == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualTMDbMatchByID(ctx context.Context, id int, mediaType string) *Match {
|
||||||
|
if s.tmdb == nil || !s.tmdb.Enabled() || id <= 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if mediaType == "tv" || mediaType == "anime" || mediaType == "variety" {
|
||||||
|
if match, err := s.tmdb.GetTVMatch(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if match, err := s.tmdb.GetMovieMatch(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
if match, err := s.tmdb.GetTVMatch(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualTMDbMatchByIDForType(ctx context.Context, id int, mediaType string) *Match {
|
||||||
|
if s.tmdb == nil || !s.tmdb.Enabled() || id <= 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
switch normalizeMediaType(mediaType, "", "") {
|
||||||
|
case "tv", "anime", "variety":
|
||||||
|
if match, err := s.tmdb.GetTVMatch(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
case "movie", "adult":
|
||||||
|
if match, err := s.tmdb.GetMovieMatch(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualDoubanMatch(ctx context.Context, query string) *Match {
|
||||||
|
if s.douban == nil || !s.douban.Enabled() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if id, ok := parsePositiveIDString(query); ok {
|
||||||
|
if match, err := s.douban.GetMatchByID(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
}
|
||||||
|
match, err := s.douban.SearchMatch(ctx, query)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualBangumiMatch(ctx context.Context, query string) *Match {
|
||||||
|
if s.bangumi == nil || !s.bangumi.Enabled() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if id, ok := parsePositiveInt(query); ok {
|
||||||
|
if match, err := s.bangumi.GetSubject(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
}
|
||||||
|
match, err := s.bangumi.Search(ctx, query)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ScraperService) manualTheTVDBMatch(ctx context.Context, query string) *Match {
|
||||||
|
if s.thetvdb == nil || !s.thetvdb.Enabled() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if id, ok := parsePositiveIDString(normalizeTheTVDBSeriesID(query)); ok {
|
||||||
|
if match, err := s.thetvdb.GetSeriesMatchByID(ctx, id); err == nil && match != nil {
|
||||||
|
return match
|
||||||
|
}
|
||||||
|
}
|
||||||
|
match, err := s.thetvdb.SearchSeries(ctx, query)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return match
|
||||||
|
}
|
||||||
@@ -0,0 +1,157 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/ShukeBta/MediaStationGo/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (s *ScraperService) ManualSearch(ctx context.Context, media *model.Media, query, provider, mediaType string) ([]ExternalMediaResult, error) {
|
||||||
|
if s == nil || media == nil {
|
||||||
|
return nil, errors.New("media required")
|
||||||
|
}
|
||||||
|
lib, _ := s.repo.Library.FindByID(ctx, media.LibraryID)
|
||||||
|
queries := manualSearchQueries(media, lib, query)
|
||||||
|
if len(queries) == 0 {
|
||||||
|
return nil, errors.New("search query required")
|
||||||
|
}
|
||||||
|
if mediaType == "" {
|
||||||
|
if mediaIsEpisodic(media, lib) {
|
||||||
|
mediaType = "tv"
|
||||||
|
} else if lib != nil {
|
||||||
|
mediaType = lib.Type
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mediaType = normalizeMediaType(mediaType, queries[0], "")
|
||||||
|
providers := manualSearchProviderSet(provider)
|
||||||
|
year := mediaYearHint(media)
|
||||||
|
if year <= 0 {
|
||||||
|
_, year = CleanQuery(queries[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
out := make([]ExternalMediaResult, 0, 6)
|
||||||
|
add := func(source, typ string, match *Match) {
|
||||||
|
if match == nil || strings.TrimSpace(match.Title) == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
out = append(out, ExternalMediaResult{
|
||||||
|
Source: source,
|
||||||
|
MediaType: typ,
|
||||||
|
Title: match.Title,
|
||||||
|
OriginalName: match.OriginalName,
|
||||||
|
Overview: match.Overview,
|
||||||
|
PosterURL: match.PosterURL,
|
||||||
|
BackdropURL: match.BackdropURL,
|
||||||
|
Year: match.Year,
|
||||||
|
Rating: match.Rating,
|
||||||
|
TMDbID: match.TMDbID,
|
||||||
|
BangumiID: match.BangumiID,
|
||||||
|
DoubanID: match.DoubanID,
|
||||||
|
TheTVDBID: match.TheTVDBID,
|
||||||
|
SubscribeKeyword: buildSubscribeKeyword(match.Title, match.Year),
|
||||||
|
SubscribeAliases: buildSubscribeAliases(match.Title, match.OriginalName, match.Year),
|
||||||
|
Languages: match.Languages,
|
||||||
|
Countries: match.Countries,
|
||||||
|
Genres: match.Genres,
|
||||||
|
NSFW: match.NSFW,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
if providers.want("adult") {
|
||||||
|
for _, candidateQuery := range queries {
|
||||||
|
for _, match := range s.manualAdultMatches(ctx, media, candidateQuery) {
|
||||||
|
add("adult", "adult", match)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if providers.want("tmdb") {
|
||||||
|
for _, candidateQuery := range queries {
|
||||||
|
for _, candidate := range s.manualTMDbCandidates(ctx, candidateQuery, year, mediaType) {
|
||||||
|
add("tmdb", candidate.MediaType, candidate.Match)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if providers.want("douban") {
|
||||||
|
for _, candidateQuery := range queries {
|
||||||
|
if match := s.manualDoubanMatch(ctx, candidateQuery); match != nil {
|
||||||
|
add("douban", normalizeMediaType(mediaType, candidateQuery, ""), match)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if providers.want("bangumi") {
|
||||||
|
for _, candidateQuery := range queries {
|
||||||
|
if match := s.manualBangumiMatch(ctx, candidateQuery); match != nil {
|
||||||
|
add("bangumi", "anime", match)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if providers.want("thetvdb") {
|
||||||
|
for _, candidateQuery := range queries {
|
||||||
|
if match := s.manualTheTVDBMatch(ctx, candidateQuery); match != nil {
|
||||||
|
add("thetvdb", "tv", match)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return dedupeExternalMedia(out), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type manualSearchProviders map[string]struct{}
|
||||||
|
|
||||||
|
func manualSearchProviderSet(provider string) manualSearchProviders {
|
||||||
|
out := manualSearchProviders{}
|
||||||
|
for _, field := range strings.FieldsFunc(provider, func(r rune) bool {
|
||||||
|
return r == ',' || r == ';' || r == '|' || r == ' '
|
||||||
|
}) {
|
||||||
|
field = strings.ToLower(strings.TrimSpace(field))
|
||||||
|
if field == "" || field == "all" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
out[field] = struct{}{}
|
||||||
|
}
|
||||||
|
if len(out) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p manualSearchProviders) want(provider string) bool {
|
||||||
|
if len(p) == 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
_, ok := p[provider]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func manualSearchQueries(media *model.Media, lib *model.Library, query string) []string {
|
||||||
|
seen := map[string]struct{}{}
|
||||||
|
out := make([]string, 0, 4)
|
||||||
|
add := func(value string) {
|
||||||
|
value = strings.Join(strings.Fields(strings.TrimSpace(value)), " ")
|
||||||
|
if value == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
key := strings.ToLower(value)
|
||||||
|
if _, ok := seen[key]; ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
seen[key] = struct{}{}
|
||||||
|
out = append(out, value)
|
||||||
|
}
|
||||||
|
|
||||||
|
add(query)
|
||||||
|
if strings.TrimSpace(query) == "" && media != nil {
|
||||||
|
add(firstText(media.Title, media.OriginalName))
|
||||||
|
}
|
||||||
|
if media != nil {
|
||||||
|
for _, candidate := range scrapeQueryCandidates(media, lib) {
|
||||||
|
add(candidate)
|
||||||
|
}
|
||||||
|
if len(out) == 0 {
|
||||||
|
title, _ := CleanQuery(media.Path)
|
||||||
|
add(title)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user