From 2f02940ab37265e89f62a1ea8981181f4fa9413a Mon Sep 17 00:00:00 2001
From: Aevann1 <randomname42029@gmail.com>
Date: Fri, 10 Dec 2021 02:39:39 +0200
Subject: [PATCH] localstorage for emojis

---
 files/__main__.py                       |  1 +
 files/assets/js/emoji_modal.js          | 20 ++++++++++++++++++++
 files/helpers/jinja2.py                 |  8 --------
 files/helpers/sanitize.py               | 15 ---------------
 files/templates/emoji_modal.html        |  5 +----
 files/templates/submission.html         | 15 +++++++++++++++
 files/templates/submission_listing.html |  8 ++++++++
 7 files changed, 45 insertions(+), 27 deletions(-)

diff --git a/files/__main__.py b/files/__main__.py
index a1656c799..816ecfb11 100644
--- a/files/__main__.py
+++ b/files/__main__.py
@@ -122,6 +122,7 @@ def teardown_request(error):
 
 @app.after_request
 def after_request(response):
+	if session.get("favorite_emojis"): del session["favorite_emojis"]
 
 	response.headers.add("Strict-Transport-Security", "max-age=31536000")
 	response.headers.add("X-Frame-Options", "deny")
diff --git a/files/assets/js/emoji_modal.js b/files/assets/js/emoji_modal.js
index a371c698c..c9a2a9477 100644
--- a/files/assets/js/emoji_modal.js
+++ b/files/assets/js/emoji_modal.js
@@ -790,6 +790,11 @@ function getEmoji(searchTerm) {
 	commentBox.setAttribute(TEXTAREA_POS_ATTR, newPos.toString());
 
 	if (typeof checkForRequired === "function") checkForRequired();
+
+	const favorite_emojis = JSON.parse(localStorage.getItem("favorite_emojis")) || {}
+	if (favorite_emojis[searchTerm]) favorite_emojis[searchTerm] += 1
+	else favorite_emojis[searchTerm] = 1
+	window.localStorage.setItem("favorite_emojis", JSON.stringify(favorite_emojis))	
 }
 
 function loadEmojis(form) {
@@ -856,3 +861,18 @@ function loadEmojis(form) {
 		loadEmojis(form);
 	};
 }
+
+let str = ""
+const favorite_emojis = JSON.parse(localStorage.getItem("favorite_emojis"))
+
+if (favorite_emojis)
+{
+	const sortable = Object.fromEntries(
+		Object.entries(favorite_emojis).sort(([,a],[,b]) => b-a)
+	);
+			
+	for (const emoji of Object.keys(sortable))
+		str += `<button class="btn m-1 px-0 emoji2" onclick="getEmoji('${emoji}')" data-bs-toggle="tooltip" title=":${emoji}:" delay:="0"><img loading="lazy" width=50 src="/assets/images/emojis/${emoji}.webp" alt="${emoji}-emoji"></button>`
+
+	document.getElementById('EMOJIS_favorite').innerHTML = str
+}
\ No newline at end of file
diff --git a/files/helpers/jinja2.py b/files/helpers/jinja2.py
index 70913e02e..e0318c80f 100644
--- a/files/helpers/jinja2.py
+++ b/files/helpers/jinja2.py
@@ -22,14 +22,6 @@ def post_embed(id, v):
 	
 	return render_template("submission_listing.html", listing=[p], v=v)
 
-@app.template_filter("favorite_emojis")
-def favorite_emojis(x):
-	str = ""
-	emojis = sorted(x.items(), key=lambda x: x[1], reverse=True)[:25]
-	for k, v in emojis:
-		str += f'<button class="btn m-1 px-0 emoji2" onclick="getEmoji(\'{k}\')" data-bs-toggle="tooltip" title=":{k}:" delay:="0"><img loading="lazy" width=50 src="/assets/images/emojis/{k}.webp" alt="{k}-emoji"></button>'
-	return str
-
 @app.context_processor
 def inject_constants():
 	constants = [c for c in dir(const) if not c.startswith("_")]
diff --git a/files/helpers/sanitize.py b/files/helpers/sanitize.py
index ed8f2eac0..a18c0f455 100644
--- a/files/helpers/sanitize.py
+++ b/files/helpers/sanitize.py
@@ -164,11 +164,6 @@ def sanitize(sanitized, noimages=False):
 	start = '&lt;s&gt;'
 	end = '&lt;/s&gt;' 
 
-	try:
-		if not session.get("favorite_emojis"): session["favorite_emojis"] = {}
-	except:
-		pass
-
 	if start in sanitized and end in sanitized and start in sanitized.split(end)[0] and end in sanitized.split(start)[1]: sanitized = sanitized.replace(start, '<span class="spoiler">').replace(end, '</span>')
 	
 	for i in re.finditer("[^a]>\s*(:[!#]{0,2}\w+:\s*)+<\/", sanitized):
