From dd940ff46f7d4a0628b13a714ebfa416bb8a2d9d Mon Sep 17 00:00:00 2001
From: knotteye <knotteye@airmail.cc>
Date: Fri, 15 Jan 2021 21:28:38 -0600
Subject: [PATCH] Fix bugs with updating viewer count

---
 src/http.ts   | 1 +
 src/server.ts | 9 ++-------
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/src/http.ts b/src/http.ts
index e154fc8..697e378 100644
--- a/src/http.ts
+++ b/src/http.ts
@@ -630,6 +630,7 @@ async function initChat() {
 			let rooms = Object.keys(socket.rooms);
 			for(let i=1;i<rooms.length;i++){
 				io.to(rooms[i]).emit('ALERT', socket.nick+' disconnected');
+				db.query('update user_meta set viewers = viewers - 1 where username='+db.raw.escape(rooms[i]));
 			}
 			if(Array.isArray(store.get(socket.nick))) {
 				store.set(socket.nick, store.get(socket.nick).filter(item => item !== socket.id))
diff --git a/src/server.ts b/src/server.ts
index 8f05596..924926f 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -129,13 +129,8 @@ function init () {
 		let session = nms.getSession(id);
 		let app: string = StreamPath.split("/")[1];
 		let key: string = StreamPath.split("/")[2];
-		if(!session.isLocal && app === config['media']['publicEndpoint']) {
-			if(keystore[key]){
-				// increment viewer count
-				db.query('update user_meta set viewers = viewers - 1 where username='+db.raw.escape(key));
-				return true;
-			}
-		}
+		// decrement viewer count
+		db.query('update user_meta,users set user_meta.viewers = user_meta.viewers - 1 where users.stream_key='+db.raw.escape(key));
 	});
 }