7385b31f13
When the client only closes the input pipe, we are still sending protocol messages over the output pipe. This could probably lead to some errors, e.g. write buffer being full.
71 lines
2.6 KiB
TypeScript
71 lines
2.6 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.
|
|
*/
|
|
|
|
import * as fs from 'fs';
|
|
import * as util from 'util';
|
|
import { installDebugController } from './debug/debugController';
|
|
import { DispatcherConnection } from './dispatchers/dispatcher';
|
|
import { PlaywrightDispatcher } from './dispatchers/playwrightDispatcher';
|
|
import { installBrowsersWithProgressBar } from './install/installer';
|
|
import { Transport } from './protocol/transport';
|
|
import { Playwright } from './server/playwright';
|
|
import { gracefullyCloseAll } from './server/processLauncher';
|
|
import { installHarTracer } from './trace/harTracer';
|
|
import { installTracer } from './trace/tracer';
|
|
|
|
|
|
const readFileAsync = util.promisify(fs.readFile);
|
|
const writeFileAsync = util.promisify(fs.writeFile);
|
|
|
|
export async function copyPrintDeps(destination: string) {
|
|
const content = await readFileAsync(require.resolve('../bin/PrintDeps.exe'));
|
|
await writeFileAsync(destination, content);
|
|
}
|
|
|
|
export async function installWithProgressBar(location: string) {
|
|
await installBrowsersWithProgressBar(location);
|
|
}
|
|
|
|
export async function apiJson(): Promise<string> {
|
|
return (await readFileAsync(require.resolve('../docs/api.json'))).toString();
|
|
}
|
|
|
|
export function runServer() {
|
|
installDebugController();
|
|
installTracer();
|
|
installHarTracer();
|
|
|
|
const dispatcherConnection = new DispatcherConnection();
|
|
const transport = new Transport(process.stdout, process.stdin);
|
|
transport.onmessage = message => dispatcherConnection.dispatch(JSON.parse(message));
|
|
dispatcherConnection.onmessage = message => transport.send(JSON.stringify(message));
|
|
transport.onclose = async () => {
|
|
// Drop any messages during shutdown on the floor.
|
|
dispatcherConnection.onmessage = () => {};
|
|
// Force exit after 30 seconds.
|
|
setTimeout(() => process.exit(0), 30000);
|
|
// Meanwhile, try to gracefully close all browsers.
|
|
await gracefullyCloseAll();
|
|
process.exit(0);
|
|
};
|
|
|
|
const playwright = new Playwright(__dirname, require('../browsers.json')['browsers']);
|
|
new PlaywrightDispatcher(dispatcherConnection.rootDispatcher(), playwright);
|
|
}
|
|
|
|
if (process.argv[2] === 'serve')
|
|
runServer();
|