@apocaliss92/nodelink-js 0.4.1 → 0.4.2

package/dist/cli/rtsp-server.cjs

@@ -8070,6 +8070,7 @@ var BaichuanRtspServer = class _BaichuanRtspServer extends import_node_events2.E
   flow;
   deviceId;
   dedicatedSessionRelease;
+  externalListener;
   // Authentication
   authCredentials = [];
   requireAuth;
@@ -8237,6 +8238,7 @@ var BaichuanRtspServer = class _BaichuanRtspServer extends import_node_events2.E
     this.logger = options.logger ?? console;
     this.tcpRtpFraming = options.tcpRtpFraming ?? "rfc4571";
     this.deviceId = options.deviceId;
+    this.externalListener = options.externalListener ?? false;
     this.authCredentials = options.credentials ?? [];
     this.requireAuth = options.requireAuth ?? this.authCredentials.length > 0;
     const transport = this.api.client.getTransport();
@@ -8367,41 +8369,56 @@ var BaichuanRtspServer = class _BaichuanRtspServer extends import_node_events2.E
       this.streamMetadata = { frameRate: 25 };
       this.setFlowVideoType("H264", "metadata unavailable");
     }
-    this.clientConnectionServer = net.createServer((socket) => {
-      this.handleRtspConnection(socket);
-    });
-    await new Promise((resolve, reject) => {
-      this.clientConnectionServer.listen(
-        this.listenPort,
-        this.listenHost,
-        () => {
-          const address = this.clientConnectionServer.address();
-          if (address && typeof address === "object" && "port" in address) {
-            this.listenPort = address.port;
+    if (!this.externalListener) {
+      this.clientConnectionServer = net.createServer((socket) => {
+        this.handleRtspConnection(socket);
+      });
+      await new Promise((resolve, reject) => {
+        this.clientConnectionServer.listen(
+          this.listenPort,
+          this.listenHost,
+          () => {
+            const address = this.clientConnectionServer.address();
+            if (address && typeof address === "object" && "port" in address) {
+              this.listenPort = address.port;
+            }
+            resolve();
           }
-          resolve();
-        }
-      );
-      this.clientConnectionServer.on("error", (error) => {
-        reject(error);
+        );
+        this.clientConnectionServer.on("error", (error) => {
+          reject(error);
+        });
       });
-    });
+    }
     this.active = true;
     this.logger.info(
       `[BaichuanRtspServer] RTSP server started on ${this.listenHost}:${this.listenPort}, path: ${this.path}`
     );
   }
+  /**
+   * Accept an externally-routed RTSP connection.
+   * Used in directHandoff mode where RtspProxyServer routes sockets here.
+   * @param socket - The client TCP socket (already authenticated by proxy)
+   * @param initialBuffer - Any bytes already read during path parsing/auth
+   */
+  acceptConnection(socket, initialBuffer) {
+    if (!this.active) {
+      socket.end("RTSP/1.0 503 Service Unavailable\r\n\r\n");
+      return;
+    }
+    this.handleRtspConnection(socket, initialBuffer);
+  }
   /**
    * Handle RTSP connection from a client.
    */
-  handleRtspConnection(socket) {
+  handleRtspConnection(socket, initialBuffer) {
     const clientId = `${socket.remoteAddress}:${socket.remotePort}`;
     const connectTime = Date.now();
     this.logger.info(
       `[rebroadcast] client connected  client=${clientId} path=${this.path} profile=${this.profile} channel=${this.channel}`
     );
     let sessionId = "";
-    let buffer = Buffer.alloc(0);
+    let buffer = initialBuffer ?? Buffer.alloc(0);
     let clientFfmpeg;
     let useTcpInterleaved = false;
     let clientUdpSocket = null;
@@ -8486,8 +8503,7 @@ var BaichuanRtspServer = class _BaichuanRtspServer extends import_node_events2.E
       }
       cleanup();
     });
-    socket.on("data", async (data) => {
-      buffer = Buffer.concat([buffer, data]);
+    const processBuffer = async () => {
       while (buffer.includes("\r\n\r\n")) {
         const endIndex = buffer.indexOf("\r\n\r\n");
         const requestText = buffer.subarray(0, endIndex).toString();
@@ -8762,7 +8778,14 @@ var BaichuanRtspServer = class _BaichuanRtspServer extends import_node_events2.E
           sendResponse(501, "Not Implemented");
         }
       }
+    };
+    socket.on("data", (data) => {
+      buffer = Buffer.concat([buffer, data]);
+      void processBuffer();
     });
+    if (buffer.includes("\r\n\r\n")) {
+      void processBuffer();
+    }
   }
   /**
    * Generate SDP (Session Description Protocol) for RTSP DESCRIBE.
@@ -9730,21 +9753,27 @@ var BaichuanRtspServer = class _BaichuanRtspServer extends import_node_events2.E
           res.seenFirstVideoKeyframe = false;
           res.rtpSentVideoConfig = false;
         }
-        if (this.dedicatedSessionRelease) {
-          const release = this.dedicatedSessionRelease;
-          this.dedicatedSessionRelease = void 0;
-          release().catch(() => {
-          });
-        }
         this.logger.info(
           `[rebroadcast] native stream ended (camera sleeping or connection lost)  profile=${this.profile} channel=${this.channel} clients=${this.connectedClients.size}`
         );
-        if (this.connectedClients.size > 0) {
-          this.logger.info(
-            `[rebroadcast] restarting native stream for ${this.connectedClients.size} active client(s)`
-          );
-          setImmediate(() => void this.startNativeStream());
-        }
+        const releaseAndRestart = async () => {
+          if (this.dedicatedSessionRelease) {
+            const release = this.dedicatedSessionRelease;
+            this.dedicatedSessionRelease = void 0;
+            try {
+              await release();
+            } catch {
+            }
+          }
+          if (this.connectedClients.size > 0) {
+            this.logger.info(
+              `[rebroadcast] restarting native stream for ${this.connectedClients.size} active client(s)`
+            );
+            await new Promise((r) => setTimeout(r, 500));
+            void this.startNativeStream();
+          }
+        };
+        void releaseAndRestart();
       }
     });
     this.nativeFanout.start();
@@ -9968,6 +9997,31 @@ var BaichuanRtspServer = class _BaichuanRtspServer extends import_node_events2.E
   getClientCount() {
     return this.connectedClients.size;
   }
+  /**
+   * Subscribe to the raw native stream for diagnostic purposes.
+   * The subscriber receives the same frames as RTSP clients.
+   * Counts as a "consumer" for lifecycle — prevents auto-stop while subscribed.
+   * If the native stream is not active, starts it automatically.
+   */
+  async subscribeDiagnostic(id) {
+    this.connectedClients.add(`diag:${id}`);
+    if (!this.nativeStreamActive) {
+      await this.startNativeStream();
+    }
+    return this.nativeFanout.subscribe(`diag:${id}`);
+  }
+  /**
+   * Unsubscribe a diagnostic session.
+   */
+  unsubscribeDiagnostic(id) {
+    this.removeClient(`diag:${id}`);
+  }
+  /**
+   * Returns detected audio metadata (available after first audio frame).
+   */
+  getAudioInfo() {
+    return this.audioInfo;
+  }
 };
 
 // src/reolink/baichuan/ReolinkBaichuanApi.ts