@apocaliss92/nodelink-js 0.3.4 → 0.3.5

package/dist/index.js

@@ -42,7 +42,7 @@ import {
   normalizeUid,
   parseSupportXml,
   setGlobalLogger
-} from "./chunk-YSEFEQYV.js";
+} from "./chunk-UDS2UR4S.js";
 import {
   AesStreamDecryptor,
   BC_AES_IV,
@@ -4585,8 +4585,527 @@ async function createReplayHttpServer(options) {
   };
 }
 
-// src/baichuan/stream/BaichuanHttpStreamServer.ts
+// src/baichuan/stream/Go2rtcTcpServer.ts
 import { EventEmitter as EventEmitter2 } from "events";
+import * as net from "net";
+var AsyncBoundedQueue = class {
+  maxItems;
+  queue = [];
+  waiting;
+  closed = false;
+  constructor(maxItems) {
+    this.maxItems = Math.max(1, maxItems | 0);
+  }
+  push(item) {
+    if (this.closed) return;
+    if (this.waiting) {
+      const { resolve } = this.waiting;
+      this.waiting = void 0;
+      resolve({ value: item, done: false });
+      return;
+    }
+    this.queue.push(item);
+    if (this.queue.length > this.maxItems) {
+      this.queue.splice(0, this.queue.length - this.maxItems);
+    }
+  }
+  close() {
+    if (this.closed) return;
+    this.closed = true;
+    if (this.waiting) {
+      const { resolve } = this.waiting;
+      this.waiting = void 0;
+      resolve({ value: void 0, done: true });
+    }
+  }
+  async next() {
+    if (this.closed) return { value: void 0, done: true };
+    const item = this.queue.shift();
+    if (item !== void 0) return { value: item, done: false };
+    return await new Promise((resolve) => {
+      this.waiting = { resolve };
+    });
+  }
+};
+var NativeStreamFanout = class {
+  opts;
+  queues = /* @__PURE__ */ new Map();
+  source = null;
+  running = false;
+  pumpPromise = null;
+  constructor(opts) {
+    this.opts = opts;
+  }
+  start() {
+    if (this.running) return;
+    this.running = true;
+    this.source = this.opts.createSource();
+    this.pumpPromise = (async () => {
+      try {
+        for await (const frame of this.source) {
+          try {
+            this.opts.onFrame?.(frame);
+          } catch {
+          }
+          for (const q of this.queues.values()) {
+            q.push(frame);
+          }
+        }
+      } catch (e) {
+        this.opts.onError?.(e);
+      } finally {
+        for (const q of this.queues.values()) q.close();
+        this.queues.clear();
+        this.running = false;
+        this.opts.onEnd?.();
+      }
+    })();
+  }
+  subscribe(id) {
+    const q = new AsyncBoundedQueue(this.opts.maxQueueItems);
+    this.queues.set(id, q);
+    const self = this;
+    return (async function* () {
+      try {
+        while (true) {
+          const r = await q.next();
+          if (r.done) return;
+          yield r.value;
+        }
+      } finally {
+        q.close();
+        self.queues.delete(id);
+      }
+    })();
+  }
+  async stop() {
+    if (!this.running) return;
+    this.running = false;
+    const src = this.source;
+    this.source = null;
+    for (const q of this.queues.values()) q.close();
+    this.queues.clear();
+    try {
+      await src?.return(void 0);
+    } catch {
+    }
+    try {
+      await this.pumpPromise;
+    } catch {
+    }
+    this.pumpPromise = null;
+  }
+};
+var Go2rtcTcpServer = class _Go2rtcTcpServer extends EventEmitter2 {
+  api;
+  channel;
+  profile;
+  variant;
+  listenHost;
+  listenPort;
+  logger;
+  deviceId;
+  gracePeriodMs;
+  prebufferMaxMs;
+  maxBufferBytes;
+  prestartStream;
+  active = false;
+  server;
+  resolvedPort;
+  // Native stream
+  nativeFanout = null;
+  nativeStreamActive = false;
+  dedicatedSessionRelease;
+  detectedVideoType;
+  // Client tracking
+  connectedClients = /* @__PURE__ */ new Set();
+  clientSockets = /* @__PURE__ */ new Map();
+  stopGraceTimer;
+  // Prebuffer
+  prebuffer = [];
+  constructor(options) {
+    super();
+    this.api = options.api;
+    this.channel = options.channel;
+    this.profile = options.profile;
+    this.variant = options.variant ?? "default";
+    this.listenHost = options.listenHost ?? "127.0.0.1";
+    this.listenPort = options.listenPort ?? 0;
+    this.logger = options.logger ?? console;
+    this.deviceId = options.deviceId;
+    this.gracePeriodMs = options.gracePeriodMs ?? 3e4;
+    this.prebufferMaxMs = options.prebufferMs ?? 3e3;
+    this.maxBufferBytes = options.maxBufferBytes ?? 1e8;
+    this.prestartStream = options.prestartStream ?? true;
+  }
+  // -----------------------------------------------------------------------
+  // Public API
+  // -----------------------------------------------------------------------
+  /** Start listening. Resolves once the TCP server is bound. */
+  async start() {
+    if (this.active) return;
+    this.active = true;
+    this.server = net.createServer((socket) => this.handleClient(socket));
+    this.server.on("error", (err) => {
+      this.logger.error?.(`[Go2rtcTcpServer] server error: ${err.message}`);
+      this.emit("error", err);
+    });
+    await new Promise((resolve, reject) => {
+      this.server.listen(this.listenPort, this.listenHost, () => {
+        const addr = this.server.address();
+        this.resolvedPort = addr.port;
+        this.logger.info?.(
+          `[Go2rtcTcpServer] listening on ${addr.address}:${addr.port}  channel=${this.channel} profile=${this.profile}`
+        );
+        this.emit("listening", { host: addr.address, port: addr.port });
+        resolve();
+      });
+      this.server.once("error", reject);
+    });
+    if (this.prestartStream) {
+      this.logger.info?.(
+        `[Go2rtcTcpServer] pre-starting native stream  channel=${this.channel} profile=${this.profile}`
+      );
+      this.startNativeStream();
+    }
+  }
+  /** Stop the server and all active streams. */
+  async stop() {
+    if (!this.active) return;
+    this.active = false;
+    clearTimeout(this.stopGraceTimer);
+    for (const [id, sock] of this.clientSockets) {
+      sock.destroy();
+      this.connectedClients.delete(id);
+    }
+    this.clientSockets.clear();
+    await this.stopNativeStream();
+    if (this.server) {
+      await new Promise((resolve) => {
+        this.server.close(() => resolve());
+      });
+      this.server = void 0;
+    }
+    this.prebuffer = [];
+    this.resolvedPort = void 0;
+    this.emit("close");
+  }
+  /** The actual port the server is listening on (available after start()). */
+  get port() {
+    return this.resolvedPort;
+  }
+  /** The go2rtc-compatible source URL. */
+  get go2rtcSourceUrl() {
+    if (this.resolvedPort == null) return void 0;
+    return `tcp://127.0.0.1:${this.resolvedPort}`;
+  }
+  /** Number of currently connected clients. */
+  get clientCount() {
+    return this.connectedClients.size;
+  }
+  // -----------------------------------------------------------------------
+  // Client handling
+  // -----------------------------------------------------------------------
+  handleClient(socket) {
+    const clientId = `${socket.remoteAddress}:${socket.remotePort}`;
+    socket.setNoDelay(true);
+    this.connectedClients.add(clientId);
+    this.clientSockets.set(clientId, socket);
+    this.logger.info?.(
+      `[Go2rtcTcpServer] client connected  id=${clientId} total=${this.connectedClients.size}`
+    );
+    this.emit("client", clientId);
+    if (this.stopGraceTimer) {
+      clearTimeout(this.stopGraceTimer);
+      this.stopGraceTimer = void 0;
+    }
+    if (!this.nativeStreamActive) {
+      this.startNativeStream();
+    }
+    this.feedClient(clientId, socket).catch((err) => {
+      this.logger.warn?.(
+        `[Go2rtcTcpServer] feedClient error  id=${clientId}: ${err}`
+      );
+    });
+    const cleanup = () => {
+      this.removeClient(clientId);
+      socket.destroy();
+    };
+    socket.on("error", cleanup);
+    socket.on("close", cleanup);
+  }
+  async feedClient(clientId, socket) {
+    const fanoutDeadline = Date.now() + 3e4;
+    while (this.active && !this.nativeFanout) {
+      if (socket.destroyed) return;
+      if (Date.now() > fanoutDeadline) {
+        this.logger.warn?.(
+          `[Go2rtcTcpServer] fanout not ready after 30s, dropping client ${clientId}`
+        );
+        return;
+      }
+      await new Promise((r) => setTimeout(r, 100));
+    }
+    if (!this.active || !this.nativeFanout) return;
+    const subscription = this.nativeFanout.subscribe(clientId);
+    const prebufferSnap = this.prebuffer.slice();
+    let lastIdrIdx = -1;
+    for (let i = prebufferSnap.length - 1; i >= 0; i--) {
+      if (prebufferSnap[i].isKeyframe) {
+        lastIdrIdx = i;
+        break;
+      }
+    }
+    if (lastIdrIdx >= 0) {
+      const replay = prebufferSnap.slice(lastIdrIdx);
+      this.logger.info?.(
+        `[Go2rtcTcpServer] prebuffer replay  client=${clientId} frames=${replay.length}`
+      );
+      for (const entry of replay) {
+        if (socket.destroyed) return;
+        socket.write(entry.data);
+      }
+    }
+    let seenKeyframe = lastIdrIdx >= 0;
+    let liveFrameCount = 0;
+    let liveVideoWritten = 0;
+    let lastLogAt = Date.now();
+    try {
+      this.logger.info?.(
+        `[Go2rtcTcpServer] entering live loop  client=${clientId} seenKeyframe=${seenKeyframe}`
+      );
+      for await (const frame of subscription) {
+        if (socket.destroyed || !this.active) {
+          this.logger.info?.(
+            `[Go2rtcTcpServer] live loop exit  client=${clientId} destroyed=${socket.destroyed} active=${this.active}`
+          );
+          break;
+        }
+        liveFrameCount++;
+        const annexB = this.convertFrame(frame);
+        if (!annexB) continue;
+        if (!seenKeyframe) {
+          if (!this.isAnnexBKeyframe(annexB, frame.videoType)) continue;
+          seenKeyframe = true;
+          this.logger.info?.(
+            `[Go2rtcTcpServer] first live keyframe  client=${clientId} after ${liveFrameCount} frames`
+          );
+        }
+        socket.write(annexB);
+        liveVideoWritten++;
+        if (Date.now() - lastLogAt > 1e4) {
+          this.logger.info?.(
+            `[Go2rtcTcpServer] live stats  client=${clientId} received=${liveFrameCount} written=${liveVideoWritten} bufLen=${socket.writableLength}`
+          );
+          lastLogAt = Date.now();
+        }
+        if (socket.writableLength > this.maxBufferBytes) {
+          this.logger.warn?.(
+            `[Go2rtcTcpServer] buffer overflow (${socket.writableLength} bytes), dropping client ${clientId}`
+          );
+          socket.destroy();
+          break;
+        }
+      }
+      this.logger.info?.(
+        `[Go2rtcTcpServer] live loop ended naturally  client=${clientId} received=${liveFrameCount} written=${liveVideoWritten}`
+      );
+    } finally {
+      await subscription.return(void 0).catch(() => {
+      });
+    }
+  }
+  // -----------------------------------------------------------------------
+  // Frame conversion
+  // -----------------------------------------------------------------------
+  /**
+   * Convert a native frame to wire-ready Annex-B.
+   * Audio frames are skipped — raw TCP carries only video (Annex-B).
+   * go2rtc auto-detects the codec from SPS/PPS/VPS NALUs.
+   */
+  convertFrame(frame) {
+    if (frame.audio) {
+      return null;
+    }
+    if (frame.data.length === 0) return null;
+    try {
+      if (frame.videoType === "H264") {
+        return convertToAnnexB(frame.data);
+      }
+      if (frame.videoType === "H265") {
+        return convertToAnnexB2(frame.data);
+      }
+    } catch {
+    }
+    return frame.data;
+  }
+  /** Check if an Annex-B buffer contains a keyframe (IDR for H.264, IRAP for H.265). */
+  isAnnexBKeyframe(annexB, videoType) {
+    try {
+      if (videoType === "H264") {
+        const nals = _Go2rtcTcpServer.splitAnnexBNals(annexB);
+        return nals.some((n) => n.length >= 1 && (n[0] & 31) === 5);
+      }
+      if (videoType === "H265") {
+        const nals = splitAnnexBToNalPayloads2(annexB);
+        return nals.some(
+          (n) => n.length >= 2 && isH265Irap(n[0] >> 1 & 63)
+        );
+      }
+    } catch {
+    }
+    return false;
+  }
+  /** Split Annex-B byte stream into individual NAL units. */
+  static splitAnnexBNals(buf) {
+    const nals = [];
+    let i = 0;
+    while (i < buf.length) {
+      if (i + 2 < buf.length && buf[i] === 0 && buf[i + 1] === 0) {
+        let scLen;
+        if (buf[i + 2] === 1) {
+          scLen = 3;
+        } else if (i + 3 < buf.length && buf[i + 2] === 0 && buf[i + 3] === 1) {
+          scLen = 4;
+        } else {
+          i++;
+          continue;
+        }
+        const nalStart = i + scLen;
+        let nalEnd = buf.length;
+        for (let j = nalStart; j < buf.length - 2; j++) {
+          if (buf[j] === 0 && buf[j + 1] === 0 && (buf[j + 2] === 1 || j + 3 < buf.length && buf[j + 2] === 0 && buf[j + 3] === 1)) {
+            nalEnd = j;
+            break;
+          }
+        }
+        if (nalEnd > nalStart) {
+          nals.push(buf.subarray(nalStart, nalEnd));
+        }
+        i = nalEnd;
+      } else {
+        i++;
+      }
+    }
+    return nals;
+  }
+  // -----------------------------------------------------------------------
+  // Native stream management
+  // -----------------------------------------------------------------------
+  async startNativeStream() {
+    if (this.nativeStreamActive) return;
+    this.nativeStreamActive = true;
+    let dedicatedClient;
+    if (this.deviceId) {
+      try {
+        const session = await this.api.createDedicatedSession(
+          `live:${this.deviceId}:ch${this.channel}:${this.profile}`
+        );
+        dedicatedClient = session.client;
+        this.dedicatedSessionRelease = session.release;
+      } catch (e) {
+        this.logger.warn?.(
+          `[Go2rtcTcpServer] failed to acquire dedicated session, using shared socket: ${e}`
+        );
+      }
+    }
+    this.logger.info?.(
+      `[Go2rtcTcpServer] native stream starting  channel=${this.channel} profile=${this.profile} dedicated=${!!dedicatedClient}`
+    );
+    this.nativeFanout = new NativeStreamFanout({
+      maxQueueItems: 200,
+      createSource: () => createNativeStream(this.api, this.channel, this.profile, {
+        variant: this.variant,
+        ...dedicatedClient ? { client: dedicatedClient } : {}
+      }),
+      onFrame: (frame) => {
+        if (!frame.audio && (frame.videoType === "H264" || frame.videoType === "H265")) {
+          this.detectedVideoType = frame.videoType;
+        }
+        const wireData = this.convertFrame(frame);
+        if (!wireData || wireData.length === 0) return;
+        const isKeyframe = !frame.audio && this.isAnnexBKeyframe(wireData, frame.videoType);
+        this.prebuffer.push({
+          data: Buffer.from(wireData),
+          time: Date.now(),
+          isKeyframe,
+          audio: frame.audio
+        });
+        const cutoff = Date.now() - this.prebufferMaxMs;
+        let trimIdx = 0;
+        while (trimIdx < this.prebuffer.length && this.prebuffer[trimIdx].time < cutoff) {
+          trimIdx++;
+        }
+        if (trimIdx > 0) this.prebuffer.splice(0, trimIdx);
+      },
+      onError: (error) => {
+        this.logger.warn?.(`[Go2rtcTcpServer] native stream error: ${error}`);
+      },
+      onEnd: () => {
+        if (!this.nativeStreamActive) return;
+        this.nativeStreamActive = false;
+        this.nativeFanout = null;
+        if (this.dedicatedSessionRelease) {
+          this.dedicatedSessionRelease().catch(() => {
+          });
+          this.dedicatedSessionRelease = void 0;
+        }
+        if (this.active && (this.connectedClients.size > 0 || this.prestartStream)) {
+          this.logger.info?.(
+            `[Go2rtcTcpServer] native stream ended, restarting (clients=${this.connectedClients.size}, prestart=${this.prestartStream})`
+          );
+          this.startNativeStream();
+        }
+      }
+    });
+    this.nativeFanout.start();
+  }
+  async stopNativeStream() {
+    this.nativeStreamActive = false;
+    const fanout = this.nativeFanout;
+    this.nativeFanout = null;
+    if (fanout) {
+      await fanout.stop();
+    }
+    this.prebuffer = [];
+    if (this.dedicatedSessionRelease) {
+      await this.dedicatedSessionRelease().catch(() => {
+      });
+      this.dedicatedSessionRelease = void 0;
+    }
+  }
+  // -----------------------------------------------------------------------
+  // Client lifecycle
+  // -----------------------------------------------------------------------
+  removeClient(clientId) {
+    if (!this.connectedClients.has(clientId)) return;
+    this.connectedClients.delete(clientId);
+    this.clientSockets.delete(clientId);
+    this.logger.info?.(
+      `[Go2rtcTcpServer] client disconnected  id=${clientId} remaining=${this.connectedClients.size}`
+    );
+    this.emit("clientDisconnected", clientId);
+    if (this.connectedClients.size === 0 && !this.prestartStream) {
+      this.scheduleStop();
+    }
+  }
+  scheduleStop() {
+    if (this.stopGraceTimer) return;
+    this.logger.info?.(
+      `[Go2rtcTcpServer] no clients, scheduling stream stop in ${this.gracePeriodMs}ms`
+    );
+    this.stopGraceTimer = setTimeout(async () => {
+      this.stopGraceTimer = void 0;
+      if (this.connectedClients.size === 0 && this.nativeStreamActive) {
+        this.logger.info?.("[Go2rtcTcpServer] grace period expired, stopping native stream");
+        await this.stopNativeStream();
+      }
+    }, this.gracePeriodMs);
+  }
+};
+
+// src/baichuan/stream/BaichuanHttpStreamServer.ts
+import { EventEmitter as EventEmitter3 } from "events";
 import { spawn as spawn5 } from "child_process";
 import * as http4 from "http";
 var NAL_START_CODE_4B = Buffer.from([0, 0, 0, 1]);
