From 663904fb3d6a27c7bd86d8ef5cb3a9d173bb5170 Mon Sep 17 00:00:00 2001
From: Aevann1 <randomname42029@gmail.com>
Date: Sun, 21 Aug 2022 19:20:09 +0200
Subject: [PATCH] steal the parts I wanted from spidey's PR

---
 files/helpers/alerts.py   | 13 ++++---------
 files/helpers/sanitize.py | 29 +++++++++++++++--------------
 2 files changed, 19 insertions(+), 23 deletions(-)

diff --git a/files/helpers/alerts.py b/files/helpers/alerts.py
index a0eb82dbd..6e0b81733 100644
--- a/files/helpers/alerts.py
+++ b/files/helpers/alerts.py
@@ -87,15 +87,10 @@ def NOTIFY_USERS(text, v):
 		if id == 0 or v.id == id: continue
 		if word in text.lower() and id not in notify_users: notify_users.add(id)
 
-	captured = []
-	for i in mention_regex.finditer(text):
-		if v.username.lower() == i.group(2).lower(): continue
-
-		if i.group(0) in captured: continue
-		captured.append(i.group(0))
-
-		user = get_user(i.group(2), graceful=True)
-		if user and v.id != user.id and not v.any_block_exists(user): notify_users.add(user.id)
+	names = set(m.group(2) for m in mention_regex.finditer(text))
+	for user in get_users(names, graceful=True):
+		if v.id != user.id and not v.any_block_exists(user):
+			notify_users.add(user.id)
 
 	if SITE_NAME == "WPD" and 'daisy' in text.lower():
 		admin_ids = [x[0] for x in g.db.query(User.id).filter(User.admin_level > 0).all()]
diff --git a/files/helpers/sanitize.py b/files/helpers/sanitize.py
index be7d8aa45..b17136a0e 100644
--- a/files/helpers/sanitize.py
+++ b/files/helpers/sanitize.py
@@ -244,21 +244,22 @@ def sanitize(sanitized, edit=False, limit_pings=0, showmore=True):
 
 	v = getattr(g, 'v', None)
 
-	matches = [m for m in mention_regex.finditer(sanitized) if m]
-	names = set(m.group(2) for m in matches)
-	if limit_pings and len(names) > limit_pings and not v.admin_level: abort(406)
-	users = get_users(names, graceful=True)
+	names = set(m.group(2) for m in mention_regex.finditer(sanitized))
+	if len(names) > 100 and not v.admin_level: abort(406)
+	users_list = get_users(names, graceful=True)
+	users_dict = {}
+	for u in users_list:
+		users_dict[u.username.lower()] = u
+		if u.original_username:
+			users_dict[u.original_username.lower()] = u
 
-	for u in users:
-		if not u: continue
-		m = [m for m in matches if u.username.lower() == m.group(2).lower() or u.original_username.lower() == m.group(2).lower()]
-		for i in m:
-			if not (v and v.any_block_exists(u)) or (v and v.admin_level >= 2):
-				sanitized = re.sub(
-					f'{i.group(0)}($|[^a-zA-Z0-9_\-])',
-					rf'''{i.group(1)}<a href="/id/{u.id}"><img loading="lazy" src="/pp/{u.id}">@{u.username}</a>\1''',
-					sanitized
-				)
+	def replacer(m):
+		u = users_dict.get(m.group(2).lower())
+		if not u:
+			return m.group(0)
+		return f'{m.group(1)}<a href="/id/{u.id}"><img loading="lazy" src="/pp/{u.id}">@{u.username}</a>'
+
+	sanitized = mention_regex.sub(replacer, sanitized)
 
 	soup = BeautifulSoup(sanitized, 'lxml')