vsd 0.2.5

Download video streams served over HTTP from websites, HLS and DASH 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
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
78
79
80
81
82
83

use crate::subtitles::MP4Subtitles;
use anyhow::{anyhow, bail, Result};
use clap::{Args, ValueEnum};

#[derive(Debug, Clone, ValueEnum)]
pub enum Format {
    Srt,
    Vtt,
}

/// Extract subtitles embedded inside an mp4 file.
///
/// This is based on the https://github.com/xhlove/dash-subtitle-extractor
#[derive(Debug, Clone, Args)]
pub struct Extract {
    /// List of subtitles segment files where first file is init.mp4 and following files are *.m4s (segments).
    /// A single mp4 file can also be provided.
    #[arg(required = true)]
    files: Vec<String>,

    /// Subtitles output format.
    #[arg(short, long, value_enum, default_value_t = Format::Srt)]
    format: Format,

    /// Set timescale manually if no init segment is present.
    /// If timescale is set to anything then webvtt codec is used.
    #[arg(short, long)]
    timescale: Option<u32>,
    //   -segment-time SEGMENT_TIME, --segment-time SEGMENT_TIME
    //                         single segment duration, usually needed for ttml
    //                         content, calculation method: d / timescale
}

impl Extract {
    pub fn perform(&self) -> Result<()> {
        let mut files = vec![];

        for pattern in &self.files {
            for file in glob::glob(pattern)? {
                files.push(file?);
            }
        }

        if files.is_empty() {
            bail!("at least one file is required to extract subtitles")
        }

        let subtitles = if files.len() == 1 {
            let split_data =
                mp4decrypt::mp4split(&std::fs::read(&files[0])?).map_err(|x| anyhow!(x))?;

            let mut subtitles =
                MP4Subtitles::new(&split_data[0], self.timescale).map_err(|x| anyhow!(x))?;

            for data in &split_data[1..] {
                subtitles.add_cue(data).map_err(|x| anyhow!(x))?;
            }

            subtitles.to_subtitles()
        } else {
            let mut subtitles = MP4Subtitles::new(&std::fs::read(&files[0])?, self.timescale)
                .map_err(|x| anyhow!(x))?;

            for file in &files[1..] {
                subtitles
                    .add_cue(&std::fs::read(file)?)
                    .map_err(|x| anyhow!(x))?;
            }

            subtitles.to_subtitles()
        };

        print!(
            "{}",
            match &self.format {
                Format::Srt => subtitles.to_srt(),
                Format::Vtt => subtitles.to_vtt(),
            }
        );

        Ok(())
    }
}