core/datastore/repositories/network-repository.js

import { Promise } from 'es6-promise';

import { KinveyRequest, RequestMethod } from '../../request';
import { Aggregation } from '../../aggregation';
import { Repository } from './repository';
import { ensureArray } from '../../utils';
import { buildCollectionUrl } from './utils';

/**
 * @private
 * @typedef RequestOptions
 * @property {Operation} operation
 * @property {Object} options
 * @property {Query} [options.query]
 * @property {Query} [options.query]
 * @property {Object} [options.aggregation]
 * @property {Number} [options.timeout]
 * @property {Object} [options.properties]
 * @property {Boolean} [options.skipBL]
 * @property {Boolean} [options.trace]
 * @property {Number} [options.kinveyFileTTL]
 * @property {Boolean} [options.kinveyFileTLS]
 */
export class NetworkRepository extends Repository {
  read(collection, query, options = {}) {
    const requestConfig = this._buildRequestConfig(collection, RequestMethod.GET, null, query, null, null, options);
    return this._makeHttpRequest(requestConfig, options.dataOnly);
  }

  readById(collection, entityId, options) {
    const requestConfig = this._buildRequestConfig(collection, RequestMethod.GET, null, null, entityId, null, options);
    return this._makeHttpRequest(requestConfig);
  }

  create(collection, entities, options) {
    return this._processBatch(collection, RequestMethod.POST, entities, null, options);
  }

  update(collection, entities, options) {
    return this._processBatch(collection, RequestMethod.PUT, entities, options);
  }

  deleteById(collection, entityId, options) {
    const requestConfig = this._buildRequestConfig(collection, RequestMethod.DELETE, null, null, entityId, null, options);
    return this._makeHttpRequest(requestConfig)
      .then(response => response.count);
  }

  delete(collection, query, options) {
    const requestConfig = this._buildRequestConfig(collection, RequestMethod.DELETE, null, query, null, null, options);
    return this._makeHttpRequest(requestConfig)
      .then(response => response.count);
  }

  count(collection, query, options) {
    const requestConfig = this._buildRequestConfig(collection, RequestMethod.GET, null, query, null, '_count', null, options);
    return this._makeHttpRequest(requestConfig, options.dataOnly)
      .then((response) => {
        if (options.dataOnly === false) {
          return response;
        }

        return response.count;
      });
  }

  group(collection, aggregationQuery, options) {
    const requestConfig = this._buildRequestConfig(collection, RequestMethod.POST, null, aggregationQuery, null, '_group', null, options);
    return this._makeHttpRequest(requestConfig);
  }

  _processBatch(collection, method, entities, options) {
    const isSingle = !Array.isArray(entities);
    const requestPromises = ensureArray(entities).map((entity) => {
      const id = method === RequestMethod.PUT ? entity._id : null;
      const requestConfig = this._buildRequestConfig(collection, method, entity, null, id, null, options);
      return this._makeHttpRequest(requestConfig);
    });

    return Promise.all(requestPromises)
      .then(res => (isSingle ? res && res[0] : res));
  }

  _makeHttpRequest(requestConfig, dataOnly) {
    return KinveyRequest.execute(requestConfig, null, dataOnly);
  }

  /**
   * @param {String} collection
   * @param {Object} query
   * @param {RequestOptions} options
   * @param {RequestMethod} method
   * @param {Objet} data
   * @returns {RequestOptions}
   */
  _buildRequestConfig(collection, method, data, query, id, restAction, options) {
    options = options || {};
    const config = {
      method,
      pathname: buildCollectionUrl(collection, id, restAction),
      timeout: options.timeout,
      properties: options.properties,
      trace: options.trace,
      skipBL: options.skipBL,
      kinveyFileTTL: options.kinveyFileTTL,
      kinveyFileTLS: options.kinveyFileTLS
    };

    if (data) {
      config.body = data;
    }

    if (query instanceof Aggregation) {
      config.aggregation = query;
    } else {
      config.query = query;
    }

    return config;
  }
}