Issue 2169 - Unify development and production implementation of multiplexer

sitescripts/web.py

 # coding: utf-8
 
 # This file is part of the Adblock Plus web scripts,
 # Copyright (C) 2006-2015 Eyeo GmbH
 #
 # Adblock Plus is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
 # published by the Free Software Foundation.
 #
 # Adblock Plus is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 # GNU General Public License for more details.
 #
 # You should have received a copy of the GNU General Public License
 # along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
 
 import base64
 import imp
 import importlib
+import re
 from sitescripts.utils import get_config
 
 handlers = {}
 authenticated_users = {}
 
 def url_handler(url):
   def decorator(func):
     registerUrlHandler(url, func)
     return func
   return decorator
@@ skipping 20 matching lines @@
         expected_username = config.get(config_section, "basic_auth_username")
         expected_password = config.get(config_section, "basic_auth_password")
         if username == expected_username and password == expected_password:
           return f(environ, start_response)
 
   realm = get_config().get("DEFAULT", "basic_auth_realm")
   start_response("401 UNAUTHORIZED",
                  [("WWW-Authenticate", 'Basic realm="%s"' % realm)])
   return ""
 
+def multiplex(environ, start_response):
+  try:
+    path = environ["PATH_INFO"]
+    try:
+      handler = handlers[path]
+    except KeyError:
+      handler = handlers[re.sub(r"[^/]+$", "", path)]
+  except KeyError:
+    start_response("404 Not Found", [("Content-Type", "text/plain")])
+    return ["Not Found"]
+
+  return handler(environ, start_response)
+
 for module in set(get_config().options("multiplexer")) - set(get_config().defaults()):
   module_path = get_config().get("multiplexer", module)
   if module_path:
     imp.load_source(module, module_path)
   else:
     importlib.import_module(module)