Add a proper attributes value for HTML elements

1 files changed, 3 insertions, 8 deletions

diff --git a/lib/html-generator.js b/lib/html-generator.js
index 065d1d2..179d6d0 100644
--- a/lib/html-generator.js
+++ b/lib/html-generator.js
@@ -79,14 +79,9 @@ function HtmlGenerator() {
     function writeElement(stackElement) {
         stackElement.written = true;
         var element = stackElement.element;
-        if (element["class"]) {
-            var attributeString = util.format(' %s="%s"', "class", element["class"]);
-        } else {
-            var attributeString = "";
-        }
-        if (element.href) {
-            attributeString += util.format(' href="%s"', escapeHtml(element.href));
-        }
+        var attributeString = _.map(element.attributes, function(value, key) {
+            return util.format(' %s="%s"', key, value);
+        }).join("");
         fragments.push(util.format("<%s%s>", element.tagName, attributeString));
     }