From 5e19f0af4759157415bc006b77daf6fb6f37b38f Mon Sep 17 00:00:00 2001
From: Wizzard <rich@bandaholics.cash>
Date: Sun, 16 Mar 2025 00:07:25 -0400
Subject: [PATCH] Update websocket.rs and script.js

---
 src/websocket.rs   | 23 +++++++++++++++++------
 webradar/script.js | 34 +++++++++++++++++++++++++++++++++-
 2 files changed, 50 insertions(+), 7 deletions(-)

diff --git a/src/websocket.rs b/src/websocket.rs
index 0ac6459..f9b0a83 100644
--- a/src/websocket.rs
+++ b/src/websocket.rs
@@ -10,7 +10,7 @@ use std::io::Write;
 use tokio::sync::RwLock;
 use tower_http::services::ServeDir;
 
-use crate::comms::RadarData;
+use crate::comms::{RadarData, ArcRwlockRadarData};
 
 #[derive(Clone)]
 struct AppState {
@@ -34,14 +34,25 @@ async fn handle_socket(mut socket: WebSocket, state: AppState) {
                     if let Ok(json) = serde_json::to_string(&*radar_data) {
                         compression_buffer.clear();
 
-                        let mut encoder = GzEncoder::new(Vec::new(), Compression::fast());
+                        let compression_level = if json.len() > 10000 {
+                            Compression::best()
+                        } else {
+                            Compression::fast()
+                        };
+
+                        let mut encoder = GzEncoder::new(Vec::new(), compression_level);
                         if encoder.write_all(json.as_bytes()).is_ok() {
                             match encoder.finish() {
                                 Ok(compressed) => {
-                                    let mut message = vec![0x01];
-                                    message.extend_from_slice(&compressed);
-
-                                    let _ = socket.send(Message::Binary(message)).await;
+                                    if compressed.len() < json.len() {
+                                        let mut message = vec![0x01];
+                                        message.extend_from_slice(&compressed);
+                                        let _ = socket.send(Message::Binary(message)).await;
+                                    } else {
+                                        let mut uncompressed = vec![0x00];
+                                        uncompressed.extend_from_slice(json.as_bytes());
+                                        let _ = socket.send(Message::Binary(uncompressed)).await;
+                                    }
                                 },
                                 Err(_) => {
                                     let mut uncompressed = vec![0x00];
diff --git a/webradar/script.js b/webradar/script.js
index 8ec307c..34eb4ed 100644
--- a/webradar/script.js
+++ b/webradar/script.js
@@ -68,6 +68,35 @@ const websocketAddr = location.protocol === 'https:'
 const clamp = (num, min, max) => Math.min(Math.max(num, min), max);
 const degreesToRadians = (degrees) => degrees * (Math.PI / 180);
 
+const pingTracker = {
+    history: [],
+    lastRequestTime: 0,
+    maxSamples: 10,
+
+    startRequest: function () {
+        this.lastRequestTime = performance.now();
+    },
+
+    endRequest: function () {
+        if (this.lastRequestTime === 0) return;
+
+        const ping = performance.now() - this.lastRequestTime;
+        this.history.push(ping);
+
+        if (this.history.length > this.maxSamples) {
+            this.history.shift();
+        }
+
+        this.lastRequestTime = 0;
+    },
+
+    getAveragePing: function () {
+        if (this.history.length === 0) return 0;
+        const sum = this.history.reduce((a, b) => a + b, 0);
+        return sum / this.history.length;
+    }
+};
+
 function render() {
     requestAnimationFrame(render);
 
@@ -83,6 +112,7 @@ function render() {
 
     if (!isRequestPending && websocket && websocket.readyState === WebSocket.OPEN) {
         isRequestPending = true;
+        pingTracker.startRequest();
         websocket.send("requestInfo");
     }
 
@@ -117,7 +147,7 @@ function renderFrame() {
         ctx.font = "16px Arial";
         ctx.textAlign = "left";
         ctx.fillStyle = "#00FF00";
-        ctx.fillText(`${currentFps} FPS | ${freq} Hz`, 10, 20);
+        ctx.fillText(`${currentFps} FPS | ${freq} Hz | Ping: ${Math.round(pingTracker.getAveragePing())}ms`, 10, 20);
     }
 }
 
@@ -753,6 +783,8 @@ function processData(data) {
 
 function decompressData(data) {
     try {
+        pingTracker.endRequest();
+
         if (data[0] === 0x01) {
             try {
                 if (typeof pako === 'undefined') {