feat: 🎸 set depth attribute in mdastToFlat()

streamich · Mar 3, 2019 · 0f5eca7 · 0f5eca7
1 parent d75bc86
commit 0f5eca7
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 0 deletions.
diff --git a/src/__tests__/mdastToFlat.spec.ts b/src/__tests__/mdastToFlat.spec.ts
@@ -48,6 +48,14 @@ describe('structure', () => {
     });
   });
 
+  it('root node should have a depth attribute', () => {
+    const parser = create();
+    const mdast = parser.tokenizeBlock('foo');
+    const flat = mdastToFlat(mdast!);
+
+    expect(flat.nodes[0].depth).toBe(0);
+  });
+
   it('adds titles to contents list', () => {
     const parser = create();
     const mdast = parser.tokenizeBlock('# Title\n' + '\n' + '## Subtitle\n');

diff --git a/src/mdastToFlat.ts b/src/mdastToFlat.ts
@@ -49,6 +49,7 @@ export const mdastToFlat: MdastToFlat = (mdast) => {
   if (mdast) {
     traverse(mdast, 0);
 
+    // Process references.
     let footnoteCounter = 0;
     for (const node of nodes) {
       if ((node.type === 'footnoteReference') || (node.type === 'imageReference')) {
@@ -65,6 +66,10 @@ export const mdastToFlat: MdastToFlat = (mdast) => {
         }
       }
     }
+
+    // Process root node.
+    const root = doc.nodes[0];
+    root.depth = 0;
   }
 
   return doc;

diff --git a/src/replace.ts b/src/replace.ts
@@ -1,4 +1,5 @@
 import {Flat} from './types';
+import {findRoot} from './findRoot';
 
 export const replace = (into: Flat, at: number, what: Flat): Flat => {
   const mergeIdx = into.nodes.length;
@@ -33,6 +34,7 @@ export const replace = (into: Flat, at: number, what: Flat): Flat => {
     merged.nodes.push(newNode);
   }
   merged.nodes[mergeIdx].parent = at;
+  merged.nodes[mergeIdx].depth = (into.nodes[findRoot(into, at)].depth || 0) + 1;
 
   // MERGE METADATA.
   for (const idx of what.contents) {

diff --git a/src/types.ts b/src/types.ts
@@ -11,6 +11,10 @@ export interface FlatFootnotes {
 export type TNode = (IRoot | TAnyToken) & {
   idx: number;
   parent: number;
+  /**
+   * Root nodes have `depth` key, which tracks how deeply they have been merged.
+   */
+  depth?: number;
   children?: number[];
 };