pg-async/lib/makeAsyncApi.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; };

var _bluebird = require('bluebird');

var _bluebird2 = _interopRequireDefault(_bluebird);

var _debug = require('./debug');

var _debug2 = _interopRequireDefault(_debug);

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

function _asyncToGenerator(fn) { return function () { var gen = fn.apply(this, arguments); return new _bluebird2.default(function (resolve, reject) { function step(key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return _bluebird2.default.resolve(value).then(function (value) { return step("next", value); }, function (err) { return step("throw", err); }); } } return step("next"); }); }; }

const makeAsyncApi = client => {
  function query(sql) {
    for (var _len = arguments.length, values = Array(_len > 1 ? _len - 1 : 0), _key = 1; _key < _len; _key++) {
      values[_key - 1] = arguments[_key];
    }

    return query.queryArgs(sql, values);
  }

  query.query = function (sql) {
    for (var _len2 = arguments.length, values = Array(_len2 > 1 ? _len2 - 1 : 0), _key2 = 1; _key2 < _len2; _key2++) {
      values[_key2 - 1] = arguments[_key2];
    }

    return query.queryArgs(sql, values);
  };
  query.queryArgs = (sql, values) => new _bluebird2.default((resolve, reject) => {
    (0, _debug2.default)('query params: %j query: %j', values, sql);
    client.query(sql, values, (err, result) => {
      if (err) {
        (0, _debug2.default)('%s query(%j, %j)', err, sql, values);
        return reject(err);
      }
      (0, _debug2.default)('query ok: %d rows', result.rowCount);
      return resolve(result);
    });
  });

  query.rows = function (sql) {
    for (var _len3 = arguments.length, values = Array(_len3 > 1 ? _len3 - 1 : 0), _key3 = 1; _key3 < _len3; _key3++) {
      values[_key3 - 1] = arguments[_key3];
    }

    return query.rowsArgs(sql, values);
  };
  query.rowsArgs = (() => {
    var ref = _asyncToGenerator(function* (sql, values) {
      return (yield query.queryArgs(sql, values)).rows;
    }),
        _this = undefined;

    return function (_x, _x2) {
      return ref.apply(_this, arguments);
    };
  })();

  query.row = function (sql) {
    for (var _len4 = arguments.length, values = Array(_len4 > 1 ? _len4 - 1 : 0), _key4 = 1; _key4 < _len4; _key4++) {
      values[_key4 - 1] = arguments[_key4];
    }

    return query.rowArgs(sql, values);
  };
  query.rowArgs = (() => {
    var ref = _asyncToGenerator(function* (sql, values) {
      const result = yield query.queryArgs(sql, values);
      if (result.rowCount !== 1) throw new Error(`SQL: Expected exactly one row result but ${ result.rowCount } returned`);
      return result.rows[0];
    }),
        _this = undefined;

    return function (_x3, _x4) {
      return ref.apply(_this, arguments);
    };
  })();

  query.value = (() => {
    var ref = _asyncToGenerator(function* (sql) {
      for (var _len5 = arguments.length, values = Array(_len5 > 1 ? _len5 - 1 : 0), _key5 = 1; _key5 < _len5; _key5++) {
        values[_key5 - 1] = arguments[_key5];
      }

      return query.valueArgs(sql, values);
    }),
        _this = undefined;

    return function (_x5, _x6) {
      return ref.apply(_this, arguments);
    };
  })();
  query.valueArgs = (() => {
    var ref = _asyncToGenerator(function* (sql, values) {
      // will be better to use require('pgDriver/lib/utils').normalizeQueryConfig
      // but it is private API
      const opts = typeof sql === 'string' ? { text: sql, rowMode: 'array' } : _extends({}, sql, { rowMode: 'array' });
      const result = yield query.rowArgs(opts, values);
      if (result.length !== 1) throw new Error(`SQL: Expected exactly one column but ${ result.length } returned`);
      return result[0];
    }),
        _this = undefined;

    return function (_x7, _x8) {
      return ref.apply(_this, arguments);
    };
  })();

  query.inTransaction = false;

  query.startTransaction = () => {
    query.inTransaction = true;
    return query.query('BEGIN');
  };

  query.commit = () => {
    query.inTransaction = false;
    return query.query('COMMIT');
  };

  query.rollback = () => {
    query.inTransaction = false;
    return query.query('ROLLBACK');
  };

  query.end = _asyncToGenerator(function* () {
    if (query.inTransaction) {
      yield query.rollback();
      throw new Error('Transaction started manually but not closed. Automatic rollback');
    }
  });

  return query;
};

exports.default = makeAsyncApi;
module.exports = exports['default'];