typescript-node-scripts/build/lib/processHandler.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.AppController = void 0;
const tslib_1 = require("tslib");
const chalk_1 = (0, tslib_1.__importDefault)(require("chalk"));
const child_process_1 = require("child_process");
const pstree_remy_1 = (0, tslib_1.__importStar)(require("pstree.remy"));
const platform_1 = require("../util/platform");
function invariant(condition, message) {
    if (!condition) {
        const err = new Error(message);
        err.name = 'Invariant Violation';
        throw err;
    }
}
class AppController {
    constructor(executablePath, argv) {
        this.state = {
            appRunning: false,
            proc: null,
            pid: null,
        };
        this.executablePath = executablePath;
        this.argv = argv;
    }
    stopApp() {
        return (0, tslib_1.__awaiter)(this, void 0, void 0, function* () {
            return new Promise((resolve) => {
                if (this.state.appRunning) {
                    console.log('Sending SIGTERM signal to app...');
                    const { pid } = this.state;
                    const closeHandler = (code, signal) => {
                        this.setState({
                            appRunning: false,
                            proc: null,
                            pid: null,
                        });
                        resolve();
                    };
                    if ((0, platform_1.isWindows)()) {
                        (0, child_process_1.exec)('taskkill /pid ' + pid + ' /T /F').on('close', closeHandler);
                    }
                    else {
                        // pstree is used to kill the full subtree of a spawned app
                        (0, pstree_remy_1.default)(pid, (_, children) => {
                            if (pstree_remy_1.hasPS) {
                                // we now send SIGTERM to the spawned process
                                (0, child_process_1.spawn)('kill', ['-s', 'SIGTERM', pid].concat(children)).on('close', closeHandler);
                            }
                            else {
                                const pids = children.concat(pid).sort();
                                pids.forEach((childPid) => {
                                    // 15 is for SIGTERM
                                    (0, child_process_1.exec)('kill -15 ' + childPid).on('close', closeHandler);
                                });
                            }
                        });
                    }
                }
                else {
                    resolve();
                }
            });
        });
    }
    runApp() {
        invariant(!this.state.appRunning, "Can't start app when app is already running!");
        // spawn the process
        const proc = (0, child_process_1.spawn)('node', [this.executablePath, ...this.argv]);
        // attach event listeners
        proc.on('error', (e) => {
            console.log(chalk_1.default.red('Failed to start app: ' + e.message));
            this.setState({
                appRunning: false,
                proc: null,
                pid: null,
            });
        });
        proc.on('exit', (code, signal) => {
            if (code !== null) {
                console.log(code > 0
                    ? chalk_1.default.red('App exited with code ' + code + '.')
                    : chalk_1.default.green('App exited with code ' + code + '.'));
            }
            if (signal !== null) {
                console.log(signal !== 'SIGTERM'
                    ? chalk_1.default.red('App killed with signal ' + signal + '.')
                    : chalk_1.default.green('App killed with signal SIGTERM.'));
            }
            this.setState({
                appRunning: false,
                proc: null,
                pid: null,
            });
        });
        // pipe spawned output stream to current process's output streams
        proc.stdout.pipe(process.stdout);
        proc.stderr.pipe(process.stderr);
        if (proc.pid) {
            console.log(chalk_1.default.green('App started!'));
            this.setState({
                proc,
                pid: proc.pid.toString(),
                appRunning: true,
            });
        }
    }
    setState(state) {
        this.state = state;
    }
}
exports.AppController = AppController;
//# sourceMappingURL=processHandler.js.map