add support for cookies in the websocket

flask_sockets.py

@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
 from werkzeug.routing import Map, Rule
 from werkzeug.exceptions import NotFound
+from werkzeug.http import parse_cookie
+from flask import request
 
 def log_request(self):
     log = self.server.log
@@ -37,9 +39,15 @@ def __call__(self, environ, start_response):
         try:
             handler, values = adapter.match()
             environment = environ['wsgi.websocket']
+            cookie = None
+            if 'HTTP_COOKIE' in environ:
+                cookie = parse_cookie(environ['HTTP_COOKIE'])
 
             with self.app.app_context():
                 with self.app.request_context(environ):
+                    # add cookie to the request to have correct session handling
+                    request.cookie = cookie
+
                     handler(environment, **values)
                     return []
         except (NotFound, KeyError):