mirror of
https://github.com/truewhile/MeBox.git
synced 2026-10-08 14:26:37 +08:00
fix: stabilize library paths and subscription matching
This commit is contained in:
ShukeBta
@@ -0,0 +1,197 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"github.com/ShukeBta/MediaStationGo/internal/model"
|
||||
"github.com/ShukeBta/MediaStationGo/internal/repository"
|
||||
)
|
||||
|
||||
const (
|
||||
RecognitionWordsEnabledKey = "recognition_words.enabled"
|
||||
RecognitionWordsLocalTextKey = "recognition_words.local_text"
|
||||
RecognitionWordsSharedURLsKey = "recognition_words.shared_urls"
|
||||
RecognitionWordsSharedTextKey = "recognition_words.shared_text"
|
||||
RecognitionWordsSyncedAtKey = "recognition_words.synced_at"
|
||||
)
|
||||
|
||||
var DefaultRecognitionWordURLs = []string{
|
||||
"https://raw.githubusercontent.com/Putarku/MoviePilot-Help/main/Words/general.txt",
|
||||
"https://raw.githubusercontent.com/Putarku/MoviePilot-Help/main/Words/TV.txt",
|
||||
"https://raw.githubusercontent.com/Putarku/MoviePilot-Help/main/Words/anime.txt",
|
||||
}
|
||||
|
||||
type RecognitionWordsService struct {
|
||||
log *zap.Logger
|
||||
repo *repository.Container
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
type RecognitionWordsConfig struct {
|
||||
Enabled bool `json:"enabled"`
|
||||
LocalText string `json:"local_text"`
|
||||
SharedURLs []string `json:"shared_urls"`
|
||||
SharedText string `json:"shared_text,omitempty"`
|
||||
SyncedAt string `json:"synced_at,omitempty"`
|
||||
RuleCount int `json:"rule_count"`
|
||||
}
|
||||
|
||||
type RecognitionWordsTestResult struct {
|
||||
Input string `json:"input"`
|
||||
Output string `json:"output"`
|
||||
Title string `json:"title"`
|
||||
Year int `json:"year"`
|
||||
Changed bool `json:"changed"`
|
||||
}
|
||||
|
||||
func NewRecognitionWordsService(log *zap.Logger, repo *repository.Container) *RecognitionWordsService {
|
||||
return &RecognitionWordsService{
|
||||
log: log,
|
||||
repo: repo,
|
||||
client: recognitionWordHTTPClient(),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *RecognitionWordsService) Config(ctx context.Context) RecognitionWordsConfig {
|
||||
cfg := recognitionWordsConfig(ctx, s.repo)
|
||||
cfg.RuleCount = len(parseRecognitionWordRules(recognitionWordsCombinedText(cfg)))
|
||||
return cfg
|
||||
}
|
||||
|
||||
func (s *RecognitionWordsService) SaveConfig(ctx context.Context, cfg RecognitionWordsConfig) error {
|
||||
if s == nil || s.repo == nil || s.repo.Setting == nil {
|
||||
return fmt.Errorf("setting repository unavailable")
|
||||
}
|
||||
if err := s.repo.Setting.Set(ctx, RecognitionWordsEnabledKey, strconv.FormatBool(cfg.Enabled)); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := s.repo.Setting.Set(ctx, RecognitionWordsLocalTextKey, cfg.LocalText); err != nil {
|
||||
return err
|
||||
}
|
||||
rawURLs, err := json.Marshal(normalizeRecognitionWordURLs(cfg.SharedURLs))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return s.repo.Setting.Set(ctx, RecognitionWordsSharedURLsKey, string(rawURLs))
|
||||
}
|
||||
|
||||
func (s *RecognitionWordsService) SyncShared(ctx context.Context) (RecognitionWordsConfig, error) {
|
||||
cfg := s.Config(ctx)
|
||||
urls := cfg.SharedURLs
|
||||
if len(urls) == 0 {
|
||||
urls = DefaultRecognitionWordURLs
|
||||
}
|
||||
var combined []string
|
||||
for _, rawURL := range urls {
|
||||
text, err := s.fetchSharedWords(ctx, rawURL)
|
||||
if err != nil {
|
||||
return cfg, err
|
||||
}
|
||||
combined = append(combined, "# "+rawURL, text)
|
||||
}
|
||||
now := time.Now().Format(time.RFC3339)
|
||||
if err := s.repo.Setting.Set(ctx, RecognitionWordsSharedTextKey, strings.Join(combined, "\n")); err != nil {
|
||||
return cfg, err
|
||||
}
|
||||
if err := s.repo.Setting.Set(ctx, RecognitionWordsSyncedAtKey, now); err != nil {
|
||||
return cfg, err
|
||||
}
|
||||
return s.Config(ctx), nil
|
||||
}
|
||||
|
||||
func (s *RecognitionWordsService) Test(ctx context.Context, input string) RecognitionWordsTestResult {
|
||||
output := ApplyRecognitionWords(ctx, s.repo, input)
|
||||
title, year := CleanQuery(output)
|
||||
return RecognitionWordsTestResult{
|
||||
Input: input,
|
||||
Output: output,
|
||||
Title: title,
|
||||
Year: year,
|
||||
Changed: strings.TrimSpace(input) != strings.TrimSpace(output),
|
||||
}
|
||||
}
|
||||
|
||||
func ApplyRecognitionWords(ctx context.Context, repo *repository.Container, raw string) string {
|
||||
cfg := recognitionWordsConfig(ctx, repo)
|
||||
if !cfg.Enabled {
|
||||
return raw
|
||||
}
|
||||
rules := parseRecognitionWordRules(recognitionWordsCombinedText(cfg))
|
||||
return applyRecognitionWordRules(raw, rules)
|
||||
}
|
||||
|
||||
func CleanQueryWithRecognition(ctx context.Context, repo *repository.Container, raw string) (string, int) {
|
||||
return CleanQuery(ApplyRecognitionWords(ctx, repo, raw))
|
||||
}
|
||||
|
||||
func recognitionWordsConfig(ctx context.Context, repo *repository.Container) RecognitionWordsConfig {
|
||||
cfg := RecognitionWordsConfig{Enabled: true, SharedURLs: DefaultRecognitionWordURLs}
|
||||
if repo == nil || repo.DB == nil || repo.Setting == nil || !repo.DB.Migrator().HasTable(&model.Setting{}) {
|
||||
return cfg
|
||||
}
|
||||
if value, err := repo.Setting.Get(ctx, RecognitionWordsEnabledKey); err == nil && strings.TrimSpace(value) != "" {
|
||||
cfg.Enabled = parseBoolSetting(value, true)
|
||||
}
|
||||
if value, err := repo.Setting.Get(ctx, RecognitionWordsLocalTextKey); err == nil {
|
||||
cfg.LocalText = value
|
||||
}
|
||||
if value, err := repo.Setting.Get(ctx, RecognitionWordsSharedURLsKey); err == nil && strings.TrimSpace(value) != "" {
|
||||
cfg.SharedURLs = parseRecognitionWordURLs(value)
|
||||
}
|
||||
if value, err := repo.Setting.Get(ctx, RecognitionWordsSharedTextKey); err == nil {
|
||||
cfg.SharedText = value
|
||||
}
|
||||
if value, err := repo.Setting.Get(ctx, RecognitionWordsSyncedAtKey); err == nil {
|
||||
cfg.SyncedAt = value
|
||||
}
|
||||
return cfg
|
||||
}
|
||||
|
||||
func recognitionWordsCombinedText(cfg RecognitionWordsConfig) string {
|
||||
return strings.TrimSpace(cfg.LocalText + "\n" + cfg.SharedText)
|
||||
}
|
||||
|
||||
func parseRecognitionWordURLs(raw string) []string {
|
||||
var values []string
|
||||
if err := json.Unmarshal([]byte(raw), &values); err == nil {
|
||||
return normalizeRecognitionWordURLs(values)
|
||||
}
|
||||
return normalizeRecognitionWordURLs(strings.FieldsFunc(raw, func(r rune) bool {
|
||||
return r == '\n' || r == '\r' || r == ',' || r == ';'
|
||||
}))
|
||||
}
|
||||
|
||||
func normalizeRecognitionWordURLs(values []string) []string {
|
||||
seen := map[string]struct{}{}
|
||||
out := make([]string, 0, len(values))
|
||||
for _, value := range values {
|
||||
value = strings.TrimSpace(value)
|
||||
if value == "" {
|
||||
continue
|
||||
}
|
||||
if _, ok := seen[value]; ok {
|
||||
continue
|
||||
}
|
||||
seen[value] = struct{}{}
|
||||
out = append(out, value)
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
type recognitionWordRule struct {
|
||||
raw string
|
||||
block string
|
||||
replaceFrom string
|
||||
replaceTo string
|
||||
offsetLeft string
|
||||
offsetRight string
|
||||
offsetExpr string
|
||||
}
|
||||
Reference in New Issue
Block a user