Use arrays for status/action-bar selectors

Fixes #8: No icon in replies on a single status page
prplecake · Feb 6, 2022 · 9ab16dc · 9ab16dc
1 parent 03575eb
commit 9ab16dc
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/js/mastodon/mastopinner.js b/js/mastodon/mastopinner.js
@@ -1,4 +1,7 @@
-const STATUS_SELECTOR = '.item-list .status';
+const STATUS_SELECTORS = [
+    '.item-list .status',
+    '.scrollable .status'
+];
 
 class MastoPinner {
     constructor() {
@@ -17,7 +20,12 @@ class MastoPinner {
         }
 
         // inject the initially visible statuses
-        let statuses = document.querySelectorAll(STATUS_SELECTOR);
+        let statuses = [];
+        STATUS_SELECTORS.forEach((selector) => {
+            document.querySelectorAll(selector).forEach((s) => {
+                statuses.push(s);
+            });
+        });
         if (statuses.length > 0) {
             this._inject();
         }
@@ -30,7 +38,12 @@ class MastoPinner {
     }
 
     async _inject() {
-        let statuses = document.querySelectorAll(STATUS_SELECTOR);
+        let statuses = [];
+        STATUS_SELECTORS.forEach((selector) => {
+            document.querySelectorAll(selector).forEach((s) => {
+                statuses.push(s);
+            });
+        });
         if (statuses.length == 0) {
             console.warn(`MastoPinner: couldn't inject the UI: no statuses found.`);
             return;

diff --git a/js/mastodon/mastopinner_ui.js b/js/mastodon/mastopinner_ui.js
@@ -1,3 +1,7 @@
+const ACTION_BAR_SELECTORS = [
+    'div.status__action-bar'
+];
+
 class MastoPinnerUI {
     constructor(status, { send }) {
         this.status = status;
@@ -6,7 +10,7 @@ class MastoPinnerUI {
     }
 
     render() {
-        this.actionBar = this.status.querySelector('div.status__action-bar');
+        this.actionBar = this.status.querySelector(ACTION_BAR_SELECTORS);
 
         // define node to insert
         let nodeHtml = `<button id="mastopinner-button"