mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-25 07:07:48 +01:00
98 lines
No EOL
2.8 KiB
Python
98 lines
No EOL
2.8 KiB
Python
from quart import g, abort, render_template, jsonify, websocket
|
|
from http import HTTPStatus
|
|
|
|
from lnbits.decorators import check_user_exists, validate_uuids
|
|
|
|
from . import copilot_ext
|
|
from .crud import get_copilot
|
|
|
|
from quart import g, abort, render_template, jsonify, websocket
|
|
from functools import wraps
|
|
import trio
|
|
import shortuuid
|
|
from . import copilot_ext
|
|
|
|
connected_websockets = {}
|
|
|
|
|
|
@copilot_ext.websocket("/ws/panel/<copilot_id>")
|
|
async def ws_panel(copilot_id):
|
|
global connected_websockets
|
|
while True:
|
|
data = await websocket.receive()
|
|
connected_websockets[copilot_id] = shortuuid.uuid() + "-" + data
|
|
|
|
@copilot_ext.websocket("/ws/compose/<copilot_id>")
|
|
async def ws_compose(copilot_id):
|
|
global connected_websockets
|
|
print("poo")
|
|
while True:
|
|
print("poo")
|
|
data = await websocket.receive()
|
|
await websocket.send(connected_websockets[copilot_id])
|
|
|
|
|
|
@copilot_ext.route("/")
|
|
@validate_uuids(["usr"], required=True)
|
|
@check_user_exists()
|
|
async def index():
|
|
return await render_template("copilot/index.html", user=g.user)
|
|
|
|
|
|
@copilot_ext.route("/cp/<copilot_id>")
|
|
async def compose(copilot_id):
|
|
copilot = await get_copilot(copilot_id) or abort(
|
|
HTTPStatus.NOT_FOUND, "Copilot link does not exist."
|
|
)
|
|
if copilot.lnurl_toggle:
|
|
return await render_template(
|
|
"copilot/compose.html",
|
|
copilot=copilot,
|
|
lnurl=copilot.lnurl,
|
|
lnurl_toggle=copilot.lnurl_toggle,
|
|
)
|
|
return await render_template(
|
|
"copilot/compose.html", copilot=copilot, lnurl_toggle=copilot.lnurl_toggle
|
|
)
|
|
|
|
|
|
@copilot_ext.route("/<copilot_id>")
|
|
async def panel(copilot_id):
|
|
copilot = await get_copilot(copilot_id) or abort(
|
|
HTTPStatus.NOT_FOUND, "Copilot link does not exist."
|
|
)
|
|
return await render_template("copilot/panel.html", copilot=copilot)
|
|
|
|
|
|
@copilot_ext.route("/api/v1/copilot/hook/<copilot_id>/<amount>", methods=["GET"])
|
|
async def api_copilot_hooker(copilot_id, amount):
|
|
copilot = await get_copilot(copilot_id)
|
|
print("amount")
|
|
if not copilot:
|
|
return (
|
|
jsonify({"message": "Copilot link link does not exist."}),
|
|
HTTPStatus.NOT_FOUND,
|
|
)
|
|
print(amount)
|
|
if (
|
|
copilot.animation1threshold
|
|
and int(amount) > copilot.animation1threshold
|
|
):
|
|
data = copilot.animation1
|
|
if (
|
|
copilot.animation2threshold
|
|
and int(amount) > copilot.animation2threshold
|
|
):
|
|
data = copilot.animation2
|
|
if (
|
|
copilot.animation3threshold
|
|
and int(amount) > copilot.animation3threshold
|
|
):
|
|
data = copilot.animation3
|
|
print(data)
|
|
async with websocket(
|
|
"/ws/compose/" + copilot_id
|
|
) as the_websocket:
|
|
await the_websocket.send(data)
|
|
print(data)
|
|
return "", HTTPStatus.OK |