snapshot of https://github.com/toeverything/AFFiNE.git (canary, 2026-09-16; blobs > 131072B omitted)

blocksuite/affine/blocks/latex/src/adapters/extension.ts

@@ -0,0 +1,11 @@
+import type { ExtensionType } from '@blocksuite/store';
+
+import { LatexMarkdownAdapterExtensions } from './markdown/index.js';
+import { LatexBlockNotionHtmlAdapterExtension } from './notion-html.js';
+import { LatexBlockPlainTextAdapterExtension } from './plain-text.js';
+
+export const LatexBlockAdapterExtensions: ExtensionType[] = [
+  LatexMarkdownAdapterExtensions,
+  LatexBlockNotionHtmlAdapterExtension,
+  LatexBlockPlainTextAdapterExtension,
+].flat();

blocksuite/affine/blocks/latex/src/adapters/index.ts

@@ -0,0 +1,3 @@
+export * from './markdown/index.js';
+export * from './notion-html.js';
+export * from './plain-text.js';

blocksuite/affine/blocks/latex/src/adapters/markdown/index.ts

@@ -0,0 +1,12 @@
+import type { ExtensionType } from '@blocksuite/store';
+
+import { LatexBlockMarkdownAdapterExtension } from './markdown.js';
+import { LatexMarkdownPreprocessorExtension } from './preprocessor.js';
+
+export * from './markdown.js';
+export * from './preprocessor.js';
+
+export const LatexMarkdownAdapterExtensions: ExtensionType[] = [
+  LatexMarkdownPreprocessorExtension,
+  LatexBlockMarkdownAdapterExtension,
+];

blocksuite/affine/blocks/latex/src/adapters/markdown/markdown.ts

@@ -0,0 +1,55 @@
+import { LatexBlockSchema } from '@blocksuite/affine-model';
+import {
+  BlockMarkdownAdapterExtension,
+  type BlockMarkdownAdapterMatcher,
+  type MarkdownAST,
+} from '@blocksuite/affine-shared/adapters';
+import { nanoid } from '@blocksuite/store';
+
+const isLatexNode = (node: MarkdownAST) => node.type === 'math';
+
+export const latexBlockMarkdownAdapterMatcher: BlockMarkdownAdapterMatcher = {
+  flavour: LatexBlockSchema.model.flavour,
+  toMatch: o => isLatexNode(o.node),
+  fromMatch: o => o.node.flavour === LatexBlockSchema.model.flavour,
+  toBlockSnapshot: {
+    enter: (o, context) => {
+      const latex = 'value' in o.node ? o.node.value : '';
+      const { walkerContext } = context;
+      walkerContext
+        .openNode(
+          {
+            type: 'block',
+            id: nanoid(),
+            flavour: 'affine:latex',
+            props: {
+              latex,
+            },
+            children: [],
+          },
+          'children'
+        )
+        .closeNode();
+    },
+  },
+  fromBlockSnapshot: {
+    enter: (o, context) => {
+      const latex =
+        'latex' in o.node.props ? (o.node.props.latex as string) : '';
+      const { walkerContext } = context;
+      walkerContext
+        .openNode(
+          {
+            type: 'math',
+            value: latex,
+          },
+          'children'
+        )
+        .closeNode();
+    },
+  },
+};
+
+export const LatexBlockMarkdownAdapterExtension = BlockMarkdownAdapterExtension(
+  latexBlockMarkdownAdapterMatcher
+);

blocksuite/affine/blocks/latex/src/adapters/markdown/preprocessor.ts

