Allow CSS injection inside 'styled' tagged template literals

atom · Oct 30, 2018 · a2f66fc · a2f66fc
1 parent 61cc764
commit a2f66fc
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/lib/main.js b/lib/main.js
@@ -8,10 +8,10 @@ exports.activate = function () {
       const {firstChild} = callExpression
       switch (firstChild.type) {
         case 'identifier':
-          return firstChild.text
+          return languageStringForTemplateTag(firstChild.text)
         case 'member_expression':
           if (firstChild.startPosition.row === firstChild.endPosition.row) {
-            return firstChild.text
+            return languageStringForTemplateTag(firstChild.text)
           }
       }
     },
@@ -50,3 +50,13 @@ exports.activate = function () {
     content (regex) { return regex }
   })
 }
+
+const STYLED_REGEX = /\bstyled\b/i
+
+function languageStringForTemplateTag (tag) {
+  if (STYLED_REGEX.test(tag)) {
+    return 'CSS'
+  } else {
+    return tag
+  }
+}