from threading import Thread
from typing import Callable

from loguru import logger
from websocket import WebSocketApp

from lnbits.app import settings

from .. import send_req_queue
from .event import NostrEvent


async def publish_nostr_event(e: NostrEvent):
    print('### publish_nostr_event', e.dict())
    await send_req_queue.put(["EVENT", e.dict()])


async def connect_to_nostrclient_ws(
    on_open: Callable, on_message: Callable
) -> WebSocketApp:
    def on_error(_, error):
        logger.warning(error)

    logger.debug(f"Subscribing to websockets for nostrclient extension")
    ws = WebSocketApp(
        f"ws://localhost:{settings.port}/nostrclient/api/v1/relay",
        on_message=on_message,
        on_open=on_open,
        on_error=on_error,
    )

    wst = Thread(target=ws.run_forever)
    wst.daemon = True
    wst.start()

    return ws