Merge pull request #32 from bitvora/dev-profileRefresh

customize the trust network by follower count
This commit is contained in:
Barry Deen 2024-09-12 15:24:57 -04:00 committed by GitHub
commit 37a9e1c0cc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 27 additions and 14 deletions

View File

@ -8,10 +8,11 @@ RELAY_URL="wss://wot.utxo.one"
DB_PATH="db"
# where we should store the index.html and static files
INDEX_PATH="templates/index.html"
STATIC_PATH="templates/static"
INDEX_PATH="/mnt/dev/bitvora/wot-relay/templates/index.html"
STATIC_PATH="/mnt/dev/bitvora/wot-relay/templates/static/"
# relay behavior
# how often to refresh the relay's view of the WoT in HOURS
REFRESH_INTERVAL_HOURS=1
REFRESH_INTERVAL_HOURS=1
MINIMUM_FOLLOWERS=5

View File

@ -50,6 +50,7 @@ DB_PATH="/home/ubuntu/wot-relay/db" # any path you would like the database to be
INDEX_PATH="/home/ubuntu/wot-relay/templates/index.html" # path to the index.html file
STATIC_PATH="/home/ubuntu/wot-relay/templates/static" # path to the static folder
REFRESH_INTERVAL_HOURS=24 # interval in hours to refresh the web of trust
MINIMUM_FOLLOWERS=3 #how many followers before they're allowed in the WoT
```
### 4. Build the project

33
main.go
View File

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