From 1a714d214b1bfa81d95d7d661592de94bdc1e8e8 Mon Sep 17 00:00:00 2001
From: Lim Chee Aun <cheeaun@gmail.com>
Date: Sat, 16 Sep 2023 15:45:09 +0800
Subject: [PATCH] Fix not all classes removed

This is due to DomTokenList being dynamic, looping it while removing items from it cause wrong indices
---
 src/utils/enhance-content.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/utils/enhance-content.js b/src/utils/enhance-content.js
index 9ccc110a..2573a24b 100644
--- a/src/utils/enhance-content.js
+++ b/src/utils/enhance-content.js
@@ -1,6 +1,7 @@
 import emojifyText from './emojify-text';
 
 const fauxDiv = document.createElement('div');
+const whitelistLinkClasses = ['u-url', 'mention', 'hashtag'];
 
 function enhanceContent(content, opts = {}) {
   const { emojis, postEnhanceDOM = () => {} } = opts;
@@ -22,10 +23,9 @@ function enhanceContent(content, opts = {}) {
 
     // Remove all classes except `u-url`, `mention`, `hashtag`
     const links = Array.from(dom.querySelectorAll('a[class]'));
-    const whitelistClasses = ['u-url', 'mention', 'hashtag'];
     links.forEach((link) => {
-      link.classList.forEach((c) => {
-        if (!whitelistClasses.includes(c)) {
+      Array.from(link.classList).forEach((c) => {
+        if (!whitelistLinkClasses.includes(c)) {
           link.classList.remove(c);
         }
       });