From 79255e6425b584ab68d02a1cecb1fbf8cfeb9e54 Mon Sep 17 00:00:00 2001
From: Fredrik Unger <fred@tree.se>
Date: Sun, 10 Apr 2011 16:24:37 +0200
Subject: [PATCH] Implementing generation of a menu, currently only the full
 menu with and without Page marker.

---
 src/tree-cutter.py | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/src/tree-cutter.py b/src/tree-cutter.py
index 0216439..b199534 100755
--- a/src/tree-cutter.py
+++ b/src/tree-cutter.py
@@ -146,15 +146,15 @@ class Page():
         os.chdir(cwd)
 
     def template(self,sitemap):
-        htmlmenu =  sitemap.gen_menu(self._lang,None)
-        levelmenu = sitemap.gen_menu(self._lang,self)
+        htmlmenu =  sitemap.gen_menu(self._lang,None,None)
+        levelmenu = sitemap.gen_menu(self._lang,self,"tree")
         template = Template(file=style_tmpl,
                             searchList=[{'title':self._title},
                                         {'menu':htmlmenu},
                                         {'article':self._rendered_article},
                                         {'levelmenu':levelmenu},
-                                        {'levelname':levelname}])
-        outfile = tmptarget+self._file+'.'+self._lang+'.html'
+                                        {'levelname':'Menu'}])
+        outfile = tmptarget+'html'.join(self._file.rsplit('xml',1))
         mkdir_p(os.path.dirname(outfile))
         out = open(outfile, 'w')
         out.write(str(template))
@@ -267,6 +267,24 @@ class Trie():
 #        G.draw('g.png')
 #        print G.string()
 
+    def _menu(self, trie, lang, page, css):
+        html = "<ul%s>\n" % css
+        for l in trie:
+            sel = ''
+            if l.value().page(lang) == page:
+                sel = ' class="selected"'
+            html += '<li%s><a href="%s">%s</a>\n' \
+            % (sel,l.value().link(),l.value().page(lang).menu())
+            html += self._menu(l.children(), lang, page, "")
+        html += "</ul>\n"
+        return html
+
+    def menu(self,lang,page,cssclass):
+        css = ''
+        if cssclass:
+            css = ' class="'+cssclass+'"'
+        return self._menu(self._root, lang, page, css)
+
 class Sitemap():
     """Class keeping the internal site structure"""
     def __init__(self):
-- 
2.30.2