From bee91538efc41a2ed25f185120e454a92fed1e49 Mon Sep 17 00:00:00 2001
From: Tanguy Cizain <tanguycizain@gmail.com>
Date: Mon, 21 Jun 2021 17:14:24 +0200
Subject: [PATCH] builder: add multiple local address support (#591)

---
 libp2p/builders.nim | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/libp2p/builders.nim b/libp2p/builders.nim
index 7ee4129..7d2e797 100644
--- a/libp2p/builders.nim
+++ b/libp2p/builders.nim
@@ -36,7 +36,7 @@ type
 
   SwitchBuilder* = ref object
     privKey: Option[PrivateKey]
-    address: MultiAddress
+    addresses: seq[MultiAddress]
     secureManagers: seq[SecureProtocol]
     mplexOpts: MplexOpts
     tcpTransportOpts: TcpTransportOpts
@@ -56,7 +56,7 @@ proc new*(T: type[SwitchBuilder]): T =
 
   SwitchBuilder(
     privKey: none(PrivateKey),
-    address: address,
+    addresses: @[address],
     secureManagers: @[],
     tcpTransportOpts: TcpTransportOpts(),
     maxConnections: MaxConnections,
@@ -71,9 +71,14 @@ proc withPrivateKey*(b: SwitchBuilder, privateKey: PrivateKey): SwitchBuilder =
   b
 
 proc withAddress*(b: SwitchBuilder, address: MultiAddress): SwitchBuilder =
-  b.address = address
+  b.addresses = @[address]
   b
 
+proc withAddresses*(b: SwitchBuilder, addresses: seq[MultiAddress]): SwitchBuilder =
+  b.addresses = addresses
+  b
+
+
 proc withMplex*(b: SwitchBuilder, inTimeout = 5.minutes, outTimeout = 5.minutes): SwitchBuilder =
   proc newMuxer(conn: Connection): Muxer =
     Mplex.init(
@@ -143,7 +148,7 @@ proc build*(b: SwitchBuilder): Switch
   let
     peerInfo = PeerInfo.init(
       seckey,
-      [b.address],
+      b.addresses,
       protoVersion = b.protoVersion,
       agentVersion = b.agentVersion)