refactor media classifier helpers

internal/service/media_classifier_source_hints.go

@@ -0,0 +1,77 @@
+package service
+
+import "strings"
+
+type sourceCategoryHintDef struct {
+	Key       string
+	Fallback  string
+	MediaType string
+}
+
+var sourceCategoryHints = []sourceCategoryHintDef{
+	{Key: "animation_movie", Fallback: "动画电影", MediaType: "movie"},
+	{Key: "chinese_movie", Fallback: "华语电影", MediaType: "movie"},
+	{Key: "jk_movie", Fallback: "日韩电影", MediaType: "movie"},
+	{Key: "euus_movie", Fallback: "欧美电影", MediaType: "movie"},
+	{Key: "foreign_movie", Fallback: "外语电影", MediaType: "movie"},
+	{Key: "domestic_tv", Fallback: "国产剧", MediaType: "tv"},
+	{Key: "euus_tv", Fallback: "欧美剧", MediaType: "tv"},
+	{Key: "jk_tv", Fallback: "日韩剧", MediaType: "tv"},
+	{Key: "cn_anime", Fallback: "国漫", MediaType: "anime"},
+	{Key: "jp_anime", Fallback: "日番", MediaType: "anime"},
+	{Key: "euus_anime", Fallback: "欧美动漫", MediaType: "anime"},
+	{Key: "variety", Fallback: "综艺", MediaType: "variety"},
+	{Key: "documentary", Fallback: "纪录片", MediaType: "tv"},
+	{Key: "children", Fallback: "儿童", MediaType: "tv"},
+	{Key: "adult", Fallback: "成人", MediaType: "adult"},
+	{Key: "adult_9kg", Fallback: "9KG", MediaType: "adult"},
+	{Key: "adult_jav", Fallback: "番号", MediaType: "adult"},
+}
+
+func sourceCategoryHint(category, mediaType string, categories map[string]string) string {
+	tokens := sourceCategoryTokens(category)
+	if len(tokens) == 0 {
+		return ""
+	}
+	for _, hint := range sourceCategoryHints {
+		if !sourceCategoryCompatible(mediaType, hint.MediaType) {
+			continue
+		}
+		for _, name := range []string{hint.Fallback, categoryName(categories, hint.Key, hint.Fallback)} {
+			if _, ok := tokens[strings.ToLower(strings.TrimSpace(name))]; ok {
+				return categoryName(categories, hint.Key, hint.Fallback)
+			}
+		}
+	}
+	return ""
+}
+
+func sourceCategoryTokens(category string) map[string]struct{} {
+	category = strings.TrimSpace(category)
+	if category == "" {
+		return nil
+	}
+	normalized := strings.NewReplacer("\\", " ", "/", " ", "|", " ", ",", " ", ";", " ").Replace(category)
+	out := map[string]struct{}{
+		strings.ToLower(category): {},
+	}
+	for _, field := range strings.Fields(normalized) {
+		out[strings.ToLower(strings.TrimSpace(field))] = struct{}{}
+	}
+	return out
+}
+
+func sourceCategoryCompatible(mediaType, categoryMediaType string) bool {
+	mediaType = strings.ToLower(strings.TrimSpace(mediaType))
+	categoryMediaType = strings.ToLower(strings.TrimSpace(categoryMediaType))
+	if mediaType == "" || categoryMediaType == "" || mediaType == categoryMediaType {
+		return true
+	}
+	if mediaType == "tv" && (categoryMediaType == "anime" || categoryMediaType == "variety") {
+		return true
+	}
+	if categoryMediaType == "adult" {
+		return true
+	}
+	return false
+}