🚀 Initializing HAVEN

2025-12-18 14:24:18 +01:00 · 2024-09-22 23:28:25 -04:00
commit 5097bb716c
12 changed files with 1218 additions and 0 deletions
--- a/import.go
+++ b/import.go
@@ -0,0 +1,137 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/fiatjaf/eventstore"
+	"github.com/nbd-wtf/go-nostr"
+)
+
+const layout = "2006-01-02"
+
+var ownerImportedNotes = 0
+var taggedImportedNotes = 0
+
+func importOwnerNotes() {
+	ctx := context.Background()
+	pool = nostr.NewSimplePool(ctx)
+	wdb := eventstore.RelayWrapper{Store: &outboxDB}
+
+	startTime, err := time.Parse(layout, config.ImportStartDate)
+	if err != nil {
+		fmt.Println("Error parsing start date:", err)
+		return
+	}
+	endTime := startTime.Add(240 * time.Hour)
+
+	for {
+		ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
+		defer cancel()
+
+		startTimestamp := nostr.Timestamp(startTime.Unix())
+		endTimestamp := nostr.Timestamp(endTime.Unix())
+
+		filters := []nostr.Filter{{
+			Authors: []string{nPubToPubkey(config.OwnerNpub)},
+			Since:   &startTimestamp,
+			Until:   &endTimestamp,
+		}}
+
+		for ev := range pool.SubManyEose(ctx, config.ImportSeedRelays, filters) {
+			wdb.Publish(ctx, *ev.Event)
+			ownerImportedNotes++
+		}
+		log.Println("📦 imported", ownerImportedNotes, "owner notes")
+		time.Sleep(5 * time.Second)
+
+		startTime = startTime.Add(240 * time.Hour)
+		endTime = endTime.Add(240 * time.Hour)
+
+		if startTime.After(time.Now()) {
+			log.Println("✅ owner note import complete! ")
+			break
+		}
+	}
+}
+
+func importTaggedNotes() {
+	ctx := context.Background()
+	pool = nostr.NewSimplePool(ctx)
+	wdb := eventstore.RelayWrapper{Store: &inboxDB}
+
+	startTime, err := time.Parse(layout, config.ImportStartDate)
+	if err != nil {
+		fmt.Println("Error parsing start date:", err)
+		return
+	}
+	endTime := startTime.Add(240 * time.Hour)
+
+	for {
+		ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
+		defer cancel()
+
+		startTimestamp := nostr.Timestamp(startTime.Unix())
+		endTimestamp := nostr.Timestamp(endTime.Unix())
+
+		filters := []nostr.Filter{{
+			Tags: nostr.TagMap{
+				"p": {nPubToPubkey(config.OwnerNpub)},
+			},
+			Since: &startTimestamp,
+			Until: &endTimestamp,
+		}}
+
+		for ev := range pool.SubManyEose(ctx, config.ImportSeedRelays, filters) {
+			wdb.Publish(ctx, *ev.Event)
+			taggedImportedNotes++
+		}
+		log.Println("📦 imported", taggedImportedNotes, "tagged notes")
+		time.Sleep(5 * time.Second)
+
+		startTime = startTime.Add(240 * time.Hour)
+		endTime = endTime.Add(240 * time.Hour)
+
+		if startTime.After(time.Now()) {
+			log.Println("✅ tagged import complete. please restart the relay")
+			break
+		}
+	}
+}
+
+func subscribeInbox() {
+	ctx := context.Background()
+	wdb := eventstore.RelayWrapper{Store: &inboxDB}
+	filters := []nostr.Filter{{
+		Tags: nostr.TagMap{
+			"p": {nPubToPubkey(config.OwnerNpub)},
+		},
+	}}
+
+	log.Println("📢 subscribing to inbox")
+	for ev := range pool.SubMany(ctx, config.ImportSeedRelays, filters) {
+		if !wotMap[ev.Event.PubKey] {
+			continue
+		}
+
+		wdb.Publish(ctx, *ev.Event)
+		switch ev.Event.Kind {
+		case nostr.KindTextNote:
+			log.Println("📰 new note in your inbox")
+		case nostr.KindReaction:
+			log.Println(ev.Event.Content, "new reaction in your inbox")
+		case nostr.KindZap:
+			log.Println("⚡️ new zap in your inbox")
+		case nostr.KindEncryptedDirectMessage:
+			log.Println("🔒 new encrypted message in your inbox")
+		case nostr.KindRepost:
+			log.Println("🔁 new repost in your inbox")
+		case nostr.KindFollowList:
+			// do nothing
+		default:
+			log.Println("📦 new event in your inbox")
+		}
+	}
+}