@walkhi/code-relax 0.1.0-beta.1

package/node_modules/node-datachannel/src/cpp/media-track-wrapper.cpp

@@ -0,0 +1,458 @@
+#include "media-track-wrapper.h"
+#include "media-direction.h"
+#include "media-mediahandler-helper.h"
+
+#include "plog/Log.h"
+
+Napi::FunctionReference TrackWrapper::constructor = Napi::FunctionReference();
+std::unordered_set<TrackWrapper *> TrackWrapper::instances;
+
+void TrackWrapper::CloseAll()
+{
+  auto copy(instances);
+  for (auto inst : copy)
+    inst->doClose();
+}
+
+void TrackWrapper::CleanupAll()
+{
+  PLOG_DEBUG << "CleanupAll() called";
+  auto copy(instances);
+  for (auto inst : copy)
+    inst->doCleanup();
+}
+
+Napi::Object TrackWrapper::Init(Napi::Env env, Napi::Object exports)
+{
+  Napi::HandleScope scope(env);
+
+  Napi::Function func = DefineClass(env, "Track",
+                                    {
+                                        InstanceMethod("direction", &TrackWrapper::direction),
+                                        InstanceMethod("mid", &TrackWrapper::mid),
+                                        InstanceMethod("type", &TrackWrapper::type),
+                                        InstanceMethod("close", &TrackWrapper::close),
+                                        InstanceMethod("sendMessage", &TrackWrapper::sendMessage),
+                                        InstanceMethod("sendMessageBinary", &TrackWrapper::sendMessageBinary),
+                                        InstanceMethod("isOpen", &TrackWrapper::isOpen),
+                                        InstanceMethod("isClosed", &TrackWrapper::isClosed),
+                                        InstanceMethod("maxMessageSize", &TrackWrapper::maxMessageSize),
+                                        InstanceMethod("requestBitrate", &TrackWrapper::requestBitrate),
+                                        InstanceMethod("requestKeyframe", &TrackWrapper::requestKeyframe),
+                                        InstanceMethod("setMediaHandler", &TrackWrapper::setMediaHandler),
+                                        InstanceMethod("onOpen", &TrackWrapper::onOpen),
+                                        InstanceMethod("onClosed", &TrackWrapper::onClosed),
+                                        InstanceMethod("onError", &TrackWrapper::onError),
+                                        InstanceMethod("onMessage", &TrackWrapper::onMessage),
+                                    });
+
+  // If this is not the first call, we don't want to reassign the constructor (hot-reload problem)
+  if (constructor.IsEmpty())
+  {
+    constructor = Napi::Persistent(func);
+    constructor.SuppressDestruct();
+  }
+
+  exports.Set("Track", func);
+  return exports;
+}
+
+TrackWrapper::TrackWrapper(const Napi::CallbackInfo &info) : Napi::ObjectWrap<TrackWrapper>(info)
+{
+  PLOG_DEBUG << "Constructor called";
+  mTrackPtr = *(info[0].As<Napi::External<std::shared_ptr<rtc::Track>>>().Data());
+  PLOG_DEBUG << "Track created";
+
+  // Closed callback must be set to trigger cleanup
+  mOnClosedCallback =
+      std::make_unique<ThreadSafeCallback>(Napi::Function::New(info.Env(), [](const Napi::CallbackInfo &) {}));
+
+  instances.insert(this);
+}
+
+TrackWrapper::~TrackWrapper() { doClose(); }
+
+void TrackWrapper::doClose()
+{
+  if (mTrackPtr)
+  {
+    try
+    {
+      mTrackPtr->close();
+      mTrackPtr.reset();
+    }
+    catch (std::exception &ex)
+    {
+      std::cerr << std::string("libdatachannel error while closing track: ") + ex.what() << std::endl;
+      return;
+    }
+  }
+
+  mOnOpenCallback.reset();
+  mOnErrorCallback.reset();
+  mOnMessageCallback.reset();
+}
+
+void TrackWrapper::doCleanup()
+{
+  PLOG_DEBUG << "doCleanup() called";
+  mOnClosedCallback.reset();
+  instances.erase(this);
+}
+
+void TrackWrapper::close(const Napi::CallbackInfo &info) { doClose(); }
+
+Napi::Value TrackWrapper::direction(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "direction() called on destroyed track").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  Napi::Env env = info.Env();
+  return Napi::String::New(env, directionToStr(mTrackPtr->direction()));
+}
+
+Napi::Value TrackWrapper::mid(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "mid() called on destroyed track").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  Napi::Env env = info.Env();
+  return Napi::String::New(env, mTrackPtr->mid());
+}
+
+Napi::Value TrackWrapper::type(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "type() called on destroyed track").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  Napi::Env env = info.Env();
+  return Napi::String::New(env, mTrackPtr->description().type());
+}
+
+Napi::Value TrackWrapper::sendMessage(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "sendMessage() called on destroyed track").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  // Allow call with NULL
+  if (length < 1 || (!info[0].IsString() && !info[0].IsNull()))
+  {
+    Napi::TypeError::New(env, "String or Null expected").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  try
+  {
+    return Napi::Boolean::New(info.Env(), mTrackPtr->send(info[0].As<Napi::String>().ToString()));
+  }
+  catch (std::exception &ex)
+  {
+    Napi::Error::New(env, std::string("libdatachannel error while sending track message: ") + ex.what())
+        .ThrowAsJavaScriptException();
+    return Napi::Boolean::New(info.Env(), false);
+  }
+}
+
+Napi::Value TrackWrapper::sendMessageBinary(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "sendMessageBinary() called on destroyed track").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (length < 1 || !info[0].IsBuffer())
+  {
+    Napi::TypeError::New(env, "Buffer expected").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  try
+  {
+    Napi::Uint8Array buffer = info[0].As<Napi::Uint8Array>();
+    return Napi::Boolean::New(info.Env(), mTrackPtr->send((std::byte *)buffer.Data(), buffer.ByteLength()));
+  }
+  catch (std::exception &ex)
+  {
+    Napi::Error::New(env, std::string("libdatachannel error while sending track message: ") + ex.what())
+        .ThrowAsJavaScriptException();
+    return Napi::Boolean::New(info.Env(), false);
+  }
+}
+
+Napi::Value TrackWrapper::isOpen(const Napi::CallbackInfo &info)
+{
+  Napi::Env env = info.Env();
+
+  if (!mTrackPtr)
+  {
+    return Napi::Boolean::New(env, false);
+  }
+
+  return Napi::Boolean::New(env, mTrackPtr->isOpen());
+}
+
+Napi::Value TrackWrapper::isClosed(const Napi::CallbackInfo &info)
+{
+  Napi::Env env = info.Env();
+
+  if (!mTrackPtr)
+  {
+    return Napi::Boolean::New(env, true);
+  }
+
+  return Napi::Boolean::New(env, mTrackPtr->isClosed());
+}
+
+Napi::Value TrackWrapper::maxMessageSize(const Napi::CallbackInfo &info)
+{
+  Napi::Env env = info.Env();
+
+  if (!mTrackPtr)
+  {
+    return Napi::Number::New(info.Env(), 0);
+  }
+
+  try
+  {
+    return Napi::Number::New(info.Env(), mTrackPtr->maxMessageSize());
+  }
+  catch (std::exception &ex)
+  {
+    Napi::Error::New(env, std::string("libdatachannel error: ") + ex.what()).ThrowAsJavaScriptException();
+    return Napi::Number::New(info.Env(), 0);
+  }
+}
+
+Napi::Value TrackWrapper::requestBitrate(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "requestBitrate() called on destroyed track").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (length < 1 || !info[0].IsNumber())
+  {
+    Napi::TypeError::New(env, "Number expected").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  unsigned int bitrate = static_cast<uint32_t>(info[0].As<Napi::Number>());
+  return Napi::Boolean::New(env, mTrackPtr->requestBitrate(bitrate));
+}
+
+Napi::Value TrackWrapper::requestKeyframe(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "requestKeyframe() called on destroyed track").ThrowAsJavaScriptException();
+    return info.Env().Null();
+  }
+
+  Napi::Env env = info.Env();
+  return Napi::Boolean::New(env, mTrackPtr->requestKeyframe());
+}
+
+void TrackWrapper::setMediaHandler(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "setMediaHandler() called on destroyed track").ThrowAsJavaScriptException();
+    return;
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (length < 1 || !info[0].IsObject())
+  {
+    Napi::TypeError::New(env, "MediaHandler class instance expected").ThrowAsJavaScriptException();
+    return;
+  }
+
+  std::shared_ptr<rtc::MediaHandler> handler = asMediaHandler(info[0].As<Napi::Object>());
+  if (!handler)
+  {
+    Napi::TypeError::New(env, "MediaHandler class instance expected. If this error is unexpected, please report a bug!")
+      .ThrowAsJavaScriptException();
+    return;
+  }
+  mTrackPtr->setMediaHandler(handler);
+}
+
+void TrackWrapper::onOpen(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "onOpen() called on destroyed track").ThrowAsJavaScriptException();
+    return;
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (length < 1 || !info[0].IsFunction())
+  {
+    Napi::TypeError::New(env, "Function expected").ThrowAsJavaScriptException();
+    return;
+  }
+
+  // Callback
+  mOnOpenCallback = std::make_unique<ThreadSafeCallback>(info[0].As<Napi::Function>());
+
+  mTrackPtr->onOpen(
+      [&]()
+      {
+    if (mOnOpenCallback)
+      mOnOpenCallback->call(
+          [this](Napi::Env env, std::vector<napi_value> &args)
+          {
+        // Check the track is not closed
+        if (instances.find(this) == instances.end())
+          throw ThreadSafeCallback::CancelException();
+
+        // This will run in main thread and needs to construct the
+        // arguments for the call
+        args = {};
+      });
+  });
+}
+
+void TrackWrapper::onClosed(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "onClosed() called on destroyed track").ThrowAsJavaScriptException();
+    return;
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (length < 1 || !info[0].IsFunction())
+  {
+    Napi::TypeError::New(env, "Function expected").ThrowAsJavaScriptException();
+    return;
+  }
+
+  // Callback
+  mOnClosedCallback = std::make_unique<ThreadSafeCallback>(info[0].As<Napi::Function>());
+
+  mTrackPtr->onClosed(
+      [&]()
+      {
+    if (mOnClosedCallback)
+      mOnClosedCallback->call(
+          [this](Napi::Env env, std::vector<napi_value> &args)
+          {
+        // Do not check if the data channel has been closed here
+
+        // This will run in main thread and needs to construct the
+        // arguments for the call
+        args = {};
+          },
+          [this] { doCleanup(); });
+  });
+}
+
+void TrackWrapper::onError(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "onError() called on destroyed track").ThrowAsJavaScriptException();
+    return;
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (length < 1 || !info[0].IsFunction())
+  {
+    Napi::TypeError::New(env, "Function expected").ThrowAsJavaScriptException();
+    return;
+  }
+
+  // Callback
+  mOnErrorCallback = std::make_unique<ThreadSafeCallback>(info[0].As<Napi::Function>());
+
+  mTrackPtr->onError(
+      [&](const std::string &error)
+      {
+    if (mOnErrorCallback)
+      mOnErrorCallback->call(
+          [this, error](Napi::Env env, std::vector<napi_value> &args)
+          {
+        // Check the track is not closed
+        if (instances.find(this) == instances.end())
+          throw ThreadSafeCallback::CancelException();
+
+        // This will run in main thread and needs to construct the
+        // arguments for the call
+        args = {Napi::String::New(env, error)};
+      });
+  });
+}
+
+void TrackWrapper::onMessage(const Napi::CallbackInfo &info)
+{
+  if (!mTrackPtr)
+  {
+    Napi::Error::New(info.Env(), "onMessage() called on destroyed track").ThrowAsJavaScriptException();
+    return;
+  }
+
+  Napi::Env env = info.Env();
+  int length = info.Length();
+
+  if (length < 1 || !info[0].IsFunction())
+  {
+    Napi::TypeError::New(env, "Function expected").ThrowAsJavaScriptException();
+    return;
+  }
+
+  // Callback
+  mOnMessageCallback = std::make_unique<ThreadSafeCallback>(info[0].As<Napi::Function>());
+
+  mTrackPtr->onMessage(
+      [&](std::variant<rtc::binary, std::string> message)
+      {
+    if (mOnMessageCallback)
+      mOnMessageCallback->call(
+          [this, message = std::move(message)](Napi::Env env, std::vector<napi_value> &args)
+          {
+        // Check the track is not closed
+        if (instances.find(this) == instances.end())
+          throw ThreadSafeCallback::CancelException();
+
+        // This will run in main thread and needs to construct the
+        // arguments for the call
+        if (std::holds_alternative<rtc::binary>(message)) // Track will always receive messages as binary
+        {
+          auto bin = std::get<rtc::binary>(std::move(message));
+          args = {Napi::Buffer<std::byte>::Copy(env, bin.data(), bin.size())};
+        }
+      });
+  });
+}

