fix(problems): 修复查找编辑器元素的逻辑

XYShaoKang · Jul 15, 2023 · 379a757 · 379a757
1 parent b55a2e4
commit 379a757
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/content/pages/problems/Timer.tsx b/src/content/pages/problems/Timer.tsx
@@ -7,6 +7,7 @@ import {
   SuccessCheckReturnType,
   findElement,
   findElementByXPath,
+  findAllElement,
 } from '@/utils'
 import { useEvent, useHover, useObserverAncestor } from '@/hooks'
 import { ToolTip } from '@/components/ToolTip'
@@ -375,7 +376,11 @@ const Timer: FC<TimerProps> = ({ beta, root, dynamicLayout }) => {
     let editEl: HTMLElement
 
     if (beta) {
-      editEl = await findElement('.monaco-editor')
+      const editEls = await findAllElement('.monaco-editor')
+      editEl = editEls.find(
+        el => el.parentElement?.dataset.modeId !== 'plaintext'
+      )!
+      if (!editEl) throw new Error('未找到编辑器元素')
     } else {
       editEl = await findElement('.euyvu2f0')
     }