feat: --no-fail

Cargo.toml

@@ -16,6 +16,7 @@ github = ["reqwest"]
 [dependencies]
 clap = { version = "4.5.21", features = ["derive", "color", "error-context", "help", "std", "usage"], default-features = false }
 colored = "2.1.0"
+lazy_static = "1.5.0"
 reqwest = { version = "0.12.9", features = ["__tls", "charset", "default-tls", "h2", "http2", "json"], default-features = false, optional = true }
 serde = { version = "1.0.215", features = ["derive"] }
 serde_json = "1.0.132"

src/api/aur.rs

@@ -15,14 +15,15 @@ pub fn get_latest(package: String, _: String) -> crate::api::ReleaseFuture {
         match result.status() {
             StatusCode::OK => (),
             status => {
-                crate::custom_error("GET request didn't return 200", format!("\n{}", status));
+                crate::custom_error("GET request didn't return 200", format!("\n{}", status), "");
+                return None;
             }
         }
 
         let json: serde_json::Value = result.json().await.unwrap();
         let first_result = json.get("results").unwrap().get(0).unwrap();
 
-        crate::api::Release {
+        Some(crate::api::Release {
             tag_name: first_result
                 .get("Version")
                 .unwrap()
@@ -33,6 +34,6 @@ pub fn get_latest(package: String, _: String) -> crate::api::ReleaseFuture {
                 .replace("\"", "")
                 .to_string(),
             html_url: first_result.get("URL").unwrap().to_string(),
-        }
+        })
     })
 }

src/api/github.rs

@@ -3,7 +3,7 @@ use reqwest::{
     StatusCode,
 };
 
-pub fn get_latest(_: String, repo: String) -> crate::api::ReleaseFuture {
+pub fn get_latest(package: String, repo: String) -> crate::api::ReleaseFuture {
     Box::pin(async move {
         let url = format!("https://api.github.com/repos/{}/releases/latest", repo);
         let mut headers = HeaderMap::new();
@@ -22,11 +22,24 @@ pub fn get_latest(_: String, repo: String) -> crate::api::ReleaseFuture {
 
         match result.status() {
             StatusCode::OK => (),
+            StatusCode::FORBIDDEN => {
+                crate::custom_error(
+                    "GET request returned 430: ",
+                    format!("{}\nwe might be getting rate-limited here", package),
+                    "",
+                );
+                return None;
+            }
             status => {
-                crate::custom_error("GET request didn't return 200", format!("\n{}", status));
+                crate::custom_error(
+                    "GET request didn't return 200: ",
+                    format!("{}\n{}", package, status),
+                    "",
+                );
+                return None;
             }
         }
 
-        result.json().await.unwrap()
+        Some(result.json().await.unwrap())
     })
 }

src/api/mod.rs

@@ -11,7 +11,8 @@ pub struct Release {
     pub html_url: String,
 }
 
-pub type ReleaseFuture = std::pin::Pin<Box<dyn std::future::Future<Output = Release> + Send>>;
+pub type ReleaseFuture =
+    std::pin::Pin<Box<dyn std::future::Future<Output = Option<Release>> + Send>>;
 
 pub struct Api {
     pub name: &'static str,

src/config.rs

@@ -59,7 +59,7 @@ pub fn load(custom_path: Option<String>) -> (Config, PathBuf) {
                 PathBuf::from(config_path),
             );
         } else {
-            crate::custom_error("specified config file not found", String::new());
+            crate::custom_error("specified config file not found", String::new(), "");
         }
     }
 
@@ -93,7 +93,7 @@ pub fn load(custom_path: Option<String>) -> (Config, PathBuf) {
         crate::custom_error(
             "no config found",
             "\nconfig file locations:\n ~/.config/nvrs.toml\n ./nvrs.toml\nmake sure the file is not empty".to_string(),
-        );
+         "");
     }
 
     (

src/main.rs

@@ -1,11 +1,16 @@
 use clap::Parser;
 use colored::Colorize;
+use std::sync::Mutex;
 use std::time::{SystemTime, UNIX_EPOCH};
 
 mod api;
 pub mod config;
 mod verfiles;
 
+lazy_static::lazy_static! {
+    static ref MSG_NOEXIT: Mutex<bool> = Mutex::new(false);
+}
+
 #[derive(Parser)]
 #[command(version, about)]
 struct Cli {
@@ -41,6 +46,9 @@ struct Cli {
     )]
     custom_config: Option<String>,
 
+    #[arg(long, help = "Don't exit the program on recoverable errors")]
+    no_fail: bool,
+
     #[arg(long, help = "Display copyright information")]
     copyright: bool,
 }
