From 965c51c2d45a98db10543c5108ac486e4fc6459e Mon Sep 17 00:00:00 2001
From: Mario <mario@mariovavti.com>
Date: Fri, 15 Nov 2019 20:29:58 +0000
Subject: sse: implement notifications for anonymous visitors (info, notice and
 pubs) and fix a potential memory leak

---
 Zotlabs/Module/Sse.php    | 31 +++++++++++++++++++++++++------
 Zotlabs/Module/Sse_bs.php | 32 +++++++++++++++++++++++++-------
 2 files changed, 50 insertions(+), 13 deletions(-)

(limited to 'Zotlabs/Module')

diff --git a/Zotlabs/Module/Sse.php b/Zotlabs/Module/Sse.php
index 556fe2853..b227a396d 100644
--- a/Zotlabs/Module/Sse.php
+++ b/Zotlabs/Module/Sse.php
@@ -11,22 +11,41 @@ class Sse extends Controller {
 
 	public static $uid;
 	public static $ob_hash;
+	public static $sse_id;
 	public static $vnotify;
 
 	function init() {
 
+		if((observer_prohibited(true))) {
+			killme();
+		}
+
+		if(! intval(get_config('system','open_pubstream',1))) {
+			if(! get_observer_hash()) {
+				killme();
+			}
+		}
+
 		// this is important!
 		session_write_close();
 
-		$sys = get_sys_channel();
-
 		self::$uid = local_channel();
 		self::$ob_hash = get_observer_hash();
-		self::$vnotify = get_pconfig(self::$uid, 'system', 'vnotify');
+		self::$sse_id = false;
 
-		if(! self::$ob_hash)
-			return;
+		if(! self::$ob_hash) {
+			if(session_id()) {
+				self::$sse_id = true;
+				self::$ob_hash = 'sse_id.' . session_id();
+			}
+			else {
+				return;
+			}
+		}
 
+		self::$vnotify = get_pconfig(self::$uid, 'system', 'vnotify');
+
+		$sys = get_sys_channel();
 		$sleep_seconds = 3;
 
 		header("Content-Type: text/event-stream");
@@ -40,7 +59,7 @@ class Sse extends Controller {
 			 * Update chat presence indication (if applicable)
 			 */
 
-			if(self::$ob_hash) {
+			if(! self::$sse_id) {
 				$r = q("select cp_id, cp_room from chatpresence where cp_xchan = '%s' and cp_client = '%s' and cp_room = 0 limit 1",
 					dbesc(self::$ob_hash),
 					dbesc($_SERVER['REMOTE_ADDR'])
diff --git a/Zotlabs/Module/Sse_bs.php b/Zotlabs/Module/Sse_bs.php
index c7fcd0542..270e8b9b9 100644
--- a/Zotlabs/Module/Sse_bs.php
+++ b/Zotlabs/Module/Sse_bs.php
@@ -11,6 +11,7 @@ class Sse_bs extends Controller {
 
 	public static $uid;
 	public static $ob_hash;
+	public static $sse_id;
 	public static $vnotify;
 	public static $evdays;
 	public static $limit;
@@ -21,16 +22,26 @@ class Sse_bs extends Controller {
 
 		self::$uid = local_channel();
 		self::$ob_hash = get_observer_hash();
+		self::$sse_id = false;
+
+		if(! self::$ob_hash) {
+			if(session_id()) {
+				self::$sse_id = true;
+				self::$ob_hash = 'sse_id.' . session_id();
+			}
+			else {
+				return;
+			}
+		}
+
 		self::$vnotify = get_pconfig(self::$uid, 'system', 'vnotify');
 		self::$evdays = intval(get_pconfig(self::$uid, 'system', 'evdays'));
 		self::$limit = 100;
 		self::$offset = 0;
 		self::$xchans = '';
 
-		if(self::$ob_hash) {
-			set_xconfig(self::$ob_hash, 'sse', 'timestamp', datetime_convert());
-			set_xconfig(self::$ob_hash, 'sse', 'language', App::$language);
-		}
+		set_xconfig(self::$ob_hash, 'sse', 'timestamp', datetime_convert());
+		set_xconfig(self::$ob_hash, 'sse', 'language', App::$language);
 
 		if(!empty($_GET['nquery']) && $_GET['nquery'] !== '%') {
 			$nquery = $_GET['nquery'];
@@ -69,9 +80,6 @@ class Sse_bs extends Controller {
 			default:
 		}
 
-		//hz_syslog('init: ' . argv(1));
-		//hz_syslog('offset: ' . argv(2));
-
 		if(self::$offset && $f) {
 			$result = self::$f(true);
 			json_return_and_die($result);
@@ -227,6 +235,16 @@ class Sse_bs extends Controller {
 		$result['pubs']['notifications'] = [];
 		$result['pubs']['count'] = 0;
 
+		if((observer_prohibited(true))) {
+			return $result;
+		}
+
+		if(! intval(get_config('system','open_pubstream',1))) {
+			if(! get_observer_hash()) {
+				return $result;
+			}
+		}
+
 		if(! isset($_SESSION['static_loadtime']))
 			$_SESSION['static_loadtime'] = datetime_convert();
 
-- 
cgit v1.2.3