feat: Refactor WebSocketPushSender to improve handler retrieval and add session check

Signed-off-by: Mario Serrano <mario@dynamiasoluciones.com>

Author: marioserrano09

zk/src/main/java/tools/dynamia/zk/websocket/WebSocketPushSender.java

@@ -58,7 +58,7 @@ public static boolean sendPushCommand(Desktop desktop, String command) {
      * @return true if command is sended successfull. Check log for false response
      */
     public static boolean sendPushCommand(Desktop desktop, String command, Map<String, Object> payload) {
-        WebSocketGlobalCommandHandler handler = Containers.get().findObject(WebSocketGlobalCommandHandler.class);
+        WebSocketGlobalCommandHandler handler = getHandler();
         if (handler != null) {
             try {
                 Map<String, Object> data = new HashMap<>();
@@ -99,7 +99,7 @@ public static boolean sendPushCommand(String command) {
      *
      */
     public static void broadcastCommand(String command) {
-        WebSocketGlobalCommandHandler handler = Containers.get().findObject(WebSocketGlobalCommandHandler.class);
+        WebSocketGlobalCommandHandler handler = getHandler();
         if (handler != null) {
             handler.getAllSessions().forEach(s -> {
                 try {
@@ -111,8 +111,12 @@ public static void broadcastCommand(String command) {
         }
     }
 
+    private static WebSocketGlobalCommandHandler getHandler() {
+        return Containers.get().findObject(WebSocketGlobalCommandHandler.class);
+    }
+
     /**
-     * Init WS connection with client
+     * Initialize WebSocket connection on client side by requesting it via Clients.evalJavaScript
      */
     public static void initWS() {
         if (ZKUtil.isInEventListener()) {
@@ -124,4 +128,17 @@ public static void initWS() {
             }
         }
     }
+
+    /**
+     * Check if desktop has an open WS session, if not try to init WS connection
+     */
+    public static void checkDesktop(Desktop desktop) {
+        var handler = getHandler();
+        if (handler != null) {
+            var session = handler.findSession(desktop);
+            if (session == null || !session.isOpen()) {
+                initWS();
+            }
+        }
+    }
 }