From c15f25e27bd78b523b2b7483c2148eb145518ac4 Mon Sep 17 00:00:00 2001
From: ayaka <ayaka@kitty.community>
Date: Tue, 9 Apr 2024 01:27:40 +1200
Subject: [PATCH] Handle preview embedding better

---
 src/utils.rs     | 33 ++++++++++++++++-----------------
 static/style.css |  2 ++
 2 files changed, 18 insertions(+), 17 deletions(-)

diff --git a/src/utils.rs b/src/utils.rs
index 9b67e0b..63e679d 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -877,8 +877,8 @@ pub fn format_url(url: &str) -> String {
 static REDDIT_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r#"href="(https|http|)://(www\.|old\.|np\.|amp\.|new\.|)(reddit\.com|redd\.it)/"#).unwrap());
 static REDDIT_PREVIEW_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r"https?://(external-preview|preview)\.redd\.it(.*)[^?]").unwrap());
 static REDDIT_EMOJI_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r"https?://(www|).redditstatic\.com/(.*)").unwrap());
-static REDLIB_PREVIEW_LINK_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r#"/preview/(pre|external-pre)/(.*)">"#).unwrap());
-static REDLIB_PREVIEW_TEXT_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r">(.*)</a>").unwrap());
+static REDLIB_PREVIEW_LINK_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r#"/preview/(pre|external-pre)/(.*?)>"#).unwrap());
+static REDLIB_PREVIEW_TEXT_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r">(.*?)</a>").unwrap());
 
 // Rewrite Reddit links to Redlib in body of text
 pub fn rewrite_urls(input_text: &str) -> String {
@@ -894,24 +894,23 @@ pub fn rewrite_urls(input_text: &str) -> String {
 		.replace("\\_", "_");
 
 	// Rewrite external media previews to Redlib
-	if REDDIT_PREVIEW_REGEX.is_match(&text1) {
-		loop {
-			if REDDIT_PREVIEW_REGEX.find(&text1).is_none() {
-				return text1;
-			} else {
-				let formatted_url = format_url(REDDIT_PREVIEW_REGEX.find(&text1).map(|x| x.as_str()).unwrap_or_default());
+	loop {
+		if REDDIT_PREVIEW_REGEX.find(&text1).is_none() {
+			return text1;
+		} else {
+			let formatted_url = format_url(REDDIT_PREVIEW_REGEX.find(&text1).map(|x| x.as_str()).unwrap_or_default());
 
-				let image_url = REDLIB_PREVIEW_LINK_REGEX.find(&formatted_url).map_or("", |m| m.as_str()).to_string();
-				let image_text = REDLIB_PREVIEW_TEXT_REGEX.find(&formatted_url).map_or("", |m| m.as_str()).to_string();
+			let image_url = REDLIB_PREVIEW_LINK_REGEX.find(&formatted_url).map_or("", |m| m.as_str()).to_string();
+			let image_text = REDLIB_PREVIEW_TEXT_REGEX.find(&formatted_url).map_or("", |m| m.as_str()).to_string();
 
-				text1 = REDDIT_PREVIEW_REGEX
-					.replace(&text1, format_url(REDDIT_PREVIEW_REGEX.find(&text1).map(|x| x.as_str()).unwrap_or_default()))
-					.replace(&image_text, &format!("><img src=\"{image_url}</a>"))
-					.to_string()
-			}
+			let image_to_replace = format!("<a href=\"{image_url}{image_text}").replace(">>", ">");
+			let image_replacement = format!("<a href=\"{image_url}<img src=\"{image_url}</a>");
+
+			text1 = REDDIT_PREVIEW_REGEX
+				.replace(&text1, formatted_url)
+				.replace(&image_to_replace, &image_replacement)
+				.to_string()
 		}
-	} else {
-		text1
 	}
 }
 
diff --git a/static/style.css b/static/style.css
index 5e8236d..a85ab75 100644
--- a/static/style.css
+++ b/static/style.css
@@ -1177,11 +1177,13 @@ a.search_subreddit:hover {
 
 .comment img {
 	max-width: 50%;
+	height: auto;
 }
 
 @media screen and (max-width: 500px) {
 	.comment img {
 		max-width: 80%;
+		height: auto;
 	}
 }