From 634297772b0398ac46dbf3ef312b267b0db6f282 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Augusto=20C=C3=A9sar?= <augusto.silva@mentimeter.com>
Date: Wed, 30 Oct 2024 14:49:02 +0100
Subject: [PATCH] feat: check running services when using paid tunnel (#113)

When we run free tunnels, we check if there is any local service already
running to warn the user about it. We are currently not doing this for
paid tunnels as well.

This PR adds the same check for paid tunnels.
---
 Cargo.lock              |  2 +-
 linkup-cli/Cargo.toml   |  2 +-
 linkup-cli/src/start.rs | 22 ++++++++++++++++------
 3 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 470c82a..d77ffbf 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -1091,7 +1091,7 @@ dependencies = [
 
 [[package]]
 name = "linkup-cli"
-version = "1.2.0"
+version = "1.2.1"
 dependencies = [
  "base64",
  "clap",
diff --git a/linkup-cli/Cargo.toml b/linkup-cli/Cargo.toml
index 069b614..1700710 100644
--- a/linkup-cli/Cargo.toml
+++ b/linkup-cli/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "linkup-cli"
-version = "1.2.0"
+version = "1.2.1"
 edition = "2021"
 
 [[bin]]
diff --git a/linkup-cli/src/start.rs b/linkup-cli/src/start.rs
index 91a86b1..2ee66bf 100644
--- a/linkup-cli/src/start.rs
+++ b/linkup-cli/src/start.rs
@@ -3,6 +3,8 @@ use std::{
     path::{Path, PathBuf},
 };
 
+use colored::Colorize;
+
 use crate::{
     background_booting::{BackgroundServices, RealBackgroundServices},
     env_files::write_to_env_file,
@@ -108,6 +110,8 @@ fn start_paid_tunnel(
         boot.boot_local_dns(state.domain_strings(), state.linkup.session_name.clone())?;
     }
 
+    check_local_not_started(&state)?;
+
     Ok(())
 }
 
@@ -121,9 +125,10 @@ fn start_free_tunnel(state: LocalState, no_tunnel: bool) -> Result<(), CliError>
     }
 
     let background_service = RealBackgroundServices {};
-    background_service.boot_background_services(state)?;
+    let state = background_service.boot_background_services(state)?;
+
+    check_local_not_started(&state)?;
 
-    check_local_not_started()?;
     Ok(())
 }
 
@@ -195,14 +200,19 @@ fn set_service_env(directory: String, config_path: String) -> Result<(), CliErro
     Ok(())
 }
 
-fn check_local_not_started() -> Result<(), CliError> {
-    let state = LocalState::load()?;
-    for service in state.services {
+fn check_local_not_started(state: &LocalState) -> Result<(), CliError> {
+    for service in &state.services {
         if service.local == service.remote {
             continue;
         }
+
         if server_status(service.local.to_string(), None) == ServerStatus::Ok {
-            println!("⚠️  Service {} is already running locally!! You need to restart it for linkup's environment variables to be loaded.", service.name);
+            let warning = format!(
+                "⚠️  Service {} is already running locally!! You need to restart it for linkup's environment variables to be loaded.",
+                service.name
+            ).yellow();
+
+            println!("{}", warning);
         }
     }
     Ok(())