Use Object.create to simplify element handling in HtmlGenerator

1 files changed, 12 insertions, 14 deletions

diff --git a/lib/html-generator.js b/lib/html-generator.js
index 3825f9b..3705767 100644
--- a/lib/html-generator.js
+++ b/lib/html-generator.js
@@ -47,20 +47,19 @@ function HtmlGenerator() {
     }
     
     function stylePartMatchesStackElement(stylePart, stackElement) {
-        return stackElement && stylePart.tagName === stackElement.htmlPathElement.tagName;
+        return stackElement && stylePart.tagName === stackElement.tagName;
     }
     
-    function open(element) {
-        stack.push({
-            htmlPathElement: element,
-            written: false
-        });
+    function open(htmlPathElement) {
+        var stackElement = Object.create(htmlPathElement);
+        stackElement.written = false;
+        stack.push(stackElement);
     }
     
     function close() {
-        var stackElement = stack.pop();
-        if (stackElement.written) {
-            fragments.push(util.format("</%s>", stackElement.htmlPathElement.tagName));
+        var element = stack.pop();
+        if (element.written) {
+            fragments.push(util.format("</%s>", element.tagName));
         }
     }
     
@@ -76,11 +75,10 @@ function HtmlGenerator() {
         }).forEach(writeElement);
     }
     
-    function writeElement(stackElement) {
-        stackElement.written = true;
-        var htmlPathElement = stackElement.htmlPathElement;
-        var attributeString = generateAttributeString(htmlPathElement.attributes)
-        fragments.push(util.format("<%s%s>", htmlPathElement.tagName, attributeString));
+    function writeElement(element) {
+        element.written = true;
+        var attributeString = generateAttributeString(element.attributes)
+        fragments.push(util.format("<%s%s>", element.tagName, attributeString));
     }
     
     function selfClosing(element) {