chore: tiny refactor

This commit is contained in:
David Dias 2018-02-07 06:40:00 +00:00
parent 9c77a69ae3
commit de1d7fe75c
4 changed files with 16 additions and 10 deletions

View File

@ -4,19 +4,25 @@ const multiaddr = require('multiaddr')
const Address6 = require('ip-address').Address6 const Address6 = require('ip-address').Address6
module.exports = (socket) => { module.exports = (socket) => {
let mh let ma
if (socket.remoteFamily === 'IPv6') { if (socket.remoteFamily === 'IPv6') {
var addr = new Address6(socket.remoteAddress) const addr = new Address6(socket.remoteAddress)
if (addr.v4) { if (addr.v4) {
var ip4 = addr.to4().correctForm() const ip4 = addr.to4().correctForm()
mh = multiaddr('/ip4/' + ip4 + '/tcp/' + socket.remotePort) ma = multiaddr('/ip4/' + ip4 +
'/tcp/' + socket.remotePort
)
} else { } else {
mh = multiaddr('/ip6/' + socket.remoteAddress + '/tcp/' + socket.remotePort) ma = multiaddr('/ip6/' + socket.remoteAddress +
'/tcp/' + socket.remotePort
)
} }
} else { } else {
mh = multiaddr('/ip4/' + socket.remoteAddress + '/tcp/' + socket.remotePort) ma = multiaddr('/ip4/' + socket.remoteAddress +
'/tcp/' + socket.remotePort)
} }
return mh return ma
} }

View File

@ -57,7 +57,7 @@ class TCP {
options = {} options = {}
} }
handler = handler || (() => {}) handler = handler || noop
return createListener(handler) return createListener(handler)
} }

View File

@ -3,12 +3,12 @@
const tests = require('interface-transport') const tests = require('interface-transport')
const multiaddr = require('multiaddr') const multiaddr = require('multiaddr')
const Tcp = require('../src') const TCP = require('../src')
describe('interface-transport compliance', () => { describe('interface-transport compliance', () => {
tests({ tests({
setup (cb) { setup (cb) {
let tcp = new Tcp() let tcp = new TCP()
const addrs = [ const addrs = [
multiaddr('/ip4/127.0.0.1/tcp/9091'), multiaddr('/ip4/127.0.0.1/tcp/9091'),
multiaddr('/ip4/127.0.0.1/tcp/9092'), multiaddr('/ip4/127.0.0.1/tcp/9092'),