nostrmarket/nostr/nostr_client.py
2023-03-15 23:08:37 +02:00

58 lines
1.7 KiB
Python

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
async def subscribe_to_direct_messages(public_key: str, since: int):
in_messages_filter = {"kind": 4, "#p": [public_key]}
out_messages_filter = {"kind": 4, "authors": [public_key]}
if since != 0:
in_messages_filter["since"] = since
out_messages_filter["since"] = since
print("### in_messages_filter", in_messages_filter)
print("### out_messages_filter", out_messages_filter)
await send_req_queue.put(
["REQ", f"direct-messages-in:{public_key}", in_messages_filter]
)
await send_req_queue.put(
["REQ", f"direct-messages-out:{public_key}", out_messages_filter]
)
async def unsubscribe_from_direct_messages(public_key: str):
await send_req_queue.put(["CLOSE", f"direct-messages-in:{public_key}"])
await send_req_queue.put(["CLOSE", f"direct-messages-out:{public_key}"])