amis2/packages/amis-formula/scripts/genDoc.ts

import path = require('path');
import fs = require('fs');
import doctrine = require('doctrine');

const workDir = path.resolve(path.dirname(__dirname));
const jsFile = path.join(workDir, 'src/evalutor.ts');
const outputFile = path.join(workDir, 'lib/doc.js');
const outputMD = path.join(workDir, 'lib/doc.md');

function getFormulaComments(contents: string) {
  const comments: Array<{
    fnName: string;
    comments: string;
  }> = [];

  contents.replace(/\/\*[\s\S]+?\*\//g, (_, index, input) => {
    const pos = index + _.length;
    const following = input.substring(pos, pos + 200);

    if (/^\s*fn(\w+)\(/.test(following)) {
      comments.push({
        fnName: RegExp.$1,
        comments: _
      });
    }

    return _;
  });

  return comments;
}

function formatType(tag: any): string {
  // console.log(tag);
  if (tag.type === 'RestType') {
    return `...${formatType(tag.expression)}`;
  } else if (tag.type === 'TypeApplication') {
    return `Array<${tag.applications
      .map((item: any) => formatType(item))
      .join(',')}>`;
  }

  return tag.name;
}

async function main(...params: Array<any>) {
  const contents = fs.readFileSync(jsFile, 'utf8');

  const comments = getFormulaComments(contents);
  const result = comments.map(item => {
    const ast = doctrine.parse(item.comments, {
      unwrap: true
    });
    const result: any = {
      name: item.fnName,
      description: ast.description
    };

    let example = '';
    let params: Array<any> = [];
    let returns: any = undefined;
    let namespace = '';
    ast.tags.forEach(tag => {
      if (tag.title === 'example') {
        example = tag.description!;
      } else if (tag.title === 'namespace') {
        namespace = tag.name!;
      } else if (tag.title === 'param') {
        params.push({
          type: formatType(tag.type),
          name: tag.name,
          description: tag.description
        });
      } else if (tag.title === 'returns') {
        returns = {
          type: formatType(tag.type),
          description: tag.description
        };
      }
    });

    result.example = example;
    result.params = params;
    result.returns = returns;
    result.namespace = namespace;

    return result;
  });

  fs.writeFileSync(
    outputFile,
    `/**\n * 公式文档\n */\nexports.doc = ${JSON.stringify(
      result,
      null,
      2
    )}`.replace(/\"(\w+)\"\:/g, (_, key) => `${key}:`),
    'utf8'
  );
  console.log(`公式文档生成 > ${outputFile}`);
  fs.writeFileSync(
    outputFile.replace(/\.js$/, '.d.ts'),
    //  可以通过以下命令生成
    // tsc --declaration --emitDeclarationOnly --allowJs doc.js
    [
      `export var doc: {`,
      `  name: string;`,
      `  description: string;`,
      `  example: string;`,
      `  params: {`,
      `      type: string;`,
      `      name: string;`,
      `      description: string;`,
      `  }[];`,
      `  returns: {`,
      `      type: string;`,
      `      description: string;`,
      `  };`,
      `  namespace: string;`,
      `}[];`
    ].join('\n'),
    'utf8'
  );

  const grouped: any = {};
  result.forEach((item: any) => {
    const scope = item.namespace || '其他';
    const arr = grouped[scope] || (grouped[scope] = []);

    arr.push(item);
  });

  let md = '';
  Object.keys(grouped).forEach(key => {
    md += `## ${key}\n\n`;

    grouped[key].forEach((item: any) => {
      md += `### ${item.name}\n\n`;

      md += `用法：\`${item.example}\`\n\n`;

      if (item.params.length) {
        // md += `参数：\n`;

        item.params.forEach((param: any) => {
          md += ` * \`${param.name}:${param.type}\` ${param.description}\n`;
        });

        if (item.returns) {
          md += `\n返回：\`${item.returns.type}\` ${
            item.returns.description || ''
          }\n\n`;
        }
      }

      md += `${item.description}\n\n`;
    });
  });
  fs.writeFileSync(outputMD, md, 'utf8');
  console.log(`公式md生成 > ${outputMD}`);
}

main().catch(e => console.error(e));