Fix incorrect image for reddit albums

qsniyg · Feb 1, 2025 · 506899d · 506899d
1 parent 19c1290
commit 506899d
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 2 deletions.
diff --git a/src/userscript.ts b/src/userscript.ts
@@ -26578,6 +26578,9 @@ var $$IMU_EXPORT$$;
 	};
 
 	common_functions["update_album_info_links"] = function(obj, cmp) {
+		if (!obj.album_info)
+			return obj;
+
 		var links = obj.album_info.links;
 
 		var found = false;
@@ -60288,6 +60291,7 @@ var $$IMU_EXPORT$$;
 
 							album_links.push({
 								url: url,
+								x_id: mediaitem.media_id,
 								is_current: first_url === url
 							});
 						});
@@ -60314,6 +60318,27 @@ var $$IMU_EXPORT$$;
 
 			newsrc = website_query({
 				website_regex: /^[a-z]+:\/\/[^/]+\/+(?:(?:(?:r|u(?:ser)?)\/+[^/]+\/+comments|gallery)\/+)?([a-z0-9]+)(?:\/+.*)?(?:[?#].*)?$/,
+				override_cb: function(cb, data) {
+					if (!is_array(data))
+						data = [data];
+
+					let id = null;
+					let imu_id = src.match(/#imu-id=(.*)/);
+					if (imu_id)
+						id = imu_id[1];
+
+					if (id) {
+						id = id.replace(/^.*\/([0-9a-z]+)\.[a-z0-9]+(?:[?#].*)?$/, "$1");
+					}
+
+					for (let sobj of data) {
+						common_functions["update_album_info_links"](sobj, function(x) {
+							return x === id;
+						});
+					}
+
+					cb(data);
+				},
 				run: function(cb, match) {
 					var id = match[1];
 					// t3 = link
@@ -60333,8 +60358,12 @@ var $$IMU_EXPORT$$;
 			var get_reddit_link = function(id) {
 				id = id.replace(/^t[0-9]+_/, "");
 
+				let url = "https://redd.it/" + id;
+				if (src)
+					url += "#imu-id=" + src;
+
 				return {
-					url: "https://redd.it/" + id,
+					url,
 					is_pagelink: true
 				};
 			};

diff --git a/userscript.user.js b/userscript.user.js
@@ -24393,6 +24393,8 @@ var $$IMU_EXPORT$$;
 		return fillobj_urls(urls, obj);
 	};
 	common_functions["update_album_info_links"] = function(obj, cmp) {
+		if (!obj.album_info)
+			return obj;
 		var links = obj.album_info.links;
 		var found = false;
 		array_foreach(links, function(link, i) {
@@ -54884,6 +54886,7 @@ var $$IMU_EXPORT$$;
 								first_url = url;
 							album_links.push({
 								url: url,
+								x_id: mediaitem.media_id,
 								is_current: first_url === url
 							});
 						});
@@ -54906,6 +54909,24 @@ var $$IMU_EXPORT$$;
 			};
 			newsrc = website_query({
 				website_regex: /^[a-z]+:\/\/[^/]+\/+(?:(?:(?:r|u(?:ser)?)\/+[^/]+\/+comments|gallery)\/+)?([a-z0-9]+)(?:\/+.*)?(?:[?#].*)?$/,
+				override_cb: function(cb, data) {
+					if (!is_array(data))
+						data = [data];
+					var id = null;
+					var imu_id = src.match(/#imu-id=(.*)/);
+					if (imu_id)
+						id = imu_id[1];
+					if (id) {
+						id = id.replace(/^.*\/([0-9a-z]+)\.[a-z0-9]+(?:[?#].*)?$/, "$1");
+					}
+					for (var _i = 0, data_2 = data; _i < data_2.length; _i++) {
+						var sobj = data_2[_i];
+						common_functions["update_album_info_links"](sobj, function(x) {
+							return x === id;
+						});
+					}
+					cb(data);
+				},
 				run: function(cb, match) {
 					var id = match[1];
 					// t3 = link
@@ -54924,8 +54945,11 @@ var $$IMU_EXPORT$$;
 			host_domain_nosub === "reddit.com" && options.element) {
 			var get_reddit_link = function(id) {
 				id = id.replace(/^t[0-9]+_/, "");
+				var url = "https://redd.it/" + id;
+				if (src)
+					url += "#imu-id=" + src;
 				return {
-					url: "https://redd.it/" + id,
+					url: url,
 					is_pagelink: true
 				};
 			};