@commercetools/customer-groups-exporter/lib/main.js

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = void 0;

var _nodeFetch = _interopRequireDefault(require("node-fetch"));

var _sdkClient = require("@commercetools/sdk-client");

var _apiRequestBuilder = require("@commercetools/api-request-builder");

var _sdkMiddlewareHttp = require("@commercetools/sdk-middleware-http");

var _sdkMiddlewareAuth = require("@commercetools/sdk-middleware-auth");

var _sdkMiddlewareUserAgent = require("@commercetools/sdk-middleware-user-agent");

var _JSONStream = _interopRequireDefault(require("JSONStream"));

var _silentLogger = _interopRequireDefault(require("./utils/silent-logger"));

var _package = _interopRequireDefault(require("../package.json"));

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

function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; var ownKeys = Object.keys(source); if (typeof Object.getOwnPropertySymbols === 'function') { ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function (sym) { return Object.getOwnPropertyDescriptor(source, sym).enumerable; })); } ownKeys.forEach(function (key) { _defineProperty(target, key, source[key]); }); } return target; }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

class CustomerGroupsExporter {
  // Set type annotations
  constructor(options) {
    if (!options.apiConfig) throw new Error('The constructor must be passed an `apiConfig` object');
    this.apiConfig = options.apiConfig;
    this.client = (0, _sdkClient.createClient)({
      middlewares: [(0, _sdkMiddlewareAuth.createAuthMiddlewareWithExistingToken)(options.accessToken ? `Bearer ${options.accessToken}` : ''), (0, _sdkMiddlewareAuth.createAuthMiddlewareForClientCredentialsFlow)(_objectSpread({}, this.apiConfig, {
        fetch: _nodeFetch.default
      })), (0, _sdkMiddlewareUserAgent.createUserAgentMiddleware)({
        libraryName: _package.default.name,
        libraryVersion: _package.default.version
      }), (0, _sdkMiddlewareHttp.createHttpMiddleware)({
        host: this.apiConfig.apiUrl,
        fetch: _nodeFetch.default
      })]
    });
    this.predicate = options.predicate;
    this.logger = _objectSpread({}, _silentLogger.default, options.logger);
  }

  run(outputStream) {
    this.logger.info('Starting Export');

    const jsonStream = _JSONStream.default.stringify();

    jsonStream.pipe(outputStream);
    this.handleOutput(outputStream, jsonStream, this.apiConfig.projectKey);
  }

  handleOutput(outputStream, pipeStream, projectKey) {
    this.fetchGroups(pipeStream, projectKey).then(() => {
      this.logger.info('Export operation completed successfully');
      if (outputStream !== process.stdout) pipeStream.end();
    }).catch(e => {
      outputStream.emit('error', e);
    });
  }

  fetchGroups(output, projectKey) {
    const request = CustomerGroupsExporter.buildRequest(projectKey, this.predicate);
    return this.client.process(request, ({
      statusCode,
      body
    }) => {
      if (statusCode !== 200) return Promise.reject(new Error(`Request returned error ${statusCode}`));
      body.results.forEach(object => {
        output.write(object);
      });
      this.logger.debug(`Successfully exported ${body.count} customer %s`, body.count > 1 ? 'groups' : 'group');
      return Promise.resolve();
    }, {
      accumulate: false
    });
  }

  static buildRequest(projectKey, predicate) {
    const uri = CustomerGroupsExporter.buildUri(projectKey, predicate);
    return {
      uri,
      method: 'GET'
    };
  }

  static buildUri(projectKey, predicate) {
    const service = (0, _apiRequestBuilder.createRequestBuilder)({
      projectKey
    }).customerGroups;
    if (predicate) service.where(predicate);
    return service.build();
  }

}

exports.default = CustomerGroupsExporter;