119 lines
3.6 KiB
Go
119 lines
3.6 KiB
Go
package recognize
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"unicode"
|
|
|
|
"git.vakhrushev.me/av/jellybit/internal/metadata"
|
|
)
|
|
|
|
// matchMetadata сверяет план с включёнными базами: ищет по названию+году и,
|
|
// если ровно один кандидат уверенно совпадает (название и год), возвращает
|
|
// матч с официальным id и каноническим именем. Несколько кандидатов или их
|
|
// отсутствие → nil (тогда авто-раскладки не будет). Ошибки провайдера не
|
|
// валят распознавание — просто нет матча.
|
|
func (r *Recognizer) matchMetadata(ctx context.Context, plan Plan) *Match {
|
|
if len(r.providers) == 0 {
|
|
return nil
|
|
}
|
|
mt := metadata.Movie
|
|
if plan.Type == MediaSeries {
|
|
mt = metadata.Series
|
|
}
|
|
|
|
searchTitle := plan.ProviderHint
|
|
if strings.TrimSpace(searchTitle) == "" {
|
|
searchTitle = plan.Title
|
|
}
|
|
matchTitles := normSet(plan.Title, plan.OriginalTitle)
|
|
|
|
for _, p := range r.providers {
|
|
cands, err := p.Search(ctx, metadata.Query{Type: mt, Title: searchTitle, Year: plan.Year})
|
|
if err != nil {
|
|
r.log.Warn("recognize: metadata search failed", "provider", p.Name(), "err", err)
|
|
continue
|
|
}
|
|
strong := strongMatches(cands, plan.Year, matchTitles)
|
|
if len(strong) != 1 {
|
|
continue
|
|
}
|
|
c := strong[0]
|
|
match := &Match{Provider: c.Provider, ProviderID: c.ID, Title: c.Title, Year: c.Year}
|
|
if mt == metadata.Series {
|
|
if counts, cerr := p.SeasonEpisodeCounts(ctx, c.ID); cerr == nil {
|
|
match.SeasonEpisodeCounts = counts
|
|
} else {
|
|
r.log.Warn("recognize: episode counts failed",
|
|
"provider", p.Name(), "id", c.ID, "err", cerr)
|
|
}
|
|
}
|
|
return match
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// strongMatches оставляет кандидатов, чьё название совпадает с одним из
|
|
// названий плана (после нормализации) и год бьётся (±1 год), дедуплицируя
|
|
// по id.
|
|
func strongMatches(cands []metadata.Candidate, year int, titles map[string]bool) []metadata.Candidate {
|
|
seen := map[string]bool{}
|
|
var out []metadata.Candidate
|
|
for _, c := range cands {
|
|
if !yearMatches(year, c.Year) {
|
|
continue
|
|
}
|
|
if !titles[normalize(c.Title)] && !titles[normalize(c.OriginalTitle)] {
|
|
continue
|
|
}
|
|
if seen[c.ID] {
|
|
continue
|
|
}
|
|
seen[c.ID] = true
|
|
out = append(out, c)
|
|
}
|
|
return out
|
|
}
|
|
|
|
// yearMatches: год известен у обоих и расходится не больше чем на 1 (разные
|
|
// базы по-разному датируют релиз), либо где-то год неизвестен.
|
|
func yearMatches(a, b int) bool {
|
|
if a == 0 || b == 0 {
|
|
return true
|
|
}
|
|
d := a - b
|
|
if d < 0 {
|
|
d = -d
|
|
}
|
|
return d <= 1
|
|
}
|
|
|
|
// normSet — множество нормализованных непустых названий.
|
|
func normSet(titles ...string) map[string]bool {
|
|
out := map[string]bool{}
|
|
for _, t := range titles {
|
|
if n := normalize(t); n != "" {
|
|
out[n] = true
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
// normalize приводит название к сравнимому виду: нижний регистр, только
|
|
// буквы/цифры (юникод), одиночные пробелы.
|
|
func normalize(s string) string {
|
|
var b strings.Builder
|
|
prevSpace := false
|
|
for _, r := range strings.ToLower(s) {
|
|
switch {
|
|
case unicode.IsLetter(r) || unicode.IsDigit(r):
|
|
b.WriteRune(r)
|
|
prevSpace = false
|
|
case !prevSpace:
|
|
b.WriteByte(' ')
|
|
prevSpace = true
|
|
}
|
|
}
|
|
return strings.TrimSpace(b.String())
|
|
}
|