From 02f27f0de351e662f3e51cd7e136226bddc1cfa9 Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.com>
Date: Mon, 30 Sep 2013 20:51:43 -0400
Subject: [PATCH] import flask-websocket

---
 flask_websocket.py | 58 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 flask_websocket.py

diff --git a/flask_websocket.py b/flask_websocket.py
new file mode 100644
index 0000000..adda2a1
--- /dev/null
+++ b/flask_websocket.py
@@ -0,0 +1,58 @@
+import gevent
+from flask import request
+from websocket import handle_websocket
+
+from geventwebsocket.gunicorn.workers import GeventWebSocketWorker as Worker
+
+def log_request(self):
+    log = self.server.log
+    if log:
+        if hasattr(log, "info"):
+            log.info(self.format_request() + '\n')
+        else:
+            log.write(self.format_request() + '\n')
+
+
+class SocketMiddleware(object):
+
+    def __init__(self, wsgi_app, socket):
+        self.ws = socket
+        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]
+            environment = environ["wsgi.websocket"]
+
+            handler(environment)
+        else:
+            return self.app(environ, start_response)
+
+
+class WebSocket(object):
+
+    def __init__(self, app=None, patch=True):
+        self.url_map = {}
+        if app:
+            self.init_app(app, patch=patch)
+
+    def init_app(self, app, patch=True):
+        if patch:
+            # Monkey-patch log_request handler for Gevent/Gunicorn compatability.
+            if hasattr(gevent, 'pywsgi'):
+                gevent.pywsgi.WSGIHandler.log_request = log_request
+
+        app.wsgi_app = SocketMiddleware(app.wsgi_app, self)
+
+    def route(self, rule, **options):
+
+        def decorator(f):
+            endpoint = options.pop('endpoint', None)
+            self.add_url_rule(rule, endpoint, f, **options)
+            return f
+        return decorator
+
+    def add_url_rule(self, rule, _, f, **options):
+        self.url_map[rule] = f
\ No newline at end of file