git.basschouten.com Git - openhab-addons.git/blob

   1 /**
   2  * Copyright (c) 2010-2023 Contributors to the openHAB project
   3  *
   4  * See the NOTICE file(s) distributed with this work for additional
   5  * information.
   6  *
   7  * This program and the accompanying materials are made available under the
   8  * terms of the Eclipse Public License 2.0 which is available at
   9  * http://www.eclipse.org/legal/epl-2.0
  10  *
  11  * SPDX-License-Identifier: EPL-2.0
  12  */
  13 package org.openhab.binding.yioremote.internal.utils;
  14
  15 import java.io.IOException;
  16
  17 import org.eclipse.jdt.annotation.NonNullByDefault;
  18 import org.eclipse.jdt.annotation.Nullable;
  19 import org.eclipse.jetty.websocket.api.Session;
  20 import org.eclipse.jetty.websocket.api.StatusCode;
  21 import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose;
  22 import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
  23 import org.eclipse.jetty.websocket.api.annotations.OnWebSocketError;
  24 import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
  25 import org.eclipse.jetty.websocket.api.annotations.WebSocket;
  26 import org.slf4j.Logger;
  27 import org.slf4j.LoggerFactory;
  28
  29 /**
  30  * The {@link Websocket} is responsible for the Websocket Connection
  31  *
  32  * @author Michael Loercher - Initial contribution
  33  */
  34
  35 @NonNullByDefault
  36 @WebSocket
  37 public class Websocket {
  38
  39     private @Nullable Session session;
  40     private final Logger logger = LoggerFactory.getLogger(Websocket.class);
  41     private @Nullable WebsocketInterface websocketHandler;
  42
  43     public void addMessageHandler(WebsocketInterface yioremotedockwebsocketinterfacehandler) {
  44         this.websocketHandler = yioremotedockwebsocketinterfacehandler;
  45     }
  46
  47     @OnWebSocketMessage
  48     public void onText(Session session, String receivedMessage) {
  49         if (websocketHandler != null) {
  50             websocketHandler.onMessage(receivedMessage);
  51         }
  52     }
  53
  54     @OnWebSocketConnect
  55     public void onConnect(Session session) {
  56         this.session = session;
  57         if (websocketHandler != null) {
  58             websocketHandler.onConnect(true);
  59         }
  60     }
  61
  62     @OnWebSocketError
  63     public void onError(Throwable cause) {
  64         logger.debug("WebSocketError {}", cause.getMessage());
  65         if (websocketHandler != null) {
  66             websocketHandler.onError(cause);
  67         }
  68     }
  69
  70     @OnWebSocketClose
  71     public void onClose(int statusCode, String reason) {
  72         if (statusCode != StatusCode.NORMAL) {
  73             logger.debug("WebSocket Connection closed: {} - {}", statusCode, reason);
  74         }
  75         if (session != null) {
  76             if (!session.isOpen()) {
  77                 if (session != null) {
  78                     session.close();
  79                 }
  80             }
  81             session = null;
  82         }
  83         if (websocketHandler != null) {
  84             websocketHandler.onClose();
  85         }
  86     }
  87
  88     public void sendMessage(String str) {
  89         if (session != null) {
  90             try {
  91                 session.getRemote().sendString(str);
  92             } catch (IOException e) {
  93                 logger.warn("Error during sendMessage function {}", e.getMessage());
  94             }
  95         }
  96     }
  97
  98     public void closeWebsocketSession() {
  99         if (session != null) {
 100             session.close();
 101         }
 102     }
 103 }