Vue组件文档生成工具库的方法

目录

• 解析.vue 文件
• 提取文档信息
• 提取组件名称、描述、props、methods、model
• 提取描述
• 提取 methods
• 提取 props
• 提取 model
• 提取组件Events
• 提取组件Slots
• 结语

const compiler = require('vue-template-compiler')
const result = compiler.parseComponent(vueStr, [options])
// parseComponent 返回  template、script、style内容，
export interface SFCDescriptor {
  template: SFCBlock | undefined;
  script: SFCBlock | undefined;
  styles: SFCBlock[];
  customBlocks: SFCBlock[];
}

const compiler = require('vue-template-compiler')
//vueStr .vue 文件内容
const vue = compiler.parseComponent(vueStr)
//生成html部分的 ast 
let template = compiler.compile(vue.template.content, {
    preserveWhitespace: false,
    comments: true // 生成注释信息
})

const parse = require('@babel/parser');
//生成js部分的 ast
let jsAst = parse.parse(vue.script.content, {
    allowImportExportEverywhere: true 
})

const traverse = require('@babel/traverse');
traverse.default(jsAst, {
  enter(path){ // 开始
  },
  // 支持自定义节点 比如当节点类型 为 ExportDefaultDeclaration 时掉这个方法
  ExportDefaultDeclaration(){
  }
})

let componentInfo = {}
traverse.default(jsAst, {
  ExportDefaultDeclaration(path){
    path.node.declaration.properties.forEach(item => {
        switch (item.key.name) {
            case 'props':
                componentInfo.props = extractProps(item) // 提取 props
                break;
            case 'methods':
                componentInfo.methods = extractMethods(item)  // 提取 methods
                break
            case 'name':
                componentInfo.name = item.value.value // 获取组件名称
                break
            case 'model':
                componentInfo.model = extractModel(item)  // 提取 model
                break
            default:
                break;
        }
    });
  }
})

/** 
 * 多行备注
 * 用来上传文档信息
 */
// 单行备注
export default {
}
// 结尾注释

// ExportDefaultDeclaration 插入如下代码 
if (path.node.leadingComments) {
    componentInfo.desc = path.node.leadingComments.map(item => {
        if (item.type === 'CommentLine') {
            return item.value.trim()
        } else {
            return item.value.split('\n').map(item => item.replace(/[\s\*]/g, '')).filter(Boolean)
        }
    }).toString()
}

/**
 * 方法描述
 * @param {Bool} type 参数描述
 * @returns 返回值描述
 */

{
    propA: Number, // 只有类型
    propB: [String, Number], // 只有类型但是支持多种
    propC: { 
      type: String,
      required: true
    },
    propD: {
      type: Number,
      default: 100 // 带有默认值 
    },
    propE: {
      type: Object,
      default () { // 默认值 需要函数返回
        return { message: 'hello' }
      }
    },
    propF: {
      default: function () { // 默认值 需要函数返回 和上面的 default 的 ast 节点类型是不同的
        return { message: 'hello' }
      }
      validator: function (value) { // 校验
        return ['success', 'warning', 'danger'].indexOf(value) !== -1
      }
    }
}

// 获取Props默认值
function getDefaultVal (node) {
    if (types.isRegExpLiteral(node) || types.isBooleanLiteral(node) || types.isNumericLiteral(node) || types.isStringLiteral(node)) {
        return node.value
    } else if (types.isFunctionExpression(node) || types.isArrowFunctionExpression(node) || types.isObjectMethod(node)) {
        try {
            let code = generate.default(types.isObjectMethod(node) ? node.body : node).code
            let fun = eval(**0,${types.isObjectMethod(node) ? 'function ()' : ''} $[code]**)
            return JSON.stringify(fun())
        } catch (error) {
        }
    }
}

const extractEvents = (path) => {
    // 第一个元素是事件名称
    const eventName = path.parent.arguments[0]; 
    let comments = path.parentPath.parent.leadingComments
    return {
        name: eventName.value,
        desc: comments ? comments.map(item => item.value.trim()).toString() : '——'
    }
}
MemberExpression (path) {
    // 判断是不是event
    if (path.node.property.name === '$emit') {
        let event = extractEvents(path)
        !componentInfo.events && (componentInfo.events = {});
        if (componentInfo.events[event.name]) {
            componentInfo.events[event.name].desc = event.desc ? event.desc : componentInfo.events[event.name].desc
        } else {
            componentInfo.events[event.name] = event
        }
    }
}

const traverserTemplateAst = (ast, visitor = {}) => {
    function traverseArray (array, parent) {
        array.forEach(child => {
            traverseNode(child, parent);
        });
    }
    function traverseNode (node, parent) {
        visitor.enter && visitor.enter(node, parent);
        visitor[node.tag] && visitor[node.tag](node, parent);
        node.children && traverseArray(node.children, node);
        visitor.exit && visitor.exit(node, parent);
    }
    traverseNode(ast, null);
}

traverserTemplateAst(template.ast, {
    slot (node, parent) {
        !componentInfo.slots && (componentInfo.slots = {})
        // 获取节点位置
        let index = parent.children.findIndex(item => item === node)
        let desc = '无描述', name = '-';
        if (index > 0) {
            let tag = parent.children[index - 1]
            // isComment 判断是否是 注释
            if (tag.isComment) {
                desc = tag.text.trim()
            }
        }
        if (node.slotName) name = node.attrsMap.name
        componentInfo.slots[name] = {
            name,
            desc
        }
    }
})