fix media workflows and download organization

2026-10-10 23:36:37 +08:00 · 2026-06-26 08:42:16 +08:00
parent 0da96f7e4e
commit 65adca04df
102 changed files with 4063 additions and 437 deletions
@@ -0,0 +1,105 @@
+package service
+
+import (
+	"context"
+	"strings"
+	"time"
+
+	"go.uber.org/zap"
+
+	"github.com/ShukeBta/MediaStationGo/internal/model"
+)
+
+func (s *SubscriptionService) prepareSubscriptionForRun(ctx context.Context, sub *model.Subscription) {
+	if s == nil || sub == nil {
+		return
+	}
+	normalizeSubscriptionDefaults(sub)
+	updates := map[string]any{}
+	if s.fillSubscriptionRunMetadata(ctx, sub, updates); len(updates) > 0 && s.repo != nil && s.repo.DB != nil {
+		if err := s.repo.DB.WithContext(ctx).Model(&model.Subscription{}).Where("id = ?", sub.ID).Updates(updates).Error; err != nil && s.log != nil {
+			s.log.Debug("subscription metadata prepare persist failed", zap.String("id", sub.ID), zap.Error(err))
+		}
+	}
+}
+
+func (s *SubscriptionService) fillSubscriptionRunMetadata(ctx context.Context, sub *model.Subscription, updates map[string]any) {
+	if sub == nil {
+		return
+	}
+	if needsSubscriptionMetadataLookup(sub) {
+		query := subscriptionMetadataPrepareQuery(sub)
+		lookupCtx, cancel := context.WithTimeout(ctx, 12*time.Second)
+		defer cancel()
+		if match := s.lookupSubscriptionMetadata(lookupCtx, strings.TrimSpace(sub.MediaType), query, sub); match != nil {
+			applySubscriptionMetadataMatch(sub, match, updates)
+		}
+	}
+	if isSubscriptionSeriesType(strings.TrimSpace(sub.MediaType)) && sub.TotalEpisodes <= 0 {
+		if total := s.resolveSubscriptionTotalEpisodes(ctx, sub, 0); total > 0 {
+			sub.TotalEpisodes = total
+			updates["total_episodes"] = total
+		}
+	}
+}
+
+func needsSubscriptionMetadataLookup(sub *model.Subscription) bool {
+	if sub == nil {
+		return false
+	}
+	return strings.TrimSpace(sub.MediaType) == "" ||
+		strings.TrimSpace(sub.OriginalName) == "" ||
+		sub.Year <= 0
+}
+
+func subscriptionMetadataPrepareQuery(sub *model.Subscription) string {
+	if sub == nil {
+		return ""
+	}
+	if value := strings.TrimSpace(sub.Filter); value != "" {
+		return value
+	}
+	return strings.TrimSpace(sub.Name)
+}
+
+func applySubscriptionMetadataMatch(sub *model.Subscription, match *Match, updates map[string]any) {
+	if sub == nil || match == nil {
+		return
+	}
+	if strings.TrimSpace(sub.MediaType) == "" {
+		if mediaType := normalizeMetadataMatchSubscriptionType(match); mediaType != "" {
+			sub.MediaType = mediaType
+			updates["media_type"] = mediaType
+		}
+	}
+	if strings.TrimSpace(sub.OriginalName) == "" {
+		if value := strings.TrimSpace(match.OriginalName); value != "" {
+			sub.OriginalName = value
+			updates["original_name"] = value
+		}
+	}
+	if sub.Year <= 0 && match.Year > 0 {
+		sub.Year = match.Year
+		updates["year"] = match.Year
+	}
+}
+
+func normalizeMetadataMatchSubscriptionType(match *Match) string {
+	if match == nil {
+		return ""
+	}
+	switch normalizeOrganizeMediaType(match.MediaType) {
+	case "movie":
+		return "movie"
+	case "tv":
+		return "tv"
+	case "anime":
+		return "anime"
+	case "variety":
+		return "variety"
+	case "adult":
+		return "adult"
+	default:
+		return ""
+	}
+}