move everything to the same directory.

subscription.go

@@ -0,0 +1,90 @@
+package nostr
+
+import (
+	"github.com/gorilla/websocket"
+)
+
+type Subscription struct {
+	channel string
+	relays  map[string]*websocket.Conn
+
+	filters EventFilters
+	Events  chan EventMessage
+
+	started      bool
+	UniqueEvents chan Event
+}
+
+type EventMessage struct {
+	Event Event
+	Relay string
+}
+
+func (subscription Subscription) Unsub() {
+	for _, ws := range subscription.relays {
+		ws.WriteJSON([]interface{}{
+			"CLOSE",
+			subscription.channel,
+		})
+	}
+
+	if subscription.Events != nil {
+		close(subscription.Events)
+	}
+	if subscription.UniqueEvents != nil {
+		close(subscription.UniqueEvents)
+	}
+}
+
+func (subscription Subscription) Sub(filters EventFilters) {
+	for _, ws := range subscription.relays {
+		message := []interface{}{
+			"REQ",
+			subscription.channel,
+		}
+		for _, filter := range subscription.filters {
+			message = append(message, filter)
+		}
+
+		ws.WriteJSON(message)
+	}
+
+	if !subscription.started {
+		go subscription.startHandlingUnique()
+	}
+}
+
+func (subscription Subscription) startHandlingUnique() {
+	seen := make(map[string]struct{})
+	for em := range subscription.Events {
+		if _, ok := seen[em.Event.ID]; ok {
+			continue
+		}
+		seen[em.Event.ID] = struct{}{}
+		subscription.UniqueEvents <- em.Event
+	}
+}
+
+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
+
+	message := []interface{}{
+		"REQ",
+		subscription.channel,
+	}
+	for _, filter := range subscription.filters {
+		message = append(message, filter)
+	}
+
+	ws.WriteJSON(message)
+}