From 394e4058912ae670bf138f2e76daf8fd0d6b942b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tobias=20Fr=C3=B6lich?= <tobias@blueshoe.de>
Date: Tue, 4 Jul 2023 11:05:55 +0200
Subject: [PATCH] feat: add sunset date

---
 src-tauri/Cargo.lock            |  2 +-
 src-tauri/src/heartbeat.rs      | 21 ++++++++-------------
 src/components/ClusterTable.vue | 22 ++++++++++++++++++++--
 3 files changed, 29 insertions(+), 16 deletions(-)

diff --git a/src-tauri/Cargo.lock b/src-tauri/Cargo.lock
index adadc0f..a7cec4a 100644
--- a/src-tauri/Cargo.lock
+++ b/src-tauri/Cargo.lock
@@ -152,7 +152,7 @@ checksum = "604178f6c5c21f02dc555784810edfb88d34ac2c73b2eae109655649ee73ce3d"
 
 [[package]]
 name = "beiboot_desktop"
-version = "0.0.2"
+version = "0.0.3"
 dependencies = [
  "bollard",
  "futures",
diff --git a/src-tauri/src/heartbeat.rs b/src-tauri/src/heartbeat.rs
index 02d44ce..cfc372e 100644
--- a/src-tauri/src/heartbeat.rs
+++ b/src-tauri/src/heartbeat.rs
@@ -1,17 +1,12 @@
-use tokio::time::{self, Duration};
 use beiboot_desktop::connection::ConnectError;
 
-pub async fn establish_heartbeat_connection(cluster_id: &str, token: &str) -> Result<(), ConnectError> {
-    let mut interval = time::interval(Duration::from_secs(250));
-    loop {
-        let resp = reqwest::Client::new()
-            .post(format!("https://api.getdeck.dev/clusters/{}/heartbeat", cluster_id).as_str())
-            .header(reqwest::header::AUTHORIZATION, format!("Bearer {}", token))
-            .send()
-            .await
-            .expect("Failed to send");
-        println!("Heartbeat status: {}", resp.status());
-        interval.tick().await;
-    } 
+pub async fn establish_heartbeat_connection(cluster_id: &str, token: &str) -> Result<String, ConnectError> {
+    let resp = reqwest::Client::new()
+        .post(format!("https://api.getdeck.dev/clusters/{}/heartbeat", cluster_id).as_str())
+        .header(reqwest::header::AUTHORIZATION, format!("Bearer {}", token))
+        .send()
+        .await
+        .expect("Failed to send");
+    Ok(format!("Heartbeat status: {}", resp.status()))
 }
 
diff --git a/src/components/ClusterTable.vue b/src/components/ClusterTable.vue
index e1ca978..f7bb7e8 100644
--- a/src/components/ClusterTable.vue
+++ b/src/components/ClusterTable.vue
@@ -4,6 +4,8 @@
             <tr>
                 <th>Name</th>
                 <th>State</th>
+                <th>Sunset</th>
+                <th>Lifetime/Timeout</th>
                 <th>Actions</th>
             </tr>
         </thead>
@@ -16,6 +18,8 @@
                         {{ cluster.state }}
                     </v-chip>
                 </td>
+                <td>{{ getClusterSunset(cluster.sunset) }}</td>
+                <td>{{ cluster.max_lifetime.value }} / {{ cluster.max_session_timeout.value }}</td>
                 <td>
                     <v-tooltip text="Connect">
                     <template v-slot:activator="{ props }">
@@ -92,7 +96,6 @@ const getChipColor = (state: string) => {
 const getClusterList = () => {
     ClustersService.clusterListClustersGet().then((res) => {
         clusterList.value = res.items;
-        console.log(res.items);
     });
 };
 
@@ -137,6 +140,7 @@ const clusterDisconnect = (clusterName: string) => {
       store.connection.clusterName = "";
       store.connection.kubeconfigPath = "";
       store.connection.connected = false;
+      store.connection.clusterId = "";
     }).catch((err) => {
         console.log(err);
     });
@@ -158,7 +162,21 @@ onMounted(() => {
     getClusterList();
     setInterval(() => {
         getClusterList();
-        checkRunningConnects().then((res) => console.log(res));
+        checkRunningConnects();
+        if (store.connection.clusterId) {
+          invoke("establish_heartbeat_connection", {
+            clusterId: store.connection.clusterId,
+            // @ts-ignore
+            token: store.auth.token
+        }).catch(err => console.error(err));
+
+        }
     }, 5000);
 });
+
+const getClusterSunset = (sunset: string) => {
+  if (sunset) {
+    return new Date(`${sunset}Z`).toLocaleTimeString()
+  }
+};
 </script>