feat: add a http_retries setting to define number of retry attempts

Cargo.toml

@@ -149,6 +149,7 @@ tera = "1"
 terminal_size = "0.4"
 thiserror = "2"
 tokio = { version = "1", features = ["full"] }
+tokio-retry = "0.3"
 toml = { version = "0.8", features = ["parse"] }
 toml_edit = { version = "0.22", features = ["parse"] }
 ubi = { version = "0.7.3", default-features = false }

schema/mise.json

@@ -608,6 +608,11 @@
           "description": "Timeout in seconds for all HTTP requests in mise.",
           "type": "string"
         },
+        "http_retries": {
+          "default": 0,
+          "description": "Number of retries for HTTP requests in mise.",
+          "type": "number"
+        },
         "idiomatic_version_file": {
           "default": true,
           "description": "Set to false to disable the idiomatic version files such as .node-version, .ruby-version, etc.",

settings.toml

@@ -486,6 +486,15 @@ description = "Timeout in seconds for all HTTP requests in mise."
 env = "MISE_HTTP_TIMEOUT"
 type = "Duration"
 
+[http_retries]
+default = 0
+description = "Number of retries for HTTP requests in mise."
+docs = """
+Uses an exponential backoff strategy. The duration is calculated by taking the base (10ms) to the n-th power.
+"""
+env = "MISE_HTTP_RETRIES"
+type = "Integer"
+
 [idiomatic_version_file]
 default = true
 deprecated = "This has been replaced with the idiomatic_version_file_enable_tools setting."

src/http.rs

@@ -7,6 +7,8 @@ use regex::Regex;
 use reqwest::header::{HeaderMap, HeaderValue};
 use reqwest::{ClientBuilder, IntoUrl, Method, Response};
 use std::sync::LazyLock as Lazy;
+use tokio_retry::Retry;
+use tokio_retry::strategy::{ExponentialBackoff, jitter};
 use url::Url;
 
 use crate::cli::version;
@@ -201,11 +203,12 @@ impl Client {
     ) -> Result<()> {
         let url = url.into_url()?;
         debug!("GET Downloading {} to {}", &url, display_path(path));
-
-        let mut resp = self.get_async_with_headers(url, headers).await?;
+        let mut resp = self.get_async_with_headers(url.clone(), headers).await?;
         if let Some(length) = resp.content_length() {
             if let Some(pr) = pr {
+                // Reset progress on each attempt
                 pr.set_length(length);
+                pr.set_position(0);
             }
         }
 
@@ -229,18 +232,29 @@ impl Client {
         headers: &HeaderMap,
         verb_label: &str,
     ) -> Result<Response> {
-        match self
-            .send_once(method.clone(), url.clone(), headers, verb_label)
-            .await
-        {
-            Ok(resp) => Ok(resp),
-            Err(_) if url.scheme() == "http" => {
-                let mut url = url;
-                url.set_scheme("https").unwrap();
-                self.send_once(method, url, headers, verb_label).await
-            }
-            Err(err) => Err(err),
-        }
+        Retry::spawn(
+            default_backoff_strategy(Settings::get().http_retries),
+            || {
+                let method = method.clone();
+                let url = url.clone();
+                let headers = headers.clone();
+                async move {
+                    match self
+                        .send_once(method.clone(), url.clone(), &headers, verb_label)
+                        .await
+                    {
+                        Ok(resp) => Ok(resp),
+                        Err(_err) if url.scheme() == "http" => {
+                            let mut url = url;
+                            url.set_scheme("https").unwrap();
+                            self.send_once(method, url, &headers, verb_label).await
+                        }
+                        Err(err) => Err(err),
+                    }
+                }
+            },
+        )
+        .await
     }
 
     async fn send_once(
@@ -416,6 +430,12 @@ fn display_github_rate_limit(resp: &Response) {
     }
 }
 
+fn default_backoff_strategy(retries: i64) -> impl Iterator<Item = std::time::Duration> {
+    ExponentialBackoff::from_millis(10)
+        .map(jitter)
+        .take(retries.max(0) as usize)
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;