Skip to content

Commit 14b5f8a

Browse files
committed
refactor(circuit): consolidate get peer info
1 parent e993239 commit 14b5f8a

File tree

2 files changed

+3
-23
lines changed

2 files changed

+3
-23
lines changed

src/circuit/circuit/utils.js

Lines changed: 2 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ const multiaddr = require('multiaddr')
44
const PeerInfo = require('peer-info')
55
const PeerId = require('peer-id')
66
const proto = require('../protocol')
7+
const { getPeerInfo } = require('../../get-peer-info')
78

89
module.exports = function (swarm) {
910
/**
@@ -28,32 +29,11 @@ module.exports = function (swarm) {
2829
* Helper to make a peer info from a multiaddrs
2930
*
3031
* @param {Multiaddr|PeerInfo|PeerId} ma
31-
* @param {Swarm} swarm
3232
* @return {PeerInfo}
3333
* @private
3434
*/
35-
// TODO: this is ripped off of libp2p, should probably be a generally available util function
3635
function peerInfoFromMa (peer) {
37-
let p
38-
// PeerInfo
39-
if (PeerInfo.isPeerInfo(peer)) {
40-
p = peer
41-
// Multiaddr instance (not string)
42-
} else if (multiaddr.isMultiaddr(peer)) {
43-
const peerIdB58Str = peer.getPeerId()
44-
try {
45-
p = swarm._peerBook.get(peerIdB58Str)
46-
} catch (err) {
47-
p = new PeerInfo(PeerId.createFromB58String(peerIdB58Str))
48-
}
49-
p.multiaddrs.add(peer)
50-
// PeerId
51-
} else if (PeerId.isPeerId(peer)) {
52-
const peerIdB58Str = peer.toB58String()
53-
p = swarm._peerBook.has(peerIdB58Str) ? swarm._peerBook.get(peerIdB58Str) : peer
54-
}
55-
56-
return p
36+
return getPeerInfo(peer, swarm._peerBook)
5737
}
5838

5939
/**

src/get-peer-info.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ function getPeerInfo (peer, peerBook) {
3838

3939
addr && peer.multiaddrs.add(addr)
4040

41-
return peerBook.put(peer)
41+
return peerBook ? peerBook.put(peer) : peer
4242
}
4343

4444
/**

0 commit comments

Comments
 (0)