From fa0b2cbbb627be8efea918ed03a0a33862affba7 Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Sat, 21 Jan 2012 21:09:53 -0500
Subject: [PATCH] make extra nav and footer follow theme

---
 www/kojiweb/includes/footer.chtml |  2 +-
 www/kojiweb/includes/header.chtml |  2 +-
 www/lib/kojiweb/util.py           | 42 ++++++++++++++++++-------------
 3 files changed, 27 insertions(+), 19 deletions(-)

diff --git a/www/kojiweb/includes/footer.chtml b/www/kojiweb/includes/footer.chtml
index 058c88b9..9e8b2d9d 100644
--- a/www/kojiweb/includes/footer.chtml
+++ b/www/kojiweb/includes/footer.chtml
@@ -6,7 +6,7 @@
           <a href="https://fedorahosted.org/koji/"><img src="$util.themePath('images/powered-by-koji.png')" alt="Powered By Koji" id="PoweredByKojiLogo"/></a>
         </p>
 
-#set $localfooterpath="/usr/share/koji-web/local/extra-footer.html"
+#set $localfooterpath=$util.themePath("extra-footer.html", local=True)
 #if os.path.exists($localfooterpath)
 #set $localfooter="".join(open($localfooterpath).readlines())
 $localfooter
diff --git a/www/kojiweb/includes/header.chtml b/www/kojiweb/includes/header.chtml
index 850250d5..ff04063f 100644
--- a/www/kojiweb/includes/header.chtml
+++ b/www/kojiweb/includes/header.chtml
@@ -47,7 +47,7 @@
         <!-- HEADER -->
         <div id="header">
           <img src="$util.themePath('images/koji.png')" alt="Koji Logo" id="kojiLogo"/>
-#set $localnavpath="/usr/share/koji-web/local/extra-nav.html"
+#set $localnavpath=$util.themePath("extra-nav.html", local=True)
 #if os.path.exists($localnavpath)
 #set $localnav="".join(open($localnavpath).readlines())
 $localnav
diff --git a/www/lib/kojiweb/util.py b/www/lib/kojiweb/util.py
index ec8e7463..8f9c673a 100644
--- a/www/lib/kojiweb/util.py
+++ b/www/lib/kojiweb/util.py
@@ -32,31 +32,39 @@ def _initValues(req, title='Build System Info', pageID='summary'):
     values['title'] = title
     values['pageID'] = pageID
     values['currentDate'] = str(datetime.datetime.now())
-    theme = req.get_options().get('KojiTheme', None)
-    if theme:
-        themeCache.clear()
-        themeInfo['name'] = theme
-        themeInfo['staticdir'] = req.get_options().get('KojiStaticDir', '/usr/share/koji-web/static')
-    else:
-        themeInfo.clear()
+    themeCache.clear()
+    themeInfo.clear()
+    themeInfo['name'] = req.get_options().get('KojiTheme', None)
+    themeInfo['staticdir'] = req.get_options().get('KojiStaticDir', '/usr/share/koji-web/static')
 
     req._values = values
 
     return values
 
-def themePath(path):
+def themePath(path, local=False):
     global themeInfo
     global themeCache
-    if not themeInfo:
-        return "/koji-static/%s" % path
-    if path in themeCache:
-        return themeCache[path]
-    themepath = os.path.join(themeInfo['staticdir'], 'themes', themeInfo['name'], path)
-    if os.path.exists(themepath):
-        ret = "/koji-static/themes/%s/%s" % (themeInfo['name'], path)
+    local = bool(local)
+    if (path, local) in themeCache:
+        return themeCache[path, local]
+    if not themeInfo['name']:
+        if local:
+            ret = os.path.join(themeInfo['staticdir'], path)
+        else:
+            ret = "/koji-static/%s" % path
     else:
-        ret = "/koji-static/%s" % path
-    themeCache[path] = ret
+        themepath = os.path.join(themeInfo['staticdir'], 'themes', themeInfo['name'], path)
+        if os.path.exists(themepath):
+            if local:
+                ret = themepath
+            else:
+                ret = "/koji-static/themes/%s/%s" % (themeInfo['name'], path)
+        else:
+            if local:
+                ret = os.path.join(themeInfo['staticdir'], path)
+            else:
+                ret = "/koji-static/%s" % path
+    themeCache[path, local] = ret
     return ret
 
 class DecodeUTF8(Cheetah.Filters.Filter):