pot-js/lib/utils/createCommand.js

Process management module

'use strict';

Object.defineProperty(exports, "__esModule", {
	value: true
});

var _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; };

exports.default = createCommand;

var _potLogger = require('pot-logger');

var _resolveConfig = require('./resolveConfig');

var _resolveConfig2 = _interopRequireDefault(_resolveConfig);

var _getCliOptionsBySchema = require('./getCliOptionsBySchema');

var _getCliOptionsBySchema2 = _interopRequireDefault(_getCliOptionsBySchema);

var _validateBySchema = require('./validateBySchema');

var _validateBySchema2 = _interopRequireDefault(_validateBySchema);

var _lodash = require('lodash');

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

function _objectWithoutProperties(obj, keys) { var target = {}; for (var i in obj) { if (keys.indexOf(i) >= 0) continue; if (!Object.prototype.hasOwnProperty.call(obj, i)) continue; target[i] = obj[i]; } return target; }

function _asyncToGenerator(fn) { return function () { var gen = fn.apply(this, arguments); return new Promise(function (resolve, reject) { function step(key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { return Promise.resolve(value).then(function (value) { step("next", value); }, function (err) { step("throw", err); }); } } return step("next"); }); }; } /* A helper function to create yargs command */

const builder = function builder(yargs) {
	const middlewares = this.middlewares,
	      optional = this.optional,
	      demanded = this.demanded,
	      original = this.original;

	const potStore = middlewares[0] || {};
	const schema = potStore.schema,
	      getBuilder = potStore.getBuilder;

	const spec = (0, _getCliOptionsBySchema2.default)(schema);

	if ((0, _lodash.isFunction)(getBuilder)) {
		return getBuilder.call(this, yargs, spec);
	} else if ((0, _lodash.isObject)(spec)) {
		const args = yargs.usage(`$0 ${original}`);
		const positions = (0, _lodash.flatten)([...demanded.map((_ref) => {
			let cmd = _ref.cmd;
			return cmd;
		}), ...optional.map((_ref2) => {
			let cmd = _ref2.cmd;
			return cmd;
		})]);
		positions.forEach(key => {
			if (spec[key]) args.positional(key, (0, _lodash.omit)(spec[key], ['alias']));
		});
		args.options((0, _lodash.omit)(spec, positions));
		return args;
	}

	return yargs.argv;
};

const handler = (() => {
	var _ref3 = _asyncToGenerator(function* (argv) {
		try {
			const potStore = this.middlewares[0] || {};
			const operator = potStore.operator,
			      configFileKey = potStore.configFile,
			      schema = potStore.schema,
			      validate = potStore.validate;


			if ((0, _lodash.isFunction)(operator)) {
				if (configFileKey) {
					argv = yield (0, _resolveConfig2.default)(argv, configFileKey, schema);
				}
				if (validate && schema) {
					(0, _validateBySchema2.default)(schema, argv);
				}
				yield operator(argv);
			}
		} catch (err) {
			(0, _potLogger.setLoggers)('logLevel', argv.logLevel || 'INFO');
			_potLogger.logger.error(err.message);
			_potLogger.logger.debug(err);
		}
	});

	function handler(_x) {
		return _ref3.apply(this, arguments);
	}

	return handler;
})();

function createCommand() {
	let options = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
	const operator = options.operator,
	      schema = options.schema,
	      getBuilder = options.getBuilder,
	      configFile = options.configFile,
	      validate = options.validate;
	var _options$middlewares = options.middlewares;

	const middlewares = _options$middlewares === undefined ? [] : _options$middlewares,
	      other = _objectWithoutProperties(options, ['operator', 'schema', 'getBuilder', 'configFile', 'validate', 'middlewares']);

	// A trick middleware to store configs


	const potStoreMiddleware = function potStoreMiddleware(a) {
		return a;
	};
	Object.assign(potStoreMiddleware, {
		operator,
		schema,
		getBuilder,
		configFile,
		validate
	});
	middlewares.unshift(potStoreMiddleware);

	return _extends({
		builder,
		handler
	}, other, {
		middlewares
	});
}