From 5cb5f85b45ad24410c8362bb36bf58592eae02d3 Mon Sep 17 00:00:00 2001
From: srgooglo <srgooglo@ragestudio.net>
Date: Thu, 12 May 2022 14:59:14 +0200
Subject: [PATCH] get public streams methods

---
 .../src/managers/SessionsManager/index.js       | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/packages/streaming-server/src/managers/SessionsManager/index.js b/packages/streaming-server/src/managers/SessionsManager/index.js
index 60e4360a..82526867 100644
--- a/packages/streaming-server/src/managers/SessionsManager/index.js
+++ b/packages/streaming-server/src/managers/SessionsManager/index.js
@@ -1,3 +1,5 @@
+import lodash from "lodash"
+
 export default class SessionsManager {
     constructor() {
         this.sessions = {}
@@ -26,7 +28,18 @@ export default class SessionsManager {
         this.publicStreams.push(payload)
     }
 
-    unpublishStream = (id) => {
-        this.publicStreams = this.publicStreams.filter(stream => stream.id !== id)
+    unpublishStream = (stream_key) => {
+        this.publicStreams = this.publicStreams.filter(stream => stream.stream_key !== stream_key)
+    }
+
+    getPublicStreams = () => {
+        // return this.publicStreams but without stream_key property
+        return lodash.map(this.publicStreams, stream => {
+            return lodash.omit(stream, "stream_key")
+        })
+    }
+
+    getStreamsByUserId = (user_id) => {
+        return lodash.filter(this.publicStreams, stream => stream.user_id === user_id)
     }
 }
\ No newline at end of file