adapt relaypool to the new nip-01.

relaypool/subscription.go

@@ -0,0 +1,62 @@
+package relaypool
+
+import (
+	"github.com/fiatjaf/go-nostr/event"
+	"github.com/fiatjaf/go-nostr/filter"
+	"github.com/gorilla/websocket"
+)
+
+type Subscription struct {
+	channel string
+	relays  map[string]*websocket.Conn
+
+	filter *filter.EventFilter
+	Events chan EventMessage
+}
+
+func (subscription Subscription) Unsub() {
+	for _, ws := range subscription.relays {
+		ws.WriteJSON([]interface{}{
+			"CLOSE",
+			subscription.channel,
+		})
+	}
+}
+
+func (subscription Subscription) Sub(filter *filter.EventFilter) {
+	if filter != nil {
+		subscription.filter = filter
+	}
+
+	for _, ws := range subscription.relays {
+		ws.WriteJSON([]interface{}{
+			"REQ",
+			subscription.channel,
+			subscription.filter,
+		})
+	}
+}
+
+func (subscription Subscription) removeRelay(relay string) {
+	if ws, ok := subscription.relays[relay]; ok {
+		delete(subscription.relays, relay)
+		ws.WriteJSON([]interface{}{
+			"CLOSE",
+			subscription.channel,
+		})
+	}
+}
+
+func (subscription Subscription) addRelay(relay string, ws *websocket.Conn) {
+	subscription.relays[relay] = ws
+	ws.WriteJSON([]interface{}{
+		"REQ",
+		subscription.channel,
+		subscription.filter,
+	})
+}
+
+type EventMessage struct {
+	Event event.Event
+	Relay string
+}