feather-ws -> wowlet-backend

wowlet_backend/routes.py

@@ -0,0 +1,69 @@
+# SPDX-License-Identifier: BSD-3-Clause
+# Copyright (c) 2020, The Monero Project.
+# Copyright (c) 2020, dsc@xmr.pm
+
+import asyncio
+import json
+
+from quart import websocket, jsonify
+
+from wowlet_backend.factory import app
+from wowlet_backend.wsparse import WebsocketParse
+from wowlet_backend.utils import collect_websocket, feather_data
+
+
+@app.route("/")
+async def root():
+    data = await feather_data()
+    return jsonify(data)
+
+
+@app.websocket('/ws')
+@collect_websocket
+async def ws(queue):
+    data = await feather_data()
+
+    # blast available data on connect
+    for task_key, task_value in data.items():
+        if not task_value:
+            continue
+        await websocket.send(json.dumps({"cmd": task_key, "data": task_value}).encode())
+
+    async def rx():
+        while True:
+            buffer = await websocket.receive()
+            try:
+                blob = json.loads(buffer)
+                if "cmd" not in blob:
+                    continue
+                cmd = blob.get('cmd')
+                _data = blob.get('data')
+                result = await WebsocketParse.parser(cmd, _data)
+                if result:
+                    rtn = json.dumps({"cmd": cmd, "data": result}).encode()
+                    await websocket.send(rtn)
+            except Exception as ex:
+                continue
+
+    async def tx():
+        while True:
+            data = await queue.get()
+            payload = json.dumps(data).encode()
+            await websocket.send(payload)
+
+    # bidirectional async rx and tx loops
+    consumer_task = asyncio.ensure_future(rx())
+    producer_task = asyncio.ensure_future(tx())
+    try:
+        await asyncio.gather(consumer_task, producer_task)
+    finally:
+        consumer_task.cancel()
+        producer_task.cancel()
+
+
+@app.errorhandler(403)
+@app.errorhandler(404)
+@app.errorhandler(405)
+@app.errorhandler(500)
+def page_not_found(e):
+    return ":)", 500