shyft-storage-connector/lib/helpers.js

Storage connector for Shyft

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.runTestPlaceholderQuery = exports.serializeValues = exports.parseValuesMap = exports.parseValues = void 0;

var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));

var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));

var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));

var _StorageTypePostgres = _interopRequireDefault(require("./StorageTypePostgres"));

var _shyft = require("shyft");

var _lodash = _interopRequireDefault(require("lodash"));

var parseValues = function parseValues(entity, data, model, context) {
  if (!data) {
    return data;
  }

  var entityAttributes = entity.getAttributes();

  _lodash["default"].forEach(entityAttributes, function (attribute) {
    var attributeName = attribute.name;
    var value = data[attributeName];

    if (typeof value === 'undefined' || attribute.mutationInput) {
      return;
    }

    var attributeType;

    if ((0, _shyft.isEntity)(attribute.type) || (0, _shyft.isShadowEntity)(attribute.type)) {
      var primaryAttribute = attribute.type.getPrimaryAttribute();
      attributeType = primaryAttribute.type;
    } else {
      attributeType = attribute.type;
    }

    var storageDataType = _StorageTypePostgres["default"].convertToStorageDataType(attributeType);

    if (storageDataType.parse) {
      data[attributeName] = storageDataType.parse(value, data, entity, model, context);
    }
  });

  return data;
};

exports.parseValues = parseValues;

var parseValuesMap = function parseValuesMap(entity, rows, model, context) {
  return rows.map(function (row) {
    return parseValues(entity, row, model, context);
  });
};

exports.parseValuesMap = parseValuesMap;

var serializeValues = function serializeValues(entity, mutation, data, model, context) {
  var entityAttributes = entity.getAttributes();
  var mutationAttributes = mutation.attributes || [];

  _lodash["default"].forEach(entityAttributes, function (attribute) {
    if (attribute.mutationInput) {
      return;
    }

    var attributeName = attribute.name;
    var value = data[attributeName];
    var attributeType;

    if ((0, _shyft.isEntity)(attribute.type) || (0, _shyft.isShadowEntity)(attribute.type)) {
      var primaryAttribute = attribute.type.getPrimaryAttribute();
      attributeType = primaryAttribute.type;
    } else {
      attributeType = attribute.type;
    }

    var storageDataType = _StorageTypePostgres["default"].convertToStorageDataType(attributeType);

    if (!mutationAttributes.includes(attributeName) && !storageDataType.enforceSerialize) {
      return;
    }

    if (storageDataType.serialize) {
      data[attributeName] = storageDataType.serialize(value, data, entity, mutation, model, context);
    }
  });

  return data;
};

exports.serializeValues = serializeValues;

var runTestPlaceholderQuery = /*#__PURE__*/function () {
  var _ref = (0, _asyncToGenerator2["default"])( /*#__PURE__*/_regenerator["default"].mark(function _callee(cmd, vars) {
    var storageInstance, manager, _storageInstance$driv, _storageInstance$driv2, query, parameters;

    return _regenerator["default"].wrap(function _callee$(_context) {
      while (1) {
        switch (_context.prev = _context.next) {
          case 0:
            storageInstance = _StorageTypePostgres["default"].getStorageInstance();
            manager = storageInstance.manager;
            _storageInstance$driv = storageInstance.driver.escapeQueryWithParameters(cmd, vars, {}), _storageInstance$driv2 = (0, _slicedToArray2["default"])(_storageInstance$driv, 2), query = _storageInstance$driv2[0], parameters = _storageInstance$driv2[1];
            return _context.abrupt("return", manager.query(query, parameters));

          case 4:
          case "end":
            return _context.stop();
        }
      }
    }, _callee);
  }));

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

exports.runTestPlaceholderQuery = runTestPlaceholderQuery;