Write thead and th for headers in table

1 files changed, 44 insertions, 11 deletions

diff --git a/lib/document-to-html.js b/lib/document-to-html.js
index d56ad1d..5e7079a 100644
--- a/lib/document-to-html.js
+++ b/lib/document-to-html.js
@@ -79,16 +79,17 @@ function DocumentConversion(options, comments) {
         });
     }
     
-    function convertElements(elements, messages) {
+    function convertElements(elements, messages, options) {
         return flatMap(elements, function(element) {
-            return elementToHtml(element, messages);
+            return elementToHtml(element, messages, options);
         });
     }
 
-    function elementToHtml(element, messages) {
+    function elementToHtml(element, messages, options) {
+        options = options || {};
         var handler = elementConverters[element.type];
         if (handler) {
-            return handler(element, messages);
+            return handler(element, messages, options);
         } else {
             return [];
         }
@@ -213,14 +214,46 @@ function DocumentConversion(options, comments) {
     }
     
     function convertTable(element, messages) {
-        return wrapChildrenInFreshElement(element, "table", messages);
+        var bodyIndex = _.findIndex(element.children, function(child) {
+            return !child.isHeader;
+        });
+        if (bodyIndex === -1) {
+            bodyIndex = element.children.length;
+        }
+        var children;
+        if (bodyIndex === 0) {
+            children = convertElements(
+                element.children,
+                messages,
+                {isTableHeader: false}
+            );
+        } else {
+            var headRows = convertElements(
+                element.children.slice(0, bodyIndex),
+                messages,
+                {isTableHeader: true}
+            );
+            var bodyRows = convertElements(
+                element.children.slice(bodyIndex),
+                messages,
+                {isTableHeader: false}
+            );
+            children = [
+                Html.freshElement("thead", {}, headRows),
+                Html.freshElement("tbody", {}, bodyRows)
+            ];
+        }
+        return [
+            Html.freshElement("table", {}, [Html.forceWrite].concat(children))
+        ];
     }
     
-    function convertTableRow(element, messages) {
-        return wrapChildrenInFreshElement(element, "tr", messages);
+    function convertTableRow(element, messages, options) {
+        return wrapChildrenInFreshElement(element, "tr", messages, options);
     }
     
-    function convertTableCell(element, messages) {
+    function convertTableCell(element, messages, options) {
+        var tagName = options.isTableHeader ? "th" : "td";
         var children = convertElements(element.children, messages);
         var attributes = {};
         if (element.colSpan !== 1) {
@@ -231,7 +264,7 @@ function DocumentConversion(options, comments) {
         }
             
         return [
-            Html.freshElement("td", attributes, [Html.forceWrite].concat(children))
+            Html.freshElement(tagName, attributes, [Html.forceWrite].concat(children))
         ];
     }
     
@@ -292,8 +325,8 @@ function DocumentConversion(options, comments) {
         }
     }
     
-    function wrapChildrenInFreshElement(element, wrapElementName, messages) {
-        var children = convertElements(element.children, messages);
+    function wrapChildrenInFreshElement(element, wrapElementName, messages, options) {
+        var children = convertElements(element.children, messages, options);
         return [
             Html.freshElement(wrapElementName, {}, [Html.forceWrite].concat(children))
         ];