From f82524fd154a1bf4a7277db868147afdb0cd507a Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 27 Jun 2024 12:30:09 +0800
Subject: feat: implement native rag url loader (#660)

---
 src/utils/mod.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'src/utils')

diff --git a/src/utils/mod.rs b/src/utils/mod.rs
index 7d1d07a..76ab208 100644
--- a/src/utils/mod.rs
+++ b/src/utils/mod.rs
@@ -14,6 +14,7 @@ pub use self::prompt_input::*;
 pub use self::render_prompt::render_prompt;
 pub use self::spinner::{create_spinner, Spinner};
 
+use anyhow::{Context, Result};
 use fancy_regex::Regex;
 use is_terminal::IsTerminal;
 use lazy_static::lazy_static;
@@ -180,6 +181,28 @@ pub fn safe_join_path<T1: AsRef<Path>, T2: AsRef<Path>>(
     }
 }
 
+pub fn set_proxy(
+    builder: reqwest::ClientBuilder,
+    proxy: Option<&String>,
+) -> Result<reqwest::ClientBuilder> {
+    let proxy = if let Some(proxy) = proxy {
+        if proxy.is_empty() || proxy == "-" {
+            return Ok(builder);
+        }
+        proxy.clone()
+    } else if let Some(proxy) = ["HTTPS_PROXY", "https_proxy", "ALL_PROXY", "all_proxy"]
+        .into_iter()
+        .find_map(|v| env::var(v).ok())
+    {
+        proxy
+    } else {
+        return Ok(builder);
+    };
+    let builder = builder
+        .proxy(reqwest::Proxy::all(&proxy).with_context(|| format!("Invalid proxy `{proxy}`"))?);
+    Ok(builder)
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
-- 
cgit v1.2.3