From 6b2de29fa78ae2666e79c5f383e38acde12aeb44 Mon Sep 17 00:00:00 2001
From: John Mulhausen <john@docker.com>
Date: Mon, 10 Oct 2016 20:15:50 -0700
Subject: [PATCH] Fix for nesting of TOC

---
 js/menu.js | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/js/menu.js b/js/menu.js
index 163fd827ea..0f875cdb05 100644
--- a/js/menu.js
+++ b/js/menu.js
@@ -57,7 +57,6 @@ jQuery(document).ready(function(){
     var output = "";
     $("h2, h3, h4").each(function() {
         var li= "<li><a href='" + window.location + "#" + $(this).attr('id') + "'>" + $(this).text().replace("¶","") + "</a></li>";
-        lastHeader = currentHeader;
         if( $(this).is("h2") ){
           // h2
           currentHeader = 2;
@@ -68,18 +67,19 @@ jQuery(document).ready(function(){
           // h4
           currentHeader = 4;
         }
-        if (currentHeader > lastHeader)
-        {
+        if (currentHeader > lastHeader) {
             // nest further
-            output += "<ul>" + li;
-        } else if (lastHeader < currentHeader)
-        {
-            // close nesting
-            output += "</ul>" + li
-        } else {
-            // continue, no change in nesting
-            output += li;
+            output += "<ul>"
         }
+        if (currentHeader < lastHeader && lastHeader > 0) {
+            // close nesting
+            for (i=0; i < (lastHeader - currentHeader); i++)
+            {
+              output += "</ul>"
+            }
+        }
+        output += li;
+        lastHeader = currentHeader;
         /*
         if( $(this).is("h2") ){
             prevH2List = $("<ul></ul>");