@@ -4633,7 +5152,7 @@ function isH264KeyframeFromAnnexB(annexB) {
   }
   return false;
 }
-var BaichuanHttpStreamServer = class extends EventEmitter2 {
+var BaichuanHttpStreamServer = class extends EventEmitter3 {
   videoStream;
   listenPort;
   path;
@@ -4900,15 +5419,15 @@ var BaichuanHttpStreamServer = class extends EventEmitter2 {
 };
 
 // src/baichuan/stream/BaichuanMjpegServer.ts
-import { EventEmitter as EventEmitter4 } from "events";
+import { EventEmitter as EventEmitter5 } from "events";
 import * as http5 from "http";
 
 // src/baichuan/stream/MjpegTransformer.ts
-import { EventEmitter as EventEmitter3 } from "events";
+import { EventEmitter as EventEmitter4 } from "events";
 import { spawn as spawn6 } from "child_process";
 var JPEG_SOI = Buffer.from([255, 216]);
 var JPEG_EOI = Buffer.from([255, 217]);
-var MjpegTransformer = class extends EventEmitter3 {
+var MjpegTransformer = class extends EventEmitter4 {
   options;
   ffmpeg = null;
   started = false;
@@ -5105,7 +5624,7 @@ Content-Length: ${frame.length}\r
 }
 
 // src/baichuan/stream/BaichuanMjpegServer.ts
-var BaichuanMjpegServer = class extends EventEmitter4 {
+var BaichuanMjpegServer = class extends EventEmitter5 {
   options;
   clients = /* @__PURE__ */ new Map();
   httpServer = null;
@@ -5386,7 +5905,7 @@ var BaichuanMjpegServer = class extends EventEmitter4 {
 };
 
 // src/baichuan/stream/BaichuanWebRTCServer.ts
-import { EventEmitter as EventEmitter5 } from "events";
+import { EventEmitter as EventEmitter6 } from "events";
 function parseAnnexBNalUnits(annexB) {
   const nalUnits = [];
   let offset = 0;
@@ -5421,7 +5940,7 @@ function getH264NalType(nalUnit) {
 function getH265NalType2(nalUnit) {
   return nalUnit[0] >> 1 & 63;
 }
-var BaichuanWebRTCServer = class extends EventEmitter5 {
+var BaichuanWebRTCServer = class extends EventEmitter6 {
   options;
   sessions = /* @__PURE__ */ new Map();
   sessionIdCounter = 0;
@@ -6323,7 +6842,7 @@ Error: ${err}`
 };
 
 // src/baichuan/stream/BaichuanHlsServer.ts
-import { EventEmitter as EventEmitter6 } from "events";
+import { EventEmitter as EventEmitter7 } from "events";
 import fs from "fs";
 import fsp from "fs/promises";
 import os from "os";
@@ -6400,7 +6919,7 @@ function getNalTypes(codec, annexB) {
     }
   });
 }
-var BaichuanHlsServer = class extends EventEmitter6 {
+var BaichuanHlsServer = class extends EventEmitter7 {
   api;
   channel;
   profile;
@@ -6818,10 +7337,10 @@ var BaichuanHlsServer = class extends EventEmitter6 {
 };
 
 // src/multifocal/compositeRtspServer.ts
-import { EventEmitter as EventEmitter7 } from "events";
+import { EventEmitter as EventEmitter8 } from "events";
 import { spawn as spawn8 } from "child_process";
-import * as net from "net";
-var CompositeRtspServer = class extends EventEmitter7 {
+import * as net2 from "net";
+var CompositeRtspServer = class extends EventEmitter8 {
   options;
   compositeStream = null;
   rtspServer = null;
@@ -6887,7 +7406,7 @@ var CompositeRtspServer = class extends EventEmitter7 {
     const width = widerStreamInfo?.width ?? 1920;
     const height = widerStreamInfo?.height ?? 1080;
     const fps = widerStreamInfo?.frameRate ?? 25;
-    this.rtspServer = net.createServer((socket) => {
+    this.rtspServer = net2.createServer((socket) => {
       this.handleRtspConnection(socket);
     });
     await new Promise((resolve, reject) => {
@@ -7162,6 +7681,7 @@ export {
   DUAL_LENS_DUAL_MOTION_MODELS,
   DUAL_LENS_MODELS,
   DUAL_LENS_SINGLE_MOTION_MODELS,
+  Go2rtcTcpServer,
   H264RtpDepacketizer,
   H265RtpDepacketizer,
   HlsSessionManager,