WebTransportStream sends frames to internal server.

Author: jianjunz

source/agent/addons/common/MediaFramePipelineWrapper.h

@@ -6,10 +6,10 @@
 #ifndef MEDIAFRAMEPIPELINEWRAPPER_H
 #define MEDIAFRAMEPIPELINEWRAPPER_H
 
+#include <MediaFramePipeline.h>
+#include <nan.h>
 #include <node.h>
 #include <node_object_wrap.h>
-#include <MediaFramePipeline.h>
-
 
 /*
  * Wrapper class of owt_base::FrameDestination
@@ -30,5 +30,13 @@ class FrameSource : public node::ObjectWrap{
   owt_base::FrameSource* src;
 };
 
+/*
+ * Nan::ObjectWrap of owt_base::FrameSource and owt_base::FrameDestination, represents a node in the media or data pipeline.
+ */
+class NanFrameNode : public Nan::ObjectWrap {
+public:
+    virtual owt_base::FrameSource* FrameSource() = 0;
+    virtual owt_base::FrameDestination* FrameDestination() = 0;
+};
 
 #endif

source/agent/addons/quic/QuicTransportStream.h

@@ -9,12 +9,13 @@
 
 #include "owt/quic/web_transport_stream_interface.h"
 #include "../../core/owt_base/MediaFramePipeline.h"
+#include "../common/MediaFramePipelineWrapper.h"
 #include <logger.h>
 #include <nan.h>
 #include <mutex>
 #include <string>
 
-class QuicTransportStream : public owt_base::FrameSource, public owt_base::FrameDestination, public Nan::ObjectWrap, public owt::quic::WebTransportStreamInterface::Visitor {
+class QuicTransportStream : public owt_base::FrameSource, public owt_base::FrameDestination, public NanFrameNode, public owt::quic::WebTransportStreamInterface::Visitor {
     DECLARE_LOGGER();
 
 public:
@@ -45,6 +46,10 @@ class QuicTransportStream : public owt_base::FrameSource, public owt_base::Frame
     void onFrame(const owt_base::Frame&) override;
     void onVideoSourceChanged() override;
 
+    // Overrides NanFrameNode.
+    owt_base::FrameSource* FrameSource() override { return this; }
+    owt_base::FrameDestination* FrameDestination() override { return this; }
+
     static Nan::Persistent<v8::Function> s_constructor;
 
 protected:

source/agent/internalConnectionRouter.js

@@ -167,8 +167,8 @@ class InternalConnectionRouter {
    * @param {FrameSource} source Wrapper class for FrameSource
    */
   addLocalSource(id, type, source) {
-    log.debug('addLocalSource:', id, type);
-    this.internalServer.addSource(id, source);
+    const isNativeSource = (type === 'quic');
+    this.internalServer.addSource(id, source, isNativeSource);
     return this.connections.addConnection(id, type, '', source, 'in');
   }
 
@@ -191,7 +191,11 @@ class InternalConnectionRouter {
         return this.removeLocalSource(id);
       } else if (conn.direction === 'out') {
         return this.removeLocalDestination(id);
+      } else {
+        return Promise.reject('Unexpected direction '+conn.direction);
       }
+    } else {
+      return Promise.reject('Cannot find connection.');
     }
   }