mirror of
https://github.com/lnbits/lnbits.git
synced 2025-10-09 20:12:34 +02:00
111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
from quart import g, abort, render_template, jsonify, websocket
|
|
from http import HTTPStatus
|
|
import httpx
|
|
|
|
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
|
|
|
|
while True:
|
|
|
|
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>/<comment>", methods=["GET"]
|
|
)
|
|
async def api_copilot_hooker(copilot_id, amount, comment):
|
|
|
|
data = ""
|
|
webhook = ""
|
|
copilot = await get_copilot(copilot_id)
|
|
|
|
if not copilot:
|
|
return (
|
|
jsonify({"message": "Copilot link link does not exist."}),
|
|
HTTPStatus.NOT_FOUND,
|
|
)
|
|
if int(copilot.animation1threshold) and int(amount) > copilot.animation1threshold:
|
|
data = copilot.animation1
|
|
webhook = copilot.animation1webhook
|
|
if (
|
|
int(copilot.animation2threshold)
|
|
and int(amount) > copilot.animation2threshold
|
|
):
|
|
data = copilot.animation2
|
|
webhook = copilot.animation1webhook
|
|
if (
|
|
int(copilot.animation3threshold)
|
|
and int(amount) > copilot.animation3threshold
|
|
):
|
|
data = copilot.animation3
|
|
webhook = copilot.animation1webhook
|
|
if webhook:
|
|
async with httpx.AsyncClient() as client:
|
|
await client.post(
|
|
webhook,
|
|
json={
|
|
copilot,
|
|
},
|
|
timeout=40,
|
|
)
|
|
global connected_websockets
|
|
connected_websockets[copilot_id] = shortuuid.uuid() + "-" + data + "-" + comment
|
|
print(connected_websockets)
|
|
return "", HTTPStatus.OK
|