vsd 0.5.0

A command-line utility and library for downloading streams from DASH manifests and HLS playlists.
Documentation 
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
77

use crate::error::Result;
use clap::{Args, ValueEnum};
use std::path::PathBuf;
use tokio::fs;
use vsd_mp4::sub::{StppSubsParser, WvttSubsParser};

/// Extract subtitles from a fragmented mp4 file.
#[derive(Args, Clone, Debug)]
pub struct Extract {
    /// Fragmented mp4 file path containing either wvtt (webvtt) or stpp (ttml) boxes.
    ///
    /// For multiple segments, use merge sub-command to combine them into a single file first.
    #[arg(required = true)]
    input: PathBuf,

    /// Codec for output subtitles.
    #[arg(short, long, value_enum, default_value_t = Codec::Webvtt)]
    codec: Codec,

    /// Output file path for the extracted subtitles.
    ///
    /// The codec is inferred from the file extension (.srt or .vtt).
    #[arg(short, long, value_name = "PATH")]
    output: Option<PathBuf>,
}

#[derive(Clone, Debug, ValueEnum)]
enum Codec {
    Subrip,
    Webvtt,
}

impl Extract {
    pub async fn execute(self) -> Result<()> {
        let data = fs::read(self.input).await?;
        let subtitles;

        if let Ok(vtt) = WvttSubsParser::from_init(&data) {
            subtitles = vtt.parse(&data, None)?;
        } else if let Ok(ttml) = StppSubsParser::from_init(&data) {
            subtitles = ttml.parse(&data)?;
        } else {
            bail!(
                "Unable to determine the subtitle codec because neither wvtt nor stpp boxes were found."
            );
        }

        if let Some(path) = self.output {
            let ext = path
                .extension()
                .and_then(|x| match x.to_str() {
                    Some("srt") => Some(Codec::Subrip),
                    Some("vtt") => Some(Codec::Webvtt),
                    _ => None,
                })
                .unwrap_or(self.codec);
            fs::write(
                &path,
                match ext {
                    Codec::Subrip => subtitles.as_srt(),
                    Codec::Webvtt => subtitles.as_vtt(),
                },
            )
            .await?;
        } else {
            print!(
                "{}",
                match &self.codec {
                    Codec::Subrip => subtitles.as_srt(),
                    Codec::Webvtt => subtitles.as_vtt(),
                }
            );
        }

        Ok(())
    }
}