feat: graft/prune events and mesh peer tagging

src/index.ts

@@ -1471,7 +1471,7 @@
     const now = Date.now()
     let doPX = this.opts.doPX
 
-    graft.forEach(({ topicID }) => {
+    graft.forEach(async ({ topicID }) => {
       if (!topicID) {
         return
       }
@@ -1545,22 +1545,22 @@
       peersInMesh.add(id)
 
       this.metrics?.onAddToMesh(topicID, InclusionReason.Subscribed, 1)
+
+      if (this.opts?.taggingEnabled ?? false) {
+        await this.components.peerStore.merge(peerIdFromString(id), {
+          tags: {
+            [topicID]: {
+              value: 100 // value should be 0-100
+            }
+          }
+        })
+      }
     })
 
     if (!prune.length) {
       return []
     }
 
-    if (this.opts?.taggingEnabled ?? false) {
-      await this.components.peerStore.merge(peerIdFromString(id), {
-        tags: {
-          'gossipsub-mesh-peer': {
-            value: 100 // value should be 0-100
-          }
-        }
-      })
-    }
-
     const onUnsubscribe = false
     return await Promise.all(prune.map((topic) => this.makePrune(id, topic, doPX, onUnsubscribe)))
   }
@@ -1609,14 +1609,14 @@
         }
         await this.pxConnect(peers)
       }
-    }
 
-    if (this.opts?.taggingEnabled ?? false) {
-      await this.components.peerStore.save(peerIdFromString(id), {
-        tags: {
-          'gossipsub-mesh-peer': {}
-        }
-      })
+      if (this.opts?.taggingEnabled ?? false) {
+        await this.components.peerStore.save(peerIdFromString(id), {
+          tags: {
+            [topicID]: {}
+          }
+        })
+      }
     }
   }
 
@@ -2225,7 +2225,7 @@
     if (this.opts?.taggingEnabled ?? false) {
       await this.components.peerStore.merge(peerIdFromString(id), {
         tags: {
-          'gossipsub-mesh-peer': {
+          [topic]: {
             value: 100 // value should be 0-100
           }
         }
@@ -2246,7 +2246,7 @@
     if (this.opts.taggingEnabled ?? false) {
       await this.components.peerStore.save(peerIdFromString(id), {
         tags: {
-          'gossipsub-mesh-peer': {}
+          [topic]: {}
         }
       })
     }
@@ -2500,7 +2500,7 @@
 
         try {
           peerInfo = await this.components.peerStore.get(id)
         } catch (err: any) {
           if (err.code !== 'ERR_NOT_FOUND') {
             throw err
           }