From 91cee59a1ab45be18d1cbb3de7545055de353d0e Mon Sep 17 00:00:00 2001
From: Jules Olleon <jolleon@gmail.com>
Date: Fri, 18 Apr 2014 20:50:07 -0700
Subject: [PATCH] add werkzeug routing: allows '/endpoint/<some_id>' style
 routing

---
 flask_sockets.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/flask_sockets.py b/flask_sockets.py
index 0b74dc0..2670091 100644
--- a/flask_sockets.py
+++ b/flask_sockets.py
@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from werkzeug.routing import Map, Rule
+from werkzeug.exceptions import NotFound
 
 def log_request(self):
     log = self.server.log
@@ -30,22 +32,20 @@ class SocketMiddleware(object):
         self.app = wsgi_app
 
     def __call__(self, environ, start_response):
-        path = environ['PATH_INFO']
-
-        if path in self.ws.url_map:
-            handler = self.ws.url_map[path]
+        adapter = self.ws.url_map.bind_to_environ(environ)
+        try:
+            handler, values = adapter.match()
             environment = environ['wsgi.websocket']
-
-            handler(environment)
+            handler(environment, **values)
             return []
-        else:
+        except NotFound:
             return self.app(environ, start_response)
 
 
 class Sockets(object):
 
     def __init__(self, app=None):
-        self.url_map = {}
+        self.url_map = Map()
         if app:
             self.init_app(app)
 
@@ -61,7 +61,7 @@ class Sockets(object):
         return decorator
 
     def add_url_rule(self, rule, _, f, **options):
-        self.url_map[rule] = f
+        self.url_map.add(Rule(rule, endpoint=f))
 
 # CLI sugar.
 if 'Worker' in locals():