From 155cf5c5a1b45aafe7a5632f2ee6ecf957f04dde Mon Sep 17 00:00:00 2001
From: Bob Zeleznik <zzzman@gmail.com>
Date: Tue, 3 Sep 2019 23:07:42 -0400
Subject: fixes for bullets

---
 src/client/util/ProsemirrorExampleTransfer.ts | 59 +++++++++++++++++++--------
 1 file changed, 41 insertions(+), 18 deletions(-)

(limited to 'src/client/util/ProsemirrorExampleTransfer.ts')

diff --git a/src/client/util/ProsemirrorExampleTransfer.ts b/src/client/util/ProsemirrorExampleTransfer.ts
index 419311df8..5016e72c6 100644
--- a/src/client/util/ProsemirrorExampleTransfer.ts
+++ b/src/client/util/ProsemirrorExampleTransfer.ts
@@ -93,35 +93,63 @@ export default function buildKeymap<S extends Schema<any>>(schema: S, mapKeys?:
 
     bind("Mod-s", TooltipTextMenu.insertStar);
 
+    let updateOrderedList = (start: number, rangeStart: any, delta: number, tx2: Transaction, forward: boolean) => {
+        let bs = rangeStart.attrs.bulletStyle;
+        bs + delta > 0 && tx2.setNodeMarkup(start, rangeStart.type, { mapStyle: rangeStart.attrs.mapStyle, bulletStyle: rangeStart.attrs.bulletStyle + delta }, rangeStart.marks);
+
+        let brk = false;
+        rangeStart.descendants((node: any, offset: any, index: any) => {
+            if (node.type === schema.nodes.ordered_list || node.type === schema.nodes.list_item) {
+                if (!brk && (bs !== node.attrs.bulletStyle || delta > 0 || (forward && bs > 1))) {
+                    tx2.setNodeMarkup(start + offset + 1, node.type, { mapStyle: node.attrs.mapStyle, bulletStyle: node.attrs.bulletStyle + delta }, node.marks);
+                } else {
+                    brk = true;
+                }
+            }
+        });
+    }
+
     let updateBullets = (tx2: Transaction, refStart: number, delta: number) => {
-        for (let i = refStart; i > 0; i--) {
+        let i = refStart;
+        for (let i = refStart; i >= 0; i--) {
             let testPos = tx2.doc.nodeAt(i);
-            if (testPos && testPos.type === schema.nodes.list_item) {
+            if (!testPos) {
+                for (let i = refStart + 1; i <= tx2.doc.nodeSize; i++) {
+                    try {
+                        let testPos = tx2.doc.nodeAt(i);
+                        if (testPos && testPos.type === schema.nodes.ordered_list) {
+                            updateOrderedList(i, testPos, delta, tx2, true);
+                            break;
+                        }
+                    } catch (e) {
+                        break;
+                    }
+                }
+                break;
+            }
+            if ((testPos.type === schema.nodes.list_item || testPos.type === schema.nodes.ordered_list)) {
                 let start = i;
                 let preve = i > 0 && tx2.doc.nodeAt(start - 1);
                 if (preve && preve.type === schema.nodes.ordered_list) {
                     start = start - 1;
                 }
                 let rangeStart = tx2.doc.nodeAt(start);
-                if (rangeStart && rangeStart.type === schema.nodes.ordered_list) {
-                    tx2.setNodeMarkup(start, rangeStart.type, { ...rangeStart.attrs, bulletStyle: rangeStart.attrs.bulletStyle + delta }, rangeStart.marks);
+                if (rangeStart) {
+                    updateOrderedList(start, rangeStart, delta, tx2, false);
                 }
-                rangeStart && rangeStart.descendants((node: any, offset: any, index: any) => {
-                    if (node.type === schema.nodes.ordered_list) {
-                        tx2.setNodeMarkup(start + offset + 1, node.type, { ...node.attrs, bulletStyle: node.attrs.bulletStyle + delta }, node.marks);
-                    }
-                });
                 break;
             }
         }
     }
 
+
     bind("Tab", (state: EditorState<S>, dispatch: (tx: Transaction<S>) => void) => {
         var ref = state.selection;
         var range = ref.$from.blockRange(ref.$to);
         var marks = state.storedMarks || (state.selection.$to.parentOffset && state.selection.$from.marks());
         if (!sinkListItem(schema.nodes.list_item)(state, (tx2: Transaction) => {
-            updateBullets(tx2, range!.start, 1);
+            var range = state.selection.$from.blockRange(state.selection.$to);
+            updateBullets(tx2, range!.start - 1, 1);
             marks && tx2.ensureMarks([...marks]);
             marks && tx2.setStoredMarks([...marks]);
             dispatch(tx2);
@@ -129,13 +157,7 @@ export default function buildKeymap<S extends Schema<any>>(schema: S, mapKeys?:
             let sxf = state.tr.setSelection(TextSelection.create(state.doc, range!.start, range!.end));
             let newstate = state.applyTransaction(sxf);
             if (!wrapInList(schema.nodes.ordered_list)(newstate.state, (tx2: Transaction) => {
-                for (let i = range!.start; i >= 0; i--) {
-                    let rangeStart = tx2.doc.nodeAt(i);
-                    if (rangeStart && rangeStart.type === schema.nodes.ordered_list) {
-                        tx2.setNodeMarkup(i, rangeStart.type, { ...rangeStart.attrs, bulletStyle: 1 }, rangeStart.marks);
-                        break;
-                    }
-                }
+                updateBullets(tx2, range!.start, 1);
                 // when promoting to a list, assume list will format things so don't copy the stored marks.
                 marks && tx2.ensureMarks([...marks]);
                 marks && tx2.setStoredMarks([...marks]);
@@ -151,9 +173,10 @@ export default function buildKeymap<S extends Schema<any>>(schema: S, mapKeys?:
         var marks = state.storedMarks || (state.selection.$to.parentOffset && state.selection.$from.marks());
 
         let tr = state.tr;
-        updateBullets(tr, range!.start, -1);
 
         if (!liftListItem(schema.nodes.list_item)(tr, (tx2: Transaction) => {
+            var range = tx2.selection.$from.blockRange(tx2.selection.$to);
+            updateBullets(tx2, range!.start, -1);
             marks && tx2.ensureMarks([...marks]);
             marks && tx2.setStoredMarks([...marks]);
             dispatch(tx2);
-- 
cgit v1.2.3-70-g09d2