implement text parser (#12)

Co-authored-by: DN.Wangchuk <dung.nt@yeah1.vn>

Author: bluzky

src/js/index.js

@@ -3,9 +3,32 @@ import MindmapLayouts from "./layouts"
 import MindMap from "./mindmap"
 import MindMapViewer from './viewer'
 import sample from './sample'
+import { TextParser } from './parser'
 
 // import * as $ from 'jquery'
 
+let text = `
+Root
+
+    - branch 1
+        +branch 1.1
+
+    - branch 2
+        branch 2.1
+        * branch 2.2
+            branch 2.2.1
+
+    -Branch 3
+        - alo
+        - ola
+        - ollala
+
+    -Branch 4
+        - Branch 4.1
+            - Branch 4.1.1
+        - Branch 4.2
+        - Branch 4.3`
+
 
 const formNode = document.getElementById('layout-props')
 const layoutTimeNode = document.getElementById('layout-time')
@@ -19,11 +42,12 @@ const viewer = new MindMapViewer('#drawing', {})
 function render() {
     const count = formNode.dataSize.value
     const layoutType = formNode.layoutType.value
-    const root = randomTree(count)
+    // const root = randomTree(count)
     const MindmapLayout = MindmapLayouts[layoutType]
+    const test = TextParser.parse(text)
 
     const t0 = window.performance.now()
-    const mindMap = new MindMap(sample.root, MindmapLayout, {})
+    const mindMap = new MindMap(test.root, MindmapLayout, {})
     mindMap.build()
     const t1 = window.performance.now()
     viewer.render(mindMap)

src/js/mindmap.js

@@ -6,7 +6,6 @@ class MindMap {
     constructor(data, layout, options) {
         this.data = data
         this.options = options
-        console.log(options)
         this.theme = options["theme"] || new Theme(data["theme"])
         options["theme"] = this.theme
         this.rootNode = new Node(data, options)

src/js/parser/index.js

@@ -0,0 +1,5 @@
+import TextParser from './text-parser'
+
+export {
+    TextParser
+}

src/js/parser/text-parser.js

@@ -0,0 +1,65 @@
+
+
+function parse(text) {
+    let lines = text.match(/[^\r\n]+/g);
+    let root = null
+    let levelStack = []
+    let currentLevel = 0
+
+    for (let line of lines) {
+        if (root == null) {
+            root = {
+                content: getContent(line),
+                children: []
+            }
+            levelStack.push({ node: root, level: 0 })
+        } else {
+            let content = getContent(line)
+            let level = getLevel(line)
+            let node = {
+                content: content,
+                children: []
+            }
+
+            if (level == 0) {
+                break
+            }
+
+            let lastIndex = levelStack.length - 1
+
+            while (levelStack[levelStack.length - 1].level > level) {
+                levelStack.pop()
+                currentLevel -= 1
+            }
+
+            if (level != currentLevel) {
+                levelStack[levelStack.length - 1].node.children.push(node)
+                levelStack.push({ node: node, level: level })
+                currentLevel = level
+            } else {
+                levelStack.pop()
+                levelStack[levelStack.length - 1].node.children.push(node)
+                levelStack.push({ node: node, level: level })
+            }
+        }
+    }
+    console.log(root)
+    return {
+        root
+    }
+}
+
+function getLevel(line) {
+    line = line.replace(/\t/, "    ")
+    let leadingWs = line.match(/^\s*/)
+    leadingWs = leadingWs[0] || ""
+    return Math.round(leadingWs.length / 4)
+}
+
+function getContent(line) {
+    line = line.trim()
+    line = line.replace(/^[+\-_*]\s*/, "")
+    return line
+}
+
+export default { parse: parse }