Transform tagged template

joe-sky · Nov 27, 2018 · 9db4040 · 9db4040
1 parent ebc7c1f
commit 9db4040
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 10 deletions.
diff --git a/packages/babel-plugin-nornj-in-jsx/package-lock.json b/packages/babel-plugin-nornj-in-jsx/package-lock.json
diff --git a/packages/babel-plugin-nornj-in-jsx/src/index.js b/packages/babel-plugin-nornj-in-jsx/src/index.js
@@ -5,7 +5,7 @@ const nj = require('nornj').default;
 // const transformWith = require('./withTag');
 const transformExTag = require('./exTag');
 const transformExAttr = require('./exAttr');
-const transformExpression = require('./expression');
+const transformTaggedTemplate = require('./taggedTemplate');
 const astUtil = require('./util/ast');
 const utils = require('./util/utils');
 
@@ -19,7 +19,8 @@ module.exports = function (babel) {
   // };
   const exTagHandler = transformExTag(babel);
   const exAttrHandler = transformExAttr(babel);
-  const expressionHandler = transformExpression(babel);
+  const taggedTemplateHandler = transformTaggedTemplate(babel);
+  let TaggedTemplates = ['nj', 'n', 't', 's'];
 
   const visitor = {
     JSXElement: {
@@ -57,8 +58,9 @@ module.exports = function (babel) {
       }
     },
     TaggedTemplateExpression(path, state) {
-      if (path.node.tag.name === 'n') {
-        path.replaceWith(expressionHandler(path.node, path.hub.file, state));
+      const taggedName = path.node.tag.name;
+      if (TaggedTemplates.indexOf(taggedName) >= 0) {
+        path.replaceWith(taggedTemplateHandler(path.node, path.hub.file, state, taggedName));
       }
     },
     ImportDeclaration(path, state) {
@@ -120,7 +122,8 @@ module.exports = function (babel) {
             [
               types.importDefaultSpecifier(types.identifier('nj')),
               types.importSpecifier(types.identifier('n'), types.identifier('expression')),
-              types.importSpecifier(types.identifier('t'), types.identifier('template'))
+              types.importSpecifier(types.identifier('t'), types.identifier('template')),
+              types.importSpecifier(types.identifier('s'), types.identifier('css'))
             ],
             types.stringLiteral('nornj')
           ));

diff --git a/...bel-plugin-nornj-in-jsx/src/expression.js → ...plugin-nornj-in-jsx/src/taggedTemplate.js b/...bel-plugin-nornj-in-jsx/src/expression.js → ...plugin-nornj-in-jsx/src/taggedTemplate.js
@@ -5,8 +5,8 @@ const generate = require('./util/generate');
 module.exports = function (babel) {
   const types = babel.types;
 
-  return function (node, file, state) {
+  return function (node, file, state, taggedName) {
     const { quasis, expressions } = node.quasi;
-    return generate.createRenderTmpl(babel, quasis, expressions, state.opts, { isExpresson: true });
+    return generate.createRenderTmpl(babel, quasis, expressions, state.opts, taggedName);
   };
 };
diff --git a/packages/babel-plugin-nornj-in-jsx/src/util/generate.js b/packages/babel-plugin-nornj-in-jsx/src/util/generate.js
@@ -185,11 +185,11 @@ const CTX_DATAS = 'datas';
 const CTX_DATA = 'data';
 const CTX_GET_DATA = 'getData';
 
-function createRenderTmpl(babel, quasis, expressions, opts, taggedTmplConfig) {
+function createRenderTmpl(babel, quasis, expressions, opts, taggedName) {
   const types = babel.types;
 
   let tmplStr = '';
-  if (!taggedTmplConfig) {
+  if (!taggedName) {
     let paramCount = 0;
     quasis.forEach((q, i) => {
       tmplStr += q.value.cooked;
@@ -203,6 +203,16 @@ function createRenderTmpl(babel, quasis, expressions, opts, taggedTmplConfig) {
     });
   }
   else {
+    let taggedTmplConfig = {};
+    switch (taggedName) {
+      case 'n':
+        taggedTmplConfig = { isExpresson: true };
+        break;
+      case 's':
+        taggedTmplConfig = { isCss: true };
+        break;
+    }
+
     tmplStr = Object.assign({ quasis: quasis.map(q => q.value.cooked) }, taggedTmplConfig);
   }