Handle new Bublé template string compilation. (#155)

bublejs/buble#67 changed Bublé template strings to compile similarly to Babel ones, for spec reasons.
With this patch, the new style as well as the old style are detected correctly.

Author: goto-bus-stop

package.json

@@ -23,6 +23,7 @@
   "dependencies": {
     "@choojs/findup": "^0.2.0",
     "acorn-node": "^1.3.0",
+    "estree-is-member-expression": "^1.0.0",
     "fast-json-parse": "^1.0.2",
     "insert-css": "^2.0.0",
     "map-limit": "0.0.1",

transform.js

@@ -1,3 +1,4 @@
+const isMemberExpression = require('estree-is-member-expression')
 const cssResolve = require('style-resolve').sync
 const transformAst = require('transform-ast')
 const staticEval = require('static-eval')
@@ -18,6 +19,13 @@ function isBabelTemplateDefinition (node) {
     node.callee.type === 'Identifier' && node.callee.name === '_taggedTemplateLiteral'
 }
 
+function isBubleTemplateDefinition (node) {
+  return node.type === 'CallExpression' &&
+    isMemberExpression(node.callee, 'Object.freeze') &&
+    node.arguments[0] && node.arguments[0].type === 'ArrayExpression' &&
+    node.arguments[0].elements.every(function (el) { return el.type === 'Literal' })
+}
+
 // inline sheetify transform for browserify
 // obj -> (str, opts) -> str
 function transform (filename, options) {
@@ -61,6 +69,7 @@ function transform (filename, options) {
     // judge us too harshly, we'll work on perf ✨soon✨ -yw
     const nodes = []
     const babelTemplateObjects = {}
+    const bubleTemplateObjects = {}
     const src = Buffer.concat(bufs).toString('utf8')
     var mname = null
     var ast
@@ -117,11 +126,17 @@ function transform (filename, options) {
       var css
       var elements
 
-      if (node.type === 'VariableDeclarator' && node.init && isBabelTemplateDefinition(node.init)) {
-        // Babel generates helper calls like
-        //    _taggedTemplateLiteral([":host .class { color: hotpink; }"], [":host .class { color: hotpink; }"])
-        // we only keep the "cooked" part
-        babelTemplateObjects[node.id.name] = node.init.arguments[0]
+      if (node.type === 'VariableDeclarator' && node.init) {
+        if (isBabelTemplateDefinition(node.init)) {
+          // Babel generates helper calls like
+          //    _taggedTemplateLiteral([":host .class { color: hotpink; }"], [":host .class { color: hotpink; }"])
+          // we only keep the "cooked" part
+          babelTemplateObjects[node.id.name] = node.init.arguments[0]
+        } else if (isBubleTemplateDefinition(node.init)) {
+          // Buble generates helper calls like
+          //    Object.freeze([":host .class { color: hotpink; }"])
+          bubleTemplateObjects[node.id.name] = node.init.arguments[0]
+        }
       }
 
       if (node.type === 'TemplateLiteral' && node.parent && node.parent.tag) {
@@ -146,7 +161,12 @@ function transform (filename, options) {
         } else if (node.arguments[0] && node.arguments[0].type === 'Identifier') {
           // Babel generates code like
           //    sheetify(_templateObject)
-          elements = babelTemplateObjects[node.arguments[0].name].elements
+          var arg = node.arguments[0].name
+          if (babelTemplateObjects[arg]) {
+            elements = babelTemplateObjects[arg].elements
+          } else if (bubleTemplateObjects[arg]) {
+            elements = bubleTemplateObjects[arg].elements
+          }
         }
 
         if (elements) {