tcjlj/MeBox
1
0
Fork 0
mirror of https://github.com/truewhile/MeBox.git synced 2026-10-07 22:06:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix media organize and subscription workflows

Browse Source
This commit is contained in:
ShukeBta
2026-06-30 20:18:02 +08:00
parent b41f62d572
commit 941bb273b9
107 changed files with 5420 additions and 526 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/service/subscription_score.go
+36 -6
View File
@@ -7,12 +7,25 @@ import (
"github.com/ShukeBta/MediaStationGo/internal/model"
)
var (
dolbyVisionTokenRE = regexp.MustCompile(`\bdv\b`)
webDLTokenRE = regexp.MustCompile(`\bweb[\s._-]?dl\b`)
webRipTokenRE = regexp.MustCompile(`\bweb[\s._-]?rip\b`)
bluRayTokenRE = regexp.MustCompile(`\b(?:blu[\s._-]?ray|bdrip|bdremux|uhd[\s._-]?blu[\s._-]?ray)\b`)
)
const (
defaultSubscriptionFreePromotionScore = 50_000
washSubscriptionFreePromotionScore = 25
)
func subscriptionCandidateScore(sub *model.Subscription, item SearchResult) int {
title := strings.ToLower(subscriptionSearchResultText(item))
score := item.Seeders
if sub == nil || !sub.WashEnabled {
if !subscriptionAllowsWash(sub) {
score += detectDefaultSubscriptionQualityScore(title)*1_000_000 + detectResolutionScore(title)*100_000
if item.Free {
score += 25
score += defaultSubscriptionFreePromotionScore
}
return score
}
@@ -37,7 +50,7 @@ func subscriptionCandidateScore(sub *model.Subscription, item SearchResult) int
score += resolutionScore*500 + qualityScore*300 + effectScore*150
}
if item.Free {
score += 25
score += washSubscriptionFreePromotionScore
}
return score
}
@@ -52,7 +65,7 @@ func containsAnyEffect(titleFold, csv string) bool {
}
switch token {
case "dolby-vision", "dolby vision", "dv":
if strings.Contains(titleFold, "dolby vision") || strings.Contains(titleFold, "dovi") || regexp.MustCompile(`\bdv\b`).MatchString(titleFold) {
if strings.Contains(titleFold, "dolby vision") || strings.Contains(titleFold, "dovi") || dolbyVisionTokenRE.MatchString(titleFold) {
return true
}
default:
@@ -80,9 +93,11 @@ func titleMatchesResolution(titleFold, resolution string) bool {
func titleMatchesQuality(titleFold, quality string) bool {
switch strings.ToLower(strings.TrimSpace(quality)) {
case "webdl", "web-dl":
return strings.Contains(titleFold, "web-dl") || strings.Contains(titleFold, "webdl")
return webDLTokenRE.MatchString(titleFold)
case "webrip", "web-rip":
return webRipTokenRE.MatchString(titleFold)
case "bluray", "blu-ray":
return strings.Contains(titleFold, "bluray") || strings.Contains(titleFold, "blu-ray") || strings.Contains(titleFold, "bdrip")
return bluRayTokenRE.MatchString(titleFold)
case "remux":
return strings.Contains(titleFold, "remux")
case "hdtv":
@@ -92,6 +107,21 @@ func titleMatchesQuality(titleFold, quality string) bool {
}
}
func detectDefaultSubscriptionQualityScore(titleFold string) int {
switch {
case titleMatchesQuality(titleFold, "web-dl"):
return 5
case titleMatchesQuality(titleFold, "web-rip"):
return 4
case titleMatchesQuality(titleFold, "bluray"), titleMatchesQuality(titleFold, "remux"):
return 3
case titleMatchesQuality(titleFold, "hdtv"):
return 2
default:
return 1
}
}
func detectResolutionScore(titleFold string) int {
switch {
case titleMatchesResolution(titleFold, "2160p"):