add env variables to match other sdks

Signed-off-by: Jess Frazelle <[email protected]>
KittyCAD · Oct 25, 2024 · 419c168 · 419c168
1 parent 96d68a2
commit 419c168
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "zoo-kcl"
-version = "0.1.42"
+version = "0.1.43"
 edition = "2021"
 repository = "https://github.com/kittycad/kcl.py"
 

diff --git a/src/lib.rs b/src/lib.rs
@@ -198,12 +198,22 @@ async fn new_context(units: UnitLength) -> Result<ExecutorContext> {
         .tcp_keepalive(std::time::Duration::from_secs(600))
         .http1_only();
 
-    let token = std::env::var("KITTYCAD_API_TOKEN").expect("KITTYCAD_API_TOKEN not set");
+    let token = if let Ok(token) = std::env::var("KITTYCAD_API_TOKEN") {
+        token
+    } else if let Ok(token) = std::env::var("ZOO_API_TOKEN") {
+        token
+    } else {
+        return Err(anyhow::anyhow!(
+            "No API token found in environment variables. Use KITTYCAD_API_TOKEN or ZOO_API_TOKEN"
+        ));
+    };
 
     // Create the client.
     let mut client = kittycad::Client::new_from_reqwest(token, http_client, ws_client);
     // Set a local engine address if it's set.
-    if let Ok(addr) = std::env::var("KITTYCAD_HOST") {
+    if let Ok(addr) = std::env::var("ZOO_HOST") {
+        client.set_base_url(addr);
+    } else if let Ok(addr) = std::env::var("KITTYCAD_HOST") {
         client.set_base_url(addr);
     }