@@ -49,6 +57,10 @@ struct Cli {
 async fn main() {
     let cli = Cli::parse();
 
+    if cli.no_fail {
+        *MSG_NOEXIT.lock().unwrap() = true;
+    }
+
     if cli.copyright {
         let current_year = 1970
             + (SystemTime::now()
@@ -115,7 +127,7 @@ copies or substantial portions of the Software.",
                     oldver.data.data.insert(package_name, package.1.clone());
                 }
             } else {
-                custom_error_noexit("package not in newver: ", package_name);
+                custom_error("package not in newver: ", package_name, "noexit");
             }
         }
 
@@ -129,7 +141,7 @@ copies or substantial portions of the Software.",
             if config_content.packages.contains_key(&package_name) {
                 config_content.packages.remove(&package_name);
             } else {
-                custom_error_noexit("package not in config: ", package_name.clone());
+                custom_error("package not in config: ", package_name.clone(), "noexit");
             }
             newver.data.data.remove(&package_name);
             oldver.data.data.remove(&package_name);
@@ -144,35 +156,33 @@ copies or substantial portions of the Software.",
 
         for package in config_content.packages {
             if let Some(pkg) = newver.data.data.iter_mut().find(|p| p.0 == &package.0) {
-                let latest = run_source(package.clone())
-                    .await
-                    .unwrap()
-                    .tag_name
-                    .replacen(&package.1.prefix, "", 1);
+                if let Some(latest) = run_source(package.clone()).await {
+                    let latest_tag = latest.tag_name.replacen(&package.1.prefix, "", 1);
 
-                if pkg.1.version != latest {
-                    println!(
-                        "| {} {} -> {}",
-                        package.0.blue(),
-                        pkg.1.version.red(),
-                        latest.green()
-                    );
-                    pkg.1.version = latest;
+                    if pkg.1.version != latest_tag {
+                        println!(
+                            "| {} {} -> {}",
+                            package.0.blue(),
+                            pkg.1.version.red(),
+                            latest_tag.green()
+                        );
+                        pkg.1.version = latest_tag;
+                    }
                 }
             } else {
-                let latest = run_source(package.clone()).await.unwrap();
-
-                let tag = latest.tag_name.replacen(&package.1.prefix, "", 1);
-
-                println!("| {} {} -> {}", package.0.blue(), "NONE".red(), tag.green());
-                newver.data.data.insert(
-                    package.0,
-                    verfiles::Package {
-                        version: tag,
-                        gitref: format!("refs/tags/{}", latest.tag_name),
-                        url: latest.html_url,
-                    },
-                );
+                if let Some(latest) = run_source(package.clone()).await {
+                    let tag = latest.tag_name.replacen(&package.1.prefix, "", 1);
+
+                    println!("| {} {} -> {}", package.0.blue(), "NONE".red(), tag.green());
+                    newver.data.data.insert(
+                        package.0,
+                        verfiles::Package {
+                            version: tag,
+                            gitref: format!("refs/tags/{}", latest.tag_name),
+                            url: latest.html_url,
+                        },
+                    );
+                }
             }
         }
 
@@ -183,17 +193,17 @@ copies or substantial portions of the Software.",
 async fn run_source(package: (String, config::Package)) -> Option<api::Release> {
     let source = package.1.source.clone();
     if let Some(api_used) = api::API_LIST.iter().find(|a| a.name == source) {
-        Some((api_used.func)(package.0, package.1.get_api_arg(api_used.name).unwrap()).await)
+        Some((api_used.func)(package.0, package.1.get_api_arg(api_used.name).unwrap()).await?)
     } else {
-        custom_error("api not found: ", source);
+        custom_error("api not found: ", source, "");
         None
     }
 }
 
-pub fn custom_error(message: &'static str, message_ext: String) {
-    custom_error_noexit(message, message_ext);
-    std::process::exit(1);
-}
-pub fn custom_error_noexit(message: &'static str, message_ext: String) {
-    println!("! {}{}", message.red(), message_ext);
+pub fn custom_error(message: &'static str, message_ext: String, override_exit: &str) {
+    println!("! {}{}", message.red(), message_ext.replace("\n", "\n  "));
+
+    if override_exit != "noexit" && !*MSG_NOEXIT.lock().unwrap() {
+        std::process::exit(1);
+    }
 }

src/verfiles.rs

@@ -36,7 +36,7 @@ pub struct Verfile {
 
 pub fn load(config_table: Option<ConfigTable>) -> Option<(Verfile, Verfile)> {
     if config_table.is_none() {
-        crate::custom_error(CONFIG_NONE_M, CONFIG_NONE_E.to_string());
+        crate::custom_error(CONFIG_NONE_M, CONFIG_NONE_E.to_string(), "");
     }
     let config_table = config_table.unwrap();
 
@@ -50,12 +50,13 @@ pub fn load(config_table: Option<ConfigTable>) -> Option<(Verfile, Verfile)> {
             crate::custom_error(
                 "unsupported verfile version",
                 "\nplease update your verfiles".to_string(),
+                "",
             );
         }
 
         Some((oldver, newver))
     } else {
-        crate::custom_error(XVER_NONE_M, CONFIG_NONE_E.to_string());
+        crate::custom_error(XVER_NONE_M, CONFIG_NONE_E.to_string(), "");
         None
     }
 }