mirror of
https://github.com/microsoft/playwright.git
synced 2025-06-26 21:40:17 +00:00
58 lines
2.1 KiB
TypeScript
58 lines
2.1 KiB
TypeScript
|
/**
|
||
|
|
* Copyright (c) Microsoft Corporation.
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
export class SigIntWatcher {
|
||
|
|
private _hadSignal: boolean = false;
|
||
|
|
private _sigintPromise: Promise<void>;
|
||
|
|
private _sigintHandler: () => void;
|
||
|
|
constructor() {
|
||
|
|
let sigintCallback: () => void;
|
||
|
|
this._sigintPromise = new Promise<void>(f => sigintCallback = f);
|
||
|
|
this._sigintHandler = () => {
|
||
|
|
// We remove the handler so that second Ctrl+C immediately kills the runner
|
||
|
|
// via the default sigint handler. This is handy in the case where our shutdown
|
||
|
|
// takes a lot of time or is buggy.
|
||
|
|
//
|
||
|
|
// When running through NPM we might get multiple SIGINT signals
|
||
|
|
// for a single Ctrl+C - this is an NPM bug present since at least NPM v6.
|
||
|
|
// https://github.com/npm/cli/issues/1591
|
||
|
|
// https://github.com/npm/cli/issues/2124
|
||
|
|
//
|
||
|
|
// Therefore, removing the handler too soon will just kill the process
|
||
|
|
// with default handler without printing the results.
|
||
|
|
// We work around this by giving NPM 1000ms to send us duplicate signals.
|
||
|
|
// The side effect is that slow shutdown or bug in our runner will force
|
||
|
|
// the user to hit Ctrl+C again after at least a second.
|
||
|
|
setTimeout(() => process.off('SIGINT', this._sigintHandler), 1000);
|
||
|
|
this._hadSignal = true;
|
||
|
|
sigintCallback();
|
||
|
|
};
|
||
|
|
process.on('SIGINT', this._sigintHandler);
|
||
|
|
}
|
||
|
|
|
||
|
|
promise(): Promise<void> {
|
||
|
|
return this._sigintPromise;
|
||
|
|
}
|
||
|
|
|
||
|
|
hadSignal(): boolean {
|
||
|
|
return this._hadSignal;
|
||
|
|
}
|
||
|
|
|
||
|
|
disarm() {
|
||
|
|
process.off('SIGINT', this._sigintHandler);
|
||
|
|
}
|
||
|
|
}
|