From c714a2641de21adfe18ef7c5f0167f2b03169715 Mon Sep 17 00:00:00 2001
From: ben <ben@arc.wales>
Date: Tue, 2 May 2023 12:53:36 +0100
Subject: [PATCH] Adds websocket for payments

---
 lnbits/core/tasks.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lnbits/core/tasks.py b/lnbits/core/tasks.py
index e11f764b9..72b01f41e 100644
--- a/lnbits/core/tasks.py
+++ b/lnbits/core/tasks.py
@@ -9,6 +9,7 @@ from lnbits.tasks import SseListenersDict, register_invoice_listener
 from . import db
 from .crud import get_balance_notify
 from .models import Payment
+from .services import websocketUpdater
 
 api_invoice_listeners: Dict[str, asyncio.Queue] = SseListenersDict(
     "api_invoice_listeners"
@@ -37,6 +38,7 @@ async def wait_for_paid_invoices(invoice_paid_queue: asyncio.Queue):
         logger.trace("received invoice paid event")
         # send information to sse channel
         await dispatch_api_invoice_listeners(payment)
+        await websocketUpdater(payment.wallet_id, payment)
 
         # dispatch webhook
         if payment.webhook and not payment.webhook_status: