mirror of
https://github.com/puppeteer/puppeteer
synced 2024-06-14 14:02:48 +00:00
819fa355f4
This patch: - converts all var's to let's throughout the codebase - enforces the let's over var's via the eslint rule
84 lines
2.3 KiB
JavaScript
84 lines
2.3 KiB
JavaScript
/**
|
|
* Copyright 2017 Google Inc. All rights reserved.
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
let http = require('http');
|
|
let await = require('./utilities').await;
|
|
|
|
class WebServer {
|
|
constructor() {
|
|
this._server = http.createServer();
|
|
this.objectName = 'WebServer';
|
|
this.listenOnPort = this.listen;
|
|
this.newRequest = function(req, res) { };
|
|
Object.defineProperty(this, 'port', {
|
|
get: () => {
|
|
if (!this._server.listening)
|
|
return '';
|
|
return this._server.address().port + '';
|
|
},
|
|
enumerable: true,
|
|
configurable: false
|
|
});
|
|
}
|
|
|
|
close() {
|
|
this._server.close();
|
|
}
|
|
|
|
/**
|
|
* @param {nubmer} port
|
|
* @return {boolean}
|
|
*/
|
|
listen(port, callback) {
|
|
if (this._server.listening)
|
|
return false;
|
|
this.newRequest = callback;
|
|
this._server.listen(port);
|
|
let errorPromise = new Promise(x => this._server.once('error', x));
|
|
let successPromise = new Promise(x => this._server.once('listening', x));
|
|
await(Promise.race([errorPromise, successPromise]));
|
|
if (!this._server.listening)
|
|
return false;
|
|
|
|
this._server.on('request', (req, res) => {
|
|
res.close = res.end.bind(res);
|
|
let headers = res.getHeaders();
|
|
res.headers = [];
|
|
for (let key in headers) {
|
|
res.headers.push({
|
|
name: key,
|
|
value: headers[key]
|
|
});
|
|
}
|
|
res.header = res.getHeader;
|
|
res.setHeaders = headers => {
|
|
for (let key in headers)
|
|
res.setHeader(key, headers[key]);
|
|
};
|
|
Object.defineProperty(res, 'statusCode', {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: res.statusCode
|
|
});
|
|
this.newRequest.call(null, req, res);
|
|
});
|
|
return true;
|
|
}
|
|
}
|
|
|
|
module.exports = WebServer;
|