@yugu/gogocode/src/html-core/get-selector.js

The simplest tool to parse/transform/generate code on ast

// 把简单的api转换成ast
const parse = require('./parse');
const filterProps = require('./filter-prop.js');

function getSelector(selectorCode, parseOptions, expando = 'g123o456g789o') {
    const selector = { nodeType: '', structure: {} };
    if (typeof selectorCode != 'string') {
        // 如果是通过builders造出来的ast结构，比如return语句
        selector.nodeType = selectorCode.nodeType;
        filterProps(selectorCode, selector.structure);
        selector.type = selectorCode.nodeType; // 兼容只用type匹配的选择器
        return selector;
    } else {
        selectorCode = selectorCode
            .trim()
            .replace(/\$_\$/g, expando)
            .replace(/\$\$\$/g, expando.slice(0, -1) + '$3')
    }
    let selectorAst = parse(selectorCode, parseOptions);
    if (selectorAst.content && selectorAst.content.children && selectorAst.content.children[0]) {
        filterProps(selectorAst.content.children[0], selector.structure);
        selector.nodeType = selectorAst.content.children[0].nodeType;
    }
    return selector;
}

module.exports = getSelector;