kelpsget 0.1.3

A modern, lightweight wget clone written in Rust for fast and reliable file downloads from the command line.
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

use clap::Parser;
use std::error::Error;
use crate::download::download;
use crate::advanced_download::AdvancedDownloader;
use crate::config::Config;
use crate::optimization::Optimizer;

#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    /// URL do arquivo para download
    url: String,

    /// Nome do arquivo de saída
    #[arg(short = 'O', long = "output")]
    output: Option<String>,

    /// Modo silencioso (sem barra de progresso)
    #[arg(short = 'q', long = "quiet")]
    quiet: bool,

    /// Usar download avançado (paralelo e resumível)
    #[arg(short = 'a', long = "advanced")]
    advanced: bool,

    /// URL do proxy (ex: http://proxy:8080)
    #[arg(short = 'p', long = "proxy")]
    proxy: Option<String>,

    /// Usuário do proxy
    #[arg(long = "proxy-user")]
    proxy_user: Option<String>,

    /// Senha do proxy
    #[arg(long = "proxy-pass")]
    proxy_pass: Option<String>,

    /// Tipo de proxy (http, https, socks5)
    #[arg(long = "proxy-type", default_value = "http")]
    proxy_type: String,

    /// Limite de velocidade em bytes/segundo
    #[arg(short = 'l', long = "limit")]
    speed_limit: Option<u64>,

    /// Desabilitar compressão
    #[arg(long = "no-compress")]
    no_compress: bool,

    /// Desabilitar cache
    #[arg(long = "no-cache")]
    no_cache: bool,
}

mod download;
mod progress;
mod utils;
mod advanced_download;
mod config;
mod optimization;

fn main() -> Result<(), Box<dyn Error>> {
    let args = Args::parse();
    let mut config = Config::load()?;

    // Atualizar configuração com argumentos da linha de comando
    if let Some(proxy_url) = args.proxy {
        config.proxy.enabled = true;
        config.proxy.url = Some(proxy_url);
    }
    if let Some(user) = args.proxy_user {
        config.proxy.username = Some(user);
    }
    if let Some(pass) = args.proxy_pass {
        config.proxy.password = Some(pass);
    }
    config.proxy.proxy_type = match args.proxy_type.to_lowercase().as_str() {
        "https" => crate::config::ProxyType::Https,
        "socks5" => crate::config::ProxyType::Socks5,
        _ => crate::config::ProxyType::Http,
    };

    if let Some(limit) = args.speed_limit {
        config.optimization.speed_limit = Some(limit);
    }
    if args.no_compress {
        config.optimization.compression = false;
    }
    if args.no_cache {
        config.optimization.cache_enabled = false;
    }

    // Salvar configuração atualizada
    config.save()?;

    let optimizer = Optimizer::new(config.optimization.clone());

    if args.advanced {
        let downloader = AdvancedDownloader::new(
            args.url,
            args.output.unwrap_or_else(|| "output".to_string()),
            args.quiet,
            config.proxy,
            optimizer,
        );
        downloader.download()?;
    } else {
        download(&args.url, args.quiet, args.output, config.proxy, optimizer)?;
    }

    Ok(())
}