From 2f0e4500f688e41eb73b7722dd36ae0d4caed187 Mon Sep 17 00:00:00 2001
From: Snowflyt <gaoge011022@163.com>
Date: Tue, 19 Dec 2023 15:19:23 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9E=20fix:=20Corrupting=20content=20in?=
 =?UTF-8?q?=20formula=20block?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main.ts | 51 ++++++++++++++++++++++++++++-----------------------
 1 file changed, 28 insertions(+), 23 deletions(-)

diff --git a/src/main.ts b/src/main.ts
index 53734b6..3c99aa1 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -179,30 +179,35 @@ const main = async () => {
           character: range.end.character - startPos.character,
         },
       };
-      // Get the code block starter "```" or "~~~"
-      const codeBlockStarter = state.markdown
-        .split(File.useCRLF ? "\r\n" : "\n")
-        [startPos.line - 1]!.slice(startPos.character, startPos.character + 3);
-      // Get only completion text before code block starter, as in Typora code block it is not possible
-      // to insert a new code block or end one using "```" or "~~~"
-      const indexOfCodeBlockStarter = subCmCompletion.text.indexOf(codeBlockStarter);
-      if (indexOfCodeBlockStarter !== -1) {
-        subCmCompletion.displayText = subCmCompletion.displayText.slice(
-          0,
-          subCmCompletion.displayText.indexOf(codeBlockStarter),
-        );
-        subCmCompletion.text = subCmCompletion.text.slice(0, indexOfCodeBlockStarter);
-        const textAfterCodeBlockStarter = subCmCompletion.text.slice(indexOfCodeBlockStarter);
-        // Reduce `subCmCompletion.range` to only include text before code block starter
-        const rows = textAfterCodeBlockStarter.split(File.useCRLF ? "\r\n" : "\n").length - 1;
-        subCmCompletion.range.end.line -= rows;
-        subCmCompletion.range.end.character = textAfterCodeBlockStarter
-          .split(File.useCRLF ? "\r\n" : "\n")
-          .pop()!.length;
+      // Get starter of CodeMirror to determine whether it is a code block, formula, etc.
+      let cmStarter = state.markdown.split(File.useCRLF ? "\r\n" : "\n")[startPos.line - 1];
+
+      if (cmStarter) {
+        if (cmStarter.startsWith("```") || cmStarter.startsWith("~~~")) {
+          // * Code block *
+          cmStarter = cmStarter.slice(0, 3);
+          // Get only completion text before code block starter, as in Typora code block it is not possible
+          // to insert a new code block or end one using "```" or "~~~"
+          const indexOfCodeBlockStarter = subCmCompletion.text.indexOf(cmStarter);
+          if (indexOfCodeBlockStarter !== -1) {
+            subCmCompletion.displayText = subCmCompletion.displayText.slice(
+              0,
+              subCmCompletion.displayText.indexOf(cmStarter),
+            );
+            subCmCompletion.text = subCmCompletion.text.slice(0, indexOfCodeBlockStarter);
+            const textAfterCodeBlockStarter = subCmCompletion.text.slice(indexOfCodeBlockStarter);
+            // Reduce `subCmCompletion.range` to only include text before code block starter
+            const rows = textAfterCodeBlockStarter.split(File.useCRLF ? "\r\n" : "\n").length - 1;
+            subCmCompletion.range.end.line -= rows;
+            subCmCompletion.range.end.character = textAfterCodeBlockStarter
+              .split(File.useCRLF ? "\r\n" : "\n")
+              .pop()!.length;
+          }
+          insertCompletionTextToCodeMirror(cm, subCmCompletion);
+
+          return;
+        }
       }
-      insertCompletionTextToCodeMirror(cm, subCmCompletion);
-
-      return;
     }
 
     const focusedElem = document.querySelector(`[cid=${editor.focusCid}]`);