support .attrs({}) call. Fixes #28

Author: undeadcat

src/main/kotlin/com/intellij/StyledComponents/Patterns.kt

@@ -1,5 +1,6 @@
 package com.intellij.StyledComponents
 
+import com.intellij.lang.javascript.psi.JSCallExpression
 import com.intellij.lang.javascript.psi.JSExpression
 import com.intellij.lang.javascript.psi.JSReferenceExpression
 import com.intellij.lang.javascript.psi.ecma6.ES6TaggedTemplateExpression
@@ -34,6 +35,7 @@ fun withReferenceName(name: String): ElementPattern<JSReferenceExpression> {
 }
 
 fun referenceExpression() = PlatformPatterns.psiElement(JSReferenceExpression::class.java)!!
+fun callExpression() = PlatformPatterns.psiElement(JSCallExpression::class.java)!!
 
 fun withNameStartingWith(vararg components: String): ElementPattern<JSReferenceExpression> {
     val componentsList = ContainerUtil.list(*components)

src/main/kotlin/com/intellij/StyledComponents/StyledComponentsInjector.kt

@@ -22,6 +22,7 @@ class StyledComponentsInjector : MultiHostInjector {
                         PlatformPatterns.psiElement(JSExpression::class.java)
                                 .withFirstChild(styledPattern))), "div {", "}"),
                 PlaceInfo(taggedTemplate(withReferenceName("extend")), "div {", "}"),
+                PlaceInfo(taggedTemplate(callExpression().withChild(withReferenceName("attrs"))), "div {", "}"),
                 PlaceInfo(taggedTemplate("css"), "div {", "}"),
                 PlaceInfo(taggedTemplate("injectGlobal")),
                 PlaceInfo(taggedTemplate("keyframes"), "@keyframes foo {", "}")

src/test/InjectionTest.kt

@@ -105,6 +105,15 @@ class InjectionTest : LightCodeInsightFixtureTestCase() {
                         "}")
     }
 
+    fun testComponentAttrs() {
+        doTest("const div = styled.div;\n" +
+                "const FilterIcon = div.attrs({ iconName: 'filter' })`\n" +
+                "  line-height: 0;\n" +
+                "`", "div {\n" +
+                "  line-height: 0;\n" +
+                "}")
+    }
+
     fun testInjectGlobal() {
         doTest("injectGlobal`\n" +
                 "  div{\n" +