refactor: use static base url for tree sitter

amirabbas-gh · amirabbas-gh · commit d6ed997143c8 · 2025-07-21T21:40:15.000+03:30
diff --git a/crates/codemod-sandbox/Cargo.toml b/crates/codemod-sandbox/Cargo.toml
@@ -2,6 +2,7 @@
 name = "codemod-sandbox"
 version.workspace = true
 edition = "2021"
+build = "src/build.rs"
 
 [dependencies]
 rquickjs-registry = { package = "rquickjs", version = "0.9.0", default-features = false, optional = true }
diff --git a/crates/codemod-sandbox/src/build.rs b/crates/codemod-sandbox/src/build.rs
@@ -0,0 +1,5 @@
+fn main() {
+    let url = std::env::var("TREE_SITTER_BASE_URL")
+        .unwrap_or_else(|_| "https://tree-sitter-parsers.s3.us-east-1.amazonaws.com".to_string());
+    println!("cargo:rustc-env=TREE_SITTER_BASE_URL={url}");
+}
diff --git a/crates/codemod-sandbox/src/tree_sitter/mod.rs b/crates/codemod-sandbox/src/tree_sitter/mod.rs
@@ -17,6 +17,8 @@ struct ReadyLang {
     lib_path: PathBuf,
 }
 
+pub const BASE_URL: &str = env!("TREE_SITTER_BASE_URL");
+
 pub async fn load_tree_sitter(languages: &[SupportedLanguage]) -> Result<Vec<DynamicLang>, String> {
     let mut ready_langs = HashSet::new();
     for language in languages {
@@ -54,10 +56,10 @@ pub async fn load_tree_sitter(languages: &[SupportedLanguage]) -> Result<Vec<Dyn
                 std::fs::create_dir_all(parent)
                     .map_err(|e| format!("Failed to create directory: {e}"))?;
             }
-            let base_url = std::env::var("TREE_SITTER_BASE_URL").unwrap_or_else(|_| {
-                "https://tree-sitter-parsers.s3.us-east-1.amazonaws.com".to_string()
-            });
-            let url = format!("{base_url}/tree-sitter/parsers/tree-sitter-{language}/latest/{os}-{arch}.{extension}");
+            println!("base_url: {BASE_URL}");
+            let url = format!(
+                "{BASE_URL}/tree-sitter/parsers/tree-sitter-{language}/latest/{os}-{arch}.{extension}"
+            );
 
             let client = reqwest::Client::builder()
                 .timeout(std::time::Duration::from_secs(30))
