fix: set linkup env vars in .envs

mentimeter · Jun 10, 2024 · 58f4937 · 58f4937
1 parent 00f9f2c
commit 58f4937
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 9 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/linkup-cli/Cargo.toml b/linkup-cli/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "linkup-cli"
-version = "1.0.2"
+version = "1.0.3"
 edition = "2021"
 
 [[bin]]

diff --git a/linkup-cli/src/start.rs b/linkup-cli/src/start.rs
@@ -23,6 +23,7 @@ pub fn start(config_arg: &Option<String>, no_tunnel: bool) -> Result<(), CliErro
     env_logger::init();
     let is_paid = use_paid_tunnels();
     let state = load_and_save_state(config_arg, no_tunnel, is_paid)?;
+    set_linkup_env(state.clone())?;
     if is_paid {
         start_paid_tunnel(
             &RealSystem,
@@ -37,6 +38,17 @@ pub fn start(config_arg: &Option<String>, no_tunnel: bool) -> Result<(), CliErro
     Ok(())
 }
 
+fn set_linkup_env(state: LocalState) -> Result<(), CliError> {
+    // Set env vars to linkup
+    for service in &state.services {
+        match &service.directory {
+            Some(d) => set_service_env(d.clone(), state.linkup.config_path.clone())?,
+            None => {}
+        }
+    }
+    Ok(())
+}
+
 fn use_paid_tunnels() -> bool {
     env::var("LINKUP_CLOUDFLARE_ACCOUNT_ID").is_ok()
         && env::var("LINKUP_CLOUDFLARE_ZONE_ID").is_ok()
@@ -102,13 +114,6 @@ fn start_paid_tunnel(
 
 fn start_free_tunnel(state: LocalState, no_tunnel: bool) -> Result<(), CliError> {
     println!("Starting free tunnel");
-    // Set env vars to linkup
-    for service in &state.services {
-        match &service.directory {
-            Some(d) => set_service_env(d.clone(), state.linkup.config_path.clone())?,
-            None => {}
-        }
-    }
 
     if no_tunnel && !linkup_file_path(LINKUP_LOCALDNS_INSTALL).exists() {
         println!("Run `linkup local-dns install` before running without a tunnel");