diff --git a/src/webpage/highlighter/index.ts b/src/webpage/highlighter/index.ts index ac4f746..af94639 100644 --- a/src/webpage/highlighter/index.ts +++ b/src/webpage/highlighter/index.ts @@ -7,7 +7,14 @@ export async function highlight( skipStart: number = 0, skipEnd: number = 0, ) { - await new Promise((res) => setTimeout(res)); + let i = 0; + while (!document.body.contains(elm)) { + for (let i = 0; i < 4; i++) { + await new Promise((res) => queueMicrotask(res)); + } + i++; + if (i === 100) break; + } lang = lang.trim(); let content = elm.textContent; if (skipStart) {