From f6f73a21ebf15d1811fb95cc3972f950c0f52789 Mon Sep 17 00:00:00 2001
From: Chad Nehemiah <chad.nehemiah94@gmail.com>
Date: Thu, 16 Feb 2023 10:19:38 -0500
Subject: [PATCH] feat: updated connection manager and dialer with new
 interface methods (#1586)

---
 src/connection-manager/dialer/index.ts | 18 ++++++++++++++++++
 src/connection-manager/index.ts        |  4 ++++
 2 files changed, 22 insertions(+)

diff --git a/src/connection-manager/dialer/index.ts b/src/connection-manager/dialer/index.ts
index 7744a05e29..4aa81e971d 100644
--- a/src/connection-manager/dialer/index.ts
+++ b/src/connection-manager/dialer/index.ts
@@ -217,6 +217,24 @@ export class DefaultDialer implements Startable, Dialer {
     }
   }
 
+  /**
+   * Get the current dial targets which are pending
+   */
+  getPendingDialTargets (): Map<string, AbortController> {
+    return this.pendingDialTargets
+  }
+
+  /**
+   * Returns true if the peer id is in the pending dials
+   */
+  hasPendingDial (peerId: PeerId | Multiaddr): boolean {
+    if (isMultiaddr(peerId)) {
+      return this.pendingDials.has(peerId.getPeerId() ?? '')
+    }
+
+    return this.pendingDials.has(peerId.toString())
+  }
+
   /**
    * Creates a DialTarget. The DialTarget is used to create and track
    * the DialRequest to a given peer.
diff --git a/src/connection-manager/index.ts b/src/connection-manager/index.ts
index 0c40953863..c22e59ec5b 100644
--- a/src/connection-manager/index.ts
+++ b/src/connection-manager/index.ts
@@ -470,6 +470,10 @@ export class DefaultConnectionManager extends EventEmitter<ConnectionManagerEven
     return conns
   }
 
+  getConnectionsMap (): Map<string, Connection[]> {
+    return this.connections
+  }
+
   async openConnection (peerIdOrMultiaddr: PeerId | Multiaddr, options: AbortOptions = {}): Promise<Connection> {
     const { peerId, multiaddr } = getPeerAddress(peerIdOrMultiaddr)