strothj / babel-plugin-react-docgen-typescript Goto Github PK

const rdt = require('react-docgen-typescript');
const template = require('@babel/template');
const ts = require('typescript');
const glob = require('glob');
const util = require('util');

const files = glob.sync('src/**/*.{ts,tsx}', { absolute: true });
const program = ts.createProgram(files, {});

module.exports = function() {
 return {
   pre() {
     this.fileProcessed = false;
   },
   visitor: {
     Program(path, state) {
       if (this.fileProcessed) {
         return;
       }
       this.fileProcessed = true;
       const { filename } = state.file.opts;
       if (files.indexOf(filename) === -1) {
         files.push(file);
         program = ts.createProgram(files, {});
       }

       const docs = rdt
         .withCompilerOptions({}, { shouldExtractLiteralValuesFromEnum: true })
         .parseWithProgramProvider(filename, () => program);

       docs.forEach((doc) => {
         path.node.body.push(
           template.default.ast(`
           try {
             ${doc.displayName}.__docgenInfo = ${util.inspect(
             doc,
             null,
             10,
           )};
           } catch (e) {}
         `),
         );
       });
     },
   },
 };
};

const useESM = process.env.ESM === '1';

module.exports = function(api) {
  api.cache(true);

  return {
    presets: [
      '@babel/preset-typescript',
      ['@babel/preset-env', { modules: useESM ? false : 'commonjs' }],
    ],
    plugins: [
      '@babel/plugin-transform-react-jsx',
      './docgen',
      '@babel/plugin-transform-runtime',
    ],
  };
};