foy/lib/exec.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const tslib_1 = require("tslib");
const execa = require("execa");
exports.execa = execa;
const pathLib = require("path");
const utils_1 = require("./utils");
const fs_1 = require("./fs");
const stream_1 = require("stream");
function exec(commands, options) {
    return tslib_1.__awaiter(this, void 0, void 0, function* () {
        if (utils_1.Is.str(commands)) {
            return execa.shell(commands, options);
        }
        let rets = [];
        for (let cmd of commands) {
            rets.push(yield execa.shell(cmd, options));
        }
        return rets;
    });
}
exports.exec = exec;
exports.spawn = execa;
class ShellContext {
    constructor() {
        this._cwd = process.cwd();
        this._env = Object.assign({}, process.env);
        this.logCommand = false;
        this.sleep = utils_1.sleep;
    }
    cwd(cwd = this._cwd) {
        return (this._cwd = cwd);
    }
    cd(path) {
        this._cwd = pathLib.resolve(this._cwd, path);
        return this;
    }
    exec(commands, options) {
        let logger = require('./logger').logger;
        if (this.logCommand) {
            logger.info('$', commands);
        }
        let p = exec(commands, Object.assign({ cwd: this._cwd, env: this._env, stdio: this.redirectLog ? 'pipe' : 'inherit' }, options));
        let redirectLogStream = this._getRedirectLogFile();
        if (redirectLogStream) {
            p.then(p => {
                p.stdout && redirectLogStream.write(p.stdout);
                p.stderr && redirectLogStream.write(p.stderr);
                redirectLogStream.end();
            });
        }
        p.catch(err => {
            logger.error('Exec failed: ', commands);
            redirectLogStream.end();
            throw err;
        });
        return p;
    }
    spawn(file, args = [], options) {
        let logger = require('./logger').logger;
        const command = file + ' ' + args.map(a => `"${a.replace(/"/g, '\\"')}"`).join(' ');
        if (this.logCommand) {
            logger.info('Exec: ', command);
        }
        let p = exports.spawn(file, args, Object.assign({ cwd: this._cwd, env: this._env, stdio: this.redirectLog ? 'pipe' : 'inherit' }, options));
        let redirectLogStream = this._getRedirectLogFile();
        if (redirectLogStream) {
            p.then(p => {
                p.stdout && redirectLogStream.write(p.stdout);
                p.stderr && redirectLogStream.write(p.stderr);
                redirectLogStream.end();
            });
        }
        p.catch(err => {
            logger.error('Exec failed: ', command);
            redirectLogStream.end();
            throw err;
        });
        return p;
    }
    env(key, val) {
        if (arguments.length === 1) {
            return this._env[key];
        }
        this._env[key] = val;
        return this;
    }
    _getRedirectLogFile() {
        if (!this.redirectLog)
            return null;
        let stream;
        if (this.redirectLog instanceof stream_1.Writable) {
            return this.redirectLog;
        }
        let logFile = utils_1.Is.str(this.redirectLog) ? this.redirectLog : utils_1.DefaultLogFile;
        return fs_1.fs.createWriteStream(logFile, { mode: fs_1.fs.constants.O_APPEND });
    }
}
exports.ShellContext = ShellContext;
function shell(callback) {
    return tslib_1.__awaiter(this, void 0, void 0, function* () {
        return callback(new ShellContext());
    });
}
exports.shell = shell;
//# sourceMappingURL=exec.js.map