@@ -0,0 +1,88 @@
+import {
+  type MarkdownAdapterPreprocessor,
+  MarkdownPreprocessorExtension,
+} from '@blocksuite/affine-shared/adapters';
+
+function escapeBrackets(text: string) {
+  const pattern =
+    /(```[\S\s]*?```|`.*?`)|\\\[([\S\s]*?[^\\])\\]|\\\((.*?)\\\)/g;
+  return text.replaceAll(
+    pattern,
+    (match, codeBlock, squareBracket, roundBracket) => {
+      if (codeBlock) {
+        return codeBlock;
+      } else if (squareBracket) {
+        return `$$${squareBracket}$$`;
+      } else if (roundBracket) {
+        return `$${roundBracket}$`;
+      }
+      return match;
+    }
+  );
+}
+
+function escapeMhchem(text: string) {
+  return text.replaceAll('$\\ce{', '$\\\\ce{').replaceAll('$\\pu{', '$\\\\pu{');
+}
+
+/**
+ * Preprocess the content to protect code blocks and LaTeX expressions
+ * reference issue: https://github.com/remarkjs/react-markdown/issues/785
+ * reference comment: https://github.com/remarkjs/react-markdown/issues/785#issuecomment-2307567823
+ * @param content - The content to preprocess
+ * @returns The preprocessed content
+ */
+function preprocessLatex(content: string) {
+  // Protect code blocks
+  const codeBlocks: string[] = [];
+  let preprocessedContent = content;
+  preprocessedContent = preprocessedContent.replace(
+    /(```[\s\S]*?```|`[^`\n]+`)/g,
+    (_, code) => {
+      codeBlocks.push(code);
+      return `<<CODE_BLOCK_${codeBlocks.length - 1}>>`;
+    }
+  );
+
+  // Protect existing LaTeX expressions
+  const latexExpressions: string[] = [];
+  preprocessedContent = preprocessedContent.replace(
+    /(\$\$[\s\S]*?\$\$|\\\[[\s\S]*?\\\]|\\\(.*?\\\))/g,
+    match => {
+      latexExpressions.push(match);
+      return `<<LATEX_${latexExpressions.length - 1}>>`;
+    }
+  );
+
+  // Escape dollar signs that are likely currency indicators
+  preprocessedContent = preprocessedContent.replace(/\$(?=\d)/g, '\\$');
+
+  // Restore LaTeX expressions
+  preprocessedContent = preprocessedContent.replace(
+    /<<LATEX_(\d+)>>/g,
+    (_, index) => latexExpressions[parseInt(index)]
+  );
+
+  // Restore code blocks
+  preprocessedContent = preprocessedContent.replace(
+    /<<CODE_BLOCK_(\d+)>>/g,
+    (_, index) => codeBlocks[parseInt(index)]
+  );
+
+  // Apply additional escaping functions
+  preprocessedContent = escapeBrackets(preprocessedContent);
+  preprocessedContent = escapeMhchem(preprocessedContent);
+
+  return preprocessedContent;
+}
+
+const latexPreprocessor: MarkdownAdapterPreprocessor = {
+  name: 'latex',
+  levels: ['block', 'slice', 'doc'],
+  preprocess: content => {
+    return preprocessLatex(content);
+  },
+};
+
+export const LatexMarkdownPreprocessorExtension =
+  MarkdownPreprocessorExtension(latexPreprocessor);

blocksuite/affine/blocks/latex/src/adapters/notion-html.ts

@@ -0,0 +1,50 @@
+import { LatexBlockSchema } from '@blocksuite/affine-model';
+import {
+  BlockNotionHtmlAdapterExtension,
+  type BlockNotionHtmlAdapterMatcher,
+  HastUtils,
+} from '@blocksuite/affine-shared/adapters';
+import { nanoid } from '@blocksuite/store';
+
+export const latexBlockNotionHtmlAdapterMatcher: BlockNotionHtmlAdapterMatcher =
+  {
+    flavour: LatexBlockSchema.model.flavour,
+    toMatch: o => {
+      return (
+        HastUtils.isElement(o.node) &&
+        o.node.tagName === 'figure' &&
+        !!HastUtils.querySelector(o.node, '.equation-container')
+      );
+    },
+    fromMatch: () => false,
+    toBlockSnapshot: {
+      enter: (o, context) => {
+        if (!HastUtils.isElement(o.node)) {
+          return;
+        }
+        const { walkerContext } = context;
+        const latex = HastUtils.getTextContent(
+          HastUtils.querySelector(o.node, 'annotation')
+        );
+        walkerContext
+          .openNode(
+            {
+              type: 'block',
+              id: nanoid(),
+              flavour: LatexBlockSchema.model.flavour,
+              props: {
+                latex,
+              },
+              children: [],
+            },
+            'children'
+          )
+          .closeNode();
+        walkerContext.skipAllChildren();
+      },
+    },
+    fromBlockSnapshot: {},
+  };
+
+export const LatexBlockNotionHtmlAdapterExtension =
+  BlockNotionHtmlAdapterExtension(latexBlockNotionHtmlAdapterMatcher);

blocksuite/affine/blocks/latex/src/adapters/plain-text.ts

@@ -0,0 +1,29 @@
+import { LatexBlockSchema } from '@blocksuite/affine-model';
+import {
+  BlockPlainTextAdapterExtension,
+  type BlockPlainTextAdapterMatcher,
+} from '@blocksuite/affine-shared/adapters';
+
+const latexPrefix = 'LaTex, with value: ';
+
+export const latexBlockPlainTextAdapterMatcher: BlockPlainTextAdapterMatcher = {
+  flavour: LatexBlockSchema.model.flavour,
+  toMatch: () => false,
+  fromMatch: o => o.node.flavour === LatexBlockSchema.model.flavour,
+  toBlockSnapshot: {},
+  fromBlockSnapshot: {
+    enter: (o, context) => {
+      const latex =
+        'latex' in o.node.props ? (o.node.props.latex as string) : '';
+
+      const { textBuffer } = context;
+      if (latex) {
+        textBuffer.content += `${latexPrefix}${latex}`;
+        textBuffer.content += '\n';
+      }
+    },
+  },
+};
+
+export const LatexBlockPlainTextAdapterExtension =
+  BlockPlainTextAdapterExtension(latexBlockPlainTextAdapterMatcher);