Adds No_MEDIA and NO_WEBSOCKET options

Author: murat-dogan

src/cpp/main.cpp

@@ -2,24 +2,38 @@
 #include "rtc-wrapper.h"
 #include "peer-connection-wrapper.h"
 #include "data-channel-wrapper.h"
+
+#if RTC_ENABLE_MEDIA == 1
 #include "media-rtcpreceivingsession-wrapper.h"
 #include "media-track-wrapper.h"
 #include "media-video-wrapper.h"
 #include "media-audio-wrapper.h"
+#endif
+
+#if RTC_ENABLE_WEBSOCKET == 1
 #include "web-socket-wrapper.h"
 #include "web-socket-server-wrapper.h"
+#endif
 
 Napi::Object InitAll(Napi::Env env, Napi::Object exports)
 {
     RtcWrapper::Init(env, exports);
+
+#if RTC_ENABLE_MEDIA == 1
     RtcpReceivingSessionWrapper::Init(env, exports);
     TrackWrapper::Init(env, exports);
     VideoWrapper::Init(env, exports);
     AudioWrapper::Init(env, exports);
+#endif
+
     DataChannelWrapper::Init(env, exports);
     PeerConnectionWrapper::Init(env, exports);
+
+#if RTC_ENABLE_WEBSOCKET == 1
     WebSocketWrapper::Init(env, exports);
     WebSocketServerWrapper::Init(env, exports);
+#endif
+
     return exports;
 }
 

src/cpp/peer-connection-wrapper.cpp

@@ -1,8 +1,11 @@
 #include "peer-connection-wrapper.h"
 #include "data-channel-wrapper.h"
+
+#if RTC_ENABLE_MEDIA == 1
 #include "media-track-wrapper.h"
 #include "media-video-wrapper.h"
 #include "media-audio-wrapper.h"
+#endif
 
 #include "plog/Log.h"
 
@@ -43,7 +46,11 @@ Napi::Object PeerConnectionWrapper::Init(Napi::Env env, Napi::Object exports)
             InstanceMethod("remoteDescription", &PeerConnectionWrapper::remoteDescription),
             InstanceMethod("addRemoteCandidate", &PeerConnectionWrapper::addRemoteCandidate),
             InstanceMethod("createDataChannel", &PeerConnectionWrapper::createDataChannel),
+
+#if RTC_ENABLE_MEDIA == 1
             InstanceMethod("addTrack", &PeerConnectionWrapper::addTrack),
+            InstanceMethod("onTrack", &PeerConnectionWrapper::onTrack),
+#endif
             InstanceMethod("hasMedia", &PeerConnectionWrapper::hasMedia),
             InstanceMethod("state", &PeerConnectionWrapper::state),
             InstanceMethod("iceState", &PeerConnectionWrapper::iceState),
@@ -56,7 +63,6 @@ Napi::Object PeerConnectionWrapper::Init(Napi::Env env, Napi::Object exports)
             InstanceMethod("onSignalingStateChange", &PeerConnectionWrapper::onSignalingStateChange),
             InstanceMethod("onGatheringStateChange", &PeerConnectionWrapper::onGatheringStateChange),
             InstanceMethod("onDataChannel", &PeerConnectionWrapper::onDataChannel),
-            InstanceMethod("onTrack", &PeerConnectionWrapper::onTrack),
             InstanceMethod("bytesSent", &PeerConnectionWrapper::bytesSent),
             InstanceMethod("bytesReceived", &PeerConnectionWrapper::bytesReceived),
             InstanceMethod("rtt", &PeerConnectionWrapper::rtt),
@@ -1049,6 +1055,7 @@ std::string PeerConnectionWrapper::candidateTransportTypeToString(const rtc::Can
     }
 }
 
+#if RTC_ENABLE_MEDIA == 1
 Napi::Value PeerConnectionWrapper::addTrack(const Napi::CallbackInfo &info)
 {
     PLOG_DEBUG << "addTrack() called";
@@ -1132,6 +1139,7 @@ void PeerConnectionWrapper::onTrack(const Napi::CallbackInfo &info)
                 args = {instance};
             }); });
 }
+#endif
 
 Napi::Value PeerConnectionWrapper::hasMedia(const Napi::CallbackInfo &info)
 {

src/cpp/peer-connection-wrapper.h

@@ -27,7 +27,12 @@ class PeerConnectionWrapper : public Napi::ObjectWrap<PeerConnectionWrapper>
   Napi::Value remoteDescription(const Napi::CallbackInfo &info);
   void addRemoteCandidate(const Napi::CallbackInfo &info);
   Napi::Value createDataChannel(const Napi::CallbackInfo &info);
+
+#if RTC_ENABLE_MEDIA == 1
   Napi::Value addTrack(const Napi::CallbackInfo &info);
+  void onTrack(const Napi::CallbackInfo &info);
+#endif
+
   Napi::Value hasMedia(const Napi::CallbackInfo &info);
   Napi::Value state(const Napi::CallbackInfo &info);
   Napi::Value iceState(const Napi::CallbackInfo &info);
@@ -42,7 +47,6 @@ class PeerConnectionWrapper : public Napi::ObjectWrap<PeerConnectionWrapper>
   void onSignalingStateChange(const Napi::CallbackInfo &info);
   void onGatheringStateChange(const Napi::CallbackInfo &info);
   void onDataChannel(const Napi::CallbackInfo &info);
-  void onTrack(const Napi::CallbackInfo &info);
 
   // Stats
   Napi::Value bytesSent(const Napi::CallbackInfo &info);

src/cpp/rtc-wrapper.cpp

@@ -1,9 +1,15 @@
 #include "rtc-wrapper.h"
 #include "peer-connection-wrapper.h"
 #include "data-channel-wrapper.h"
+
+#if RTC_ENABLE_MEDIA == 1
 #include "media-track-wrapper.h"
+#endif
+
+#if RTC_ENABLE_WEBSOCKET == 1
 #include "web-socket-wrapper.h"
 #include "web-socket-server-wrapper.h"
+#endif
 
 #include "plog/Log.h"
 
@@ -118,9 +124,15 @@ void RtcWrapper::cleanup(const Napi::CallbackInfo &info)
     {
         PeerConnectionWrapper::CloseAll();
         DataChannelWrapper::CloseAll();
+
+#if RTC_ENABLE_MEDIA == 1
         TrackWrapper::CloseAll();
+#endif
+
+#if RTC_ENABLE_WEBSOCKET == 1
         WebSocketWrapper::CloseAll();
         WebSocketServerWrapper::StopAll();
+#endif
 
         const auto timeout = std::chrono::seconds(10);
         if (rtc::Cleanup().wait_for(std::chrono::seconds(timeout)) == std::future_status::timeout)
@@ -129,8 +141,14 @@ void RtcWrapper::cleanup(const Napi::CallbackInfo &info)
         // Cleanup the instances
         PeerConnectionWrapper::CleanupAll();
         DataChannelWrapper::CleanupAll();
+
+#if RTC_ENABLE_MEDIA == 1
         TrackWrapper::CleanupAll();
+#endif
+
+#if RTC_ENABLE_WEBSOCKET == 1
         WebSocketWrapper::CleanupAll();
+#endif
 
         if (logCallback)
             logCallback.reset();