@mindhive/mock-aws/dist/dynamodbConnection.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.freePort = exports.serverExists = exports.clientFactory = undefined;

var _regenerator = require('babel-runtime/regenerator');

var _regenerator2 = _interopRequireDefault(_regenerator);

var _asyncToGenerator2 = require('babel-runtime/helpers/asyncToGenerator');

var _asyncToGenerator3 = _interopRequireDefault(_asyncToGenerator2);

var _net = require('net');

var _net2 = _interopRequireDefault(_net);

var _dynamodb = require('aws-sdk/clients/dynamodb');

var _dynamodb2 = _interopRequireDefault(_dynamodb);

var _global = require('aws-sdk/global');

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

var IPV4_LOCALHOST = '127.0.0.1'; // Has to be IPv4

var clientFactory = exports.clientFactory = function clientFactory(port) {
  var options = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
  return new _dynamodb2.default(Object.assign({
    region: 'localhost',
    endpoint: 'http://' + IPV4_LOCALHOST + ':' + port,
    credentials: new _global.Credentials('dummyKey', 'dummySecret')
  }, options));
};

var serverExists = exports.serverExists = function () {
  var _ref = (0, _asyncToGenerator3.default)( /*#__PURE__*/_regenerator2.default.mark(function _callee(port) {
    var client;
    return _regenerator2.default.wrap(function _callee$(_context) {
      while (1) {
        switch (_context.prev = _context.next) {
          case 0:
            client = clientFactory(port, { maxRetries: 0 });
            _context.prev = 1;
            _context.next = 4;
            return client.describeLimits().promise();

          case 4:
            return _context.abrupt('return', true);

          case 7:
            _context.prev = 7;
            _context.t0 = _context['catch'](1);
            return _context.abrupt('return', false);

          case 10:
          case 'end':
            return _context.stop();
        }
      }
    }, _callee, undefined, [[1, 7]]);
  }));

  return function serverExists(_x2) {
    return _ref.apply(this, arguments);
  };
}();

var freePort = exports.freePort = function freePort() {
  return new Promise(function (resolve, reject) {
    var server = _net2.default.createServer().once('error', function (e) {
      reject(e);
    }).once('listening', function () {
      var _server$address = server.address(),
          port = _server$address.port;

      server.close();
      resolve(port);
    }).listen(0, '0.0.0.0');
  });
};