From ca4bb76996b5dec9dc9223ab29b5fe774a2a86c5 Mon Sep 17 00:00:00 2001
From: white <122345776@umail.ucc.ie>
Date: Mon, 17 Feb 2025 12:23:32 +0000
Subject: [PATCH] UPDATE: added live viewer display on home and category pages
 (need to remove dummy values)

---
 web_server/blueprints/chat.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/web_server/blueprints/chat.py b/web_server/blueprints/chat.py
index a6f0e0e..6a989d7 100644
--- a/web_server/blueprints/chat.py
+++ b/web_server/blueprints/chat.py
@@ -23,10 +23,10 @@ def handle_join(data) -> None:
     Allow a user to join the chat of the stream they are watching.
     """
     stream_id = data.get("stream_id")
-    print(f"Stream ID GLORP: {stream_id}", flush=True)
     if stream_id:
         join_room(stream_id)
         num_viewers = len(list(socketio.server.manager.get_participants("/", stream_id)))
+        update_viewers(stream_id, num_viewers)
         emit("status", 
             {
                 "message": f"Welcome to the chat, stream_id: {stream_id}",
@@ -44,6 +44,7 @@ def handle_leave(data) -> None:
     if stream_id:
         leave_room(stream_id)
         num_viewers = len(list(socketio.server.manager.get_participants("/", stream_id)))
+        update_viewers(stream_id, num_viewers)
         emit("status", 
             {
                 "message": f"Welcome to the chat, stream_id: {stream_id}",
@@ -120,7 +121,7 @@ def save_chat(chatter_id, stream_id, message):
                     VALUES (?, ?, ?);""", (chatter_id, stream_id, message))
     db.close_connection()
 
-def update_viewers(stream_id, num_viewers):
+def update_viewers(user_id, num_viewers):
     """
         Live Update the number of viewers in the stream to be
         displayed in the homepage or discovery pages
@@ -130,4 +131,5 @@ def update_viewers(stream_id, num_viewers):
                 UPDATE streams
                 SET num_viewers = ?
                 WHERE user_id = ?;
-                """)
\ No newline at end of file
+                """, (num_viewers, user_id))
+    db.close_connection
\ No newline at end of file