v0.5.3

Cargo.lock

@@ -30,7 +30,7 @@ dependencies = [
 
 [[package]]
 name = "alterware-launcher"
-version = "0.5.2"
+version = "0.5.3"
 dependencies = [
  "colored",
  "http_req",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "alterware-launcher"
-version = "0.5.2"
+version = "0.5.3"
 edition = "2021"
 build = "res/build.rs"
 

src/config.rs

@@ -13,7 +13,19 @@ pub fn load(config_path: PathBuf) -> Config {
 }
 
 pub fn save(config_path: PathBuf, config: Config) {
-    fs::write(config_path, serde_json::to_string_pretty(&config).unwrap()).unwrap();
+    match fs::write(
+        config_path.clone(),
+        serde_json::to_string_pretty(&config).unwrap(),
+    ) {
+        Ok(_) => (),
+        Err(e) => match e.kind() {
+            std::io::ErrorKind::NotFound => {
+                fs::create_dir_all(config_path.parent().unwrap()).unwrap();
+                save(config_path, config);
+            }
+            _ => println!("Could not save config file, got:\n{}\n", e),
+        },
+    }
 }
 
 pub fn save_value(config_path: PathBuf, key: &str, value: bool) {

src/http.rs

@@ -1,22 +1,42 @@
+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();
-    let req = http_req::request::Request::new(&url.try_into().unwrap())
+
+    match http_req::request::Request::new(&url.try_into().unwrap())
         .header(
             "User-Agent",
-            "AlterWare Launcher | github.com/mxve/alterware-launcher",
+            &format!(
+                "AlterWare Launcher | github.com/{}/{}",
+                global::GH_OWNER,
+                global::GH_REPO
+            ),
         )
         .send(&mut res)
-        .unwrap_or_else(|error| {
-            panic!("\n\n{}:\n{:?}", "Error", error);
-        });
-
-    if req.status_code() == http_req::response::StatusCode::new(302) {
+    {
+        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
 }
 
@@ -27,10 +47,38 @@ pub fn get_body_string(url: &str) -> String {
 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);
-    });
+    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\n\n{}",
+                        file_path.to_str().unwrap(),
+                        e
+                    ));
+                }
+                _ => (),
+            }
+
+            misc::fatal_error(&format!(
+                "Could not create file {}, got:\n{}",
+                file_path.to_str().unwrap(),
+                e
+            ));
+        }
+    }
 }

src/main.rs

@@ -37,26 +37,35 @@ fn get_installed_games(games: &Vec<Game>) -> Vec<(u32, PathBuf)> {
     installed_games
 }
 
+#[cfg(windows)]
+fn create_shortcut(path: &Path, target: &Path, icon: String, args: String) {
+    if let Ok(mut sl) = ShellLink::new(target) {
+        sl.set_arguments(Some(args));
+        sl.set_icon_location(Some(icon));
+        sl.create_lnk(path).unwrap_or_else(|error| {
+            println!("Error creating shortcut.\n{:#?}", error);
+        });
+    } else {
+        println!("Error creating shortcut.");
+    }
+}
+
 #[cfg(windows)]
 fn setup_client_links(game: &Game, game_dir: &Path) {
     if game.client.len() > 1 {
         println!("Multiple clients installed, use the shortcuts (launch-<client>.lnk in the game directory or on the desktop) to launch a specific client.");
     }
 
-    let target = game_dir.join("alterware-launcher.exe");
-
     for c in game.client.iter() {
-        let lnk = game_dir.join(format!("launch-{}.lnk", c));
-
-        let mut sl = ShellLink::new(target.clone()).unwrap();
-        sl.set_arguments(Some(c.to_string()));
-        sl.set_icon_location(Some(
+        create_shortcut(
+            &game_dir.join(format!("launch-{}.lnk", c)),
+            &game_dir.join("alterware-launcher.exe"),
             game_dir
                 .join(format!("{}.exe", c))
                 .to_string_lossy()
                 .into_owned(),
-        ));
-        sl.create_lnk(&lnk).unwrap();
+            c.to_string(),
+        );
     }
 }
 
@@ -71,19 +80,15 @@ fn setup_desktop_links(path: &Path, game: &Game) {
             std::env::var("USERPROFILE").unwrap()
         ));
 
-        let target = path.join("alterware-launcher.exe");
-
         for c in game.client.iter() {
-            let lnk = desktop.join(format!("{}.lnk", c));
-
-            let mut sl = ShellLink::new(target.clone()).unwrap();
-            sl.set_arguments(Some(c.to_string()));
-            sl.set_icon_location(Some(
+            create_shortcut(
+                &desktop.join(format!("{}.lnk", c)),
+                &path.join("alterware-launcher.exe"),
                 path.join(format!("{}.exe", c))
                     .to_string_lossy()
                     .into_owned(),
-            ));
-            sl.create_lnk(lnk).unwrap();
+                c.to_string(),
+            );
         }
     }
 }

src/misc.rs

@@ -1,5 +1,7 @@
 use std::{fs, path::PathBuf};
 
+use colored::Colorize;
+
 pub fn get_file_sha1(path: &PathBuf) -> String {
     let mut sha1 = sha1_smol::Sha1::new();
     sha1.update(&fs::read(path).unwrap());
@@ -18,3 +20,9 @@ pub fn rev_to_int(rev: &str) -> u16 {
         .parse::<u16>()
         .unwrap_or(0)
 }
+
+pub fn fatal_error(error: &str) {
+    println!("\n\n{}:\n{}", "Error".bright_red(), error);
+    stdin();
+    std::process::exit(1);
+}

src/self_update.rs

@@ -30,6 +30,7 @@ pub fn run(update_only: bool) {
     use std::{fs, path::PathBuf};
 
     use crate::http;
+    use crate::misc;
 
     let working_dir = std::env::current_dir().unwrap();
     let files = fs::read_dir(&working_dir).unwrap();
@@ -82,9 +83,13 @@ pub fn run(update_only: bool) {
 
         self_replace::self_replace("alterware-launcher-update.exe").unwrap();
         fs::remove_file(&file_path).unwrap();
-        println!("Launcher updated. Please run it again.");
+        println!(
+            "Launcher updated. View the changelog at https://github.com/{}/{}/releases/latest",
+            GH_OWNER, GH_REPO,
+        );
+        println!("Please restart the launcher.");
         if !update_only {
-            std::io::stdin().read_line(&mut String::new()).unwrap();
+            misc::stdin();
         }
         std::process::exit(201);
     }