Allow HTML elements to be reused in HTML generation

2 files changed, 44 insertions, 6 deletions

diff --git a/lib/html-generator.js b/lib/html-generator.js
index 247ca09..5f2befa 100644
--- a/lib/html-generator.js
+++ b/lib/html-generator.js
@@ -1,5 +1,7 @@
 var util = require("util");
 
+var _ = require("underscore");
+
 exports.HtmlGenerator = HtmlGenerator;
 
 
@@ -17,10 +19,20 @@ function HtmlGenerator() {
     }
     
     function style(style) {
-        popAll();
-        style.forEach(function(tagName) {
-            push(tagName);
+        var firstFreshPartIndex = findIndex(style, function(part) {
+            return part.fresh;
         });
+        
+        var partIndex = 0;
+        while (partIndex < firstFreshPartIndex && style[partIndex].tagName === stack[partIndex]) {
+            partIndex++;
+        }
+        while (stack.length > partIndex) {
+            pop();
+        }
+        for (; partIndex < style.length; partIndex++) {
+            push(style[partIndex].tagName);
+        }
     }
     
     function push(element) {
@@ -54,3 +66,12 @@ function escapeHtml(value) {
         .replace(/</g, '&lt;')
         .replace(/>/g, '&gt;');
 }
+
+function findIndex(array, predicate) {
+    for (var i = 0; i < array.length; i++) {
+        if (predicate(array[i])) {
+            return i;
+        }
+    }
+    return -1;
+}
diff --git a/lib/styles.js b/lib/styles.js
index d3aa0a4..06d63a8 100644
--- a/lib/styles.js
+++ b/lib/styles.js
@@ -1,10 +1,27 @@
+var _ = require("underscore");
+
 exports.topLevelElement = topLevelElement;
 exports.elements = elements;
+exports.element = element;
 
 function topLevelElement(tagName) {
-    return elements([tagName]);
+    return elements([element(tagName, {fresh: true})]);
+}
+
+function elements(elementStyles) {
+    return elementStyles.map(function(elementStyle) {
+        if (_.isString(elementStyle)) {
+            return element(elementStyle);
+        } else {
+            return elementStyle;
+        }
+    });
 }
 
-function elements(tagNames) {
-    return tagNames;
+function element(tagName, options) {
+    options = options || {};
+    return {
+        tagName: tagName,
+        fresh: options.fresh
+    };
 }