goodmorningbitcoin/wot-relay
1
0
Fork 0
mirror of https://github.com/bitvora/wot-relay.git synced 2025-06-23 16:05:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

customize the trust network by follower count

Browse Source
This commit is contained in:
Barry Deen 2024-09-12 15:23:32 -04:00
parent 84ec835dcb
commit b862a0c21c
1 changed files with 22 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
main.go
View File

@ -26,6 +26,7 @@ type Config struct {
IndexPath string IndexPath string
StaticPath string StaticPath string
RefreshInterval int RefreshInterval int
MinimumFollowers int
} }
var pool *nostr.SimplePool var pool *nostr.SimplePool
@ -36,6 +37,7 @@ var seedRelays []string
var booted bool var booted bool
var oneHopNetwork []string var oneHopNetwork []string
var trustNetworkMap map[string]bool var trustNetworkMap map[string]bool
var pubkeyFollowerCount = make(map[string]int)
func main() { func main() {
nostr.InfoLogger = log.New(io.Discard, "", 0) nostr.InfoLogger = log.New(io.Discard, "", 0)
@ -162,6 +164,12 @@ func LoadConfig() Config {
refreshInterval, _ := strconv.Atoi(os.Getenv("REFRESH_INTERVAL_HOURS")) refreshInterval, _ := strconv.Atoi(os.Getenv("REFRESH_INTERVAL_HOURS"))
log.Println("🔄 refresh interval set to", refreshInterval, "hours") log.Println("🔄 refresh interval set to", refreshInterval, "hours")
if os.Getenv("MINIMUM_FOLLOWERS") == "" {
os.Setenv("MINIMUM_FOLLOWERS", "1")
}
minimumFollowers, _ := strconv.Atoi(os.Getenv("MINIMUM_FOLLOWERS"))
config := Config{ config := Config{
RelayName: getEnv("RELAY_NAME"), RelayName: getEnv("RELAY_NAME"),
RelayPubkey: getEnv("RELAY_PUBKEY"), RelayPubkey: getEnv("RELAY_PUBKEY"),
@ -171,6 +179,7 @@ func LoadConfig() Config {
IndexPath: getEnv("INDEX_PATH"), IndexPath: getEnv("INDEX_PATH"),
StaticPath: getEnv("STATIC_PATH"), StaticPath: getEnv("STATIC_PATH"),
RefreshInterval: refreshInterval, RefreshInterval: refreshInterval,
MinimumFollowers: minimumFollowers,
} }
return config return config
@ -187,11 +196,15 @@ func getEnv(key string) string {
func updateTrustNetworkFilter() { func updateTrustNetworkFilter() {
trustNetworkMap = make(map[string]bool) trustNetworkMap = make(map[string]bool)
nKeys := uint64(len(trustNetwork)) log.Println("🌐 updating trust network map")
log.Println("🌐 updating trust network map with", nKeys, "keys") for pubkey, count := range pubkeyFollowerCount {
for _, trustedPubkey := range trustNetwork { if count >= config.MinimumFollowers {
trustNetworkMap[trustedPubkey] = true trustNetworkMap[pubkey] = true
appendPubkey(pubkey)
} }
}
log.Println("🌐 trust network map updated with", len(trustNetwork), "keys")
} }
func refreshProfiles(ctx context.Context, relay *khatru.Relay) { func refreshProfiles(ctx context.Context, relay *khatru.Relay) {
@ -230,6 +243,7 @@ func refreshTrustNetwork(relay *khatru.Relay, ctx context.Context) {
log.Println("🔍 fetching owner's follows") log.Println("🔍 fetching owner's follows")
for ev := range pool.SubManyEose(timeoutCtx, seedRelays, filters) { for ev := range pool.SubManyEose(timeoutCtx, seedRelays, filters) {
for _, contact := range ev.Event.Tags.GetAll([]string{"p"}) { for _, contact := range ev.Event.Tags.GetAll([]string{"p"}) {
pubkeyFollowerCount[contact[1]]++ // Increment follower count for the pubkey
appendOneHopNetwork(contact[1]) appendOneHopNetwork(contact[1])
} }
} }
@ -251,9 +265,7 @@ func refreshTrustNetwork(relay *khatru.Relay, ctx context.Context) {
for ev := range pool.SubManyEose(timeout, seedRelays, filters) { for ev := range pool.SubManyEose(timeout, seedRelays, filters) {
for _, contact := range ev.Event.Tags.GetAll([]string{"p"}) { for _, contact := range ev.Event.Tags.GetAll([]string{"p"}) {
if len(contact) > 1 { pubkeyFollowerCount[contact[1]]++ // Increment follower count for the pubkey
appendPubkey(contact[1])
}
} }
for _, relay := range ev.Event.Tags.GetAll([]string{"r"}) { for _, relay := range ev.Event.Tags.GetAll([]string{"r"}) {
@ -264,9 +276,8 @@ func refreshTrustNetwork(relay *khatru.Relay, ctx context.Context) {
relay.AddEvent(ctx, ev.Event) relay.AddEvent(ctx, ev.Event)
} }
} }
} }
log.Println("🫂 network size:", len(trustNetwork)) log.Println("🫂 total network size:", len(pubkeyFollowerCount))
log.Println("🔗 relays discovered:", len(relays)) log.Println("🔗 relays discovered:", len(relays))
} }
@ -355,7 +366,7 @@ func archiveTrustedNotes(relay *khatru.Relay, ctx context.Context) {
case ev, ok := <-eventChan: case ev, ok := <-eventChan:
if !ok { if !ok {
log.Println("📦 SubMany channel closed") log.Println("📦 subscription channel closed")
log.Println("📦 archived", trustedNotes, "trusted notes and discarded", untrustedNotes, "untrusted notes") log.Println("📦 archived", trustedNotes, "trusted notes and discarded", untrustedNotes, "untrusted notes")
return return
} }
@ -367,7 +378,7 @@ func archiveTrustedNotes(relay *khatru.Relay, ctx context.Context) {
relay.AddEvent(ctx, ev.Event) relay.AddEvent(ctx, ev.Event)
trustedNotes++ trustedNotes++
log.Println("📦 archived note: ", ev.Event.ID) //log.Println("📦 archived note: ", ev.Event.ID)
} else { } else {
untrustedNotes++ untrustedNotes++
} }