mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-25 23:21:21 +01:00
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
from fastapi import (
|
|
APIRouter,
|
|
WebSocket,
|
|
WebSocketDisconnect,
|
|
)
|
|
|
|
from lnbits.settings import settings
|
|
|
|
from ..services import (
|
|
websocket_manager,
|
|
websocket_updater,
|
|
)
|
|
|
|
websocket_router = APIRouter(prefix="/api/v1/ws", tags=["Websocket"])
|
|
|
|
|
|
@websocket_router.websocket("/{item_id}")
|
|
async def websocket_connect(websocket: WebSocket, item_id: str):
|
|
await websocket_manager.connect(websocket, item_id)
|
|
try:
|
|
while settings.lnbits_running:
|
|
await websocket.receive_text()
|
|
except WebSocketDisconnect:
|
|
websocket_manager.disconnect(websocket)
|
|
|
|
|
|
@websocket_router.post("/{item_id}")
|
|
async def websocket_update_post(item_id: str, data: str):
|
|
try:
|
|
await websocket_updater(item_id, data)
|
|
return {"sent": True, "data": data}
|
|
except Exception:
|
|
return {"sent": False, "data": data}
|
|
|
|
|
|
@websocket_router.get("/{item_id}/{data}")
|
|
async def websocket_update_get(item_id: str, data: str):
|
|
try:
|
|
await websocket_updater(item_id, data)
|
|
return {"sent": True, "data": data}
|
|
except Exception:
|
|
return {"sent": False, "data": data}
|