feat: finish all func

This commit is contained in:
2024-01-15 05:06:51 +08:00
parent c701e1badb
commit 4ca1b62979
2 changed files with 68 additions and 21 deletions

View File

@@ -7,4 +7,7 @@ authors = ["Smart SangGe <liangjunyong06@gmail.com>"]
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies] [dependencies]
clap = "2.26.0" clap = "2.33.2"
indicatif = "0.15"
console = "0.14"
reqwest = { version = "0.11", features = ["blocking"] }

View File

@@ -1,6 +1,18 @@
extern crate clap; extern crate clap;
extern crate reqwest;
extern crate indicatif;
extern crate console;
use clap::{Arg, App}; use clap::{Arg, App};
use indicatif::{ProgressBar, ProgressStyle, HumanBytes};
use console::style;
use std::io::Read;
use reqwest::header::{CONTENT_LENGTH, CONTENT_TYPE};
use std::fs::File;
use std::io::{self, Write};
use reqwest::blocking::Client;
fn main() { fn main() {
let matches = App::new("Rget") let matches = App::new("Rget")
@@ -13,8 +25,12 @@ fn main() {
.index(1) .index(1)
.help("url to download")) .help("url to download"))
.get_matches(); .get_matches();
let url = matches.value_of("URL").unwrap(); let url = matches.value_of("URL").unwrap();
println!("{}", url);
if let Err(e) = download(url, false) {
eprintln!("Download error: {}", e);
}
} }
fn create_progress_bar(quiet_mode: bool, msg: &str, length: Option<u64>) -> ProgressBar { fn create_progress_bar(quiet_mode: bool, msg: &str, length: Option<u64>) -> ProgressBar {
@@ -40,41 +56,56 @@ fn create_progress_bar(quiet_mode: bool, msg: &str, length: Option<u64>) -> Prog
bar bar
} }
fn download(target: &str, quiet_mode: bool) -> Result<(), Box<::std::error::Error>> { fn download(target: &str, quiet_mode: bool) -> Result<(), Box<dyn (::std::error::Error)>> {
// parse url // parse url
let url = parse_url(target)?; let url = reqwest::Url::parse(target)?;
let client = Client::new().unwrap(); let client = reqwest::blocking::Client::new();
let mut resp = client.get(url)? let mut resp = client.get(url).send()?;
.send()
.unwrap();
print(format!("HTTP request sent... {}",
style(format!("{}", resp.status())).green()), if !quiet_mode {
quiet_mode); println!("HTTP request sent... {}", style(resp.status()).green());
}
if resp.status().is_success() { if resp.status().is_success() {
let headers = resp.headers().clone(); let headers = resp.headers().clone();
let ct_len = headers.get::<ContentLength>().map(|ct_len| **ct_len); let ct_len = headers.get("content-length")
.and_then(|ct_len| ct_len.to_str().ok())
.and_then(|ct_len_str| ct_len_str.parse::<u64>().ok());
let ct_type = headers.get("content-type")
.and_then(|ct_type| ct_type.to_str().ok())
.unwrap(); // Be cautious with unwrap; it's better to handle None case
let ct_type = headers.get::<ContentType>().unwrap();
match ct_len { match ct_len {
Some(len) => { Some(len) => {
print(format!("Length: {} ({})", if !quiet_mode {
style(len).green(), println!("Length: {} ({})",
style(format!("{}", HumanBytes(len))).red()), style(len).green(),
quiet_mode); style(HumanBytes(len)).red());
}
}, },
None => { None => {
print(format!("Length: {}", style("unknown").red()), quiet_mode); if !quiet_mode {
println!("Length: {}", style("unknown").red());
}
}, },
} }
print(format!("Type: {}", style(ct_type).green()), quiet_mode);
// Print content type
custom_print(format!("Type: {}", style(ct_type).green()), quiet_mode);
// Get the filename from the URL
let fname = target.split("/").last().unwrap(); let fname = target.split("/").last().unwrap();
print(format!("Saving to: {}", style(fname).green()), quiet_mode); // Print saving message
custom_print(format!("Saving to: {}", style(fname).green()), quiet_mode);
let chunk_size = match ct_len { let chunk_size = match ct_len {
Some(x) => x as usize / 99, Some(x) => x as usize / 99,
@@ -107,4 +138,17 @@ fn download(target: &str, quiet_mode: bool) -> Result<(), Box<::std::error::Erro
Ok(()) Ok(())
} }
fn save_to_file(data: &[u8], filename: &str) -> io::Result<()> {
let mut file = File::create(filename)?;
file.write_all(data)?;
Ok(())
}
fn custom_print(message: String, quiet_mode: bool) {
if !quiet_mode {
println!("{}", message);
}
}