fix: 修复有时候不出现按钮的情况

之前去获取根元素的方式有误

Author: XYShaoKang

src/index.tsx

@@ -3,44 +3,34 @@ import { render } from 'react-dom'
 import minimatch from 'minimatch'
 
 import App from './App'
+import { getElement } from './utils'
 
-if (minimatch(location.href, 'https://leetcode-cn.com/submissions/detail/**')) {
-  window.onload = () => {
-    const parent =
-      document.querySelectorAll('#lc-header+div')?.[0]?.children?.[0]
-    if (parent && parent instanceof HTMLElement) {
-      parent.style.display = 'flex'
-      parent.style.justifyContent = 'space-between'
-      const root = document.createElement('div')
-      parent.append(root)
+function loadDownload(parent: Element) {
+  if (parent && parent instanceof HTMLElement) {
+    parent.style.display = 'flex'
+    parent.style.justifyContent = 'space-between'
+    const root = document.createElement('div')
+    parent.append(root)
 
-      render(
-        <StrictMode>
-          <App />
-        </StrictMode>,
-        root
-      )
-    }
+    render(
+      <StrictMode>
+        <App />
+      </StrictMode>,
+      root
+    )
   }
 }
 
-if (minimatch(location.href, 'https://leetcode.com/submissions/detail/**')) {
-  window.onload = () => {
-    const parent =
-      document.getElementById('submission-app')?.children?.[0]?.children?.[0]
-
-    if (parent && parent instanceof HTMLElement) {
-      parent.style.display = 'flex'
-      parent.style.justifyContent = 'space-between'
-      const root = document.createElement('div')
-      parent.append(root)
+if (minimatch(location.href, 'https://leetcode-cn.com/submissions/detail/**')) {
+  window.onload = async () => {
+    const main = await getElement('.css-smuvek-Main')
+    loadDownload(main[0]?.children?.[0])
+  }
+}
 
-      render(
-        <StrictMode>
-          <App />
-        </StrictMode>,
-        root
-      )
-    }
+if (minimatch(location.href, 'https://leetcode.com/submissions/detail/**')) {
+  window.onload = async () => {
+    const parent = await getElement('#submission-app>.row>div:first-child')
+    loadDownload(parent[0])
   }
 }

src/utils.ts

@@ -11,4 +11,26 @@ function download(str: string, filename = 'contest.md'): void {
   document.body.removeChild(a)
 }
 
-export { download }
+function getElement(
+  query: string,
+  fn: (e: NodeListOf<Element>) => boolean = e => e.length > 0,
+  timeout = 10000
+): Promise<NodeListOf<Element>> {
+  const delay = 100
+  return new Promise(function (resolve, reject) {
+    const timer = setInterval(() => {
+      const element = document.querySelectorAll(query)
+      if (fn(element)) {
+        clearInterval(timer)
+        resolve(element)
+      }
+      if (timeout <= 0) {
+        clearInterval(timer)
+        reject('超时')
+      }
+      timeout -= delay
+    }, delay)
+  })
+}
+
+export { download, getElement }