Detect new class name used for chat list

Resolves #11
rantstats · Aug 24, 2023 · e13f05b · e13f05b
1 parent 5dd8879
commit e13f05b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/components/rumble/rumble.ts b/src/components/rumble/rumble.ts
@@ -7,15 +7,18 @@ import {CHAT_POPUP_REGEX} from "../../types/consts";
  */
 export const getVideoID = (): string => {
     const m = CHAT_POPUP_REGEX.exec(window.location.pathname)
-    if (m != null){
+    if (m != null) {
         return m[1]
     }
 
     const foundIds = []
     const idElements = document.querySelectorAll('[data-id]')
     idElements.forEach((element) => {
         if (
-                element.classList.contains("rumbles-vote") &&
+                (
+                        element.classList.contains("rumbles-vote") ||
+                        element.classList.contains("rumbles-vote-pill")
+                ) &&
                 element.getAttribute('data-type') === '1'
         ) {
             foundIds.push(element.getAttribute('data-id'))