chore: add support for lua, haskell style magic comments

facebook · Apr 6, 2023 · 24cde7d · 24cde7d
1 parent 46d2aa2
commit 24cde7d
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/packages/docusaurus-theme-common/src/utils/codeBlockUtils.ts b/packages/docusaurus-theme-common/src/utils/codeBlockUtils.ts
@@ -19,6 +19,7 @@ const commentPatterns = {
   jsx: {start: '\\{\\s*\\/\\*', end: '\\*\\/\\s*\\}'},
   bash: {start: '#', end: ''},
   html: {start: '<!--', end: '-->'},
+  lua: {start: '--', end: ''},
 };
 
 type CommentType = keyof typeof commentPatterns;
@@ -83,10 +84,16 @@ function getAllMagicCommentDirectiveStyles(
       // Text uses HTML, front matter uses bash
       return getCommentPattern(['html', 'jsx', 'bash'], magicCommentDirectives);
 
+    case 'lua':
+    case 'haskell':
+      return getCommentPattern(['lua'], magicCommentDirectives);
+
     default:
-      // All comment types
+      // All comment types except Lua
       return getCommentPattern(
-        Object.keys(commentPatterns) as CommentType[],
+        Object.keys(commentPatterns).filter((pattern) =>
+          !['lua'].includes(pattern)
+        ) as CommentType[],
         magicCommentDirectives,
       );
   }