MeBox/internal/service/media_classifier_source_hints.go

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
}