eval code nodes in attributes

Author: jescalan

lib/index.js

@@ -5,10 +5,25 @@ const parse = require('reshape-parser')
 module.exports = (_locals = {}) => {
   return function (tree, opts) {
     const locals = Object.assign({}, opts.locals || {}, _locals)
-    return util.modifyNodes(tree, (node) => node.type === 'code', (node) => {
-      const newNode = parse(gen.call({ runtime: opts.runtime }, node, opts)(locals))
-      newNode.location = node.location
-      return newNode
+    return util.modifyNodes(tree, (node) => {
+      return node.type === 'code' || node.type === 'tag'
+    }, (node) => {
+      if (node.type === 'tag') {
+        if (!node.attrs) return node
+        for (let k in node.attrs) {
+          node.attrs[k].map((n) => {
+            if (n.type === 'code') { node.attrs[k] = evalNode(n, locals, opts) }
+          })
+        }
+        return node
+      }
+      if (node.type === 'code') return evalNode(node, locals, opts)
     })
   }
 }
+
+function evalNode (node, locals, opts) {
+  const newNode = parse(gen.call({ runtime: opts.runtime }, node, opts)(locals))
+  if (node.location) { newNode.location = node.location }
+  return newNode
+}

test/index.js

@@ -15,6 +15,12 @@ test('loops', (t) => {
     .then((res) => { t.is(res.output(), '<p>one</p><p>two</p>') })
 })
 
+test('evals attribute code nodes', (t) => {
+  return reshape({ plugins: [expressions(), evalCode({ foo: 'bar' })] })
+    .process('<p class="{{ foo }}">content</p>')
+    .then((res) => { t.is(res.output(), '<p class="bar">content</p>') })
+})
+
 test('merges from reshape options locals', (t) => {
   return reshape({ plugins: [expressions(), evalCode()], locals: { foo: 'bar' } })
     .process('<p>{{ foo }}</p>')