biliget 0.6.8

简单的B站视频下载工具 支持免登录下载B站高清视频
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

use crate::processer::ffmpeg::{FfmpegError, convert_audio, merge_video};
use std::path::Path;
use thiserror::Error;
use tokio_util::sync::CancellationToken;

pub(crate) struct ProcessOption<'a> {
    pub video_file: Option<&'a Path>,
    pub audio_file: Option<&'a Path>,
    pub output_file: &'a Path,

    pub only_audio: bool,
}

#[allow(clippy::enum_variant_names)]
#[derive(Debug, Error)]
pub enum ProcessError {
    #[error("参数错误")]
    ParamError(),

    #[error("后处理错误")]
    ProcessError(#[from] FfmpegError),

    #[error("操作已取消")]
    Cancelled(),
}

pub async fn process(
    option: ProcessOption<'_>,
    cancel: CancellationToken,
) -> Result<(), ProcessError> {
    if option.only_audio {
        process_only_audio(
            option.audio_file.ok_or_else(ProcessError::ParamError)?,
            option.output_file,
            cancel,
        )
        .await?
    } else {
        process_default(
            Path::new(&option.video_file.ok_or_else(ProcessError::ParamError)?),
            Path::new(&option.audio_file.ok_or_else(ProcessError::ParamError)?),
            Path::new(&option.output_file),
            cancel,
        )
        .await?
    }

    Ok(())
}

async fn process_default(
    video_file: &Path,
    audio_file: &Path,
    output_file: &Path,
    cancel: CancellationToken,
) -> Result<(), ProcessError> {
    merge_video(video_file, audio_file, output_file, cancel)
        .await
        .map_err(|e| match e {
            FfmpegError::Cancelled => ProcessError::Cancelled(),
            _ => ProcessError::ProcessError(e),
        })
}

async fn process_only_audio(
    audio_file: &Path,
    output_file: &Path,
    cancel: CancellationToken,
) -> Result<(), ProcessError> {
    convert_audio(audio_file, output_file, cancel)
        .await
        .map_err(|e| match e {
            FfmpegError::Cancelled => ProcessError::Cancelled(),
            _ => ProcessError::ProcessError(e),
        })
}