File tree Expand file tree Collapse file tree 2 files changed +3
-23
lines changed Expand file tree Collapse file tree 2 files changed +3
-23
lines changed Original file line number Diff line number Diff line change @@ -4,6 +4,7 @@ const multiaddr = require('multiaddr')
4
4
const PeerInfo = require ( 'peer-info' )
5
5
const PeerId = require ( 'peer-id' )
6
6
const proto = require ( '../protocol' )
7
+ const { getPeerInfo } = require ( '../../get-peer-info' )
7
8
8
9
module . exports = function ( swarm ) {
9
10
/**
@@ -28,32 +29,11 @@ module.exports = function (swarm) {
28
29
* Helper to make a peer info from a multiaddrs
29
30
*
30
31
* @param {Multiaddr|PeerInfo|PeerId } ma
31
- * @param {Swarm } swarm
32
32
* @return {PeerInfo }
33
33
* @private
34
34
*/
35
- // TODO: this is ripped off of libp2p, should probably be a generally available util function
36
35
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 )
57
37
}
58
38
59
39
/**
Original file line number Diff line number Diff line change @@ -38,7 +38,7 @@ function getPeerInfo (peer, peerBook) {
38
38
39
39
addr && peer . multiaddrs . add ( addr )
40
40
41
- return peerBook . put ( peer )
41
+ return peerBook ? peerBook . put ( peer ) : peer
42
42
}
43
43
44
44
/**
You can’t perform that action at this time.
0 commit comments