From 0ee4e42360995c1a85e0c7b5b06695086a856d2c Mon Sep 17 00:00:00 2001
From: Ingo Oppermann <ingo@datarhei.com>
Date: Thu, 21 Jul 2022 13:46:13 +0200
Subject: [PATCH] Include SRT into viewer stats

---
 src/utils/restreamer.js | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/utils/restreamer.js b/src/utils/restreamer.js
index d5d1884..9f3fa36 100644
--- a/src/utils/restreamer.js
+++ b/src/utils/restreamer.js
@@ -1340,7 +1340,7 @@ class Restreamer {
 			bitrate_kbit: 0,
 		};
 
-		const [val, err] = await this._call(this.api.ActiveSessions, ['ffmpeg', 'hls', 'rtmp']);
+		const [val, err] = await this._call(this.api.ActiveSessions, ['ffmpeg', 'hls', 'rtmp', 'srt']);
 		if (err !== null) {
 			return sessions;
 		}
@@ -1389,6 +1389,21 @@ class Restreamer {
 			sessions.bitrate_kbit += val.rtmp[i].bandwidth_tx_kbit;
 		}
 
+		// SRT sessions
+
+		if (!val.srt) {
+			val.srt = [];
+		}
+
+		for (let i = 0; i < val.srt.length; i++) {
+			if (!val.srt[i].reference.startsWith(this.channel.channelid)) {
+				continue;
+			}
+
+			sessions.sessions++;
+			sessions.bitrate_kbit += val.srt[i].bandwidth_tx_kbit;
+		}
+
 		return sessions;
 	}