diff --git a/listener.go b/listener.go index 88d158a..c8e6c72 100644 --- a/listener.go +++ b/listener.go @@ -140,7 +140,7 @@ func (rl *Relay) notifyListeners(event *nostr.Event) { return } } - listener.ws.WriteJSON(nostr.EventEnvelope{SubscriptionID: &listener.id, Event: *event}) + listener.ws.SendEvent(listener.id, *event) } } } diff --git a/websocket.go b/websocket.go index f00a636..21f0922 100644 --- a/websocket.go +++ b/websocket.go @@ -5,6 +5,7 @@ import ( "sync" "github.com/fasthttp/websocket" + "github.com/nbd-wtf/go-nostr" ) type WebSocket struct { @@ -33,3 +34,10 @@ func (ws *WebSocket) WriteMessage(t int, b []byte) error { defer ws.mutex.Unlock() return ws.conn.WriteMessage(t, b) } + +func (ws *WebSocket) SendEvent(subscriptionId string, event nostr.Event) error { + return ws.WriteJSON(nostr.EventEnvelope{ + SubscriptionID: &subscriptionId, + Event: event, + }) +}