index.js

const { utils: docgenUtils } = require("react-docgen");
const { visit } = require("ast-types/main");
const { default: resolveHOC } = require("react-docgen/dist/utils/resolveHOC");

module.exports = function(ast) {
  const components = [];

  const exportTagged = path => {
    /**
     * If an export doesn't have leading comments, we can simply continue
     */
    const leadingComments = path.value.leadingComments;
    if (!leadingComments) {
      return false;
    }

    // Search for the @component tag in any export.
    const isTagged = leadingComments.reduce(
      (acc, comment) =>
        acc === true || comment.value.indexOf("@component") !== -1,
      [],
    );

    if (!isTagged) {
      return false;
    }

    const definitions = docgenUtils.resolveExportDeclaration(path);

    definitions.forEach(definition => {
      if (definition && components.indexOf(definition) === -1) {
        components.push(docgenUtils.resolveToValue(resolveHOC(definition)));
      }
    });

    return false;
  };

  visit(ast, {
    visitFunctionDeclaration: false,
    visitFunctionExpression: false,
    visitClassDeclaration: false,
    visitClassExpression: false,
    visitIfStatement: false,
    visitWithStatement: false,
    visitSwitchStatement: false,
    visitCatchCause: false,
    visitWhileStatement: false,
    visitDoWhileStatement: false,
    visitForStatement: false,
    visitForInStatement: false,

    visitExportDeclaration: exportTagged,
    visitExportNamedDeclaration: exportTagged,
    visitExportDefaultDeclaration: exportTagged,
  });

  return components;
};