Add support for multi-cursor

RyotaUshio · Oct 18, 2023 · 4023a54 · 4023a54
1 parent 88d240d
commit 4023a54
Showing 1 changed file with 58 additions and 53 deletions.
diff --git a/src/handlers.ts b/src/handlers.ts
@@ -1,76 +1,81 @@
-import { ChangeSpec, EditorState, Transaction } from '@codemirror/state';
+import { ChangeSpec, EditorState } from '@codemirror/state';
 import { syntaxTree } from '@codemirror/language';
 
 import { isInlineMathBegin, isInlineMathEnd, printNode } from './utils';
 
 
 export function getChangesForDeletion(state: EditorState): ChangeSpec {
 
-    const range = state.selection.main;
-    const from = range.empty ? range.from - 4 : range.from;
-    const to = range.to;
-    const text = state.sliceDoc(from, to)
-    const index = text.lastIndexOf("$");
-    // console.log(
-    //     `range: ${range.from}-${range.to}\n`
-    //     + `from = ${from} \ to = ${to}\n`
-    //     + `text = "${text}"\n`
-    //     + `index = ${index}`
-    // );
-    if (index == -1) {
-        return [];
-    }
-
-    const doc = state.doc.toString();
-    const indexNextDollar = doc.indexOf("$", from + index + 1);
-    const indexPrevDollar = doc.lastIndexOf("$", from);
-    // console.log(
-    //     `!! ${indexPrevDollar}:${indexNextDollar}: "${state.sliceDoc(indexPrevDollar, indexNextDollar)}"`
-    // );
     const tree = syntaxTree(state);
-
+    const doc = state.doc.toString();
     const changes: ChangeSpec[] = [];
-    tree.iterate({
-        from: indexPrevDollar,
-        to: indexNextDollar >= 0 ? indexNextDollar : to,
-        enter(node) {
-            // printNode(node, state);
-            if (isInlineMathBegin(node, state)
-                && state.sliceDoc(node.to, node.to + 3) == "{} ") {
-                changes.push({ from: node.to, to: node.to + 3 });
-            } else if (isInlineMathEnd(node, state)
-                && state.sliceDoc(node.from - 3, node.from) == " {}") {
-                changes.push({ from: node.from - 3, to: node.from });
-            }
+
+    for (const range of state.selection.ranges) {
+        const from = range.empty ? range.from - 4 : range.from;
+        const to = range.to;
+        const text = state.sliceDoc(from, to)
+        const index = text.lastIndexOf("$");
+        // console.log(
+        //     `range: ${range.from}-${range.to}\n`
+        //     + `from = ${from} \ to = ${to}\n`
+        //     + `text = "${text}"\n`
+        //     + `index = ${index}`
+        // );
+        if (index == -1) {
+            continue;
         }
-    });
+
+        const indexNextDollar = doc.indexOf("$", from + index + 1);
+        const indexPrevDollar = doc.lastIndexOf("$", from);
+        // console.log(
+        //     `!! ${indexPrevDollar}:${indexNextDollar}: "${state.sliceDoc(indexPrevDollar, indexNextDollar)}"`
+        // );
+
+        tree.iterate({
+            from: indexPrevDollar,
+            to: indexNextDollar >= 0 ? indexNextDollar : to,
+            enter(node) {
+                // printNode(node, state);
+                if (isInlineMathBegin(node, state)
+                    && state.sliceDoc(node.to, node.to + 3) == "{} ") {
+                    changes.push({ from: node.to, to: node.to + 3 });
+                } else if (isInlineMathEnd(node, state)
+                    && state.sliceDoc(node.from - 3, node.from) == " {}") {
+                    changes.push({ from: node.from - 3, to: node.from });
+                }
+            }
+        });
+    }
+
     return changes;
 }
 
 export function getChangesForInsertion(state: EditorState): ChangeSpec {
     const tree = syntaxTree(state);
-    const range = state.selection.main;
     const doc = state.doc.toString();
-    const indexNextDollar = doc.indexOf("$", range.to);
-    const indexPrevDollar = doc.lastIndexOf("$", range.from);
     let changes: ChangeSpec[] = [];
 
-    if (indexNextDollar >= 0) {
-        tree.iterate({
-            from: indexPrevDollar,
-            to: indexNextDollar + 1,
-            enter(node) {
-                if (isInlineMathBegin(node, state)) {
-                    if (!(state.sliceDoc(node.to, node.to + 3) == "{} ")) {
-                        changes.push({ from: node.to, insert: "{} " });
-                    }
-                } else if (isInlineMathEnd(node, state)) {
-                    if (!(state.sliceDoc(node.from - 3, node.from) == " {}")) {
-                        changes.push({ from: node.from, insert: " {}" });
+    for (const range of state.selection.ranges) {
+        const indexNextDollar = doc.indexOf("$", range.to);
+        const indexPrevDollar = doc.lastIndexOf("$", range.from);
+
+        if (indexNextDollar >= 0) {
+            tree.iterate({
+                from: indexPrevDollar,
+                to: indexNextDollar + 1,
+                enter(node) {
+                    if (isInlineMathBegin(node, state)) {
+                        if (!(state.sliceDoc(node.to, node.to + 3) == "{} ")) {
+                            changes.push({ from: node.to, insert: "{} " });
+                        }
+                    } else if (isInlineMathEnd(node, state)) {
+                        if (!(state.sliceDoc(node.from - 3, node.from) == " {}")) {
+                            changes.push({ from: node.from, insert: " {}" });
+                        }
                     }
                 }
-            }
-        });
+            });
+        }
     }
 
     return changes;