use std::collections::HashSet; use std::path::{Path, PathBuf}; use std::pin::Pin; use anyhow::{Context, Result, anyhow, bail}; use bili_sync_entity::*; use futures::stream::FuturesUnordered; use futures::{Stream, StreamExt, TryStreamExt}; use sea_orm::ActiveValue::Set; use sea_orm::TransactionTrait; use sea_orm::entity::prelude::*; use tokio::fs; use tokio::sync::Semaphore; use crate::adapter::{VideoSource, VideoSourceEnum}; use crate::bilibili::{BestStream, BiliClient, BiliError, Dimension, PageInfo, Video, VideoInfo}; use crate::config::{ARGS, PathSafeTemplate, TEMPLATE, VersionedConfig}; use crate::downloader::Downloader; use crate::error::{DownloadAbortError, ExecutionStatus, ProcessPageError}; use crate::utils::format_arg::{page_format_args, video_format_args}; use crate::utils::model::{ create_pages, create_videos, filter_unfilled_videos, filter_unhandled_video_pages, update_pages_model, update_videos_model, }; use crate::utils::nfo::NFO; use crate::utils::status::{PageStatus, STATUS_OK, VideoStatus}; /// 完整地处理某个视频来源 pub async fn process_video_source( video_source: VideoSourceEnum, bili_client: &BiliClient, connection: &DatabaseConnection, ) -> Result<()> { // 从参数中获取视频列表的 Model 与视频流 let (video_source, video_streams) = video_source.refresh(bili_client, connection).await?; // 从视频流中获取新视频的简要信息,写入数据库 refresh_video_source(&video_source, video_streams, connection).await?; // 单独请求视频详情接口,获取视频的详情信息与所有的分页,写入数据库 fetch_video_details(bili_client, &video_source, connection).await?; if ARGS.scan_only { warn!("已开启仅扫描模式,跳过视频下载.."); } else { // 从数据库中查找所有未下载的视频与分页,下载并处理 download_unprocessed_videos(bili_client, &video_source, connection).await?; } Ok(()) } /// 请求接口,获取视频列表中所有新添加的视频信息,将其写入数据库 pub async fn refresh_video_source<'a>( video_source: &VideoSourceEnum, video_streams: Pin> + 'a + Send>>, connection: &DatabaseConnection, ) -> Result<()> { video_source.log_refresh_video_start(); let latest_row_at = video_source.get_latest_row_at().and_utc(); let mut max_datetime = latest_row_at; let mut error = Ok(()); let mut video_streams = video_streams .take_while(|res| { match res { Err(e) => { error = Err(anyhow!(e.to_string())); futures::future::ready(false) } Ok(v) => { // 虽然 video_streams 是从新到旧的,但由于此处是分页请求,极端情况下可能发生访问完第一页时插入了两整页视频的情况 // 此时获取到的第二页视频比第一页的还要新,因此为了确保正确,理应对每一页的第一个视频进行时间比较 // 但在 streams 的抽象下,无法判断具体是在哪里分页的,所以暂且对每个视频都进行比较,应该不会有太大性能损失 let release_datetime = v.release_datetime(); if release_datetime > &max_datetime { max_datetime = *release_datetime; } futures::future::ready(video_source.should_take(release_datetime, &latest_row_at)) } } }) .filter_map(|res| futures::future::ready(video_source.should_filter(res, &latest_row_at))) .chunks(10); let mut count = 0; while let Some(videos_info) = video_streams.next().await { count += videos_info.len(); create_videos(videos_info, video_source, connection).await?; } // 如果获取视频分页过程中发生了错误,直接在此处返回,不更新 latest_row_at error?; if max_datetime != latest_row_at { video_source .update_latest_row_at(max_datetime.naive_utc()) .save(connection) .await?; } video_source.log_refresh_video_end(count); Ok(()) } /// 筛选出所有未获取到全部信息的视频,尝试补充其详细信息 pub async fn fetch_video_details( bili_client: &BiliClient, video_source: &VideoSourceEnum, connection: &DatabaseConnection, ) -> Result<()> { video_source.log_fetch_video_start(); let videos_model = filter_unfilled_videos(video_source.filter_expr(), connection).await?; let tasks = videos_model .into_iter() .map(|video_model| { async move { let video = Video::new(bili_client, video_model.bvid.clone()); let info: Result<_> = async { Ok((video.get_tags().await?, video.get_view_info().await?)) }.await; match info { Err(e) => { error!( "获取视频 {} - {} 的详细信息失败,错误为:{:#}", &video_model.bvid, &video_model.name, e ); if let Some(BiliError::RequestFailed(-404, _)) = e.downcast_ref::() { let mut video_active_model: bili_sync_entity::video::ActiveModel = video_model.into(); video_active_model.valid = Set(false); video_active_model.save(connection).await?; } } Ok((tags, mut view_info)) => { let VideoInfo::Detail { pages, .. } = &mut view_info else { unreachable!() }; let pages = std::mem::take(pages); let pages_len = pages.len(); let txn = connection.begin().await?; // 将分页信息写入数据库 create_pages(pages, &video_model, &txn).await?; let mut video_active_model = view_info.into_detail_model(video_model); video_source.set_relation_id(&mut video_active_model); video_active_model.single_page = Set(Some(pages_len == 1)); video_active_model.tags = Set(Some(serde_json::to_value(tags)?)); video_active_model.save(&txn).await?; txn.commit().await?; } }; Ok::<_, anyhow::Error>(()) } }) .collect::>(); tasks.try_collect::>().await?; video_source.log_fetch_video_end(); Ok(()) } /// 下载所有未处理成功的视频 pub async fn download_unprocessed_videos( bili_client: &BiliClient, video_source: &VideoSourceEnum, connection: &DatabaseConnection, ) -> Result<()> { video_source.log_download_video_start(); let semaphore = Semaphore::new(VersionedConfig::get().load().concurrent_limit.video); let downloader = Downloader::new(bili_client.client.clone()); let unhandled_videos_pages = filter_unhandled_video_pages(video_source.filter_expr(), connection).await?; let mut assigned_upper = HashSet::new(); let tasks = unhandled_videos_pages .into_iter() .map(|(video_model, pages_model)| { let should_download_upper = !assigned_upper.contains(&video_model.upper_id); assigned_upper.insert(video_model.upper_id); download_video_pages( bili_client, video_source, video_model, pages_model, connection, &semaphore, &downloader, should_download_upper, ) }) .collect::>(); let mut download_aborted = false; let mut stream = tasks // 触发风控时设置 download_aborted 标记并终止流 .take_while(|res| { if res .as_ref() .is_err_and(|e| e.downcast_ref::().is_some()) { download_aborted = true; } futures::future::ready(!download_aborted) }) // 过滤掉没有触发风控的普通 Err,只保留正确返回的 Model .filter_map(|res| futures::future::ready(res.ok())) // 将成功返回的 Model 按十个一组合并 .chunks(10); while let Some(models) = stream.next().await { update_videos_model(models, connection).await?; } if download_aborted { error!("下载触发风控,已终止所有任务,等待下一轮执行"); } video_source.log_download_video_end(); Ok(()) } #[allow(clippy::too_many_arguments)] pub async fn download_video_pages( bili_client: &BiliClient, video_source: &VideoSourceEnum, video_model: video::Model, pages: Vec, connection: &DatabaseConnection, semaphore: &Semaphore, downloader: &Downloader, should_download_upper: bool, ) -> Result { let _permit = semaphore.acquire().await.context("acquire semaphore failed")?; let mut status = VideoStatus::from(video_model.download_status); let separate_status = status.should_run(); let base_path = video_source.path().join( TEMPLATE .load() .path_safe_render("video", &video_format_args(&video_model))?, ); let upper_id = video_model.upper_id.to_string(); let base_upper_path = VersionedConfig::get() .load() .upper_path .join(upper_id.chars().next().context("upper_id is empty")?.to_string()) .join(upper_id); let is_single_page = video_model.single_page.context("single_page is null")?; // 对于单页视频,page 的下载已经足够 // 对于多页视频,page 下载仅包含了分集内容,需要额外补上视频的 poster 的 tvshow.nfo let (res_1, res_2, res_3, res_4, res_5) = tokio::join!( // 下载视频封面 fetch_video_poster( separate_status[0] && !is_single_page, &video_model, downloader, base_path.join("poster.jpg"), base_path.join("fanart.jpg"), ), // 生成视频信息的 nfo generate_video_nfo( separate_status[1] && !is_single_page, &video_model, base_path.join("tvshow.nfo"), ), // 下载 Up 主头像 fetch_upper_face( separate_status[2] && should_download_upper, &video_model, downloader, base_upper_path.join("folder.jpg"), ), // 生成 Up 主信息的 nfo generate_upper_nfo( separate_status[3] && should_download_upper, &video_model, base_upper_path.join("person.nfo"), ), // 分发并执行分 P 下载的任务 dispatch_download_page( separate_status[4], bili_client, &video_model, pages, connection, downloader, &base_path ) ); let results = [res_1, res_2, res_3, res_4, res_5] .into_iter() .map(Into::into) .collect::>(); status.update_status(&results); results .iter() .take(4) .zip(["封面", "详情", "作者头像", "作者详情"]) .for_each(|(res, task_name)| match res { ExecutionStatus::Skipped => info!("处理视频「{}」{}已成功过,跳过", &video_model.name, task_name), ExecutionStatus::Succeeded => info!("处理视频「{}」{}成功", &video_model.name, task_name), ExecutionStatus::Ignored(e) => { error!( "处理视频「{}」{}出现常见错误,已忽略: {:#}", &video_model.name, task_name, e ) } ExecutionStatus::Failed(e) | ExecutionStatus::FixedFailed(_, e) => { error!("处理视频「{}」{}失败: {:#}", &video_model.name, task_name, e) } }); if let ExecutionStatus::Failed(e) = results.into_iter().nth(4).context("page download result not found")? { if e.downcast_ref::().is_some() { return Err(e); } } let mut video_active_model: video::ActiveModel = video_model.into(); video_active_model.download_status = Set(status.into()); video_active_model.path = Set(base_path.to_string_lossy().to_string()); Ok(video_active_model) } /// 分发并执行分页下载任务,当且仅当所有分页成功下载或达到最大重试次数时返回 Ok,否则根据失败原因返回对应的错误 pub async fn dispatch_download_page( should_run: bool, bili_client: &BiliClient, video_model: &video::Model, pages: Vec, connection: &DatabaseConnection, downloader: &Downloader, base_path: &Path, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } let child_semaphore = Semaphore::new(VersionedConfig::get().load().concurrent_limit.page); let tasks = pages .into_iter() .map(|page_model| { download_page( bili_client, video_model, page_model, &child_semaphore, downloader, base_path, ) }) .collect::>(); let (mut download_aborted, mut target_status) = (false, STATUS_OK); let mut stream = tasks .take_while(|res| { match res { Ok(model) => { // 该视频的所有分页的下载状态都会在此返回,需要根据这些状态确认视频层“分 P 下载”子任务的状态 // 在过去的实现中,此处仅仅根据 page_download_status 的最高标志位来判断,如果最高标志位是 true 则认为完成 // 这样会导致即使分页中有失败到 MAX_RETRY 的情况,视频层的分 P 下载状态也会被认为是 Succeeded,不够准确 // 新版本实现会将此处取值为所有子任务状态的最小值,这样只有所有分页的子任务全部成功时才会认为视频层的分 P 下载状态是 Succeeded let page_download_status = model.download_status.try_as_ref().expect("download_status must be set"); let separate_status: [u32; 5] = PageStatus::from(*page_download_status).into(); for status in separate_status { target_status = target_status.min(status); } } Err(e) => { if e.downcast_ref::().is_some() { download_aborted = true; } } } // 仅在发生风控时终止流,其它情况继续执行 futures::future::ready(!download_aborted) }) .filter_map(|res| futures::future::ready(res.ok())) .chunks(10); while let Some(models) = stream.next().await { update_pages_model(models, connection).await?; } if download_aborted { error!("下载视频「{}」的分页时触发风控,将异常向上传递..", &video_model.name); bail!(DownloadAbortError()); } if target_status != STATUS_OK { return Ok(ExecutionStatus::FixedFailed(target_status, ProcessPageError().into())); } Ok(ExecutionStatus::Succeeded) } /// 下载某个分页,未发生风控且正常运行时返回 Ok(Page::ActiveModel),其中 status 字段存储了新的下载状态,发生风控时返回 DownloadAbortError pub async fn download_page( bili_client: &BiliClient, video_model: &video::Model, page_model: page::Model, semaphore: &Semaphore, downloader: &Downloader, base_path: &Path, ) -> Result { let _permit = semaphore.acquire().await.context("acquire semaphore failed")?; let mut status = PageStatus::from(page_model.download_status); let separate_status = status.should_run(); let is_single_page = video_model.single_page.context("single_page is null")?; let base_name = TEMPLATE .load() .path_safe_render("page", &page_format_args(video_model, &page_model))?; let (poster_path, video_path, nfo_path, danmaku_path, fanart_path, subtitle_path) = if is_single_page { ( base_path.join(format!("{}-poster.jpg", &base_name)), base_path.join(format!("{}.mp4", &base_name)), base_path.join(format!("{}.nfo", &base_name)), base_path.join(format!("{}.zh-CN.default.ass", &base_name)), Some(base_path.join(format!("{}-fanart.jpg", &base_name))), base_path.join(format!("{}.srt", &base_name)), ) } else { ( base_path .join("Season 1") .join(format!("{} - S01E{:0>2}-thumb.jpg", &base_name, page_model.pid)), base_path .join("Season 1") .join(format!("{} - S01E{:0>2}.mp4", &base_name, page_model.pid)), base_path .join("Season 1") .join(format!("{} - S01E{:0>2}.nfo", &base_name, page_model.pid)), base_path .join("Season 1") .join(format!("{} - S01E{:0>2}.zh-CN.default.ass", &base_name, page_model.pid)), // 对于多页视频,会在上一步 fetch_video_poster 中获取剧集的 fanart,无需在此处下载单集的 None, base_path .join("Season 1") .join(format!("{} - S01E{:0>2}.srt", &base_name, page_model.pid)), ) }; let dimension = match (page_model.width, page_model.height) { (Some(width), Some(height)) => Some(Dimension { width, height, rotate: 0, }), _ => None, }; let page_info = PageInfo { cid: page_model.cid, duration: page_model.duration, dimension, ..Default::default() }; let (res_1, res_2, res_3, res_4, res_5) = tokio::join!( // 下载分页封面 fetch_page_poster( separate_status[0], video_model, &page_model, downloader, poster_path, fanart_path ), // 下载分页视频 fetch_page_video( separate_status[1], bili_client, video_model, downloader, &page_info, &video_path ), // 生成分页视频信息的 nfo generate_page_nfo(separate_status[2], video_model, &page_model, nfo_path), // 下载分页弹幕 fetch_page_danmaku(separate_status[3], bili_client, video_model, &page_info, danmaku_path), // 下载分页字幕 fetch_page_subtitle(separate_status[4], bili_client, video_model, &page_info, &subtitle_path) ); let results = [res_1, res_2, res_3, res_4, res_5] .into_iter() .map(Into::into) .collect::>(); status.update_status(&results); results .iter() .zip(["封面", "视频", "详情", "弹幕", "字幕"]) .for_each(|(res, task_name)| match res { ExecutionStatus::Skipped => info!( "处理视频「{}」第 {} 页{}已成功过,跳过", &video_model.name, page_model.pid, task_name ), ExecutionStatus::Succeeded => info!( "处理视频「{}」第 {} 页{}成功", &video_model.name, page_model.pid, task_name ), ExecutionStatus::Ignored(e) => { error!( "处理视频「{}」第 {} 页{}出现常见错误,已忽略: {:#}", &video_model.name, page_model.pid, task_name, e ) } ExecutionStatus::Failed(e) | ExecutionStatus::FixedFailed(_, e) => error!( "处理视频「{}」第 {} 页{}失败: {:#}", &video_model.name, page_model.pid, task_name, e ), }); // 如果下载视频时触发风控,直接返回 DownloadAbortError if let ExecutionStatus::Failed(e) = results.into_iter().nth(1).context("video download result not found")? { if let Ok(BiliError::RiskControlOccurred) = e.downcast::() { bail!(DownloadAbortError()); } } let mut page_active_model: page::ActiveModel = page_model.into(); page_active_model.download_status = Set(status.into()); page_active_model.path = Set(Some(video_path.to_string_lossy().to_string())); Ok(page_active_model) } pub async fn fetch_page_poster( should_run: bool, video_model: &video::Model, page_model: &page::Model, downloader: &Downloader, poster_path: PathBuf, fanart_path: Option, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } let single_page = video_model.single_page.context("single_page is null")?; let url = if single_page { // 单页视频直接用视频的封面 video_model.cover.as_str() } else { // 多页视频,如果单页没有封面,就使用视频的封面 match &page_model.image { Some(url) => url.as_str(), None => video_model.cover.as_str(), } }; downloader.fetch(url, &poster_path).await?; if let Some(fanart_path) = fanart_path { fs::copy(&poster_path, &fanart_path).await?; } Ok(ExecutionStatus::Succeeded) } pub async fn fetch_page_video( should_run: bool, bili_client: &BiliClient, video_model: &video::Model, downloader: &Downloader, page_info: &PageInfo, page_path: &Path, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } let bili_video = Video::new(bili_client, video_model.bvid.clone()); let streams = bili_video .get_page_analyzer(page_info) .await? .best_stream(&VersionedConfig::get().load().filter_option)?; match streams { BestStream::Mixed(mix_stream) => downloader.fetch_with_fallback(&mix_stream.urls(), page_path).await?, BestStream::VideoAudio { video: video_stream, audio: None, } => downloader.fetch_with_fallback(&video_stream.urls(), page_path).await?, BestStream::VideoAudio { video: video_stream, audio: Some(audio_stream), } => { let (tmp_video_path, tmp_audio_path) = ( page_path.with_extension("tmp_video"), page_path.with_extension("tmp_audio"), ); let res = async { downloader .fetch_with_fallback(&video_stream.urls(), &tmp_video_path) .await?; downloader .fetch_with_fallback(&audio_stream.urls(), &tmp_audio_path) .await?; downloader.merge(&tmp_video_path, &tmp_audio_path, page_path).await } .await; let _ = fs::remove_file(tmp_video_path).await; let _ = fs::remove_file(tmp_audio_path).await; res? } } Ok(ExecutionStatus::Succeeded) } pub async fn fetch_page_danmaku( should_run: bool, bili_client: &BiliClient, video_model: &video::Model, page_info: &PageInfo, danmaku_path: PathBuf, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } let bili_video = Video::new(bili_client, video_model.bvid.clone()); bili_video .get_danmaku_writer(page_info) .await? .write(danmaku_path) .await?; Ok(ExecutionStatus::Succeeded) } pub async fn fetch_page_subtitle( should_run: bool, bili_client: &BiliClient, video_model: &video::Model, page_info: &PageInfo, subtitle_path: &Path, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } let bili_video = Video::new(bili_client, video_model.bvid.clone()); let subtitles = bili_video.get_subtitles(page_info).await?; let tasks = subtitles .into_iter() .map(|subtitle| async move { let path = subtitle_path.with_extension(format!("{}.srt", subtitle.lan)); tokio::fs::write(path, subtitle.body.to_string()).await }) .collect::>(); tasks.try_collect::>().await?; Ok(ExecutionStatus::Succeeded) } pub async fn generate_page_nfo( should_run: bool, video_model: &video::Model, page_model: &page::Model, nfo_path: PathBuf, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } let single_page = video_model.single_page.context("single_page is null")?; let nfo = if single_page { NFO::Movie(video_model.into()) } else { NFO::Episode(page_model.into()) }; generate_nfo(nfo, nfo_path).await?; Ok(ExecutionStatus::Succeeded) } pub async fn fetch_video_poster( should_run: bool, video_model: &video::Model, downloader: &Downloader, poster_path: PathBuf, fanart_path: PathBuf, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } downloader.fetch(&video_model.cover, &poster_path).await?; fs::copy(&poster_path, &fanart_path).await?; Ok(ExecutionStatus::Succeeded) } pub async fn fetch_upper_face( should_run: bool, video_model: &video::Model, downloader: &Downloader, upper_face_path: PathBuf, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } downloader.fetch(&video_model.upper_face, &upper_face_path).await?; Ok(ExecutionStatus::Succeeded) } pub async fn generate_upper_nfo( should_run: bool, video_model: &video::Model, nfo_path: PathBuf, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } generate_nfo(NFO::Upper(video_model.into()), nfo_path).await?; Ok(ExecutionStatus::Succeeded) } pub async fn generate_video_nfo( should_run: bool, video_model: &video::Model, nfo_path: PathBuf, ) -> Result { if !should_run { return Ok(ExecutionStatus::Skipped); } generate_nfo(NFO::TVShow(video_model.into()), nfo_path).await?; Ok(ExecutionStatus::Succeeded) } /// 创建 nfo_path 的父目录,然后写入 nfo 文件 async fn generate_nfo(nfo: NFO<'_>, nfo_path: PathBuf) -> Result<()> { if let Some(parent) = nfo_path.parent() { fs::create_dir_all(parent).await?; } fs::write(nfo_path, nfo.generate_nfo().await?.as_bytes()).await?; Ok(()) }