fix: fallback unknown code language

Crossbell-Box · May 21, 2024 · 9b374cc · 9b374cc
1 parent d924b0d
commit 9b374cc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/packages/shiki/shiki.server.tsx b/packages/shiki/shiki.server.tsx
@@ -16,6 +16,10 @@ export const ShikiRender: FC<ShikiCodeProps> = async ({
 }) => {
   const highlighter = await createHighlighter()
 
+  if (!Object.keys(bundledLanguages).includes(language || "")) {
+    language = "text"
+  }
+
   const rendered = highlighter.codeToHtml(code, {
     lang: language || "text",
     themes: codeTheme || {

diff --git a/src/components/ui/ShikiRemark.tsx b/src/components/ui/ShikiRemark.tsx
@@ -30,7 +30,7 @@ const pickCodeLanguage = (props: any) => {
       ? (props.children.props.className as string)
       : ""
 
-  if (className.includes("language-")) {
+  if (className?.includes("language-")) {
     return className.replace("language-", "")
   }
   return ""