From c0ebb98ec9ef8a5b062fc520dab34eff0ecfa3ff Mon Sep 17 00:00:00 2001
From: Reactor Scram <ReactorScram@users.noreply.github.com>
Date: Tue, 16 Jul 2024 09:49:12 -0500
Subject: [PATCH] chore(gui-client): measure startup time (#5864)

Refs #5026
---
 rust/gui-client/src-tauri/src/client/gui.rs | 23 ++++++++++++++-------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/rust/gui-client/src-tauri/src/client/gui.rs b/rust/gui-client/src-tauri/src/client/gui.rs
index fefa96477..bdd8a98af 100644
--- a/rust/gui-client/src-tauri/src/client/gui.rs
+++ b/rust/gui-client/src-tauri/src/client/gui.rs
@@ -12,7 +12,11 @@ use anyhow::{anyhow, bail, Context, Result};
 use connlib_client_shared::callbacks::ResourceDescription;
 use firezone_headless_client::IpcServerMsg;
 use secrecy::{ExposeSecret, SecretString};
-use std::{path::PathBuf, str::FromStr, time::Duration};
+use std::{
+    path::PathBuf,
+    str::FromStr,
+    time::{Duration, Instant},
+};
 use system_tray::Event as TrayMenuEvent;
 use tauri::{Manager, SystemTrayEvent};
 use tokio::sync::{mpsc, oneshot};
@@ -440,7 +444,7 @@ enum Status {
     /// Firezone is disconnected.
     Disconnected,
     /// Firezone is signing in and raising the tunnel.
-    Connecting,
+    Connecting { start_instant: Instant },
     /// Firezone is ready to use.
     TunnelReady { resources: Vec<ResourceDescription> },
 }
@@ -456,7 +460,7 @@ impl Status {
     fn connlib_is_up(&self) -> bool {
         match self {
             Self::Disconnected => false,
-            Self::Connecting => true,
+            Self::Connecting { .. } => true,
             Self::TunnelReady { .. } => true,
         }
     }
@@ -487,10 +491,11 @@ impl Controller {
         let api_url = self.advanced_settings.api_url.clone();
         tracing::info!(api_url = api_url.to_string(), "Starting connlib...");
 
+        let start_instant = Instant::now();
         self.ipc_client
             .connect_to_firezone(api_url.as_str(), token)
             .await?;
-        self.status = Status::Connecting;
+        self.status = Status::Connecting { start_instant };
         self.refresh_system_tray_menu()?;
 
         ran_before::set().await?;
@@ -582,7 +587,7 @@ impl Controller {
                         tracing::info!("Calling `sign_out` to cancel sign-in");
                         self.sign_out().await?;
                     }
-                    Status::Connecting => {
+                    Status::Connecting { start_instant: _ } => {
                         tracing::warn!(
                             "Connlib is already raising the tunnel, calling `sign_out` anyway"
                         );
@@ -665,13 +670,15 @@ impl Controller {
                     return Ok(());
                 }
                 tracing::debug!(len = resources.len(), "Got new Resources");
-                if !matches!(self.status, Status::TunnelReady { .. }) {
+                if let Status::Connecting { start_instant } =
+                    std::mem::replace(&mut self.status, Status::TunnelReady { resources })
+                {
+                    tracing::info!(elapsed = ?start_instant.elapsed(), "Tunnel ready");
                     os::show_notification(
                         "Firezone connected",
                         "You are now signed in and able to access resources.",
                     )?;
                 }
-                self.status = Status::TunnelReady { resources };
                 if let Err(error) = self.refresh_system_tray_menu() {
                     tracing::error!(?error, "Failed to refresh Resource list");
                 }
@@ -695,7 +702,7 @@ impl Controller {
                     tracing::error!("We have an auth session but no connlib session");
                     system_tray::Menu::SignedOut
                 }
-                Status::Connecting => system_tray::Menu::WaitingForConnlib,
+                Status::Connecting { start_instant: _ } => system_tray::Menu::WaitingForConnlib,
                 Status::TunnelReady { resources } => system_tray::Menu::SignedIn {
                     actor_name: &auth_session.actor_name,
                     resources,