githubmirror/amphtml

build-system/common/ctrlcHandler.js

58 строк1 KB

Daniel Rozenberg

Bump @types/node to v18 (#39040)

25 май 2023, 23:34

25 май 2023, 23:343cddde0

const colors = require('kleur/colors');
const {exec, execScriptAsync} = require('./exec');
const {logLocalDev} = require('./logging');

const {cyan, green} = colors;

const killCmd = process.platform == 'win32' ? 'taskkill /f /pid' : 'kill -KILL';
const killSuffix = process.platform == 'win32' ? '>NUL' : '';

/**
 * Creates an async child process that handles Ctrl + C and immediately cancels
 * the ongoing `amp` task.
 *
 * @param {string} command
 * @param {number} pid
 * @return {number}
 */
exports.createCtrlcHandler = function (command, pid = process.pid) {
  logLocalDev(
    green('Running'),
    cyan(command) + green('. Press'),
    cyan('Ctrl + C'),
    green('to cancel...')
  );
  const killMessage =
    green('\nDetected ') +
    cyan('Ctrl + C') +
    green('. Canceling ') +
    cyan(command) +
    green('.');
  const listenerCmd = `
    #!/bin/sh
    ctrlcHandler() {
      echo -e "${killMessage}"
      ${killCmd} ${pid}
      exit 1
    }
    trap 'ctrlcHandler' INT
    read _ # Waits until the process is terminated
  `;
  const {pid: handlerPid} = execScriptAsync(listenerCmd, {
    'stdio': [null, process.stdout, process.stderr],
  });
  if (!handlerPid) {
    throw new Error(`Failed to create ctrlcHandler for ${command}`);
  }
  return handlerPid;
};

/**
 * Exits the Ctrl C handler process.
 *
 * @param {string|number} handlerProcess
 */
exports.exitCtrlcHandler = function (handlerProcess) {
  const exitCmd = killCmd + ' ' + handlerProcess + ' ' + killSuffix;
  exec(exitCmd);
};