Additional fixes for #386 - Allow regular expression for an HTML attr…

…ibute value
marko-js · Nov 10, 2016 · a64044a · a64044a
1 parent da52209
commit a64044a
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/compiler/ast/HtmlAttribute/vdom/generateCode.js b/compiler/ast/HtmlAttribute/vdom/generateCode.js
@@ -7,14 +7,23 @@ module.exports = function generateCode(node, codegen, vdomUtil) {
     // node.name = codegen.generateCode(node.name);
     node.value = codegen.generateCode(node.value);
     node.isStatic = vdomUtil.isStaticValue(node.value);
-
     var name = node.name;
 
-    if (node.value && node.value.type !== 'Literal') {
-        if (name === 'class') {
-            node.value = builder.functionCall(context.helper('classAttr'), [node.value]);
-        } else if (name === 'style') {
-            node.value = builder.functionCall(context.helper('styleAttr'), [node.value]);
+    var attrValue = node.value;
+
+    if (attrValue) {
+        if (attrValue.type === 'Literal') {
+            var literalValue = attrValue.value;
+
+            if (literalValue instanceof RegExp) {
+                node.value = builder.literal(literalValue.source);
+            }
+        } else {
+            if (name === 'class') {
+                node.value = builder.functionCall(context.helper('classAttr'), [attrValue]);
+            } else if (name === 'style') {
+                node.value = builder.functionCall(context.helper('styleAttr'), [attrValue]);
+            }
         }
     }