import reactUtilsFactory from 'babel-utils-react';

export default function ({Plugin, types: t}) {
  const reactUtils = reactUtilsFactory(t);

  return new Plugin('ast-transform', {
    visitor: {
      ClassDeclaration(node) {
        if (reactUtils.isReactComponentClassDeclaration(node)) {
          // Yay, a React.Component class!
        }
      }
    }
  });
}