feat(app): add internal components batch 5

internal/app/library_fetch.go

@@ -0,0 +1,98 @@
+package app
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+)
+
+func (d *Downloader) fetchAllDramas(ctx context.Context, sourceFilter string) ([]Drama, error) {
+	if sourceFilter != "" && canonicalProviderSource(sourceFilter) != sourceHongguo {
+		return nil, errors.New("此版本仅支持红果站源")
+	}
+	if more, _ := ctx.Value(libraryMoreKey{}).(bool); more {
+		return d.fetchMoreLibrary(ctx, sourceFilter)
+	}
+	category, _ := ctx.Value(libraryCategoryKey{}).(string)
+	fmt.Println("正在获取红果剧库列表...")
+	ctx, cancel := context.WithTimeout(ctx, 10*time.Minute)
+	defer cancel()
+	dramas, err := d.fetchHongguoCategoryDramas(ctx, category)
+	dramas = onlyHongguoDramas(dramas)
+	if len(dramas) == 0 && err == nil && !hongguoCatalogInitialized(d.hongguoCatalogSnapshot()) {
+		err = errors.New("未返回可识别的视频数据")
+	}
+	reportLibraryProgress(ctx, sourceHongguo, dramas, err, true)
+	fmt.Printf(" 剧库获取完成：%d 部\n", len(dramas))
+	if err != nil {
+		fmt.Printf(" 红果获取失败: %v\n", publicError(err))
+		return dramas, &libraryLoadError{failures: map[string]error{sourceHongguo: err}}
+	}
+	return dramas, nil
+}
+
+func (d *Downloader) GetDramaChapters(ctx context.Context, seriesID string) (string, []Chapter, error) {
+	_, sourceID, valid := splitProviderDramaID(seriesID)
+	if !valid {
+		return "", nil, errors.New("仅支持红果剧集 ID，可填写 hongguo:数字ID 或数字ID")
+	}
+	title, chapters, err := d.fetchHongguoChapters(ctx, sourceID)
+	return title, uniqueChapters(chapters), err
+}
+
+func uniqueChapters(chapters []Chapter) []Chapter {
+	seen := map[string]bool{}
+	var out []Chapter
+	for i, ch := range chapters {
+		key := ch.ID
+		if key == "" {
+			key = ch.VideoURL
+		}
+		if key == "" {
+			key = fmt.Sprintf("idx_%d", i)
+		}
+		if seen[key] {
+			continue
+		}
+		seen[key] = true
+		out = append(out, ch)
+	}
+	return out
+}
+
+func (d *Downloader) BuildDramaTasks(ctx context.Context, drama Drama) ([]Task, error) {
+	return d.buildDramaTasksInDirectory(ctx, drama, "")
+}
+
+func (d *Downloader) buildDramaTasksInDirectory(ctx context.Context, drama Drama, existingDirectory string) ([]Task, error) {
+	var valid bool
+	drama, valid = normalizeHongguoDrama(drama)
+	if !valid {
+		return nil, errors.New("此版本仅支持红果剧集")
+	}
+	title, chapters, err := d.GetDramaChapters(ctx, drama.ID)
+	if err != nil {
+		return nil, err
+	}
+	if len(chapters) == 0 {
+		return nil, nil
+	}
+	drama = d.hongguoCachedDrama(drama)
+	dramaDir, err := d.downloadDramaDirectory(drama, title, existingDirectory)
+	if err != nil {
+		return nil, err
+	}
+	var tasks []Task
+	for i, ch := range chapters {
+		ep := ch.EpisodeString(i + 1)
+		epNum := padEpisode(ep)
+		fileName := safeFilename(epNum + ".mp4")
+		outPath, err := safeJoin(dramaDir, fileName)
+		if err != nil {
+			return nil, err
+		}
+		tasks = append(tasks, Task{DramaID: drama.ID, DramaTitle: title, Chapter: ch, Index: i + 1, Total: len(chapters), OutPath: outPath, ReleaseStatus: dramaReleaseStatus(drama)})
+	}
+	return tasks, nil
+}