feat: add helper functions for working with addr infos

peer/addrinfo.go

@@ -106,3 +106,12 @@ func (pi *AddrInfo) Loggable() map[string]interface{} {
 		"addrs":  pi.Addrs,
 	}
 }
+
+// AddrInfosToIDs extracts the peer IDs from the passed AddrInfos and returns them in-order.
+func AddrInfosToIDs(pis []AddrInfo) []ID {
+	ps := make([]ID, len(pis))
+	for i, pi := range pis {
+		ps[i] = pi.ID
+	}
+	return ps
+}

peerstore/helpers.go

@@ -0,0 +1,14 @@
+package peerstore
+
+import (
+	"github.com/libp2p/go-libp2p-core/peer"
+)
+
+// AddrInfos returns an AddrInfo for each specified peer ID, in-order.
+func AddrInfos(ps Peerstore, peers []peer.ID) []peer.AddrInfo {
+	pi := make([]peer.AddrInfo, len(peers))
+	for i, p := range peers {
+		pi[i] = ps.PeerInfo(p)
+	}
+	return pi
+}