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
84
85
86
87
88

use anyhow::{bail, Result};
use clap::Args;
use std::fs::File;
use std::io::Write;
use std::path::Path;

/// Merge multiple segments to a single file.
#[derive(Debug, Clone, Args)]
pub struct Merge {
    /// List of files to merge together like *.ts, *.m4s etc.
    #[arg(required = true)]
    files: Vec<String>,

    /// Path  of merged output file.
    #[arg(short, long, required = true)]
    output: String,

    /// Merge using ffmpeg instead of binary merge.
    #[arg(long)]
    ffmpeg: bool,
}

impl Merge {
    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.len() <= 1 {
            bail!("at least two files are required to merge together")
        }

        if self.ffmpeg {
            let mut concat_file = "ffmpeg-concat.txt".to_owned();

            for i in 1.. {
                if Path::new(&concat_file).exists() {
                    concat_file = format!("ffmpeg-concat-{}.txt", i);
                } else {
                    break;
                }
            }

            let mut concat = File::create(&concat_file)?;

            for file in files {
                let file = file.to_str().unwrap();

                if file != self.output {
                    concat.write_all(format!("file '{}'\n", file).as_bytes())?;
                }
            }

            let code = std::process::Command::new("ffmpeg")
                .args([
                    "-hide_banner",
                    "-y",
                    "-f",
                    "concat",
                    "-i",
                    &concat_file,
                    "-c",
                    "copy",
                    &self.output,
                ])
                .spawn()?
                .wait()?;

            if !code.success() {
                bail!("FFMPEG exited with code {}", code.code().unwrap_or(1))
            }

            std::fs::remove_file(&concat_file)?;
        } else {
            let mut merged = File::create(&self.output)?;

            for file in files {
                merged.write_all(&std::fs::read(file)?)?;
            }
        }

        Ok(())
    }
}