add pure annotation for function(object) form styled-components

Author: quantizor

src/index.js

@@ -9,6 +9,7 @@ export default function({ types: t }) {
     visitor: {
       CallExpression(path, state) {
         displayNameAndId(t)(path, state)
+        pureAnnotation(t)(path, state)
       },
       TaggedTemplateExpression(path, state) {
         minify(t)(path, state)

src/visitors/pure.js

@@ -6,13 +6,13 @@ import { isStyled, isPureHelper } from '../utils/detectors'
 export default t => (path, state) => {
   if (usePureAnnotation(state)) {
     if (
-      isStyled(path.node, state) ||
-      isStyled(path.node.callee, state) ||
-      isPureHelper(path.node.callee, state)
+      isStyled(t)(path.node, state) ||
+      isStyled(t)(path.node.callee, state) ||
+      isPureHelper(t)(path.node.tag || path.node.callee, state)
     ) {
       if (
-        path.parent.type == 'VariableDeclarator' ||
-        path.parent.type == 'TaggedTemplateExpression'
+        path.parent.type === 'VariableDeclarator' ||
+        path.parent.type === 'TaggedTemplateExpression'
       ) {
         annotateAsPure(path)
       }

test/__snapshots__/index.test.js.snap

@@ -162,6 +162,14 @@ styled(Inner).withConfig({
   displayName: \\"annotate-styled-calls-with-pure-comments__WrappedComponent\\",
   componentId: \\"sc-6tnjs3-6\\"
 })([\\"\\"]);
+const StyledObjectForm =
+/*#__PURE__*/
+styled.div.withConfig({
+  displayName: \\"annotate-styled-calls-with-pure-comments__StyledObjectForm\\",
+  componentId: \\"sc-6tnjs3-7\\"
+})({
+  color: red
+});
 const notStyled =
 /*#__PURE__*/
 styled.div('')\`\`; // not transpiled by styled components but should add pure comment

test/fixtures/annotate-styled-calls-with-pure-comments/index.js

@@ -7,5 +7,6 @@ const styles = { One: styled.div`` }
 let Component
 Component = styled.div``
 const WrappedComponent = styled(Inner)``
+const StyledObjectForm = styled.div({ color: red })
 const notStyled = styled.div('')`` // not transpiled by styled components but should add pure comment
 const normalFunc = add(5, 3)