@lng2004/node-datachannel 0.31.0-20251228

package/src/cpp/web-socket-server-wrapper.h

@@ -0,0 +1,41 @@
+#ifndef WEB_SOCKET_SERVER_WRAPPER_H
+#define WEB_SOCKET_SERVER_WRAPPER_H
+
+#include <memory>
+#include <unordered_set>
+
+#include <napi.h>
+#include <rtc/rtc.hpp>
+
+#include "thread-safe-callback.h"
+
+class WebSocketServerWrapper : public Napi::ObjectWrap<WebSocketServerWrapper>
+{
+public:
+  static Napi::Object Init(Napi::Env env, Napi::Object exports);
+  WebSocketServerWrapper(const Napi::CallbackInfo &info);
+  ~WebSocketServerWrapper();
+
+  // Functions
+  void stop(const Napi::CallbackInfo &info);
+  Napi::Value port(const Napi::CallbackInfo &info);
+
+  // Callbacks
+  void onClient(const Napi::CallbackInfo &info);
+
+  // Close all existing WebSocketServers
+  static void StopAll();
+
+private:
+  static Napi::FunctionReference constructor;
+  static std::unordered_set<WebSocketServerWrapper *> instances;
+
+  void doStop();
+
+  std::unique_ptr<rtc::WebSocketServer> mWebSocketServerPtr = nullptr;
+
+  // Callback Ptrs
+  std::unique_ptr<ThreadSafeCallback> mOnClientCallback = nullptr;
+};
+
+#endif // WEB_SOCKET_SERVER_WRAPPER_H