1 files changed, 42 insertions, 0 deletions
diff --git a/app/assets/javascripts/content_editor/extensions/code_block_highlight.js b/app/assets/javascripts/content_editor/extensions/code_block_highlight.js
index da5ac7eb158..e75178789da 100644
--- a/app/assets/javascripts/content_editor/extensions/code_block_highlight.js
+++ b/app/assets/javascripts/content_editor/extensions/code_block_highlight.js
@@ -65,6 +65,48 @@ export default CodeBlockLowlight.extend({
         tag: 'pre.js-syntax-highlight',
         preserveWhitespace: 'full',
       },
+      // Matches HTML generated by Banzai::Filter::SyntaxHighlightFilter,
+      // Banzai::Filter::MathFilter, Banzai::Filter::MermaidFilter,
+      // or Banzai::Filter::SuggestionFilter
+      {
+        tag: 'pre.code.highlight',
+        preserveWhitespace: 'full',
+      },
+      // Matches HTML generated by Banzai::Filter::MathFilter,
+      // after being transformed by ~/behaviors/markdown/render_math.js
+      {
+        tag: 'span.katex-display',
+        preserveWhitespace: 'full',
+        contentElement: 'annotation[encoding="application/x-tex"]',
+        attrs: { language: 'math' },
+      },
+      // Matches HTML generated by Banzai::Filter::MermaidFilter,
+      // after being transformed by ~/behaviors/markdown/render_sandboxed_mermaid.js
+      {
+        tag: 'svg.mermaid',
+        preserveWhitespace: 'full',
+        contentElement: 'text.source',
+        attrs: { language: 'mermaid' },
+      },
+      // Matches HTML generated by Banzai::Filter::SuggestionFilter,
+      // after being transformed by ~/vue_shared/components/markdown/suggestions.vue
+      {
+        tag: '.md-suggestion',
+        skip: true,
+      },
+      {
+        tag: '.md-suggestion-header',
+        ignore: true,
+      },
+      {
+        tag: '.md-suggestion-diff',
+        preserveWhitespace: 'full',
+        getContent: (el, schema) =>
+          [...el.querySelectorAll('.line_content.new span')].map((span) =>
+            schema.text(span.innerText),
+          ),
+        attrs: { language: 'suggestion' },
+      },
     ];
   },
   renderHTML({ HTMLAttributes }) {