From 4ebd4bc392c8481133e8625328799e5ce6fc11ab Mon Sep 17 00:00:00 2001
From: achingbrain <alex@achingbrain.net>
Date: Fri, 14 Apr 2023 12:09:02 +0100
Subject: [PATCH 1/2] feat: expose get connection map method of connection
 manager

Exposes the collection of managed connections as a peer map.
---
 packages/interface-connection-manager/package.json |  1 +
 packages/interface-connection-manager/src/index.ts | 12 ++++++++++++
 packages/interface-mocks/package.json              |  1 +
 packages/interface-mocks/src/connection-manager.ts | 14 ++++++++++++++
 4 files changed, 28 insertions(+)

diff --git a/packages/interface-connection-manager/package.json b/packages/interface-connection-manager/package.json
index 53da7837c..0511ef613 100644
--- a/packages/interface-connection-manager/package.json
+++ b/packages/interface-connection-manager/package.json
@@ -135,6 +135,7 @@
     "@libp2p/interface-connection": "^4.0.0",
     "@libp2p/interface-peer-id": "^2.0.0",
     "@libp2p/interfaces": "^3.0.0",
+    "@libp2p/peer-collections": "^3.0.1",
     "@multiformats/multiaddr": "^12.0.0"
   },
   "devDependencies": {
diff --git a/packages/interface-connection-manager/src/index.ts b/packages/interface-connection-manager/src/index.ts
index 23f4f3827..60ae288d1 100644
--- a/packages/interface-connection-manager/src/index.ts
+++ b/packages/interface-connection-manager/src/index.ts
@@ -3,6 +3,7 @@ import type { EventEmitter } from '@libp2p/interfaces/events'
 import type { Connection, MultiaddrConnection } from '@libp2p/interface-connection'
 import type { PeerId } from '@libp2p/interface-peer-id'
 import type { Multiaddr } from '@multiformats/multiaddr'
+import type { PeerMap } from '@libp2p/peer-collections'
 
 export interface ConnectionManagerEvents {
   /**
@@ -65,6 +66,17 @@ export interface ConnectionManager extends EventEmitter<ConnectionManagerEvents>
    */
   getConnections: (peerId?: PeerId) => Connection[]
 
+  /**
+   * Return a map of all connections with their associated PeerIds
+   *
+   * @example
+   *
+   * ```js
+   * const connectionsMap = libp2p.connectionManager.getConnectionsMap()
+   * ```
+   */
+  getConnectionsMap: () => PeerMap<Connection[]>
+
   /**
    * Open a connection to a remote peer
    *
diff --git a/packages/interface-mocks/package.json b/packages/interface-mocks/package.json
index aaeecf8b0..73fb4742f 100644
--- a/packages/interface-mocks/package.json
+++ b/packages/interface-mocks/package.json
@@ -154,6 +154,7 @@
     "@libp2p/interfaces": "^3.0.0",
     "@libp2p/logger": "^2.0.0",
     "@libp2p/multistream-select": "^3.0.0",
+    "@libp2p/peer-collections": "^3.0.1",
     "@libp2p/peer-id": "^2.0.0",
     "@libp2p/peer-id-factory": "^2.0.0",
     "@multiformats/multiaddr": "^12.0.0",
diff --git a/packages/interface-mocks/src/connection-manager.ts b/packages/interface-mocks/src/connection-manager.ts
index 6165f7988..2701c75e8 100644
--- a/packages/interface-mocks/src/connection-manager.ts
+++ b/packages/interface-mocks/src/connection-manager.ts
@@ -9,6 +9,7 @@ import type { Registrar } from '@libp2p/interface-registrar'
 import type { PubSub } from '@libp2p/interface-pubsub'
 import { isMultiaddr, Multiaddr } from '@multiformats/multiaddr'
 import { peerIdFromString } from '@libp2p/peer-id'
+import { PeerMap } from '@libp2p/peer-collections'
 
 export interface MockNetworkComponents {
   peerId: PeerId
@@ -86,6 +87,19 @@ class MockConnectionManager extends EventEmitter<ConnectionManagerEvents> implem
     return this.connections
   }
 
+  getConnectionsMap (): PeerMap<Connection[]> {
+    const map = new PeerMap<Connection[]>()
+
+    for (let conn of this.connections) {
+      const conns: Connection[] = map.get(conn.remotePeer) ?? []
+      conns.push(conn)
+
+      map.set(conn.remotePeer, conns)
+    }
+
+    return map
+  }
+
   async openConnection (peerId: PeerId | Multiaddr | Multiaddr[]): Promise<Connection> {
     if (this.components == null) {
       throw new CodeError('Not initialized', 'ERR_NOT_INITIALIZED')

From 16ab11703097b30178dfb37b95842d6c19c34af9 Mon Sep 17 00:00:00 2001
From: achingbrain <alex@achingbrain.net>
Date: Fri, 14 Apr 2023 12:19:09 +0100
Subject: [PATCH 2/2] chore: fix linting

---
 packages/interface-mocks/src/connection-manager.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/interface-mocks/src/connection-manager.ts b/packages/interface-mocks/src/connection-manager.ts
index 2701c75e8..b4dfba56f 100644
--- a/packages/interface-mocks/src/connection-manager.ts
+++ b/packages/interface-mocks/src/connection-manager.ts
@@ -90,7 +90,7 @@ class MockConnectionManager extends EventEmitter<ConnectionManagerEvents> implem
   getConnectionsMap (): PeerMap<Connection[]> {
     const map = new PeerMap<Connection[]>()
 
-    for (let conn of this.connections) {
+    for (const conn of this.connections) {
       const conns: Connection[] = map.get(conn.remotePeer) ?? []
       conns.push(conn)