fix: properly handle task start/stopping (#17)

lnbits/lnbits#2411

Author: dni

__init__.py

@@ -1,10 +1,11 @@
 import asyncio
+from loguru import logger
 
 from fastapi import APIRouter
 
 from lnbits.db import Database
 from lnbits.helpers import template_renderer
-from lnbits.tasks import catch_everything_and_restart
+from lnbits.tasks import create_permanent_unique_task
 
 db = Database("ext_copilot")
 
@@ -26,7 +27,17 @@ def copilot_renderer():
 from .views import *  # noqa
 from .views_api import *  # noqa
 
+scheduled_tasks: list[asyncio.Task] = []
+
+
+def copilot_stop():
+    for task in scheduled_tasks:
+        try:
+            task.cancel()
+        except Exception as ex:
+            logger.warning(ex)
+
 
 def copilot_start():
-    loop = asyncio.get_event_loop()
-    loop.create_task(catch_everything_and_restart(wait_for_paid_invoices))
+    task = create_permanent_unique_task("ext_copilot", wait_for_paid_invoices)
+    scheduled_tasks.append(task)