@@ -192,9 +187,6 @@ def sanitize(sanitized, noimages=False):
 
 			if path.isfile(f'./files/assets/images/emojis/{remoji}.webp'):
 				new = re.sub(f'(?<!"):{emoji}:', f'<img loading="lazy" data-bs-toggle="tooltip" alt=":{emoji}:" title=":{emoji}:" delay="0" {classes}src="https://{site}/assets/images/emojis/{remoji}.webp" >', new)
-
-				if remoji in session["favorite_emojis"]: session["favorite_emojis"][remoji] += 1
-				else: session["favorite_emojis"][remoji] = 1
 					
 		sanitized = sanitized.replace(old, new)
 
@@ -205,16 +197,9 @@ def sanitize(sanitized, noimages=False):
 			emoji = emoji[1:]
 			if path.isfile(f'./files/assets/images/emojis/{emoji}.webp'):
 				sanitized = re.sub(f'(?<!"):!{emoji}:', f'<img loading="lazy" data-bs-toggle="tooltip" alt=":!{emoji}:" title=":!{emoji}:" delay="0" height=30 class="emj mirrored" src="https://{site}/assets/images/emojis/{emoji}.webp">', sanitized)
-		
-				if emoji in session["favorite_emojis"]: session["favorite_emojis"][emoji] += 1
-				else: session["favorite_emojis"][emoji] = 1
 
 		elif path.isfile(f'./files/assets/images/emojis/{emoji}.webp'):
 			sanitized = re.sub(f'(?<!"):{emoji}:', f'<img loading="lazy" data-bs-toggle="tooltip" alt=":{emoji}:" title=":{emoji}:" delay="0" height=30 class="emj" src="https://{site}/assets/images/emojis/{emoji}.webp">', sanitized)
-				
-			if emoji in session["favorite_emojis"]: session["favorite_emojis"][emoji] += 1
-			else: session["favorite_emojis"][emoji] = 1
-
 
 	sanitized = sanitized.replace("https://www.", "https://").replace("https://youtu.be/", "https://youtube.com/watch?v=").replace("https://music.youtube.com/watch?v=", "https://youtube.com/watch?v=").replace("https://open.spotify.com/", "https://open.spotify.com/embed/").replace("https://streamable.com/", "https://streamable.com/e/").replace("https://youtube.com/shorts/", "https://youtube.com/watch?v=").replace("https://mobile.twitter", "https://twitter").replace("https://m.facebook", "https://facebook").replace("https://m.wikipedia", "https://wikipedia").replace("https://m.youtube", "https://youtube")
 
diff --git a/files/templates/emoji_modal.html b/files/templates/emoji_modal.html
index ccae485ac..9be7e5b4e 100644
--- a/files/templates/emoji_modal.html
+++ b/files/templates/emoji_modal.html
@@ -48,9 +48,6 @@
 					<div class="tab-content">
 						<div class="tab-pane fade show active" id="emoji-tab-favorite">
 							<div class="d-flex flex-wrap py-3 pl-2" id="EMOJIS_favorite">
-								{% if session.get("favorite_emojis") %}
-									{{session.get("favorite_emojis") | favorite_emojis | safe}}
-								{% endif %}								
 							</div>
 						</div>
 						<div class="tab-pane fade" id="emoji-tab-marsey">
@@ -84,7 +81,7 @@
 	</div>
 </div>
 
-<script src="/assets/js/emoji_modal.js?v=140"></script>
+<script src="/assets/js/emoji_modal.js?v=146"></script>
 
 <style>
 	a.emojitab {
diff --git a/files/templates/submission.html b/files/templates/submission.html
index 332b6e766..dc3857d8b 100644
--- a/files/templates/submission.html
+++ b/files/templates/submission.html
@@ -869,4 +869,19 @@
 {% endif %}
 
 
+<script>
+	(() => {
+		{% if (not v or v.highlightcomments) %}
+			showNewCommentCounts('{{p.id}}', {{p.comment_count}})
+		{% endif %}
+
+		{% if "?context" not in request.full_path %}
+			const comments = JSON.parse(localStorage.getItem("comment-counts")) || {}
+			const newTotal = {{p.comment_count}} || ((comments['{{p.id}}'] || { c: 0 }).c + 1)
+			comments['{{p.id}}'] = { c: newTotal, t: Date.now() }
+			window.localStorage.setItem("comment-counts", JSON.stringify(comments))
+		{% endif %}
+	})()
+</script>
+
 {% endblock %}
\ No newline at end of file
diff --git a/files/templates/submission_listing.html b/files/templates/submission_listing.html
index 10f77920b..af975f36a 100644
--- a/files/templates/submission_listing.html
+++ b/files/templates/submission_listing.html
@@ -55,6 +55,14 @@
 	</div>
 </div>
 
+<script>
+	(() => {
+		{% if (not v or v.highlightcomments) %}
+			showNewCommentCounts('{{p.id}}', {{p.comment_count}})
+		{% endif %}
+	})()
+</script>
+
 {% set ups=p.upvotes %}
 {% set downs=p.downvotes %}
 {% set score=ups-downs %}