@hotglue/cli 1.0.31 → 1.0.33

package/lib/commands/snapshots/download.js

@@ -1,213 +0,0 @@
-"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 _fs = require("fs");
-
-var _debug = _interopRequireDefault(require("../../helpers/debug"));
-
-var _ora = _interopRequireDefault(require("ora"));
-
-var _cliTable = _interopRequireDefault(require("cli-table"));
-
-var _awsSdk = _interopRequireDefault(require("aws-sdk"));
-
-var _descriptions = _interopRequireDefault(require("../../helpers/descriptions"));
-
-var _print = _interopRequireWildcard(require("../../helpers/print"));
-
-var _utils = require("../../helpers/utils");
-
-var _api = require("../../helpers/api");
-
-function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
-
-function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
-
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
-const debug = (0, _debug.default)('commands:snapshots:download');
-const command = 'download';
-exports.command = command;
-const desc = 'Download Snapshots';
-exports.desc = desc;
-
-const builder = async yargs => {
-  debug('builder', command);
-  return yargs.option('downloadTo', _descriptions.default.options['downloadTo'].config).demandOption('downloadTo', _descriptions.default.options['downloadTo'].demandText).option('tenant', { ..._descriptions.default.options['tenant'].config,
-    default: 'default'
-  }).option('overwrite', _descriptions.default.options['overwrite'].config).demandOption('overwrite', _descriptions.default.options['overwrite'].demandText);
-};
-
-exports.builder = builder;
-
-const handler = async argv => {
-  debug('handler', command, argv);
-  const {
-    hg,
-    json,
-    apikey,
-    env,
-    tenant,
-    downloadTo,
-    overwrite
-  } = argv;
-  const {
-    clientApiBaseUri
-  } = hg;
-  let message;
-  let spinner = (0, _ora.default)();
-
-  const folderPrefix = _path.default.resolve(process.cwd(), (0, _utils.userSetOption)('downloadTo') ? downloadTo : 'snapshots');
-
-  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: 'snapshot-download',
-      env,
-      tenant,
-      apikey
-    }); // created authenticated S3 instance
-
-    const s3 = new _awsSdk.default.S3({
-      accessKeyId,
-      secretAccessKey,
-      sessionToken
-    });
-    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary')); // 2. verify tenant is valid
-
-    message = (0, _print.themed)(`Validating tenant exists`);
-    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary'));
-
-    try {
-      // this may also throw access denied
-      const {
-        CommonPrefixes: tenantFolders
-      } = await s3.listObjectsV2({
-        Bucket: env,
-        Prefix: `${tenant}/`,
-        Delimiter: '/'
-      }).promise();
-      if (tenantFolders.length === 0) throw new Error('Invalid tenant');
-    } catch (err) {
-      debug('err', err);
-      throw new Error(`Tenant doesn't exist. Please check your tenant (-u) argument.`);
-    }
-
-    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary')); // 3. get list of downloadable files
-
-    message = (0, _print.themed)(`Scanning for downloadable files`);
-    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary'));
-    const keyPrefix = `${tenant}/snapshots/`;
-    const params = {
-      Bucket: env,
-      Prefix: keyPrefix
-    };
-    const {
-      Contents: s3Files
-    } = await s3.listObjectsV2(params).promise();
-    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary'));
-    debug('s3-list-res', s3Files); // filter unwanted files from source
-
-    const filesToDownload = s3Files ? s3Files.map(item => item.Key).filter(item => item !== keyPrefix) : [];
-
-    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 files
-
-
-    message = (0, _print.themed)(`Downloading snapshot files 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 file of filesToDownload) {
-      const subPath = file.substring(keyPrefix.length);
-
-      const fullPath = _path.default.resolve(folderPrefix, subPath);
-
-      debug('file', fullPath);
-
-      if (file.endsWith('/')) {
-        !(0, _fs.existsSync)(fullPath) && (await (0, _promises.mkdir)(fullPath, {
-          recursive: true
-        }));
-        continue;
-      } // skip if exists and overwrite is false
-
-
-      if (!overwrite) {
-        try {
-          await (0, _promises.stat)(fullPath);
-          debug('exists, skipping');
-          table.push([subPath, 'Skipped']);
-          continue;
-        } catch (err) {
-          if (err.code !== 'ENOENT') (0, _print.cl)(err);
-        }
-      }
-
-      message = (0, _print.themed)(`Downloading file: ${(0, _print.themed)(subPath, 'info')}`);
-      !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary')); // ensure destination folder exists
-
-      await (0, _promises.mkdir)(_path.default.dirname(fullPath), {
-        recursive: true
-      }); // download s3 object into destination folder
-
-      const fileStream = s3.getObject({
-        Bucket: env,
-        Key: file
-      }).createReadStream();
-      await (0, _promises.writeFile)(fullPath, await (0, _utils.streamToString)(fileStream));
-      !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary'));
-      table.push([subPath, '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;

package/lib/commands/snapshots.js

@@ -1,32 +0,0 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.handler = exports.desc = exports.command = exports.builder = void 0;
-
-var _debug = _interopRequireDefault(require("../helpers/debug"));
-
-var _baseBuilder = require("../helpers/baseBuilder");
-
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
-const debug = (0, _debug.default)('commands:snapshots');
-const command = 'snapshots <action>';
-exports.command = command;
-const desc = 'Manage tenant snapshots';
-exports.desc = desc;
-
-const builder = async function (yargs) {
-  debug('builder', command, yargs);
-  const base = await (0, _baseBuilder.baseApiBuilder)(yargs);
-  return base.commandDir('snapshots');
-};
-
-exports.builder = builder;
-
-const handler = async function (argv) {
-  debug('handler', command, argv);
-};
-
-exports.handler = handler;

package/lib/commands/tenants/custom-etl.js

@@ -1,183 +0,0 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.handler = exports.desc = exports.command = exports.builder = void 0;
-
-var _debug = _interopRequireDefault(require("../../helpers/debug"));
-
-var _ora = _interopRequireDefault(require("ora"));
-
-var _awsSdk = _interopRequireDefault(require("aws-sdk"));
-
-var _cliTable = _interopRequireDefault(require("cli-table"));
-
-var _descriptions = _interopRequireDefault(require("../../helpers/descriptions.js"));
-
-var _print = require("../../helpers/print.js");
-
-var _api = require("../../helpers/api.js");
-
-var _utils = require("../../helpers/utils.js");
-
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
-const debug = (0, _debug.default)('commands:tenants:custom-etl');
-const command = 'custom-etl';
-exports.command = command;
-const desc = 'List tenants with custom ETL in a specific flow';
-exports.desc = desc;
-
-const builder = async yargs => {
-  debug('builder', command);
-  return yargs.option('tenant', _descriptions.default.options['tenant'].config).option('flow', _descriptions.default.options['flow'].config).demandOption('flow', _descriptions.default.options['flow'].demandText);
-};
-
-exports.builder = builder;
-
-const processTenant = async ({
-  debug,
-  baseUri,
-  apikey,
-  env,
-  flow,
-  tenant,
-  supportedSources
-}) => {
-  // Get all linked sources of the tuple (env, flow, tenant)
-  const linkedSources = await (0, _api.getLinkedSources)({
-    debug,
-    baseUri,
-    env,
-    flow,
-    tenant,
-    apikey
-  });
-  let result = null;
-
-  for (const {
-    tap
-  } of [...linkedSources, ...supportedSources]) {
-    // Generate AWS credentials in order to read the S3 tap folder
-    // Note(davi): I'm using "etl-download" because it's a "read ETL" operation
-    const {
-      accessKeyId,
-      secretAccessKey,
-      sessionToken
-    } = await (0, _api.genCredentialsOnClientApi)({
-      debug,
-      baseUri,
-      task: 'etl-download',
-      env,
-      tenant,
-      flow,
-      tap,
-      apikey
-    });
-    const s3 = new _awsSdk.default.S3({
-      accessKeyId,
-      secretAccessKey,
-      sessionToken
-    });
-    const keyPrefix = `${tenant}/flows/${flow}/taps/${tap}/etl/`;
-    const params = {
-      Bucket: env,
-      Prefix: keyPrefix
-    }; // List all files inside the S3 tap folder
-
-    const {
-      Contents: s3Files
-    } = await s3.listObjectsV2(params).promise(); // Check if either "etl.py" or "etl.ipynb" exist
-
-    const hasCustomEtl = s3Files.some(({
-      Key
-    }) => Key.endsWith('etl.py') || Key.endsWith('etl.ipynb'));
-    if (hasCustomEtl) result = tenant; // If it's true, we can break already
-
-    if (result !== null) break;
-  }
-
-  return result;
-};
-
-const handler = async argv => {
-  debug('handler', command, argv);
-  const {
-    hg,
-    json,
-    apikey,
-    env,
-    flow
-  } = argv;
-  let message;
-  let spinner = (0, _ora.default)();
-
-  try {
-    message = (0, _print.themed)(`Retrieving tenants for environment ${(0, _print.themed)(env, 'info')}`);
-    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary'));
-    const [allTenants, supportedSources] = await Promise.all([(0, _api.getTenants)({
-      debug,
-      baseUri: hg.clientApiBaseUri,
-      apikey,
-      env
-    }), (0, _api.getSupportedSources)({
-      debug,
-      baseUri: hg.clientApiBaseUri,
-      apikey,
-      env,
-      flow
-    })]);
-    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary'));
-    let tenantsWithCustomEtl = [];
-    message = (0, _print.themed)(`Querying for custom ETL scripts for flow ${(0, _print.themed)(flow, 'info')}`);
-    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary')); // Break the tenants array in to an array of arrays so we can make
-    // parallel calls and make it faster
-
-    for (const tenants of (0, _utils.buildChunks)(allTenants)) {
-      const maybeTenantsWithCustomEtl = await Promise.all(tenants.map(tenant => processTenant({
-        debug,
-        baseUri: hg.clientApiBaseUri,
-        apikey,
-        env,
-        flow,
-        tenant,
-        supportedSources
-      }))); // Note(davi): we need the `.filter(Boolean)` because sometimes
-      // the function above returns `null`
-
-      tenantsWithCustomEtl = tenantsWithCustomEtl.concat(maybeTenantsWithCustomEtl.filter(Boolean));
-    }
-
-    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary'));
-
-    if (json) {
-      (0, _print.printJSON)(tenantsWithCustomEtl);
-    } else {
-      // generate results table
-      const table = new _cliTable.default({
-        head: ['Tenant ID']
-      });
-      table.push(...tenantsWithCustomEtl.map(item => [item])); // print results
-
-      console.log(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;

package/lib/commands/tenants/custom-field-map.js

@@ -1,191 +0,0 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.handler = exports.desc = exports.command = exports.builder = void 0;
-
-var _debug = _interopRequireDefault(require("../../helpers/debug"));
-
-var _ora = _interopRequireDefault(require("ora"));
-
-var _awsSdk = _interopRequireDefault(require("aws-sdk"));
-
-var _cliTable = _interopRequireDefault(require("cli-table"));
-
-var _descriptions = _interopRequireDefault(require("../../helpers/descriptions"));
-
-var _print = require("../../helpers/print");
-
-var _api = require("../../helpers/api");
-
-var _utils = require("../../helpers/utils");
-
-function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
-
-const debug = (0, _debug.default)('commands:tenants:custom-field-map');
-const command = 'custom-field-map';
-exports.command = command;
-const desc = 'List tenants with custom field map in a specific flow';
-exports.desc = desc;
-
-const builder = async yargs => {
-  debug('builder', command);
-  return yargs.option('tenant', _descriptions.default.options['tenant'].config).option('flow', _descriptions.default.options['flow'].config).demandOption('flow', _descriptions.default.options['flow'].demandText);
-};
-
-exports.builder = builder;
-
-const processTenant = async ({
-  debug,
-  baseUri,
-  apikey,
-  env,
-  flow,
-  tenant,
-  isV2Flow
-}) => {
-  // Get all linked connectors/sources of the tuple (env, flow, tenant)
-  const linkedConnectors = isV2Flow ? await (0, _api.getLinkedConnectors)({
-    debug,
-    baseUri,
-    env,
-    flow,
-    tenant,
-    apikey
-  }) : await (0, _api.getLinkedSources)({
-    debug,
-    baseUri,
-    env,
-    flow,
-    tenant,
-    apikey
-  });
-  const connectors = []; // Generate AWS credentials in order to read the S3 tap folder
-
-  const {
-    accessKeyId,
-    secretAccessKey,
-    sessionToken
-  } = await (0, _api.genCredentialsOnClientApi)({
-    debug,
-    baseUri,
-    task: 'field-map-download',
-    env,
-    tenant,
-    flow,
-    apikey
-  });
-  const s3 = new _awsSdk.default.S3({
-    accessKeyId,
-    secretAccessKey,
-    sessionToken
-  });
-
-  for (const {
-    id,
-    tap
-  } of linkedConnectors) {
-    const entityLabel = isV2Flow ? 'connectors' : 'taps';
-    const params = {
-      Bucket: env,
-      Key: `${tenant}/flows/${flow}/${entityLabel}/${isV2Flow ? id : tap}/fieldMap.json`
-    };
-
-    try {
-      await s3.headObject(params).promise();
-      connectors.push(isV2Flow ? id : tap);
-    } catch (error) {// Does not have fieldMap.json
-    }
-  }
-
-  return {
-    tenant,
-    connectors
-  };
-};
-
-const handler = async argv => {
-  debug('handler', command, argv);
-  const {
-    hg,
-    json,
-    apikey,
-    env,
-    flow
-  } = argv;
-  let message;
-  let spinner = (0, _ora.default)();
-
-  try {
-    message = (0, _print.themed)(`Retrieving tenants for environment ${(0, _print.themed)(env, 'info')}`);
-    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary'));
-    const [allTenants, supportedFlow] = await Promise.all([(0, _api.getTenants)({
-      debug,
-      baseUri: hg.clientApiBaseUri,
-      apikey,
-      env
-    }), (0, _api.getSupportedFlow)({
-      debug,
-      baseUri: hg.clientApiBaseUri,
-      apikey,
-      env,
-      flow
-    })]);
-    const isV2Flow = (supportedFlow === null || supportedFlow === void 0 ? void 0 : supportedFlow.version) === 2;
-    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary'));
-    const tenantsWithCustomFieldMap = [];
-    message = (0, _print.themed)(`Querying for custom field maps for flow ${(0, _print.themed)(flow, 'info')}`);
-    !json && spinner.start((0, _print.themed)(`In progress: ${message}...`, 'secondary')); // Break the tenants array in to an array of arrays so we can make
-    // parallel calls and make it faster
-
-    for (const tenants of (0, _utils.buildChunks)(allTenants)) {
-      const maybeTenantsWithCustomFieldMap = await Promise.all(tenants.map(tenant => processTenant({
-        debug,
-        baseUri: hg.clientApiBaseUri,
-        apikey,
-        env,
-        flow,
-        tenant,
-        isV2Flow
-      })));
-      tenantsWithCustomFieldMap.push(...maybeTenantsWithCustomFieldMap.filter(({
-        connectors
-      }) => connectors.length > 0));
-    }
-
-    !json && spinner.succeed((0, _print.themed)(`Finished: ${message}.`, 'secondary'));
-
-    if (json) {
-      (0, _print.printJSON)(tenantsWithCustomFieldMap);
-    } else {
-      // generate results table
-      const table = new _cliTable.default({
-        head: ['Tenant ID', 'Connector IDs']
-      });
-      table.push(...tenantsWithCustomFieldMap.map(({
-        tenant,
-        connectors
-      }) => [tenant, connectors.join(', ')])); // print results
-
-      console.log(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;