Merge pull request #45 from techsin/main

Update index.js
speed-highlight · Jun 2, 2024 · e93a2ec · e93a2ec
2 parents 249ad8f + d9ef6a4
commit e93a2ec
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/index.js b/src/index.js
@@ -159,9 +159,9 @@ export async function highlightElement(elm, lang = elm.className.match(/shj-lang
  * @param {ShjOptions} [opt={}] Customization options
  */
 export let highlightAll = async (opt) =>
-	document
-		.querySelectorAll('[class*="shj-lang-"]')
-		.forEach(elm => highlightElement(elm, undefined, undefined, opt))
+	await Promise.all(
+		Array.from(document.querySelectorAll('[class*="shj-lang-"]'))
+		.map(elem => highlightElement(elm, undefined, undefined, opt)))
 
 /**
  * @typedef {{ match: RegExp, type: string }
@@ -183,4 +183,4 @@ export let highlightAll = async (opt) =>
  */
 export let loadLanguage = (languageName, language) => {
 	langs[languageName] = language;
-}
+}