Ticket #264: irequestfilter-2.diff

tracnav/tracnav.py

@@ -62 +62 @@
-from trac.core import Component, implements
-from trac.wiki.api import WikiSystem, IWikiMacroProvider
-from trac.web.chrome import ITemplateProvider, add_stylesheet
+from trac.web.main import IRequestFilter, RequestDispatcher
-from trac.wiki.model import WikiPage
-from trac.wiki.formatter import Formatter, OneLinerFormatter
-from StringIO import StringIO
@@ -71 +72 @@
-LISTRULE = re.compile(r"^(?P<indent>[ \t\v]+)\* +(?P<rest>.*)$", re.M)
-ALLOWED_MACROS = ["image"]
-
+class PseudoRequest(object):
+    args = None
+    hdf = None
+    authname = None
+    perm = None
+    session = None
+    path_info = None
+    
+    def __init__(self, path):
+        self.path_info = path
-
-class TocFormatter(OneLinerFormatter):
-    """
@@ -115 +126 @@
-
-        #save for later use
-        self.env = env
+        self.config = env.config
-        self.req = req
-        self.out = out
-        self.col = 0
@@ -223 +235 @@
-        self.write('<h2><a href="%s">TracNav</a> menu</h2>\n' % TRACNAVHOME)
-
-        # add TOCs
-
+
+
-            toc = self.parse_toc(self.get_toc(name))
-            if not toc:
-                toc = self.parse_toc(' * TOC "%s" is empty!' % name)
@@ -292 +305 @@
-
-class TracNav(Component):
-
-
+, IRequestFilter
-
+    # IWikiMacroProvider methods
-    def get_macros(self):
-        yield 'TracNav'
-        yield 'JPNav' # legacy
@@ -307 +321 @@
-        from inspect import getdoc, getmodule
-        return getdoc(getmodule(self))
-
+
+    # ITemplateProvider methods
-    def get_htdocs_dirs(self):
-        from pkg_resources import resource_filename
-        return [('tracnav', resource_filename(__name__, 'htdocs'))]
@@ -314 +330 @@
-    def get_templates_dirs(self):
-        # we don't provide templates
-        return []
+
+
+    # IRequestFilter methods
+    def pre_process_request(self, req, handler):
+        return handler
+
+    def post_process_request(self, req, template, content_type):
+        status = self.config.get('tracnav', 'request_filter', 'disabled')
+        if status == 'enabled':
+            paths = self.config.get('tracnav', 'match', '').split(',')
+
+            if req.path_info == '/':
+                dispatcher = RequestDispatcher(self.env)
+                default_handler = dispatcher.default_handler
+
+                for path in paths:
+                    pseudo_req = PseudoRequest(path)
+                    if default_handler.match_request(pseudo_req):
+                        self._do_post_process(req)
+                        break
+            else:
+                for path in paths:
+                    if req.path_info.startswith(path):
+                        self._do_post_process(req)
+                        break
+
+        return template, content_type
+
+    def _do_post_process(self, req):
+        out = StringIO()
+        Invocation(self.env, req, '', out).run()
+        req.hdf.setValue('tracnavplugin.toc', out.getvalue())