vsd 0.4.3

Download video streams served over HTTP from websites, DASH (.mpd) and HLS (.m3u8) playlists.
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

use anyhow::{bail, Result};
use clap::{Args, ValueEnum};
use std::path::PathBuf;
use vsd_mp4::text::{Mp4TtmlParser, Mp4VttParser};

/// Extract subtitles from mp4 boxes.
#[derive(Args, Clone, Debug)]
pub struct Extract {
    /// Path of mp4 file which either contains WVTT or STPP box.
    /// If there are multiple fragments of same mp4 file,
    /// then merge them using merge sub-command.
    #[arg(required = true)]
    input: PathBuf,

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

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

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

        if let Ok(vtt) = Mp4VttParser::parse_init(&data) {
            subtitles = vtt.parse_media(&data, None)?;
        } else if let Ok(ttml) = Mp4TtmlParser::parse_init(&data) {
            subtitles = ttml.parse_media(&data)?;
        } else {
            bail!("cannot determine subtitles codec because neither WVTT nor STPP box is found.");
        }

        print!(
            "{}",
            match &self.codec {
                Codec::Subrip => subtitles.as_srt(),
                Codec::Webvtt => subtitles.as_vtt(),
            }
        );

        Ok(())
    }
}