init commit

bin/pac/tools/.playwright/package/lib/utils/httpServer.js

@@ -0,0 +1,195 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.HttpServer = void 0;
+var _fs = _interopRequireDefault(require("fs"));
+var _path = _interopRequireDefault(require("path"));
+var _utilsBundle = require("../utilsBundle");
+var _debug = require("./debug");
+var _network = require("./network");
+var _manualPromise = require("./manualPromise");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+/**
+ * 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.
+ */
+
+class HttpServer {
+  constructor(address = '') {
+    this._server = void 0;
+    this._urlPrefix = void 0;
+    this._port = 0;
+    this._started = false;
+    this._routes = [];
+    this._urlPrefix = address;
+    this._server = (0, _network.createHttpServer)(this._onRequest.bind(this));
+  }
+  server() {
+    return this._server;
+  }
+  routePrefix(prefix, handler) {
+    this._routes.push({
+      prefix,
+      handler
+    });
+  }
+  routePath(path, handler) {
+    this._routes.push({
+      exact: path,
+      handler
+    });
+  }
+  port() {
+    return this._port;
+  }
+  async _tryStart(port, host) {
+    const errorPromise = new _manualPromise.ManualPromise();
+    const errorListener = error => errorPromise.reject(error);
+    this._server.on('error', errorListener);
+    try {
+      this._server.listen(port, host);
+      await Promise.race([new Promise(cb => this._server.once('listening', cb)), errorPromise]);
+    } finally {
+      this._server.removeListener('error', errorListener);
+    }
+  }
+  async start(options = {}) {
+    (0, _debug.assert)(!this._started, 'server already started');
+    this._started = true;
+    const host = options.host || 'localhost';
+    if (options.preferredPort) {
+      try {
+        await this._tryStart(options.preferredPort, host);
+      } catch (e) {
+        if (!e || !e.message || !e.message.includes('EADDRINUSE')) throw e;
+        await this._tryStart(undefined, host);
+      }
+    } else {
+      await this._tryStart(options.port, host);
+    }
+    const address = this._server.address();
+    (0, _debug.assert)(address, 'Could not bind server socket');
+    if (!this._urlPrefix) {
+      if (typeof address === 'string') {
+        this._urlPrefix = address;
+      } else {
+        this._port = address.port;
+        this._urlPrefix = `http://${host}:${address.port}`;
+      }
+    }
+    return this._urlPrefix;
+  }
+  async stop() {
+    await new Promise(cb => this._server.close(cb));
+  }
+  urlPrefix() {
+    return this._urlPrefix;
+  }
+  serveFile(request, response, absoluteFilePath, headers) {
+    try {
+      for (const [name, value] of Object.entries(headers || {})) response.setHeader(name, value);
+      if (request.headers.range) this._serveRangeFile(request, response, absoluteFilePath);else this._serveFile(response, absoluteFilePath);
+      return true;
+    } catch (e) {
+      return false;
+    }
+  }
+  _serveFile(response, absoluteFilePath) {
+    const content = _fs.default.readFileSync(absoluteFilePath);
+    response.statusCode = 200;
+    const contentType = _utilsBundle.mime.getType(_path.default.extname(absoluteFilePath)) || 'application/octet-stream';
+    response.setHeader('Content-Type', contentType);
+    response.setHeader('Content-Length', content.byteLength);
+    response.end(content);
+  }
+  _serveRangeFile(request, response, absoluteFilePath) {
+    const range = request.headers.range;
+    if (!range || !range.startsWith('bytes=') || range.includes(', ') || [...range].filter(char => char === '-').length !== 1) {
+      response.statusCode = 400;
+      return response.end('Bad request');
+    }
+
+    // Parse the range header: https://datatracker.ietf.org/doc/html/rfc7233#section-2.1
+    const [startStr, endStr] = range.replace(/bytes=/, '').split('-');
+
+    // Both start and end (when passing to fs.createReadStream) and the range header are inclusive and start counting at 0.
+    let start;
+    let end;
+    const size = _fs.default.statSync(absoluteFilePath).size;
+    if (startStr !== '' && endStr === '') {
+      // No end specified: use the whole file
+      start = +startStr;
+      end = size - 1;
+    } else if (startStr === '' && endStr !== '') {
+      // No start specified: calculate start manually
+      start = size - +endStr;
+      end = size - 1;
+    } else {
+      start = +startStr;
+      end = +endStr;
+    }
+
+    // Handle unavailable range request
+    if (Number.isNaN(start) || Number.isNaN(end) || start >= size || end >= size || start > end) {
+      // Return the 416 Range Not Satisfiable: https://datatracker.ietf.org/doc/html/rfc7233#section-4.4
+      response.writeHead(416, {
+        'Content-Range': `bytes */${size}`
+      });
+      return response.end();
+    }
+
+    // Sending Partial Content: https://datatracker.ietf.org/doc/html/rfc7233#section-4.1
+    response.writeHead(206, {
+      'Content-Range': `bytes ${start}-${end}/${size}`,
+      'Accept-Ranges': 'bytes',
+      'Content-Length': end - start + 1,
+      'Content-Type': _utilsBundle.mime.getType(_path.default.extname(absoluteFilePath))
+    });
+    const readable = _fs.default.createReadStream(absoluteFilePath, {
+      start,
+      end
+    });
+    readable.pipe(response);
+  }
+  _onRequest(request, response) {
+    response.setHeader('Access-Control-Allow-Origin', '*');
+    response.setHeader('Access-Control-Request-Method', '*');
+    response.setHeader('Access-Control-Allow-Methods', 'OPTIONS, GET');
+    if (request.headers.origin) response.setHeader('Access-Control-Allow-Headers', request.headers.origin);
+    if (request.method === 'OPTIONS') {
+      response.writeHead(200);
+      response.end();
+      return;
+    }
+    request.on('error', () => response.end());
+    try {
+      if (!request.url) {
+        response.end();
+        return;
+      }
+      const url = new URL('http://localhost' + request.url);
+      for (const route of this._routes) {
+        if (route.exact && url.pathname === route.exact && route.handler(request, response)) return;
+        if (route.prefix && url.pathname.startsWith(route.prefix) && route.handler(request, response)) return;
+      }
+      response.statusCode = 404;
+      response.end();
+    } catch (e) {
+      response.end();
+    }
+  }
+}
+exports.HttpServer = HttpServer;