Innovenergy_trunk/frontend/node_modules/endpoint-utils/index.js

120 lines
2.7 KiB
JavaScript

var os = require('os');
var ip = require('ip');
var Promise = require('pinkie-promise');
var createServer = require('net').createServer;
function createServerOnFreePort () {
return new Promise(function (resolve) {
var server = createServer();
server.once('listening', function () {
resolve(server);
});
server.listen(0);
});
}
function closeServers (servers) {
return Promise.all(servers.map(function (server) {
return new Promise(function (resolve) {
server.once('close', resolve);
server.close();
});
}));
}
function checkAvailability (port, hostname) {
return new Promise(function (resolve) {
var server = createServer();
server.once('error', function () {
resolve(false);
});
server.once('listening', function () {
server.once('close', function () {
resolve(true);
});
server.close();
});
server.listen(port, hostname);
});
}
function isFreePort (port) {
return checkAvailability(port);
}
function getFreePort () {
return getFreePorts(1).then(function (ports) {
return ports[0];
});
}
function getFreePorts (count) {
var serverPromises = [];
var ports = null;
// NOTE: Sequentially collect listening
// servers to avoid interference.
for (var i = 0; i < count; i++)
serverPromises.push(createServerOnFreePort());
return Promise.all(serverPromises)
.then(function (servers) {
ports = servers.map(function (server) {
return server.address().port;
});
return servers
})
.then(closeServers)
.then(function () {
return ports;
});
}
function isMyHostname (hostname) {
return getFreePort()
.then(function (port) {
return checkAvailability(port, hostname);
});
}
function getMyHostname () {
var hostname = os.hostname();
return isMyHostname(hostname)
.then(function (mine) {
if (mine)
return hostname;
if (os.platform() === 'mac') {
hostname += '.local';
return isMyHostname(hostname)
.then(function (mine) {
return mine ? hostname : ip.address();
});
}
return ip.address();
});
}
function getIPAddress () {
return ip.address();
}
module.exports = {
isFreePort: isFreePort,
getFreePort: getFreePort,
getFreePorts: getFreePorts,
isMyHostname: isMyHostname,
getMyHostname: getMyHostname,
getIPAddress: getIPAddress
};