From 79ad3c0a844eb4d2e3720d6777ed0773a87c46ce Mon Sep 17 00:00:00 2001
From: Donald Zou <donaldzou@live.hk>
Date: Thu, 26 Jun 2025 17:56:55 +0800
Subject: [PATCH] Reconstruct Client App UI

---
 src/client.py                                 |  8 ++
 src/modules/DashboardClientsPeerAssignment.py |  1 -
 src/requirements.txt                          |  3 +-
 .../configurationComponents/peer.vue          |  1 +
 .../Configuration/configuration.vue           | 79 +++++++++++++++----
 src/static/client/src/main.js                 | 14 +++-
 src/static/client/src/stores/clientStore.js   | 16 +---
 src/static/client/src/views/index.vue         |  4 +
 8 files changed, 88 insertions(+), 38 deletions(-)

diff --git a/src/client.py b/src/client.py
index 1333f33..1312805 100644
--- a/src/client.py
+++ b/src/client.py
@@ -1,3 +1,5 @@
+from tzlocal import get_localzone
+
 from functools import wraps
 
 from flask import Blueprint, render_template, abort, request, Flask, current_app, session
@@ -91,6 +93,12 @@ def createClientBlueprint(wireguardConfigurations: dict[WireguardConfiguration],
     def ClientIndex():
         return render_template('client.html')
     
+    @client.get(f'{prefix}/api/serverInformation')
+    def ClientAPI_ServerInformation():
+        return ResponseObject(data={
+            "ServerTimezone": str(get_localzone())
+        })
+    
     @client.get(f'{prefix}/api/validateAuthentication')
     @login_required
     def ClientAPI_ValidateAuthentication():
diff --git a/src/modules/DashboardClientsPeerAssignment.py b/src/modules/DashboardClientsPeerAssignment.py
index b53ccdd..9e1e663 100644
--- a/src/modules/DashboardClientsPeerAssignment.py
+++ b/src/modules/DashboardClientsPeerAssignment.py
@@ -3,7 +3,6 @@ import uuid
 from .ConnectionString import ConnectionString
 from .DashboardLogger import DashboardLogger
 import sqlalchemy as db
-
 from .WireguardConfiguration import WireguardConfiguration
 
 
diff --git a/src/requirements.txt b/src/requirements.txt
index 586867f..91905af 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -11,4 +11,5 @@ tcconfig
 sqlalchemy
 sqlalchemy_utils
 psycopg2
-mysqlclient
\ No newline at end of file
+mysqlclient
+tzlocal
\ No newline at end of file
diff --git a/src/static/app/src/components/configurationComponents/peer.vue b/src/static/app/src/components/configurationComponents/peer.vue
index 1ecc1fa..38ace56 100644
--- a/src/static/app/src/components/configurationComponents/peer.vue
+++ b/src/static/app/src/components/configurationComponents/peer.vue
@@ -6,6 +6,7 @@ import PeerSettingsDropdown from "@/components/configurationComponents/peerSetti
 import LocaleText from "@/components/text/localeText.vue";
 import {DashboardConfigurationStore} from "@/stores/DashboardConfigurationStore.js";
 import {GetLocale} from "../../utilities/locale.js";
+
 export default {
 	name: "peer",
 	methods: {GetLocale},
diff --git a/src/static/client/src/components/Configuration/configuration.vue b/src/static/client/src/components/Configuration/configuration.vue
index e1a22d8..c06a69f 100644
--- a/src/static/client/src/components/Configuration/configuration.vue
+++ b/src/static/client/src/components/Configuration/configuration.vue
@@ -1,12 +1,50 @@
 <script setup>
-import {ref} from "vue";
+import {computed, ref} from "vue";
 import ConfigurationQRCode from "@/components/Configuration/configurationQRCode.vue";
-
+import dayjs from "dayjs";
+import Duration from 'dayjs/plugin/Duration'
+dayjs.extend(Duration);
 const props = defineProps([
 	'config'
 ])
-
 const showQRCode = ref(false)
+
+const dateJobs = computed(() => {
+	return props.config.jobs.filter(x => x.Field === 'date').sort((x, y) => {
+		if (dayjs(x.Value).isBefore(y.Value)){
+			return -1
+		}else if (dayjs(x.Value).isAfter(y.Value)){
+			return 1
+		}else{
+			return 0
+		}
+	})
+});
+
+const totalDataJobs = computed(() => {
+	return props.config.jobs.filter(x => x.Field === "total_data").sort((x, y) => {
+		return parseFloat(y.Value) - parseFloat(x.Value)
+	})
+});
+
+const dateLimit = computed(() => {
+	if (dateJobs.value.length > 0){
+		return dateJobs.value[0].Value
+	}
+	return undefined
+})
+const totalDataLimit = computed(() => {
+	if (totalDataJobs.value.length > 0){
+		return totalDataJobs.value[0].Value
+	}
+	return undefined
+})
+
+const totalDataPercentage = computed(() => {
+	if (!totalDataLimit.value) return 100
+	return ( props.config.data / totalDataLimit.value ) * 100
+})
+window.dayjs = dayjs
 </script>
 
 <template>
@@ -21,25 +59,32 @@ const showQRCode = ref(false)
 							{{ props.config.protocol === 'wg' ? 'WireGuard': 'AmneziaWG' }}
 						</span>
 		</div>
-		<div class="card-body p-3">
-			<div class="row gy-2 mb-2">
-				<div class="col-sm text-center">
-					<small class="text-muted mb-2">
+		<div class="card-body p-3 d-flex gap-3 flex-column">
+			<div>
+				<div class="mb-1 d-flex align-items-center">
+					<small class="text-muted ">
 						<i class="bi bi-bar-chart-fill me-1"></i> Data Usage
 					</small>
-					<h6 class="fw-bold ">
-						3.42 / 4.00 GB
-					</h6>
-				</div>
-				<div class="col-sm text-center">
-					<small class="text-muted mb-2">
-						<i class="bi bi-calendar me-1"></i> Valid Until
+					<small class="fw-bold ms-sm-auto">
+						{{ props.config.data.toFixed(4) }} / {{ totalDataLimit ? parseFloat(totalDataLimit).toFixed(4) : 'Unlimited'}} GB
 					</small>
-					<h6 class="fw-bold ">
-						3.42 / 4.00 GB
-					</h6>
+				</div>
+				<div class="progress" role="progressbar" aria-valuenow="25" aria-valuemin="0" aria-valuemax="100" style="height: 6px">
+					<div class="progress-bar bg-primary"
+					     :style="{'width': '' + totalDataPercentage + '%'}"></div>
 				</div>
 			</div>
+			<div>
+				<div class="mb-1 d-flex align-items-center">
+					<small class="text-muted">
+						<i class="bi bi-calendar me-1"></i> Valid Until
+					</small>
+					<small class="fw-bold ms-auto">
+						{{ dateLimit ? dateLimit : 'Unlimited Time' }}
+					</small>
+				</div>
+			</div>
+
 			<button class="btn btn-outline-body rounded-3 flex-grow-1 fw-bold w-100" @click="showQRCode = true">
 				<i class="bi bi-link-45deg me-2"></i><small>Connect</small>
 			</button>
diff --git a/src/static/client/src/main.js b/src/static/client/src/main.js
index 5aeee18..119afb2 100644
--- a/src/static/client/src/main.js
+++ b/src/static/client/src/main.js
@@ -6,8 +6,14 @@ import router from "@/router/router.js";
 import {createPinia} from "pinia";
 
 import 'bootstrap/dist/js/bootstrap.bundle.js'
+import {clientStore} from "@/stores/clientStore.js";
 
-createApp(App)
-	.use(createPinia())
-	.use(router)
-	.mount('#app')
+const app = createApp(App)
+
+app.use(createPinia())
+const store = clientStore()
+await fetch("/client/api/serverInformation").then(res => res.json()).then(res => store.serverInformation = res.data)
+
+
+app.use(router)
+app.mount("#app")
\ No newline at end of file
diff --git a/src/static/client/src/stores/clientStore.js b/src/static/client/src/stores/clientStore.js
index e92390e..c05aa29 100644
--- a/src/static/client/src/stores/clientStore.js
+++ b/src/static/client/src/stores/clientStore.js
@@ -7,6 +7,7 @@ import {axiosGet} from "@/utilities/request.js";
 
 export const clientStore = defineStore('clientStore',  {
 	state: () => ({
+		serverInformation: {},
 		notifications: [],
 		configurations: [],
 		clientProfile: {
@@ -36,21 +37,6 @@ export const clientStore = defineStore('clientStore',  {
 			const data = await axiosGet("/api/configurations")
 			if (data){
 				this.configurations = data.data
-				this.configurations.forEach(c => {
-					console.log(
-						c.jobs.sort((x, y) => {
-							if (dayjs(x.CreationDate).isBefore(y.CreationDate)){
-								return 1
-							}else if (dayjs(x.CreationDate).isAfter(y.CreationDate)){
-								return -1
-							}else{
-								return 0
-							}
-						})
-
-					)
-					console.log(c.jobs.find(x => x.Field === 'date'))
-				})
 			}else{
 				this.newNotification("Failed to fetch configurations", "danger")
 			}
diff --git a/src/static/client/src/views/index.vue b/src/static/client/src/views/index.vue
index 960a0d1..5b5efc5 100644
--- a/src/static/client/src/views/index.vue
+++ b/src/static/client/src/views/index.vue
@@ -13,6 +13,10 @@ const configurations = computed(() => {
 onMounted(async () => {
 	await store.getConfigurations()
 	loading.value = false;
+
+	setInterval(async () => {
+		await store.getConfigurations()
+	}, 5000)
 })
 </script>