alterware-launcher/src/http.rs

use crate::global;
use crate::misc;
use std::{fs, io::Write, path::Path, str};

pub fn get_body(url: &str) -> Vec<u8> {
    let mut res: Vec<u8> = Vec::new();

    match http_req::request::Request::new(&url.try_into().unwrap())
        .header(
            "User-Agent",
            &format!(
                "AlterWare Launcher | github.com/{}/{}",
                global::GH_OWNER,
                global::GH_REPO
            ),
        )
        .send(&mut res)
    {
        Ok(req) => {
            if req.status_code() == http_req::response::StatusCode::new(302)
                || req.status_code() == http_req::response::StatusCode::new(301)
            {
                let location = req.headers().get("Location").unwrap().as_str();
                return get_body(location);
            }

            if req.status_code() != http_req::response::StatusCode::new(200) {
                misc::fatal_error(&format!(
                    "Could not get body from {}, got {}",
                    url,
                    req.status_code()
                ));
            }
        }
        Err(e) => {
            misc::fatal_error(&format!("Could not get body from {}, got:\n{}", url, e));
        }
    }

    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);

    match fs::File::create(file_path) {
        Ok(mut file) => match file.write_all(&body) {
            Ok(_) => (),
            Err(e) => {
                misc::fatal_error(&format!(
                    "Could not write to file {}, got:\n{}",
                    file_path.to_str().unwrap(),
                    e
                ));
            }
        },
        Err(e) => {
            match e.kind() {
                std::io::ErrorKind::NotFound => {
                    fs::create_dir_all(file_path.parent().unwrap()).unwrap();
                    return download_file(url, file_path);
                }
                std::io::ErrorKind::PermissionDenied => {
                    misc::fatal_error(&format!(
                        "Permission to {} denied.\n  Please try:\n    1. Running the launcher as administrator.\n    2. Manually deleting the last downloaded file.\n    3. If your game is in the program files directory try moving it to another location.\n    4. Create an exception/exclusion in your Anti-Virus Software for either the last downloaded file or the entire game directory.\n\n\n{}",
                        file_path.to_str().unwrap(),
                        e
                    ));
                }
                _ => (),
            }

            misc::fatal_error(&format!(
                "Could not create file {}, got:\n{}",
                file_path.to_str().unwrap(),
                e
            ));
        }
    }
}