feat: add GitHub API proxy support with fallback

QaidVoid · QaidVoid · commit ac5911ecaee5 · 2026-03-22T10:24:29.000+05:45
diff --git a/Cargo.toml b/Cargo.toml
@@ -4,7 +4,7 @@ version = "0.1.0"
 edition = "2024"
 
 [dependencies]
-clap = { version = "4.6.0", features = ["derive"] }
+clap = { version = "4.6.0", features = ["derive", "env"] }
 hex = "0.4.3"
 serde = { version = "1.0.228", features = ["derive"] }
 serde_json = "1.0.149"
diff --git a/src/config.rs b/src/config.rs
@@ -0,0 +1,34 @@
+use std::sync::OnceLock;
+
+static GITHUB_API_PROXIES: OnceLock<Vec<String>> = OnceLock::new();
+
+pub fn init(proxies: Option<String>) {
+    let proxies = proxies
+        .or_else(|| std::env::var("GITHUB_API_PROXY").ok())
+        .map(|s| {
+            s.split(',')
+                .map(|s| s.trim().to_string())
+                .filter(|s| !s.is_empty())
+                .collect()
+        })
+        .unwrap_or_default();
+
+    let _ = GITHUB_API_PROXIES.set(proxies);
+}
+
+pub fn get_proxies() -> &'static [String] {
+    GITHUB_API_PROXIES
+        .get()
+        .map(|v| v.as_slice())
+        .unwrap_or(&[])
+}
+
+pub fn build_api_url(path: &str, proxy: Option<&str>) -> String {
+    match proxy {
+        Some(proxy) => {
+            let proxy = proxy.trim_end_matches('/');
+            format!("{}{}", proxy, path)
+        }
+        None => format!("https://api.github.com{}", path),
+    }
+}
diff --git a/src/lib.rs b/src/lib.rs
@@ -1,4 +1,5 @@
 pub mod appimage;
+pub mod config;
 pub mod error;
 pub mod update_info;
 pub mod updater;
diff --git a/src/main.rs b/src/main.rs
@@ -3,6 +3,7 @@ use std::path::PathBuf;
 use clap::Parser;
 use clap::Subcommand;
 
+use appimageupdate::config;
 use appimageupdate::{Error, Updater};
 
 #[derive(Parser)]
@@ -25,11 +26,17 @@ enum Commands {
 
         #[arg(short, long, value_name = "DIR")]
         output: Option<PathBuf>,
+
+        #[arg(long, value_name = "URL", env = "GITHUB_API_PROXY")]
+        github_api_proxy: Option<String>,
     },
 
     Check {
         #[arg(value_name = "APPIMAGE")]
         path: PathBuf,
+
+        #[arg(long, value_name = "URL", env = "GITHUB_API_PROXY")]
+        github_api_proxy: Option<String>,
     },
 }
 
@@ -64,7 +71,10 @@ fn run(cli: Cli) -> Result<(), Error> {
             path,
             overwrite,
             output,
+            github_api_proxy,
         } => {
+            config::init(github_api_proxy);
+
             let mut updater = Updater::new(&path)?;
 
             if overwrite {
@@ -120,7 +130,12 @@ fn run(cli: Cli) -> Result<(), Error> {
                 println!("Already up to date!");
             }
         }
-        Commands::Check { path } => {
+        Commands::Check {
+            path,
+            github_api_proxy,
+        } => {
+            config::init(github_api_proxy);
+
             let updater = Updater::new(&path)?;
 
             let source_path = updater.source_path();
diff --git a/src/update_info/github.rs b/src/update_info/github.rs
@@ -2,6 +2,7 @@ use std::cell::OnceCell;
 
 use serde::Deserialize;
 
+use crate::config;
 use crate::error::{Error, Result};
 
 #[derive(Debug, Clone)]
@@ -46,21 +47,37 @@ impl GitHubUpdateInfo {
     }
 
     fn resolve_url(&self) -> Result<String> {
-        let api_url = match self.tag.as_str() {
-            "latest" => format!(
-                "https://api.github.com/repos/{}/{}/releases/latest",
-                self.username, self.repo
-            ),
-            "latest-pre" | "latest-all" => format!(
-                "https://api.github.com/repos/{}/{}/releases",
-                self.username, self.repo
-            ),
+        let api_path = match self.tag.as_str() {
+            "latest" => format!("/repos/{}/{}/releases/latest", self.username, self.repo),
+            "latest-pre" | "latest-all" => {
+                format!("/repos/{}/{}/releases", self.username, self.repo)
+            }
             tag => format!(
-                "https://api.github.com/repos/{}/{}/releases/tags/{}",
+                "/repos/{}/{}/releases/tags/{}",
                 self.username, self.repo, tag
             ),
         };
 
+        let proxies = config::get_proxies();
+        let mut last_error = None;
+
+        if proxies.is_empty() {
+            return self.fetch_release_url(&api_path, None);
+        }
+
+        for proxy in proxies {
+            match self.fetch_release_url(&api_path, Some(proxy)) {
+                Ok(url) => return Ok(url),
+                Err(e) => last_error = Some(e),
+            }
+        }
+
+        Err(last_error.unwrap_or_else(|| Error::GitHubApi("All proxies failed".into())))
+    }
+
+    fn fetch_release_url(&self, api_path: &str, proxy: Option<&str>) -> Result<String> {
+        let api_url = config::build_api_url(api_path, proxy);
+
         let response = ureq::get(&api_url)
             .header("User-Agent", "appimageupdate-rs")
             .call()

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`pub mod appimage;`
	`2`	`+pub mod config;`
`2`	`3`	`pub mod error;`
`3`	`4`	`pub mod update_info;`
`4`	`5`	`pub mod updater;`