mirror of
https://git.wownero.com/wowlet/wowlet-backend.git
synced 2024-08-15 01:03:13 +00:00
78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
# SPDX-License-Identifier: BSD-3-Clause
|
|
# Copyright (c) 2020, The Monero Project.
|
|
# Copyright (c) 2020, dsc@xmr.pm
|
|
|
|
import os
|
|
import asyncio
|
|
import json
|
|
|
|
from quart import websocket, jsonify, send_from_directory
|
|
|
|
import settings
|
|
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.route("/suchwow/<path:name>")
|
|
async def suchwow(name: str):
|
|
"""Download a SuchWow.xyz image"""
|
|
base = os.path.join(settings.cwd, "data", "suchwow")
|
|
return await send_from_directory(base, name)
|
|
|
|
|
|
@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
|