Fix: unknown code should still render in the app

mrdjohnson · Apr 6, 2024 · 92de20d · 92de20d
1 parent 6fc7cc3
commit 92de20d
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/src/components/CustomCodeBlock.tsx b/src/components/CustomCodeBlock.tsx
@@ -17,20 +17,26 @@ const CustomCodeBlock = (props: React.HTMLAttributes<HTMLElement>) => {
   }, [className, multiLine])
 
   const { highlightedText, language } = useMemo(() => {
-    if (languageOverride) {
-      const hightligted = hljs.highlight(languageOverride, text)
+    try {
+      if (languageOverride) {
+        const hightligted = hljs.highlight(languageOverride, text)
+
+        return {
+          highlightedText: hightligted.value,
+          language: languageOverride,
+        }
+      }
+      const hightligted = hljs.highlightAuto(text)
 
       return {
         highlightedText: hightligted.value,
-        language: languageOverride,
+        language: hightligted.language,
+      }
+    } catch (e) {
+      return {
+        highlightedText: text,
+        language: languageOverride || undefined,
       }
-    }
-
-    const hightligted = hljs.highlightAuto(text)
-
-    return {
-      highlightedText: hightligted.value,
-      language: hightligted.language,
     }
   }, [text])