index.js

import syntax from 'babel-plugin-syntax-jsx'
import pureAnnotation from './visitors/pure'
import pureInlineCalculations from './visitors/pureInlineCalculations'
import minify from './visitors/minify'
import displayNameAndId from './visitors/displayNameAndId'
import templateLiterals from './visitors/templateLiterals'
import assignStyledRequired from './visitors/assignStyledRequired'
import transpileCssProp from './visitors/transpileCssProp'
import pureWrapStaticProps from './visitors/pureWrapStaticProps'

export default function({ types: t }) {
  return {
    inherits: syntax,
    visitor: {
      Program(path, state) {
        path.traverse(
          {
            JSXAttribute(path, state) {
              transpileCssProp(t)(path, state)
            },
            VariableDeclarator(path, state) {
              assignStyledRequired(t)(path, state)
            },
          },
          state
        )
      },
      CallExpression(path, state) {
        displayNameAndId(t)(path, state)
        pureAnnotation(t)(path, state)
      },
      TaggedTemplateExpression(path, state) {
        pureInlineCalculations(t)(path, state)
        minify(t)(path, state)
        displayNameAndId(t)(path, state)
        templateLiterals(t)(path, state)
        pureAnnotation(t)(path, state)
      },
      FunctionDeclaration(path, state) {
        // technically this is more like,
        // "mark pure if it's a function component that consumes a styled component and also has static properties",
        // but that's rather long ;)
        pureWrapStaticProps(t)(path, state)
      },
      VariableDeclarator(path, state) {
        // same thing for arrow functions
        pureWrapStaticProps(t)(path, state)
      },
    },
  }
}