haven/import.go

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")
		}
	}
}