@libp2p/devtools-metrics 1.0.1 → 1.1.0

package/src/rpc/rpc.ts

@@ -1,14 +1,14 @@
 import { enable, disable } from '@libp2p/logger'
 import { peerIdFromString } from '@libp2p/peer-id'
 import { multiaddr } from '@multiformats/multiaddr'
+import { gatherCapabilities } from '../utils/gather-capabilities.js'
 import { getPeers } from '../utils/get-peers.js'
+import { getPubSub } from '../utils/get-pubsub.js'
 import { getSelf } from '../utils/get-self.js'
 import type { MetricsRPC } from './index.js'
 import type { DevToolsMetricsComponents } from '../index.js'
 import type { PeerId } from '@libp2p/interface'
-import type { OpenConnectionOptions } from '@libp2p/interface-internal'
 import type { Multiaddr } from '@multiformats/multiaddr'
-import type { AbortOptions } from 'it-pushable'
 
 export function metricsRpc (components: DevToolsMetricsComponents): MetricsRPC {
   const log = components.logger.forComponent('libp2p:devtools-metrics:metrics-rpc')
@@ -18,10 +18,11 @@ export function metricsRpc (components: DevToolsMetricsComponents): MetricsRPC {
       return {
         self: await getSelf(components),
         peers: await getPeers(components, log),
-        debug: localStorage.getItem('debug') ?? ''
+        debug: localStorage.getItem('debug') ?? '',
+        capabilities: gatherCapabilities(components)
       }
     },
-    setDebug: async (namespace?: string) => {
+    setDebug: async (namespace?) => {
       if (namespace?.length != null && namespace?.length > 0) {
         enable(namespace)
         localStorage.setItem('debug', namespace)
@@ -30,7 +31,7 @@ export function metricsRpc (components: DevToolsMetricsComponents): MetricsRPC {
         localStorage.removeItem('debug')
       }
     },
-    openConnection: async (peerIdOrMultiaddr: string, options?: OpenConnectionOptions) => {
+    openConnection: async (peerIdOrMultiaddr, options?) => {
       let peer: PeerId | Multiaddr
 
       try {
@@ -41,7 +42,7 @@ export function metricsRpc (components: DevToolsMetricsComponents): MetricsRPC {
 
       await components.connectionManager.openConnection(peer, options)
     },
-    closeConnection: async (peerId: PeerId, options?: AbortOptions) => {
+    closeConnection: async (peerId, options?) => {
       await Promise.all(
         components.connectionManager.getConnections(peerId)
           .map(async connection => {
@@ -54,6 +55,23 @@ export function metricsRpc (components: DevToolsMetricsComponents): MetricsRPC {
       )
     },
     contentRouting: components.contentRouting,
-    peerRouting: components.peerRouting
+    peerRouting: components.peerRouting,
+    pubsub: {
+      async getTopics (component) {
+        return getPubSub(component, components).getTopics()
+      },
+      async subscribe (component, topic) {
+        getPubSub(component, components).subscribe(topic)
+      },
+      async unsubscribe (component, topic) {
+        getPubSub(component, components).unsubscribe(topic)
+      },
+      async publish (component, topic, message) {
+        await getPubSub(component, components).publish(topic, message)
+      },
+      async getSubscribers (component: string, topic: string) {
+        return getPubSub(component, components).getSubscribers(topic)
+      }
+    }
   }
 }

package/src/utils/find-capability.ts

@@ -0,0 +1,9 @@
+import { gatherCapabilities } from './gather-capabilities.js'
+
+export function findCapability (capability: string, components: any): any | undefined {
+  for (const [name, capabilities] of Object.entries(gatherCapabilities(components))) {
+    if (capabilities.includes(capability)) {
+      return components[name]
+    }
+  }
+}

package/src/utils/gather-capabilities.ts

@@ -0,0 +1,14 @@
+import { serviceCapabilities } from '@libp2p/interface'
+
+export function gatherCapabilities (components: any): Record<string, string[]> {
+  const capabilities: Record<string, string[]> = {}
+  const services: Record<string, any> = components.components ?? components
+
+  Object.entries(services).forEach(([name, component]) => {
+    if (component?.[serviceCapabilities] != null && Array.isArray(component[serviceCapabilities])) {
+      capabilities[name] = component[serviceCapabilities]
+    }
+  })
+
+  return capabilities
+}

package/src/utils/get-pubsub.ts

@@ -0,0 +1,12 @@
+import { InvalidParametersError, isPubSub } from '@libp2p/interface'
+import type { PubSub } from '@libp2p/interface'
+
+export function getPubSub (component: string, components: any): PubSub {
+  const pubsub = components[component]
+
+  if (!isPubSub(pubsub)) {
+    throw new InvalidParametersError(`Component ${component} did not implement the PubSub interface`)
+  }
+
+  return pubsub
+}