go-nostr/sdk/system.go

218 lines
6.2 KiB
Go
Raw Permalink Normal View History

package sdk
import (
"context"
"math/rand/v2"
"github.com/fiatjaf/eventstore"
"github.com/fiatjaf/eventstore/nullstore"
"github.com/graph-gophers/dataloader/v7"
"github.com/nbd-wtf/go-nostr"
"github.com/nbd-wtf/go-nostr/sdk/cache"
cache_memory "github.com/nbd-wtf/go-nostr/sdk/cache/memory"
"github.com/nbd-wtf/go-nostr/sdk/hints"
"github.com/nbd-wtf/go-nostr/sdk/hints/memoryh"
2025-01-14 23:11:37 -03:00
"github.com/nbd-wtf/go-nostr/sdk/kvstore"
kvstore_memory "github.com/nbd-wtf/go-nostr/sdk/kvstore/memory"
)
type System struct {
2025-01-14 23:11:37 -03:00
KVStore kvstore.KVStore
2025-01-01 18:16:36 -03:00
MetadataCache cache.Cache32[ProfileMetadata]
RelayListCache cache.Cache32[GenericList[Relay]]
FollowListCache cache.Cache32[GenericList[ProfileRef]]
MuteListCache cache.Cache32[GenericList[ProfileRef]]
BookmarkListCache cache.Cache32[GenericList[EventRef]]
PinListCache cache.Cache32[GenericList[EventRef]]
BlockedRelayListCache cache.Cache32[GenericList[RelayURL]]
SearchRelayListCache cache.Cache32[GenericList[RelayURL]]
2025-01-02 12:12:49 -03:00
TopicListCache cache.Cache32[GenericList[Topic]]
RelaySetsCache cache.Cache32[GenericSets[RelayURL]]
FollowSetsCache cache.Cache32[GenericSets[ProfileRef]]
TopicSetsCache cache.Cache32[GenericSets[Topic]]
2025-01-01 18:16:36 -03:00
Hints hints.HintsDB
Pool *nostr.SimplePool
RelayListRelays *RelayStream
FollowListRelays *RelayStream
MetadataRelays *RelayStream
FallbackRelays *RelayStream
JustIDRelays *RelayStream
UserSearchRelays *RelayStream
NoteSearchRelays *RelayStream
Store eventstore.Store
StoreRelay nostr.RelayStore
replaceableLoaders []*dataloader.Loader[string, *nostr.Event]
addressableLoaders []*dataloader.Loader[string, []*nostr.Event]
}
type SystemModifier func(sys *System)
2024-10-06 15:53:33 -03:00
type RelayStream struct {
URLs []string
serial int
}
func NewRelayStream(urls ...string) *RelayStream {
return &RelayStream{URLs: urls, serial: rand.Int()}
2024-10-06 15:53:33 -03:00
}
func (rs *RelayStream) Next() string {
rs.serial++
return rs.URLs[rs.serial%len(rs.URLs)]
}
func NewSystem(mods ...SystemModifier) *System {
sys := &System{
2025-01-14 23:11:37 -03:00
KVStore: kvstore_memory.NewStore(),
2025-01-01 18:16:36 -03:00
MetadataCache: cache_memory.New32[ProfileMetadata](8000),
RelayListCache: cache_memory.New32[GenericList[Relay]](8000),
FollowListCache: cache_memory.New32[GenericList[ProfileRef]](1000),
MuteListCache: cache_memory.New32[GenericList[ProfileRef]](1000),
BookmarkListCache: cache_memory.New32[GenericList[EventRef]](1000),
PinListCache: cache_memory.New32[GenericList[EventRef]](1000),
BlockedRelayListCache: cache_memory.New32[GenericList[RelayURL]](1000),
SearchRelayListCache: cache_memory.New32[GenericList[RelayURL]](1000),
2025-01-02 12:12:49 -03:00
TopicListCache: cache_memory.New32[GenericList[Topic]](1000),
RelaySetsCache: cache_memory.New32[GenericSets[RelayURL]](1000),
FollowSetsCache: cache_memory.New32[GenericSets[ProfileRef]](1000),
TopicSetsCache: cache_memory.New32[GenericSets[Topic]](1000),
2025-01-01 18:16:36 -03:00
RelayListRelays: NewRelayStream("wss://purplepag.es", "wss://user.kindpag.es", "wss://relay.nos.social"),
FollowListRelays: NewRelayStream("wss://purplepag.es", "wss://user.kindpag.es", "wss://relay.nos.social"),
MetadataRelays: NewRelayStream("wss://purplepag.es", "wss://user.kindpag.es", "wss://relay.nos.social"),
2024-10-06 15:53:33 -03:00
FallbackRelays: NewRelayStream(
"wss://relay.damus.io",
"wss://nostr.mom",
"wss://nos.lol",
"wss://mostr.pub",
"wss://relay.nostr.band",
2024-10-06 15:53:33 -03:00
),
JustIDRelays: NewRelayStream(
"wss://cache2.primal.net/v1",
"wss://relay.noswhere.com",
"wss://relay.nostr.band",
),
UserSearchRelays: NewRelayStream(
"wss://search.nos.today",
"wss://nostr.wine",
"wss://relay.nostr.band",
2024-10-06 15:53:33 -03:00
),
NoteSearchRelays: NewRelayStream(
"wss://nostr.wine",
"wss://relay.nostr.band",
2024-10-06 15:53:33 -03:00
"wss://search.nos.today",
),
Hints: memoryh.NewHintDB(),
}
sys.Pool = nostr.NewSimplePool(context.Background(),
nostr.WithAuthorKindQueryMiddleware(sys.TrackQueryAttempts),
nostr.WithEventMiddleware(sys.TrackEventHintsAndRelays),
2025-01-15 00:30:24 -03:00
nostr.WithDuplicateMiddleware(sys.TrackEventRelaysD),
nostr.WithPenaltyBox(),
)
for _, mod := range mods {
mod(sys)
}
if sys.Store == nil {
sys.Store = &nullstore.NullStore{}
sys.Store.Init()
}
sys.StoreRelay = eventstore.RelayWrapper{Store: sys.Store}
2025-01-02 12:12:49 -03:00
sys.initializeReplaceableDataloaders()
sys.initializeAddressableDataloaders()
return sys
}
2025-01-14 23:11:37 -03:00
func (sys *System) Close() {
if sys.KVStore != nil {
sys.KVStore.Close()
}
}
func WithHintsDB(hdb hints.HintsDB) SystemModifier {
return func(sys *System) {
sys.Hints = hdb
}
}
func WithRelayListRelays(list []string) SystemModifier {
return func(sys *System) {
2024-10-06 15:53:33 -03:00
sys.RelayListRelays.URLs = list
}
}
func WithMetadataRelays(list []string) SystemModifier {
return func(sys *System) {
2024-10-06 15:53:33 -03:00
sys.MetadataRelays.URLs = list
}
}
func WithFollowListRelays(list []string) SystemModifier {
return func(sys *System) {
2024-10-06 15:53:33 -03:00
sys.FollowListRelays.URLs = list
}
}
func WithFallbackRelays(list []string) SystemModifier {
return func(sys *System) {
2024-10-06 15:53:33 -03:00
sys.FallbackRelays.URLs = list
}
}
func WithJustIDRelays(list []string) SystemModifier {
return func(sys *System) {
sys.JustIDRelays.URLs = list
}
}
func WithUserSearchRelays(list []string) SystemModifier {
return func(sys *System) {
2024-10-06 15:53:33 -03:00
sys.UserSearchRelays.URLs = list
}
}
func WithNoteSearchRelays(list []string) SystemModifier {
return func(sys *System) {
2024-10-06 15:53:33 -03:00
sys.NoteSearchRelays.URLs = list
}
}
func WithStore(store eventstore.Store) SystemModifier {
return func(sys *System) {
sys.Store = store
}
}
2025-01-01 18:16:36 -03:00
func WithRelayListCache(cache cache.Cache32[GenericList[Relay]]) SystemModifier {
return func(sys *System) {
sys.RelayListCache = cache
}
}
2025-01-01 18:16:36 -03:00
func WithFollowListCache(cache cache.Cache32[GenericList[ProfileRef]]) SystemModifier {
return func(sys *System) {
sys.FollowListCache = cache
}
}
func WithMetadataCache(cache cache.Cache32[ProfileMetadata]) SystemModifier {
return func(sys *System) {
sys.MetadataCache = cache
}
}
2025-01-14 23:11:37 -03:00
func WithKVStore(store kvstore.KVStore) SystemModifier {
return func(sys *System) {
if sys.KVStore != nil {
sys.KVStore.Close()
}
sys.KVStore = store
}
}