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
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use anyhow::Result;
use std::{collections::HashMap, fs, io::Write, path::PathBuf};

pub struct Merger {
    indexed: usize,
    merger_type: MergerType,
    pos: usize,
    size: usize,
    stored_bytes: usize,
}

enum MergerType {
    Directory(PathBuf),
    File((fs::File, HashMap<usize, Vec<u8>>)),
}

impl Merger {
    pub fn new_file(size: usize, path: &PathBuf) -> Result<Self> {
        Ok(Self {
            indexed: 0,
            merger_type: MergerType::File((fs::File::create(path)?, HashMap::new())),
            pos: 0,
            size: size - 1,
            stored_bytes: 0,
        })
    }

    pub fn new_directory(size: usize, path: &PathBuf) -> Result<Self> {
        if !path.exists() {
            fs::create_dir_all(path)?;
        }

        Ok(Self {
            indexed: 0,
            merger_type: MergerType::Directory(path.to_owned()),
            pos: 0,
            stored_bytes: 0,
            size: size - 1,
        })
    }

    pub fn buffered(&self) -> bool {
        let buffers_empty = match &self.merger_type {
            MergerType::Directory(_) => true,
            MergerType::File((_, buffers)) => buffers.is_empty(),
        };
        buffers_empty && self.pos >= (self.size + 1)
    }

    pub fn flush(&mut self) -> Result<()> {
        if let MergerType::File((file, buffers)) = &mut self.merger_type {
            while self.pos <= self.size {
                if let Some(buf) = buffers.remove(&self.pos) {
                    file.write_all(&buf)?;
                    file.flush()?;
                    self.pos += 1;
                } else {
                    break;
                }
            }
        }

        Ok(())
    }

    pub fn estimate(&self) -> usize {
        if self.indexed == 0 {
            0
        } else {
            (self.stored_bytes / self.indexed) * (self.size + 1)
        }
    }

    pub fn stored(&self) -> usize {
        self.stored_bytes
    }

    pub fn write(&mut self, pos: usize, buf: &[u8]) -> Result<()> {
        match &mut self.merger_type {
            MergerType::Directory(path) => {
                let mut file = fs::File::create(path.join(format!(
                    "{}.{}",
                    pos,
                    path.extension().unwrap().to_string_lossy()
                )))?;
                file.write_all(buf)?;
                self.pos += 1;
                self.stored_bytes += buf.len();
            }
            MergerType::File((file, buffers)) => {
                if pos == 0 || (self.pos != 0 && self.pos == pos) {
                    file.write_all(buf)?;
                    file.flush()?;
                    self.pos += 1;
                    self.stored_bytes += buf.len();
                } else {
                    buffers.insert(pos, buf.to_vec());
                    self.stored_bytes += buf.len();
                }
            }
        };

        self.indexed += 1;
        Ok(())
    }
}