diff --git a/internal/service/cloud_mount.go b/internal/service/cloud_mount.go index cf74226..bca4dc9 100644 --- a/internal/service/cloud_mount.go +++ b/internal/service/cloud_mount.go @@ -1,13 +1,10 @@ package service import ( - "context" "net/url" "strings" "github.com/ShukeBta/MediaStationGo/internal/model" - "github.com/ShukeBta/MediaStationGo/internal/repository" - "github.com/ShukeBta/MediaStationGo/internal/service/cloud" ) // CloudMountInfo is the canonical identity of a mounted cloud library. ScanDir @@ -144,476 +141,3 @@ func FilterShadowedCloudLibraries(libs []model.Library) []model.Library { } return out } - -func FilterDisplayCloudLibraries(ctx context.Context, repo *repository.Container, libs []model.Library) []model.Library { - if len(libs) == 0 { - return libs - } - libs = FilterDeprecatedNativeCloudLibraries(libs) - counts := cloudLibraryMediaCounts(ctx, repo, libs) - collapsed := make([]model.Library, 0, len(libs)) - byKey := make(map[string]int, len(libs)) - for _, lib := range libs { - key, ok := cloudLibraryDisplayKey(lib) - if !ok { - collapsed = append(collapsed, lib) - continue - } - if prevIndex, exists := byKey[key]; exists { - if betterDisplayCloudLibrary(lib, collapsed[prevIndex], counts) { - collapsed[prevIndex] = lib - } - continue - } - byKey[key] = len(collapsed) - collapsed = append(collapsed, lib) - } - collapsed = FilterShadowedCloudLibraries(collapsed) - return mergeDisplayCloudLibraries(collapsed) -} - -func FilterScannableCloudLibraries(ctx context.Context, repo *repository.Container, libs []model.Library) []model.Library { - if len(libs) == 0 { - return libs - } - counts := cloudLibraryMediaCounts(ctx, repo, libs) - collapsed := make([]model.Library, 0, len(libs)) - byKey := make(map[string]int, len(libs)) - for _, lib := range libs { - if CloudLibraryAutoCategory(lib) { - continue - } - if info, ok := ParseCloudLibraryMount(lib.Path); ok && IsDeprecatedNativeCloudProvider(info.Provider) { - continue - } - key, ok := cloudLibraryDisplayKey(lib) - if !ok { - collapsed = append(collapsed, lib) - continue - } - if prevIndex, exists := byKey[key]; exists { - if betterDisplayCloudLibrary(lib, collapsed[prevIndex], counts) { - collapsed[prevIndex] = lib - } - continue - } - byKey[key] = len(collapsed) - collapsed = append(collapsed, lib) - } - return FilterShadowedCloudLibraries(collapsed) -} - -func FilterDeprecatedNativeCloudLibraries(libs []model.Library) []model.Library { - if len(libs) == 0 { - return libs - } - out := make([]model.Library, 0, len(libs)) - for _, lib := range libs { - info, ok := ParseCloudLibraryMount(lib.Path) - if ok && IsDeprecatedNativeCloudProvider(info.Provider) { - continue - } - out = append(out, lib) - } - return out -} - -func NormalizeCloudLibraryDisplayNames(libs []model.Library) []model.Library { - out := make([]model.Library, 0, len(libs)) - for _, lib := range libs { - if displayName, ok := CloudLibraryDisplayName(lib); ok && displayName != "" { - lib.Name = displayName - } - out = append(out, lib) - } - return out -} - -func cloudLibraryMediaCounts(ctx context.Context, repo *repository.Container, libs []model.Library) map[string]int64 { - counts := make(map[string]int64, len(libs)) - if repo == nil || repo.DB == nil || len(libs) == 0 { - return counts - } - ids := make([]string, 0, len(libs)) - for _, lib := range libs { - if _, ok := ParseCloudLibraryMount(lib.Path); ok { - ids = append(ids, lib.ID) - } - } - if len(ids) == 0 { - return counts - } - var rows []struct { - LibraryID string - Count int64 - } - if err := repo.DB.WithContext(ctx). - Model(&model.Media{}). - Select("library_id, COUNT(*) AS count"). - Where("library_id IN ? AND deleted_at IS NULL", ids). - Group("library_id"). - Scan(&rows).Error; err != nil { - return counts - } - for _, row := range rows { - counts[row.LibraryID] = row.Count - } - return counts -} - -func cloudLibraryDisplayKey(lib model.Library) (string, bool) { - info, ok := ParseCloudLibraryMount(lib.Path) - if !ok { - return "", false - } - dir := firstNonEmpty(info.DisplayDir, info.ScanDir) - return info.Provider + "\x00" + dir, true -} - -func betterDisplayCloudLibrary(candidate, current model.Library, counts map[string]int64) bool { - candidateCount := counts[candidate.ID] - currentCount := counts[current.ID] - if (candidateCount > 0) != (currentCount > 0) { - return candidateCount > 0 - } - if candidate.Enabled != current.Enabled { - return candidate.Enabled - } - candidateCanonical := cloudLibraryPathIsCanonical(candidate) - currentCanonical := cloudLibraryPathIsCanonical(current) - if candidateCanonical != currentCanonical { - return candidateCanonical - } - if !candidate.CreatedAt.Equal(current.CreatedAt) { - return candidate.CreatedAt.After(current.CreatedAt) - } - return candidate.ID > current.ID -} - -func cloudLibraryPathIsCanonical(lib model.Library) bool { - info, ok := ParseCloudLibraryMount(lib.Path) - if !ok { - return false - } - return BuildCloudLibraryPath(info.Provider, info.ScanDir, info.DisplayDir) == strings.TrimSpace(lib.Path) -} - -func mergeDisplayCloudLibraries(libs []model.Library) []model.Library { - if len(libs) == 0 { - return libs - } - localByKey := make(map[string]struct{}, len(libs)) - for _, lib := range libs { - if _, ok := ParseCloudLibraryMount(lib.Path); ok || !lib.Enabled { - continue - } - if key, ok := CloudLibraryMergeKey(lib); ok { - localByKey[key] = struct{}{} - } - } - out := make([]model.Library, 0, len(libs)) - for _, lib := range libs { - if displayName, ok := CloudLibraryDisplayName(lib); ok && displayName != "" { - lib.Name = displayName - if key, ok := CloudLibraryMergeKey(lib); ok { - if _, exists := localByKey[key]; exists { - continue - } - } - } - out = append(out, lib) - } - return out -} - -func CloudLibraryDisplayName(lib model.Library) (string, bool) { - info, ok := ParseCloudLibraryMount(lib.Path) - if !ok { - return "", false - } - name := stripCloudProviderDisplayPrefix(strings.TrimSpace(lib.Name), info.Provider) - dir := firstNonEmpty(info.DisplayDir, info.ScanDir) - if name == "" || strings.EqualFold(name, CloudMountProviderLabel(info.Provider)) { - if base := cloudMountDirBase(dir); base != "" { - name = base - } - } - if name == "" { - name = CloudMountProviderLabel(info.Provider) - } - return name, true -} - -func CloudLibraryMergeKey(lib model.Library) (string, bool) { - name := strings.TrimSpace(lib.Name) - if displayName, ok := CloudLibraryDisplayName(lib); ok { - name = displayName - } - name = normalizeLibraryMergeName(name) - if name == "" { - return "", false - } - return strings.ToLower(strings.TrimSpace(lib.Type)) + "\x00" + name, true -} - -func MergedLibraryIDsForLibrary(ctx context.Context, repo *repository.Container, libraryID string) ([]string, error) { - libraryID = strings.TrimSpace(libraryID) - if libraryID == "" || repo == nil || repo.Library == nil { - return []string{libraryID}, nil - } - lib, err := repo.Library.FindByID(ctx, libraryID) - if err != nil { - return nil, err - } - if lib == nil { - return []string{libraryID}, nil - } - libs, err := repo.Library.List(ctx) - if err != nil { - return nil, err - } - return MergedLibraryIDs(libs, *lib), nil -} - -func MergedLibraryIDs(libs []model.Library, lib model.Library) []string { - ids := []string{} - seen := map[string]struct{}{} - add := func(id string) { - id = strings.TrimSpace(id) - if id == "" { - return - } - if _, ok := seen[id]; ok { - return - } - seen[id] = struct{}{} - ids = append(ids, id) - } - add(lib.ID) - key, ok := CloudLibraryMergeKey(lib) - if !ok { - return ids - } - _, libIsCloud := ParseCloudLibraryMount(lib.Path) - for _, candidate := range libs { - if candidate.ID == lib.ID || !candidate.Enabled { - continue - } - candidateKey, ok := CloudLibraryMergeKey(candidate) - if !ok || candidateKey != key { - continue - } - _, candidateIsCloud := ParseCloudLibraryMount(candidate.Path) - if !libIsCloud && !candidateIsCloud { - continue - } - add(candidate.ID) - } - return ids -} - -func ExpandMediaVisibilityForMergedCloudLibraries(ctx context.Context, repo *repository.Container, visibility MediaVisibility) MediaVisibility { - if repo == nil || repo.Library == nil { - return visibility - } - libs, err := repo.Library.List(ctx) - if err != nil { - return visibility - } - if len(visibility.AllowedLibraryIDs) > 0 { - visibility.AllowedLibraryIDs = expandMergedLibraryIDsFromLibraries(libs, visibility.AllowedLibraryIDs) - } - if len(visibility.HiddenLibraryIDs) > 0 { - visibility.HiddenLibraryIDs = expandMergedLibraryIDsFromLibraries(libs, visibility.HiddenLibraryIDs) - } - visibility.HiddenLibraryIDs = appendUniqueLibraryIDs(visibility.HiddenLibraryIDs, DeprecatedNativeCloudLibraryIDs(libs)...) - return visibility -} - -func expandMergedLibraryIDs(ctx context.Context, repo *repository.Container, ids []string) []string { - if len(ids) == 0 { - return ids - } - libs, err := repo.Library.List(ctx) - if err != nil { - return ids - } - return expandMergedLibraryIDsFromLibraries(libs, ids) -} - -func expandMergedLibraryIDsFromLibraries(libs []model.Library, ids []string) []string { - byID := make(map[string]model.Library, len(libs)) - for _, lib := range libs { - byID[lib.ID] = lib - } - out := make([]string, 0, len(ids)) - for _, id := range ids { - lib, ok := byID[id] - if !ok { - out = appendUniqueLibraryIDs(out, id) - continue - } - for _, mergedID := range MergedLibraryIDs(libs, lib) { - out = appendUniqueLibraryIDs(out, mergedID) - } - } - return out -} - -func DeprecatedNativeCloudLibraryIDs(libs []model.Library) []string { - ids := make([]string, 0) - for _, lib := range libs { - info, ok := ParseCloudLibraryMount(lib.Path) - if ok && IsDeprecatedNativeCloudProvider(info.Provider) { - ids = appendUniqueLibraryIDs(ids, lib.ID) - } - } - return ids -} - -func appendUniqueLibraryIDs(ids []string, more ...string) []string { - for _, id := range more { - id = strings.TrimSpace(id) - if id == "" { - continue - } - found := false - for _, existing := range ids { - if existing == id { - found = true - break - } - } - if !found { - ids = append(ids, id) - } - } - return ids -} - -func CloudMountProviderLabel(provider string) string { - switch strings.TrimSpace(provider) { - case LegacyQuarkProvider: - return "已停用网盘" - case cloud.Type115: - return "115 网盘" - case cloud.TypeCloudDrive2: - return "CloudDrive2" - case cloud.TypeOpenList: - return "OpenList" - default: - if strings.TrimSpace(provider) == "" { - return "网盘" - } - return strings.TrimSpace(provider) - } -} - -func stripCloudProviderDisplayPrefix(name, provider string) string { - name = strings.TrimSpace(name) - if name == "" { - return "" - } - for _, label := range []string{CloudMountProviderLabel(provider), strings.TrimSpace(provider)} { - label = strings.TrimSpace(label) - if label == "" || len(name) < len(label) || !strings.EqualFold(name[:len(label)], label) { - continue - } - rest := strings.TrimSpace(name[len(label):]) - rest = strings.TrimLeft(rest, " \t\r\n·・-—–|｜:/\\") - if rest != "" { - return strings.TrimSpace(rest) - } - if strings.EqualFold(name, label) { - return "" - } - } - return name -} - -func cloudMountDirBase(dir string) string { - dir = strings.Trim(strings.TrimSpace(strings.ReplaceAll(dir, "\\", "/")), "/") - if dir == "" { - return "" - } - parts := strings.Split(dir, "/") - for i := len(parts) - 1; i >= 0; i-- { - if part := strings.TrimSpace(parts[i]); part != "" { - return part - } - } - return "" -} - -func normalizeLibraryMergeName(name string) string { - name = strings.ToLower(strings.TrimSpace(name)) - if name == "" { - return "" - } - return strings.Join(strings.Fields(name), " ") -} - -func ShadowedCloudLibraryIDSet(libs []model.Library) map[string]bool { - out := make(map[string]bool) - for _, lib := range libs { - if CloudLibraryShadowed(libs, lib) != nil { - out[lib.ID] = true - } - } - return out -} - -func InferCloudMountMediaType(dir, name string) string { - text := strings.ToLower(dir + " " + name) - switch { - case strings.Contains(text, "成人") || strings.Contains(text, "adult") || strings.Contains(text, "jav") || strings.Contains(text, "9kg"): - return "adult" - case containsAny(text, "动画电影", "华语电影", "外语电影", "欧美电影", "日韩电影", "韩国电影", "日本电影", "港台电影", "香港电影", "台湾电影", "大陆电影", "国产电影", "纪录片", "演唱会", "电影", "movie", "movies", "film", "films", "documentary", "concert"): - return "movie" - case containsAny(text, "综艺", "真人秀", "脱口秀", "晚会", "variety"): - return "variety" - case containsAny(text, "国漫", "日漫", "日番", "番剧", "动漫", "欧美动漫", "动画剧集", "anime"): - return "anime" - case containsAny(text, "国产剧", "大陆剧", "华语剧", "欧美剧", "日韩剧", "韩剧", "日剧", "港剧", "台剧", "泰剧", "英剧", "美剧", "短剧", "电视剧", "剧集", "连续剧", "series", "tv", "shows"): - return "tv" - default: - return "movie" - } -} - -func containsAny(text string, values ...string) bool { - for _, value := range values { - if strings.Contains(text, value) { - return true - } - } - return false -} - -func cloudMountAncestor(parent, child string) bool { - parent = strings.Trim(parent, "/") - child = strings.Trim(child, "/") - if parent == child { - return false - } - if parent == "" { - return child != "" - } - return strings.HasPrefix(child, parent+"/") -} - -func normalizeCloudMountDir(provider, value string) string { - value = strings.TrimSpace(value) - if decoded, err := url.PathUnescape(value); err == nil { - value = decoded - } - if decoded, err := url.QueryUnescape(value); err == nil { - value = decoded - } - value = strings.ReplaceAll(value, "\\", "/") - value = strings.Trim(strings.TrimSpace(value), "/") - if value == "." || ((provider == cloud.Type115 || provider == LegacyQuarkProvider) && value == "0") { - return "" - } - return value -} diff --git a/internal/service/cloud_mount_filter.go b/internal/service/cloud_mount_filter.go new file mode 100644 index 0000000..c66da70 --- /dev/null +++ b/internal/service/cloud_mount_filter.go @@ -0,0 +1,220 @@ +package service + +import ( + "context" + "strings" + + "github.com/ShukeBta/MediaStationGo/internal/model" + "github.com/ShukeBta/MediaStationGo/internal/repository" +) + +func FilterDisplayCloudLibraries(ctx context.Context, repo *repository.Container, libs []model.Library) []model.Library { + if len(libs) == 0 { + return libs + } + libs = FilterDeprecatedNativeCloudLibraries(libs) + counts := cloudLibraryMediaCounts(ctx, repo, libs) + collapsed := make([]model.Library, 0, len(libs)) + byKey := make(map[string]int, len(libs)) + for _, lib := range libs { + key, ok := cloudLibraryDisplayKey(lib) + if !ok { + collapsed = append(collapsed, lib) + continue + } + if prevIndex, exists := byKey[key]; exists { + if betterDisplayCloudLibrary(lib, collapsed[prevIndex], counts) { + collapsed[prevIndex] = lib + } + continue + } + byKey[key] = len(collapsed) + collapsed = append(collapsed, lib) + } + collapsed = FilterShadowedCloudLibraries(collapsed) + return mergeDisplayCloudLibraries(collapsed) +} + +func FilterScannableCloudLibraries(ctx context.Context, repo *repository.Container, libs []model.Library) []model.Library { + if len(libs) == 0 { + return libs + } + counts := cloudLibraryMediaCounts(ctx, repo, libs) + collapsed := make([]model.Library, 0, len(libs)) + byKey := make(map[string]int, len(libs)) + for _, lib := range libs { + if CloudLibraryAutoCategory(lib) { + continue + } + if info, ok := ParseCloudLibraryMount(lib.Path); ok && IsDeprecatedNativeCloudProvider(info.Provider) { + continue + } + key, ok := cloudLibraryDisplayKey(lib) + if !ok { + collapsed = append(collapsed, lib) + continue + } + if prevIndex, exists := byKey[key]; exists { + if betterDisplayCloudLibrary(lib, collapsed[prevIndex], counts) { + collapsed[prevIndex] = lib + } + continue + } + byKey[key] = len(collapsed) + collapsed = append(collapsed, lib) + } + return FilterShadowedCloudLibraries(collapsed) +} + +func FilterDeprecatedNativeCloudLibraries(libs []model.Library) []model.Library { + if len(libs) == 0 { + return libs + } + out := make([]model.Library, 0, len(libs)) + for _, lib := range libs { + info, ok := ParseCloudLibraryMount(lib.Path) + if ok && IsDeprecatedNativeCloudProvider(info.Provider) { + continue + } + out = append(out, lib) + } + return out +} + +func NormalizeCloudLibraryDisplayNames(libs []model.Library) []model.Library { + out := make([]model.Library, 0, len(libs)) + for _, lib := range libs { + if displayName, ok := CloudLibraryDisplayName(lib); ok && displayName != "" { + lib.Name = displayName + } + out = append(out, lib) + } + return out +} + +func cloudLibraryMediaCounts(ctx context.Context, repo *repository.Container, libs []model.Library) map[string]int64 { + counts := make(map[string]int64, len(libs)) + if repo == nil || repo.DB == nil || len(libs) == 0 { + return counts + } + ids := make([]string, 0, len(libs)) + for _, lib := range libs { + if _, ok := ParseCloudLibraryMount(lib.Path); ok { + ids = append(ids, lib.ID) + } + } + if len(ids) == 0 { + return counts + } + var rows []struct { + LibraryID string + Count int64 + } + if err := repo.DB.WithContext(ctx). + Model(&model.Media{}). + Select("library_id, COUNT(*) AS count"). + Where("library_id IN ? AND deleted_at IS NULL", ids). + Group("library_id"). + Scan(&rows).Error; err != nil { + return counts + } + for _, row := range rows { + counts[row.LibraryID] = row.Count + } + return counts +} + +func cloudLibraryDisplayKey(lib model.Library) (string, bool) { + info, ok := ParseCloudLibraryMount(lib.Path) + if !ok { + return "", false + } + dir := firstNonEmpty(info.DisplayDir, info.ScanDir) + return info.Provider + "\x00" + dir, true +} + +func betterDisplayCloudLibrary(candidate, current model.Library, counts map[string]int64) bool { + candidateCount := counts[candidate.ID] + currentCount := counts[current.ID] + if (candidateCount > 0) != (currentCount > 0) { + return candidateCount > 0 + } + if candidate.Enabled != current.Enabled { + return candidate.Enabled + } + candidateCanonical := cloudLibraryPathIsCanonical(candidate) + currentCanonical := cloudLibraryPathIsCanonical(current) + if candidateCanonical != currentCanonical { + return candidateCanonical + } + if !candidate.CreatedAt.Equal(current.CreatedAt) { + return candidate.CreatedAt.After(current.CreatedAt) + } + return candidate.ID > current.ID +} + +func cloudLibraryPathIsCanonical(lib model.Library) bool { + info, ok := ParseCloudLibraryMount(lib.Path) + if !ok { + return false + } + return BuildCloudLibraryPath(info.Provider, info.ScanDir, info.DisplayDir) == strings.TrimSpace(lib.Path) +} + +func mergeDisplayCloudLibraries(libs []model.Library) []model.Library { + if len(libs) == 0 { + return libs + } + localByKey := make(map[string]struct{}, len(libs)) + for _, lib := range libs { + if _, ok := ParseCloudLibraryMount(lib.Path); ok || !lib.Enabled { + continue + } + if key, ok := CloudLibraryMergeKey(lib); ok { + localByKey[key] = struct{}{} + } + } + out := make([]model.Library, 0, len(libs)) + for _, lib := range libs { + if displayName, ok := CloudLibraryDisplayName(lib); ok && displayName != "" { + lib.Name = displayName + if key, ok := CloudLibraryMergeKey(lib); ok { + if _, exists := localByKey[key]; exists { + continue + } + } + } + out = append(out, lib) + } + return out +} + +func CloudLibraryDisplayName(lib model.Library) (string, bool) { + info, ok := ParseCloudLibraryMount(lib.Path) + if !ok { + return "", false + } + name := stripCloudProviderDisplayPrefix(strings.TrimSpace(lib.Name), info.Provider) + dir := firstNonEmpty(info.DisplayDir, info.ScanDir) + if name == "" || strings.EqualFold(name, CloudMountProviderLabel(info.Provider)) { + if base := cloudMountDirBase(dir); base != "" { + name = base + } + } + if name == "" { + name = CloudMountProviderLabel(info.Provider) + } + return name, true +} + +func CloudLibraryMergeKey(lib model.Library) (string, bool) { + name := strings.TrimSpace(lib.Name) + if displayName, ok := CloudLibraryDisplayName(lib); ok { + name = displayName + } + name = normalizeLibraryMergeName(name) + if name == "" { + return "", false + } + return strings.ToLower(strings.TrimSpace(lib.Type)) + "\x00" + name, true +} diff --git a/internal/service/cloud_mount_label.go b/internal/service/cloud_mount_label.go new file mode 100644 index 0000000..20147a8 --- /dev/null +++ b/internal/service/cloud_mount_label.go @@ -0,0 +1,136 @@ +package service + +import ( + "net/url" + "strings" + + "github.com/ShukeBta/MediaStationGo/internal/model" + "github.com/ShukeBta/MediaStationGo/internal/service/cloud" +) + +func CloudMountProviderLabel(provider string) string { + switch strings.TrimSpace(provider) { + case LegacyQuarkProvider: + return "已停用网盘" + case cloud.Type115: + return "115 网盘" + case cloud.TypeCloudDrive2: + return "CloudDrive2" + case cloud.TypeOpenList: + return "OpenList" + default: + if strings.TrimSpace(provider) == "" { + return "网盘" + } + return strings.TrimSpace(provider) + } +} + +func stripCloudProviderDisplayPrefix(name, provider string) string { + name = strings.TrimSpace(name) + if name == "" { + return "" + } + for _, label := range []string{CloudMountProviderLabel(provider), strings.TrimSpace(provider)} { + label = strings.TrimSpace(label) + if label == "" || len(name) < len(label) || !strings.EqualFold(name[:len(label)], label) { + continue + } + rest := strings.TrimSpace(name[len(label):]) + rest = strings.TrimLeft(rest, " \t\r\n·・-—–|｜:/\\") + if rest != "" { + return strings.TrimSpace(rest) + } + if strings.EqualFold(name, label) { + return "" + } + } + return name +} + +func cloudMountDirBase(dir string) string { + dir = strings.Trim(strings.TrimSpace(strings.ReplaceAll(dir, "\\", "/")), "/") + if dir == "" { + return "" + } + parts := strings.Split(dir, "/") + for i := len(parts) - 1; i >= 0; i-- { + if part := strings.TrimSpace(parts[i]); part != "" { + return part + } + } + return "" +} + +func normalizeLibraryMergeName(name string) string { + name = strings.ToLower(strings.TrimSpace(name)) + if name == "" { + return "" + } + return strings.Join(strings.Fields(name), " ") +} + +func ShadowedCloudLibraryIDSet(libs []model.Library) map[string]bool { + out := make(map[string]bool) + for _, lib := range libs { + if CloudLibraryShadowed(libs, lib) != nil { + out[lib.ID] = true + } + } + return out +} + +func InferCloudMountMediaType(dir, name string) string { + text := strings.ToLower(dir + " " + name) + switch { + case strings.Contains(text, "成人") || strings.Contains(text, "adult") || strings.Contains(text, "jav") || strings.Contains(text, "9kg"): + return "adult" + case containsAny(text, "动画电影", "华语电影", "外语电影", "欧美电影", "日韩电影", "韩国电影", "日本电影", "港台电影", "香港电影", "台湾电影", "大陆电影", "国产电影", "纪录片", "演唱会", "电影", "movie", "movies", "film", "films", "documentary", "concert"): + return "movie" + case containsAny(text, "综艺", "真人秀", "脱口秀", "晚会", "variety"): + return "variety" + case containsAny(text, "国漫", "日漫", "日番", "番剧", "动漫", "欧美动漫", "动画剧集", "anime"): + return "anime" + case containsAny(text, "国产剧", "大陆剧", "华语剧", "欧美剧", "日韩剧", "韩剧", "日剧", "港剧", "台剧", "泰剧", "英剧", "美剧", "短剧", "电视剧", "剧集", "连续剧", "series", "tv", "shows"): + return "tv" + default: + return "movie" + } +} + +func containsAny(text string, values ...string) bool { + for _, value := range values { + if strings.Contains(text, value) { + return true + } + } + return false +} + +func cloudMountAncestor(parent, child string) bool { + parent = strings.Trim(parent, "/") + child = strings.Trim(child, "/") + if parent == child { + return false + } + if parent == "" { + return child != "" + } + return strings.HasPrefix(child, parent+"/") +} + +func normalizeCloudMountDir(provider, value string) string { + value = strings.TrimSpace(value) + if decoded, err := url.PathUnescape(value); err == nil { + value = decoded + } + if decoded, err := url.QueryUnescape(value); err == nil { + value = decoded + } + value = strings.ReplaceAll(value, "\\", "/") + value = strings.Trim(strings.TrimSpace(value), "/") + if value == "." || ((provider == cloud.Type115 || provider == LegacyQuarkProvider) && value == "0") { + return "" + } + return value +} diff --git a/internal/service/cloud_mount_merge.go b/internal/service/cloud_mount_merge.go new file mode 100644 index 0000000..a403dad --- /dev/null +++ b/internal/service/cloud_mount_merge.go @@ -0,0 +1,144 @@ +package service + +import ( + "context" + "strings" + + "github.com/ShukeBta/MediaStationGo/internal/model" + "github.com/ShukeBta/MediaStationGo/internal/repository" +) + +func MergedLibraryIDsForLibrary(ctx context.Context, repo *repository.Container, libraryID string) ([]string, error) { + libraryID = strings.TrimSpace(libraryID) + if libraryID == "" || repo == nil || repo.Library == nil { + return []string{libraryID}, nil + } + lib, err := repo.Library.FindByID(ctx, libraryID) + if err != nil { + return nil, err + } + if lib == nil { + return []string{libraryID}, nil + } + libs, err := repo.Library.List(ctx) + if err != nil { + return nil, err + } + return MergedLibraryIDs(libs, *lib), nil +} + +func MergedLibraryIDs(libs []model.Library, lib model.Library) []string { + ids := []string{} + seen := map[string]struct{}{} + add := func(id string) { + id = strings.TrimSpace(id) + if id == "" { + return + } + if _, ok := seen[id]; ok { + return + } + seen[id] = struct{}{} + ids = append(ids, id) + } + add(lib.ID) + key, ok := CloudLibraryMergeKey(lib) + if !ok { + return ids + } + _, libIsCloud := ParseCloudLibraryMount(lib.Path) + for _, candidate := range libs { + if candidate.ID == lib.ID || !candidate.Enabled { + continue + } + candidateKey, ok := CloudLibraryMergeKey(candidate) + if !ok || candidateKey != key { + continue + } + _, candidateIsCloud := ParseCloudLibraryMount(candidate.Path) + if !libIsCloud && !candidateIsCloud { + continue + } + add(candidate.ID) + } + return ids +} + +func ExpandMediaVisibilityForMergedCloudLibraries(ctx context.Context, repo *repository.Container, visibility MediaVisibility) MediaVisibility { + if repo == nil || repo.Library == nil { + return visibility + } + libs, err := repo.Library.List(ctx) + if err != nil { + return visibility + } + if len(visibility.AllowedLibraryIDs) > 0 { + visibility.AllowedLibraryIDs = expandMergedLibraryIDsFromLibraries(libs, visibility.AllowedLibraryIDs) + } + if len(visibility.HiddenLibraryIDs) > 0 { + visibility.HiddenLibraryIDs = expandMergedLibraryIDsFromLibraries(libs, visibility.HiddenLibraryIDs) + } + visibility.HiddenLibraryIDs = appendUniqueLibraryIDs(visibility.HiddenLibraryIDs, DeprecatedNativeCloudLibraryIDs(libs)...) + return visibility +} + +func expandMergedLibraryIDs(ctx context.Context, repo *repository.Container, ids []string) []string { + if len(ids) == 0 { + return ids + } + libs, err := repo.Library.List(ctx) + if err != nil { + return ids + } + return expandMergedLibraryIDsFromLibraries(libs, ids) +} + +func expandMergedLibraryIDsFromLibraries(libs []model.Library, ids []string) []string { + byID := make(map[string]model.Library, len(libs)) + for _, lib := range libs { + byID[lib.ID] = lib + } + out := make([]string, 0, len(ids)) + for _, id := range ids { + lib, ok := byID[id] + if !ok { + out = appendUniqueLibraryIDs(out, id) + continue + } + for _, mergedID := range MergedLibraryIDs(libs, lib) { + out = appendUniqueLibraryIDs(out, mergedID) + } + } + return out +} + +func DeprecatedNativeCloudLibraryIDs(libs []model.Library) []string { + ids := make([]string, 0) + for _, lib := range libs { + info, ok := ParseCloudLibraryMount(lib.Path) + if ok && IsDeprecatedNativeCloudProvider(info.Provider) { + ids = appendUniqueLibraryIDs(ids, lib.ID) + } + } + return ids +} + +func appendUniqueLibraryIDs(ids []string, more ...string) []string { + for _, id := range more { + id = strings.TrimSpace(id) + if id == "" { + continue + } + found := false + for _, existing := range ids { + if existing == id { + found = true + break + } + } + if !found { + ids = append(ids, id) + } + } + return ids +}