Fix: Use 'flag' link if 'hide' not present for faves

plibither8 · Dec 3, 2019 · 324135f · 324135f
1 parent c832af3
commit 324135f
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/features/more-accessible-favorite.js b/src/features/more-accessible-favorite.js
@@ -134,9 +134,17 @@ async function storyButtons(metadata) {
 	for (const item of items) {
 		const faveButton = item.querySelector('.__rhn__fave-button');
 
-		const hideUrl = item.querySelector('a[href^="hide"]').href.replace('?', '&');
-		const auth = getUrlParams('auth', hideUrl);
-		const id = getUrlParams('id', hideUrl);
+		const authStringElement =
+			item.querySelector('a[href^="flag"]') ||
+			item.querySelector('a[href^="hide"]');
+
+		if (!authStringElement) {
+			continue;
+		}
+
+		const authStringUrl = authStringElement.href.replace('?', '&');
+		const auth = getUrlParams('auth', authStringUrl);
+		const id = getUrlParams('id', authStringUrl);
 
 		let unfave = false;