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
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
107
108
109
110
111
112
113
114
115
116
117

mod decrypt;
mod extract;
mod merge;
mod save;

#[cfg(feature = "capture")]
mod capture;

#[cfg(feature = "license")]
mod license;

pub use decrypt::Decrypt;
pub use extract::Extract;
use log::LevelFilter;
pub use merge::Merge;
pub use save::Save;

#[cfg(feature = "capture")]
pub use capture::Capture;

#[cfg(feature = "license")]
pub use license::License;

use crate::{error::Result, logger::Logger};
use clap::{ArgAction, ColorChoice, Parser, Subcommand};

#[derive(Debug, Clone, Subcommand)]
pub enum Commands {
    #[cfg(feature = "capture")]
    Capture(Capture),
    Decrypt(Decrypt),
    Extract(Extract),
    #[cfg(feature = "license")]
    License(License),
    Merge(Merge),
    Save(Box<Save>),
}

#[derive(Debug, Clone, Parser)]
#[command(
    about,
    author = "clitic <clitic21@gmail.com>",
    version,
    long_version = concat!(
        env!("CARGO_PKG_VERSION"), "\n\n",
        "Enabled Features:\n",
        "capture: ", cfg!(feature = "capture"), "\n",
        "license: ", cfg!(feature = "license"), "\n",
        "rustls-tls: ", cfg!(feature = "rustls-tls"), "\n",
        "native-tls: ", cfg!(feature = "native-tls"), "\n",
        "native-tls-vendored: ", cfg!(feature = "native-tls-vendored"),
    ),
)]
pub struct Args {
    #[command(subcommand)]
    pub command: Commands,

    /// Enable colored output.
    #[arg(long, global = true, help_heading = "Global Options", default_value_t = ColorChoice::Auto)]
    pub color: ColorChoice,

    /// Disable all output except errors.
    #[arg(
        short,
        long,
        global = true,
        help_heading = "Global Options",
        conflicts_with = "verbose"
    )]
    quiet: bool,

    /// Enable more detailed logging. Use -v (debug) or -vv (trace).
    #[arg(short, long, global = true, help_heading = "Global Options", action = ArgAction::Count)]
    verbose: u8,
}

impl Args {
    pub async fn execute(self) -> Result<()> {
        let level = if self.quiet {
            LevelFilter::Error
        } else {
            match self.verbose {
                0 => LevelFilter::Info,
                1 => LevelFilter::Debug,
                _ => LevelFilter::Trace,
            }
        };

        log::set_logger(&Logger)
            .map(|()| log::set_max_level(level))
            .expect("Failed to initialize logger.");

        match self.color {
            ColorChoice::Always => colored::control::set_override(true),
            ColorChoice::Never => colored::control::set_override(false),
            _ => (),
        }

        let mut symbols = requestty::symbols::UNICODE;
        symbols.completed = '•';
        symbols.cross = 'x';
        requestty::symbols::set(symbols);

        match self.command {
            #[cfg(feature = "capture")]
            Commands::Capture(args) => args.execute().await?,
            Commands::Decrypt(args) => args.execute().await?,
            Commands::Extract(args) => args.execute().await?,
            #[cfg(feature = "license")]
            Commands::License(args) => args.execute().await?,
            Commands::Merge(args) => args.execute().await?,
            Commands::Save(args) => args.execute().await?,
        }

        Ok(())
    }
}