@antmove/alipay-wx/parse/parse.js

transform alipay miniprogram to wx miniprogram tool.

var htmlparser = require("htmlparser2");
const fs = require('fs');

function createText (text) {
    return {
        value: text,
        parent: null,
        type: 'textContent'
    };
}

function parseFile (filename) {
    return parseString(fs.readFileSync(filename, 'utf8'));
}

function parseString (code) {
    let ast = [];
    let zIndex = 0;
    let currentNode = null;
    let preTagIsText = false;     // fixbug: textContent value 被拆分的情况
    let preTextNode = null;

    var parser = new htmlparser.Parser({
        onopentag: function (name, attrs) {
            let node = {
                props: processProps(attrs),
                type: name,
                children: [],
                parent: null
            };

            let _fn = 'push';

            if (name === 'wxs') {
                _fn = 'unshift';
            }


            if (zIndex === 0) {
                ast[_fn](node);
                currentNode = node;
            } else {
                node.parent = currentNode;
                currentNode.children[_fn](node);
                currentNode = node;
            }
            zIndex++;
            preTagIsText = false;
        },
        ontext: function (text) {
            text = text.trim();
            if (text) {
                if (preTagIsText) {
                    preTextNode.value += text;
                    return false;
                }

                let node = createText(text);
                if (zIndex === 0) {
                    ast.push(node);
                } else {
                    node.parent = currentNode;
                    currentNode.children.push(node);
                }
                preTextNode = node;
                preTagIsText = true;
            }
        },
        onclosetag: function () {
            zIndex--;
            currentNode = currentNode.parent;
        }
    }, {
        decodeEntities: false,
        xmlMode: true
    });
    parser.write(code);
    parser.end();
    
    return ast;
}

function processProps (obj = {}) {
    let props = {};
    Object.keys(obj)
        .forEach(function (prop) {
            let val = obj[prop];
            props[prop] = {
                type: 'unknown',   // (val[0] === '"') ? 'double' : 'single',
                value: [val]
            };
        });
    
    return props;
}

module.exports =  {
    parseFile,
    parseString
};