add websocket client heartbeat

Author: timzaak

README.md

@@ -15,7 +15,7 @@ If you want to get all requests and responses about `https://www.exmample.com (1
 2. run commands below:
 ```shell
 # run with sbt
-sbt "runMain Main --dns=192.168.3.3:www.example.com --jks-path=jks.jks --jks-password=123456 --websocketPort=9000"
+sbt "runMain Main --dns=192.168.3.3:www.example.com --jks-path=jks.jks --jks-password=123456 --viewPort=9000"
 
 
 # run with docker

build.sbt

@@ -5,7 +5,7 @@ lazy val root = project
   .in(file("."))
   .settings(
     name := "log-http-proxy",
-    version := "0.2.2",
+    version := "0.2.3",
     scalaVersion := scala3Version,
     libraryDependencies ++=
       tapir ++ logLib ++ configLib ++

src/main/scala/Main.scala

@@ -14,7 +14,7 @@ object Main {
     @arg(doc = "jks file path") jksPath: Option[String],
     @arg(doc = "jks password") jksPassword: Option[String],
     @arg(doc = "example: 1.1.1.1,8.8.8.8") resolver: Option[String],
-    @arg(doc = "websocket port, output log via websocket, if not set, output log to cmd") websocketPort: Option[Int],
+    @arg(doc = "websocket port, output log via websocket, if not set, output log to cmd") viewerPort: Option[Int],
   ): Unit = {
     var config = AppConfig.load()
 
@@ -31,7 +31,7 @@ object Main {
         dns = dnsPairs.map((domain,ip) => DNS(ip= ip ,domain = domain)),
         resolver = resolver.map(_.split(',').toList).getOrElse(config.resolver),
         jks = jksConf.orElse(config.jks),
-        viewerPort = websocketPort.orElse(config.viewerPort),
+        viewerPort = viewerPort.orElse(config.viewerPort),
       )
 
     if(config.resolver.nonEmpty) {

src/main/scala/com/timzaak/proxy/LogWebViewer.scala

@@ -61,7 +61,10 @@ class LogWebViewer(certPath: Option[String])(using system: ActorSystem) {
         const ws_protocol = window.location.protocol === "https:" ? "wss:" : "ws:";
         const ws_host = window.location.host;
         const socket = new WebSocket(`$${ws_protocol}//$${ws_host}/api_ws?ip=$${client_ip}`);
-
+        window.socket = socket
+        let intervalId = setInterval(function() {
+            socket.send("!");
+        }, 30000);
         socket.onopen = function(event) {
             console.log("WebSocket connection established, please start sending request, make sure you have changed your host config");
         };
@@ -71,10 +74,14 @@ class LogWebViewer(certPath: Option[String])(using system: ActorSystem) {
         };
 
         socket.onerror = function(error) {
+            intervalId && clearInterval(intervalId);
+            delete intervalId;
             console.error("WebSocket Error: ", error);
         };
 
         socket.onclose = function(event) {
+            intervalId && clearInterval(intervalId);
+            delete intervalId;
             console.log("WebSocket connection closed");
         };
     </script>