resubscribe when a new relay is added

This commit is contained in:
callebtc 2023-04-11 15:47:21 +02:00
parent 0ffb158769
commit e0938cb760
3 changed files with 20 additions and 4 deletions

View file

@ -1,5 +1,6 @@
import asyncio
import ssl
import json
import threading
from .nostr.event import Event
@ -17,6 +18,13 @@ from .crud import get_relays
async def init_relays():
# we save any subscriptions teporarily to re-add them after reinitializing the client
subscriptions = {}
for relay in nostr.client.relay_manager.relays.values():
# relay.add_subscription(id, filters)
for subscription_id, filters in relay.subscriptions.items():
subscriptions[subscription_id] = filters
# reinitialize the entire client
nostr.__init__()
# get relays from db
@ -24,6 +32,16 @@ async def init_relays():
# set relays and connect to them
nostr.client.relays = list(set([r.url for r in relays.__root__ if r.url]))
nostr.client.connect()
await asyncio.sleep(2)
# re-add subscriptions
for subscription_id, subscription in subscriptions.items():
nostr.client.relay_manager.add_subscription(
subscription_id, subscription.filters
)
s = subscription.to_json_object()
json_str = json.dumps(["REQ", s["id"], s["filters"][0]])
nostr.client.relay_manager.publish_message(json_str)
return