Compare commits

...
2 changed files with 83 additions and 125 deletions
+2 -1
View File
@@ -351,7 +351,8 @@ impl BiliClient {
.read()
.get("https://api.bilibili.com/x/space/wbi/arc/search")
.query(&params)
.header("cookie", self.get_cookie());
// 不带DedeUserID会随机触发412错误
.header("cookie", format!("DedeUserID=1; {}", self.get_cookie()));
let http_resp = request.send().await?;
// 检查http响应状态码
let status = http_resp.status();
@@ -1,4 +1,4 @@
use std::{path::PathBuf, sync::Arc};
use std::{io, path::PathBuf, sync::Arc};
use eyre::{WrapErr, eyre};
use serde::{Deserialize, Serialize};
@@ -98,54 +98,29 @@ impl VideoProcessTask {
let output_path = episode_dir.join(format!("{filename}-merged.mp4"));
let (tx, rx) = tokio::sync::oneshot::channel();
let video_path_clone = video_path.clone();
let audio_path_clone = audio_path.clone();
let metadata_path_clone = metadata_path.clone();
let output_path_clone = output_path.clone();
let mut command = std::process::Command::new(ffmpeg_program);
let current_span = tracing::Span::current();
tauri::async_runtime::spawn_blocking(move || {
let _enter = current_span.enter();
let mut command = std::process::Command::new(ffmpeg_program);
command.arg("-i").arg(video_path_clone);
command.arg("-i").arg(audio_path_clone);
if let Some(metadata_path) = metadata_path_clone {
command.arg("-i").arg(metadata_path);
command.arg("-map_metadata").arg("2");
}
command.arg("-c").arg("copy");
command.arg("-map").arg("0:v:0");
command.arg("-map").arg("1:a:0");
command.arg(output_path_clone).arg("-y");
#[cfg(target_os = "windows")]
{
// 隐藏窗口
use std::os::windows::process::CommandExt;
command.creation_flags(0x0800_0000);
}
let output = command.output();
let _ = tx.send(output);
});
let output = rx.await??;
if !output.status.success() {
let stderr = String::from_utf8_lossy(&output.stderr);
let stdout = String::from_utf8_lossy(&output.stdout);
let err = eyre!(format!("STDOUT: {stdout}"))
.wrap_err(format!("STDERR: {stderr}"))
.wrap_err("原因可能是视频或音频文件损坏,建议[重来]试试");
return Err(err);
command.arg("-i").arg(&video_path);
command.arg("-i").arg(&audio_path);
if let Some(metadata_path) = &metadata_path {
command.arg("-i").arg(metadata_path);
command.arg("-map_metadata").arg("2");
}
command.arg("-c").arg("copy");
command.arg("-map").arg("0:v:0");
command.arg("-map").arg("1:a:0");
command.arg(&output_path).arg("-y");
#[cfg(target_os = "windows")]
{
// 隐藏窗口
use std::os::windows::process::CommandExt;
command.creation_flags(0x0800_0000);
}
execute_ffmpeg(command).await.wrap_err("执行FFmpeg失败")?;
std::fs::remove_file(&video_path)
.wrap_err(format!("删除视频文件`{}`失败", video_path.display()))?;
std::fs::remove_file(&audio_path)
@@ -192,49 +167,24 @@ impl VideoProcessTask {
let ffmpeg_program = utils::get_ffmpeg_program().wrap_err("获取FFmpeg程序路径失败")?;
let (tx, rx) = tokio::sync::oneshot::channel();
let video_path_clone = video_path.clone();
let audio_path_clone = audio_path.clone();
let output_path_clone = output_path.clone();
let mut command = std::process::Command::new(ffmpeg_program);
let current_span = tracing::Span::current();
tauri::async_runtime::spawn_blocking(move || {
let _enter = current_span.enter();
command.arg("-i").arg(&video_path);
command.arg("-i").arg(&audio_path);
command.arg("-c").arg("copy");
command.arg("-map").arg("0:v:0");
command.arg("-map").arg("1:a:0");
command.arg(&output_path).arg("-y");
let mut command = std::process::Command::new(ffmpeg_program);
command.arg("-i").arg(video_path_clone);
command.arg("-i").arg(audio_path_clone);
command.arg("-c").arg("copy");
command.arg("-map").arg("0:v:0");
command.arg("-map").arg("1:a:0");
command.arg(output_path_clone).arg("-y");
#[cfg(target_os = "windows")]
{
// 隐藏窗口
use std::os::windows::process::CommandExt;
command.creation_flags(0x0800_0000);
}
let output = command.output();
let _ = tx.send(output);
});
let output = rx.await??;
if !output.status.success() {
let stderr = String::from_utf8_lossy(&output.stderr);
let stdout = String::from_utf8_lossy(&output.stdout);
let err = eyre!(format!("STDOUT: {stdout}"))
.wrap_err(format!("STDERR: {stderr}"))
.wrap_err("原因可能是视频或音频文件损坏,建议[重来]试试");
return Err(err);
#[cfg(target_os = "windows")]
{
// 隐藏窗口
use std::os::windows::process::CommandExt;
command.creation_flags(0x0800_0000);
}
execute_ffmpeg(command).await.wrap_err("执行FFmpeg失败")?;
std::fs::remove_file(&video_path)
.wrap_err(format!("删除视频文件`{}`失败", video_path.display()))?;
std::fs::remove_file(&audio_path)
@@ -279,48 +229,23 @@ impl VideoProcessTask {
return Ok(());
};
let (tx, rx) = tokio::sync::oneshot::channel();
let video_path_clone = video_path.clone();
let metadata_path_clone = metadata_path.clone();
let output_path_clone = output_path.clone();
let mut command = std::process::Command::new(ffmpeg_program);
let current_span = tracing::Span::current();
tauri::async_runtime::spawn_blocking(move || {
let _enter = current_span.enter();
command.arg("-i").arg(&video_path);
command.arg("-i").arg(&metadata_path);
command.arg("-map_metadata").arg("1");
command.arg("-c").arg("copy");
command.arg(&output_path).arg("-y");
let mut command = std::process::Command::new(ffmpeg_program);
command.arg("-i").arg(video_path_clone);
command.arg("-i").arg(metadata_path_clone);
command.arg("-map_metadata").arg("1");
command.arg("-c").arg("copy");
command.arg(output_path_clone).arg("-y");
#[cfg(target_os = "windows")]
{
// 隐藏窗口
use std::os::windows::process::CommandExt;
command.creation_flags(0x0800_0000);
}
let output = command.output();
let _ = tx.send(output);
});
let output = rx.await??;
if !output.status.success() {
let stderr = String::from_utf8_lossy(&output.stderr);
let stdout = String::from_utf8_lossy(&output.stdout);
let err = eyre!(format!("STDOUT: {stdout}"))
.wrap_err(format!("STDERR: {stderr}"))
.wrap_err("原因可能是视频或音频文件损坏,建议[重来]试试");
return Err(err);
#[cfg(target_os = "windows")]
{
// 隐藏窗口
use std::os::windows::process::CommandExt;
command.creation_flags(0x0800_0000);
}
execute_ffmpeg(command).await.wrap_err("执行FFmpeg失败")?;
std::fs::remove_file(&video_path)
.wrap_err(format!("删除视频文件`{}`失败", video_path.display()))?;
std::fs::rename(&output_path, &video_path).wrap_err(format!(
@@ -367,7 +292,10 @@ impl VideoProcessTask {
let bili_client = app.get_bili_client();
let cid = Some(progress.cid);
let skip_segments = bili_client.get_skip_segments(bvid, cid).await?;
let skip_segments = bili_client
.get_skip_segments(bvid, cid)
.await
.wrap_err("获取广告片段失败,请[继续]以重试。如果重试多次依旧失败,请在设置中取消勾选[标记广告]")?;
for segment in skip_segments.0 {
if let Some(chapter_segment) = segment.into_chapter_segment() {
chapter_segments.insert(chapter_segment);
@@ -388,3 +316,32 @@ impl VideoProcessTask {
Ok(Some(metadata_path))
}
}
#[instrument(level = "error", skip_all)]
async fn execute_ffmpeg(mut command: std::process::Command) -> eyre::Result<()> {
let span = tracing::Span::current();
tauri::async_runtime::spawn_blocking(move || {
let _enter = span.enter();
let program = command.get_program().to_string_lossy().into_owned();
let output = command.output().map_err(|err| {
if err.kind() == io::ErrorKind::NotFound {
eyre!(err).wrap_err(format!("找不到FFmpeg `{program}`"))
} else {
eyre!(err)
}
})?;
if output.status.success() {
return Ok(());
}
let stderr = String::from_utf8_lossy(&output.stderr);
let stdout = String::from_utf8_lossy(&output.stdout);
Err(eyre!(format!("STDOUT: {stdout}")))
.wrap_err(format!("STDERR: {stderr}"))
.wrap_err("原因可能是视频或音频文件损坏,建议[重来]试试")
})
.await?
}