1 /* 2 * Copyright (c) 2022 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef COMMUNICATIONNETSTACK_WEBSOCKET_MODULE_H 17 #define COMMUNICATIONNETSTACK_WEBSOCKET_MODULE_H 18 19 #include "napi/native_api.h" 20 21 namespace OHOS::NetStack::Websocket { 22 class WebSocketModule final { 23 public: 24 class WebSocket { 25 public: 26 static constexpr const char *FUNCTION_CONNECT = "connect"; 27 static constexpr const char *FUNCTION_SEND = "send"; 28 static constexpr const char *FUNCTION_CLOSE = "close"; 29 static constexpr const char *FUNCTION_ON = "on"; 30 static constexpr const char *FUNCTION_OFF = "off"; 31 32 static napi_value Connect(napi_env env, napi_callback_info info); 33 34 static napi_value Send(napi_env env, napi_callback_info info); 35 36 static napi_value Close(napi_env env, napi_callback_info info); 37 38 static napi_value On(napi_env env, napi_callback_info info); 39 40 static napi_value Off(napi_env env, napi_callback_info info); 41 }; 42 43 #ifdef NETSTACK_WEBSOCKETSERVER 44 class WebSocketServer { 45 public: 46 static constexpr const char *FUNCTION_START = "start"; 47 static constexpr const char *FUNCTION_LISTALLCONNECTIONS = "listAllConnections"; 48 static constexpr const char *FUNCTION_CLOSE = "close"; 49 static constexpr const char *FUNCTION_ON = "on"; 50 static constexpr const char *FUNCTION_OFF = "off"; 51 static constexpr const char *FUNCTION_SEND = "send"; 52 static constexpr const char *FUNCTION_STOP = "stop"; 53 54 static napi_value Start(napi_env env, napi_callback_info info); 55 static napi_value ListAllConnections(napi_env env, napi_callback_info info); 56 static napi_value Close(napi_env env, napi_callback_info info); 57 static napi_value On(napi_env env, napi_callback_info info); 58 static napi_value Off(napi_env env, napi_callback_info info); 59 static napi_value Send(napi_env env, napi_callback_info info); 60 static napi_value Stop(napi_env env, napi_callback_info info); 61 }; 62 #endif 63 64 static constexpr const char *FUNCTION_CREATE_WEB_SOCKET = "createWebSocket"; 65 static constexpr const char *INTERFACE_WEB_SOCKET = "WebSocket"; 66 static constexpr const char *FUNCTION_CREATE_WEB_SOCKET_SERVER = "createWebSocketServer"; 67 static constexpr const char *INTERFACE_WEB_SOCKET_SERVER = "WebSocketServer"; 68 static napi_value InitWebSocketModule(napi_env env, napi_value exports); 69 70 private: 71 static napi_value CreateWebSocket(napi_env env, napi_callback_info info); 72 73 static void DefineWebSocketClass(napi_env env, napi_value exports); 74 75 static void FinalizeWebSocketInstance(napi_env env, void *data, void *hint); 76 77 static void InitWebSocketProperties(napi_env env, napi_value exports); 78 79 static napi_value CreateWebSocketServer(napi_env env, napi_callback_info info); 80 81 #ifdef NETSTACK_WEBSOCKETSERVER 82 static void DefineWebSocketServerClass(napi_env env, napi_value exports); 83 #endif 84 }; 85 } // namespace OHOS::NetStack::Websocket 86 #endif /* COMMUNICATIONNETSTACK_WEBSOCKET_MODULE_H */ 87