md-2-json/index.js at master · UsabilityDynamics/md-2-json · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
'use strict';

var fs = require('fs');
var marked = require('marked');
var traverse = require('traverse');

var parse = function(mdContent) {
    var json = marked.lexer(mdContent);
    var currentHeading, headings = [],
        isOrdered = true;
    var output = json.reduce(function(result, item, index, array) {
        switch (item.type) {
            case 'heading':
                if (!currentHeading || item.depth == 1) {
                    headings = [];
                    result[item.text] = {};
                    currentHeading = result[item.text];
                    headings.push(item.text);
                } else {
                    var parentHeading = getParentHeading(headings, item, result);
                    headings = parentHeading.headings;
                    currentHeading = parentHeading.parent;
                    currentHeading[item.text] = {};
                    currentHeading = currentHeading[item.text];
                }
                break;
            case 'list_start':
                isOrdered = item.ordered;
                break;
            case 'text':
                var ordered = isOrdered ? '1. ' : '- ';
                var text = ordered + item.text + '\n';
                currentHeading.raw = currentHeading.raw ? currentHeading.raw + text : text;
                break;
            case 'table':
                var tableContent = getTableContent(item);
                currentHeading.raw = currentHeading.raw ? currentHeading.raw + tableContent : tableContent;
                break;
            case 'space':
                currentHeading.raw = currentHeading.raw ? currentHeading.raw + '\n' : '\n';
                break;
            case 'paragraph':
                if (!currentHeading) {
                    currentHeading = result;
                }
                var para = checkNextLine(item.text);
                currentHeading.raw = currentHeading.raw ? currentHeading.raw + para : para;
                break;
            case 'code':
                currentHeading.raw = item.text;
                break;
            default:
                break;
        }
        return result;
    }, {});
    return output;
}
exports.parse = parse;

function getParentHeading(headings, item, result) {
    var parent, index = item.depth - 1;
    var currentHeading = headings[index];
    if (currentHeading) {
        headings.splice(index, headings.length - index);
    }
    headings.push(item.text);
    for (var i = 0; i < index; i++) {
        if (!parent) {
            parent = result[headings[i]];
        } else {
            parent = parent[headings[i]];
        }
    }
    return {
        headings: headings,
        parent: parent
    };
}

function getTableContent(item) {
    var tableHeader = '',
        tableContent = '',
        separator = '';
    for (var i = 0; i < item.header.length; i++) {
        tableHeader += item.header[i] + ' | ';
    }
    for (var i = 0; i < item.align.length; i++) {
        separator += '---------:| ';
    }
    for (var i = 0; i < item.cells.length; i++) {
        var cells = item.cells[i];
        for (var j = 0; j < cells.length; j++) {
            tableContent += cells[j] + ' | ';
        }
        var sep = i !== item.cells.length - 1 ? '| ' : '';
        tableContent += '\n' + sep;
    }
    return '| ' + tableHeader + '\n|: ' + separator + '\n| ' + tableContent + '\n';
}

function checkNextLine(mdText) {
    if (!mdText.endsWith('\n\n')) {
        mdText += '\n\n';
    }
    return mdText;
}

function toMd(jsonObject) {
    var mdText = '';
    traverse(jsonObject).reduce(function(acc, value) {
        if (this.isLeaf && this.key === 'raw') {
            mdText += value;
        } else {
            mdText += getHash(this.level) + ' ' + this.key + '\n\n';
        }
        return;
    });
    return mdText;
}
exports.toMd = toMd;

function getHash(level) {
    var hash = '';
    for (var i = 0; i < level; i++) {
        hash += '#';
    }
    return hash;
}