antd-tools/lib/jest/demoPreprocessor.js

tools for ant design

const markTwain = require('mark-twain');
const JsonML = require('jsonml.js/lib/utils');
const babel = require('babel-core');
const getBabelCommonConfig = require('../getBabelCommonConfig');
const rewriteSource = require('./rewriteSource');

const libDir = process.env.LIB_DIR || 'components';

function getCode(tree) {
  let code;
  const find = (node) => {
    if (code) return;
    if (!JsonML.isElement(node)) return;
    if (JsonML.getTagName(node) !== 'pre') {
      JsonML.getChildren(node).forEach(find);
      return;
    }
    code = JsonML.getChildren(
      JsonML.getChildren(node)[0] || ''
    )[0] || '';
  };
  find(tree);
  return code;
}

function createDemo({ types: t }) {
  return {
    visitor: {
      Program(path) {
        const importReact = t.ImportDeclaration(
          [
            t.importDefaultSpecifier(t.Identifier('React')),
          ],
          t.StringLiteral('react')
        );
        path.unshiftContainer('body', importReact);
      },

      CallExpression(path) {
        if (path.node.callee.object &&
          path.node.callee.object.name === 'ReactDOM' &&
          path.node.callee.property.name === 'render') {
          const app = t.VariableDeclaration('const', [
            t.VariableDeclarator(t.Identifier('__Demo'), path.node.arguments[0]),
          ]);
          const exportDefault = t.ExportDefaultDeclaration(t.Identifier('__Demo'));
          path.insertAfter(exportDefault);
          path.insertAfter(app);
          path.remove();
        }
      },

      ImportDeclaration(path) {
        const libPattern = new RegExp('antd(-mobile)?/lib/.+');
        if (libPattern.test(path.node.source.value)) {
          path.node.source.value = path.node.source.value.replace(/antd(-mobile)?\/lib/, '../../../components');
        }
        rewriteSource(t, path, libDir);
      },
    },
  };
}

module.exports = {
  process(src, path) {
    const markdown = markTwain(src);
    src = getCode(markdown.content);

    // @ @ secret API.
    global.__clearBabelAntdPlugin && global.__clearBabelAntdPlugin(); // eslint-disable-line

    const babelConfig = getBabelCommonConfig(true);
    babelConfig.plugins = [...babelConfig.plugins];

    babelConfig.plugins.push(createDemo);

    if (libDir !== 'dist') {
      babelConfig.plugins.push(
        [
          require.resolve('babel-plugin-import'),
          [
            {
              libraryName: 'antd',
              libraryDirectory: `../../../../${libDir}`,
            },
            {
              libraryName: 'antd-mobile',
              libraryDirectory: `../../../../${libDir}`,
            },
          ],
        ]
      );
    }

    babelConfig.filename = path;

    src = babel.transform(src, babelConfig).code;

    return src;
  },
};