fix: 找到正确的图标元素

修复在比赛进行期间,会对 WA 图标进行替换
XYShaoKang · Mar 21, 2022 · 1304e03 · 1304e03
1 parent cc2b651
commit 1304e03
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/content/pages/ranking/index.tsx b/src/content/pages/ranking/index.tsx
@@ -67,11 +67,11 @@ async function loadFileIcon() {
   trs.forEach((tr, i) => {
     const codetds = Array.from(tr.children).slice(4, 8)
     for (let j = 0; j < 4; j++) {
-      const iconEl = codetds[j]?.children?.[0]?.children?.[0]
+      const parent = codetds[j]?.querySelector('a')
+      const iconEl = parent?.querySelector('span')
 
-      if (iconEl) {
+      if (parent && iconEl) {
         if (fileIconStates[i][j]) continue
-        const parent = iconEl.parentNode! as HTMLAnchorElement
         parent.removeChild(iconEl)
 
         render(