37 lines
1.0 KiB
Rust
37 lines
1.0 KiB
Rust
use std::{fs, io::Write, path::Path, str};
|
|
|
|
pub fn get_body(url: &str) -> Vec<u8> {
|
|
let mut res: Vec<u8> = Vec::new();
|
|
let req = http_req::request::Request::new(&url.try_into().unwrap())
|
|
.header(
|
|
"User-Agent",
|
|
"AlterWare Launcher | github.com/mxve/alterware-launcher",
|
|
)
|
|
.send(&mut res)
|
|
.unwrap_or_else(|error| {
|
|
panic!("\n\n{}:\n{:?}", "Error", error);
|
|
});
|
|
|
|
if req.status_code() == http_req::response::StatusCode::new(302) {
|
|
let location = req.headers().get("Location").unwrap().as_str();
|
|
return get_body(location);
|
|
}
|
|
|
|
res
|
|
}
|
|
|
|
pub fn get_body_string(url: &str) -> String {
|
|
String::from_utf8(get_body(url)).unwrap()
|
|
}
|
|
|
|
pub fn download_file(url: &str, file_path: &Path) {
|
|
let body = get_body(url);
|
|
|
|
let mut f = fs::File::create(file_path).unwrap_or_else(|error| {
|
|
panic!("\n\n{}:\n{:?}", "Error", error);
|
|
});
|
|
f.write_all(&body).unwrap_or_else(|error| {
|
|
panic!("\n\n{}:\n{:?}", "Error", error);
|
|
});
|
|
}
|