fkget 0.2.0

A simple file downloader written in Rust
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

use std::fs::File;
use std::io::{stdout, Write};
use std::time::{Instant};
use reqwest::Url;
use std::error::Error;
use std::env;
use tokio::io::{AsyncWriteExt, AsyncReadExt};

pub async fn download_file(url: &Url) -> Result<(), Box<dyn std::error::Error>> {
    let mut response = reqwest::get(url.as_ref()).await?;
    let start_time = Instant::now();
    let file_name = url.path_segments().and_then(|segments| segments.last()).unwrap_or("file.bin");
    let mut file = File::create(file_name)?;
    let mut downloaded_size = 0;
    //let mut _buf = [0; 8192];
    while let Some(chunk) = response.chunk().await? {
        let len = chunk.len();
        downloaded_size += len;
        file.write_all(&chunk);
        let elapsed_time = start_time.elapsed();
        let elapsed_seconds = elapsed_time.as_secs_f64();
        let download_speed = downloaded_size as f64 / elapsed_seconds / 1_000_000.0;
        print!(
            "\rDownloaded {:.2} MB at {:.2} Mbps",
            downloaded_size as f64 / 1_000_000.0,
            download_speed
        );
        stdout().flush()?;
    }
    let elapsed_time = start_time.elapsed();
    let elapsed_seconds = elapsed_time.as_secs_f64();
    let file_size = downloaded_size as f64;
    let download_speed = file_size / elapsed_seconds / 1_000_000.0;
    println!(
        "\nDownloaded {:.2} MB in {:.2} seconds ({:.2} Mbps)",
        file_size / 1_000_000.0,
        elapsed_seconds,
        download_speed
    );
    Ok(())
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let url = match env::args().nth(1) {
        Some(url_str) => Url::parse(&url_str)?,
        None => Url::parse("https://speed.hetzner.de/100MB.bin")?,
    };
    download_file(&url).await?;
    Ok(())
}


// fn main() -> Result<(), Box<dyn Error>> {
//     let url = match env::args().nth(1) {
//         Some(url_str) => Url::parse(&url_str)?,
//         None => Url::parse("https://speed.hetzner.de/100MB.bin")?,
//     };
//     download_file(&url)?;
//     Ok(())
// }

// #[cfg(test)]
// mod tests {
//     use super::*;
//
//     #[test]
//     fn test_download_file() {
//         let url = Url::parse("https://speed.hetzner.de/100MB.bin").unwrap();
//         download_file(&url).unwrap();
//         let file = File::open("100MB.bin").unwrap();
//         let file_size = file.metadata().unwrap().len();
//         assert_eq!(file_size, 104_857_600);
//         std::fs::remove_file("100MB.bin").unwrap();
//     }
// }

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    async fn test_download_file() {
        let url = Url::parse("https://speed.hetzner.de/100MB.bin").unwrap();
        download_file(&url).await.unwrap();
        let file = File::open("100MB.bin").unwrap();
        let file_size = file.metadata().unwrap().len();
        assert_eq!(file_size, 104_857_600);
        std::fs::remove_file("100MB.bin").unwrap();
    }
}