吴璨
99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
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
|
|
}
|