Handle code blocks correctly

Author: alexjg

playground/src/Editor.tsx

@@ -509,8 +509,8 @@ function headingRule(nodeType: NodeType, maxLevel: number) {
 }
 
 function buildInputRules(schema: Schema) {
-  let rules = smartQuotes.concat(ellipsis, emDash),
-    type
+  const rules = smartQuotes.concat(ellipsis, emDash)
+  let type
   if ((type = schema.nodes.blockquote)) rules.push(blockQuoteRule(type))
   if ((type = schema.nodes.ordered_list)) rules.push(orderedListRule(type))
   if ((type = schema.nodes.bullet_list)) rules.push(bulletListRule(type))

playground/src/Playground.tsx

@@ -21,10 +21,15 @@ const rightRepo = new Repo({
 
 const leftHandle = leftRepo.create()
 leftHandle.change(d => {
-  d.text = "Heading"
+  d.text = ""
+  //am.splitBlock(d, ["text"], 0, {
+  //type: new am.RawString("heading"),
+  //attrs: { level: 1 },
+  //parents: [],
+  //})
   am.splitBlock(d, ["text"], 0, {
-    type: new am.RawString("heading"),
-    attrs: { level: 1 },
+    type: new am.RawString("code-block"),
+    attrs: {},
     parents: [],
   })
 })

src/traversal.ts

@@ -117,7 +117,11 @@ export function amSpliceIdxToPmIdx(
       return maxInsertableIndex
     }
     if (state.event.type === "openTag") {
-      if (state.event.tag === "paragraph" || state.event.tag === "heading") {
+      if (
+        state.event.tag === "paragraph" ||
+        state.event.tag === "heading" ||
+        state.event.tag === "code_block"
+      ) {
         maxInsertableIndex = state.after.pmIdx
       }
     } else if (state.event.type === "leafNode") {

test/traversal.spec.ts

@@ -223,6 +223,22 @@ describe("the traversal API", () => {
       // pm 0     0    1     2
       assert.equal(amSpliceIdxToPmIdx(spans, 1), 1)
     })
+
+    it("should find the first index inside a code block at the start of the document", () => {
+      const spans: am.Span[] = [
+        {
+          type: "block",
+          value: {
+            type: new am.RawString("code-block"),
+            attrs: {},
+            parents: [],
+          },
+        },
+      ]
+      const events = traverseSpans(spans)
+      console.log(printIndexTable(events))
+      assert.equal(amSpliceIdxToPmIdx(spans, 1), 1)
+    })
   })
 
   describe("the pmRangeToAmRange function", () => {