@hotglue/cli/lib/commands/env/download.js

hotglue CLI tools

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.handler = exports.desc = exports.command = exports.builder = void 0;

var _path = _interopRequireDefault(require("path"));

var _promises = require("fs/promises");

var _debug = _interopRequireDefault(require("../../helpers/debug"));

var _ora = _interopRequireDefault(require("ora"));

var _axios = _interopRequireDefault(require("axios/dist/node/axios.cjs"));

var _cliTable = _interopRequireDefault(require("cli-table"));

var _awsSdk = _interopRequireDefault(require("aws-sdk"));

var _descriptions = _interopRequireDefault(require("../../helpers/descriptions"));

var _print = require("../../helpers/print");

var _utils = require("../../helpers/utils");

var _api = require("../../helpers/api");

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

const debug = (0, _debug.default)('commands:env:download');
const command = 'download';
exports.command = command;
const desc = 'Download Environment settings';
exports.desc = desc;

const builder = async yargs => {
  debug('builder', command); // hotglue environment download -e <env> -o -d downloads

  return yargs.option('downloadTo', _descriptions.default.options['downloadTo'].config); // .option('overwrite', descriptions.options['overwrite'].config);
};

exports.builder = builder;

const handler = async argv => {
  debug('handler', command, argv);
  const {
    hg,
    json,
    apikey,
    env,
    downloadTo
  } = argv;
  const {
    clientApiBaseUri
  } = hg;

  const folderPrefix = _path.default.resolve(process.cwd(), downloadTo);

  let message;
  let spinner = (0, _ora.default)();

  try {
    // 1. get STS credentials
    message = (0, _print.themed)(`Verifying user and authorizing`);
    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary'));
    const {
      accessKeyId,
      secretAccessKey,
      sessionToken
    } = await (0, _api.genCredentialsOnClientApi)({
      debug,
      baseUri: clientApiBaseUri,
      task: 'env-download',
      env,
      apikey
    });
    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary')); // create authenticated S3 instance

    const s3 = new _awsSdk.default.S3({
      accessKeyId,
      secretAccessKey,
      sessionToken
    }); // 2. get list of files from AWS bucket

    message = (0, _print.themed)(`Scanning environment ${(0, _print.themed)(env, 'info')}`);
    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary'));
    const params = {
      Bucket: env,
      Prefix: `config/`
    };
    const {
      Contents: s3Files
    } = await s3.listObjectsV2(params).promise();
    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary'));
    debug('s3-files', s3Files); // 3. filter only downloadable files

    const filesToDownload = s3Files ? (0, _utils.filterFiles)(s3Files.map(item => item.Key), {
      pattern: '((*/requirements.txt)|(*/availableSources.json)|(*/availableTargets.json)|(*/availableConnectors.json))'
    }) : [];

    if (!filesToDownload || filesToDownload.length === 0) {
      json ? (0, _print.printJSON)({
        status: 'success',
        downloadedFiles: []
      }) : spinner.warn((0, _print.themed)(`Warning: ${(0, _print.themed)('Nothing to download!')}`, 'secondary'));
      return;
    } // 4. download environment config files


    message = (0, _print.themed)(`Downloading to ${(0, _print.themed)(folderPrefix, 'info')}`);
    !json && spinner.info((0, _print.themed)(`Info: ${message}.`, 'secondary'));
    const table = new _cliTable.default({
      head: ['File', 'Status']
    });

    for await (const item of filesToDownload) {
      // create destination folder
      const filename = _path.default.basename(item);

      const fullPath = _path.default.resolve(folderPrefix, filename);

      debug('local-file', fullPath); // ensure destination folder exists

      await (0, _promises.mkdir)(_path.default.dirname(fullPath), {
        recursive: true
      }); // download s3 object into destination folder

      message = (0, _print.themed)(`Downloading file: ${(0, _print.themed)(filename, 'info')}`);
      !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary'));
      const fileStream = s3.getObject({
        Bucket: env,
        Key: item
      }).createReadStream();
      await (0, _promises.writeFile)(fullPath, await (0, _utils.streamToString)(fileStream));
      !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary')); // debug('downloaded', env + item);

      table.push([filename, 'Downloaded']);
    }

    if (json) {
      (0, _print.printJSON)({
        status: 'success',
        downloadedFiles: filesToDownload
      });
    } else {
      // print results
      (0, _print.cl)(table.toString());
    }
  } catch (err) {
    if (json) {
      (0, _print.printJSON)({
        status: 'error',
        error: err
      });
    } else {
      spinner.fail((0, _print.themed)(`Error: ${message}.`, 'secondary'));
      (0, _print.pr)((0, _print.themed)(`Message: ${(0, _print.themed)(err.message)}`, 'secondary'));
      debug(err);

      if (err && err.response && err.response.data) {
        debug('response', err.response.data);
      }
    }
  }
};

exports.handler = handler;