From 04d7e550114ce190a0bfc8d3739aaf930e9c38e7 Mon Sep 17 00:00:00 2001
From: ThisBirchWood <dylanseandefaoite@gmail.com>
Date: Tue, 4 Mar 2025 13:57:47 +0000
Subject: [PATCH] REFACTOR: Cleaned up routes

---
 web_server/blueprints/streams.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/web_server/blueprints/streams.py b/web_server/blueprints/streams.py
index 2d36d30..c4e2375 100644
--- a/web_server/blueprints/streams.py
+++ b/web_server/blueprints/streams.py
@@ -388,18 +388,22 @@ def end_stream():
                                 FROM users 
                                 WHERE stream_key = ?""", (stream_key,))
         
+        # Return unauthorized if no user found
         if not user_info:
             print("Unauthorized - No user found from stream key", flush=True)
             return "Unauthorized", 403
             
+        # Get user info
         user_id = user_info["user_id"]
         username = user_info["username"]
     
+    # End stream
     result, message = end_user_stream(stream_key, user_id, username)
     
-    if result:
-        print(f"Stream ended: {message}", flush=True)
-        return "Stream ended", 200
-    else:
+    # Return error if stream could not be ended
+    if not result:
         print(f"Error ending stream: {message}", flush=True)
         return "Error ending stream", 500
+    
+    print(f"Stream ended: {message}", flush=True)
+    return "Stream ended", 200