package/node_modules/node-datachannel/src/cpp/media-track-wrapper.h

@@ -0,0 +1,61 @@
+#ifndef MEDIA_TRACK_WRAPPER_H
+#define MEDIA_TRACK_WRAPPER_H
+
+#include <memory>
+#include <unordered_set>
+
+#include <napi.h>
+#include <rtc/rtc.hpp>
+
+#include "thread-safe-callback.h"
+
+class TrackWrapper : public Napi::ObjectWrap<TrackWrapper>
+{
+public:
+  static Napi::FunctionReference constructor;
+  static Napi::Object Init(Napi::Env env, Napi::Object exports);
+  TrackWrapper(const Napi::CallbackInfo &info);
+  ~TrackWrapper();
+
+  // Functions
+  Napi::Value direction(const Napi::CallbackInfo &info);
+  Napi::Value mid(const Napi::CallbackInfo &info);
+  Napi::Value type(const Napi::CallbackInfo &info);
+  void close(const Napi::CallbackInfo &info);
+  Napi::Value sendMessage(const Napi::CallbackInfo &info);
+  Napi::Value sendMessageBinary(const Napi::CallbackInfo &info);
+  Napi::Value isOpen(const Napi::CallbackInfo &info);
+  Napi::Value isClosed(const Napi::CallbackInfo &info);
+  Napi::Value maxMessageSize(const Napi::CallbackInfo &info);
+  Napi::Value requestBitrate(const Napi::CallbackInfo &info);
+  Napi::Value requestKeyframe(const Napi::CallbackInfo &info);
+  void setMediaHandler(const Napi::CallbackInfo &info);
+
+  // Callbacks
+  void onOpen(const Napi::CallbackInfo &info);
+  void onClosed(const Napi::CallbackInfo &info);
+  void onError(const Napi::CallbackInfo &info);
+  void onMessage(const Napi::CallbackInfo &info);
+
+  // Close all existing tracks
+  static void CloseAll();
+
+  // Reset all Callbacks for existing tracks
+  static void CleanupAll();
+
+private:
+  static std::unordered_set<TrackWrapper *> instances;
+
+  void doClose();
+  void doCleanup();
+
+  std::shared_ptr<rtc::Track> mTrackPtr = nullptr;
+
+  // Callback Ptrs
+  std::unique_ptr<ThreadSafeCallback> mOnOpenCallback = nullptr;
+  std::unique_ptr<ThreadSafeCallback> mOnClosedCallback = nullptr;
+  std::unique_ptr<ThreadSafeCallback> mOnErrorCallback = nullptr;
+  std::unique_ptr<ThreadSafeCallback> mOnMessageCallback = nullptr;
+};
+
+#endif // MEDIA_TRACK_WRAPPER_H