79 lines
1.9 KiB
JavaScript
79 lines
1.9 KiB
JavaScript
'use strict';
|
|
|
|
const http2Server = require('http2');
|
|
const httpServer = require('http-server');
|
|
const {existsSync, statSync, createReadStream} = require('fs');
|
|
const {join} = require('path');
|
|
const argv = require('minimist')(process.argv.slice(2));
|
|
const mime = require('mime');
|
|
|
|
function sendFile(filename, response) {
|
|
response.setHeader('Content-Type', mime.lookup(filename));
|
|
response.writeHead(200);
|
|
const fileStream = createReadStream(filename);
|
|
fileStream.pipe(response);
|
|
fileStream.on('finish', response.end);
|
|
}
|
|
|
|
function createHTTP2Server(benchmark) {
|
|
const server = http2Server.createServer({}, (request, response) => {
|
|
const filename = join(
|
|
__dirname,
|
|
'benchmarks',
|
|
benchmark,
|
|
request.url
|
|
).replace(/\?.*/g, '');
|
|
|
|
if (existsSync(filename) && statSync(filename).isFile()) {
|
|
sendFile(filename, response);
|
|
} else {
|
|
const indexHtmlPath = join(filename, 'index.html');
|
|
|
|
if (existsSync(indexHtmlPath)) {
|
|
sendFile(indexHtmlPath, response);
|
|
} else {
|
|
response.writeHead(404);
|
|
response.end();
|
|
}
|
|
}
|
|
});
|
|
server.listen(8080);
|
|
return server;
|
|
}
|
|
|
|
function createHTTPServer() {
|
|
const server = httpServer.createServer({
|
|
root: join(__dirname, 'benchmarks'),
|
|
robots: true,
|
|
cache: 'no-store',
|
|
headers: {
|
|
'Access-Control-Allow-Origin': '*',
|
|
'Access-Control-Allow-Credentials': 'true',
|
|
},
|
|
});
|
|
server.listen(8080);
|
|
return server;
|
|
}
|
|
|
|
function serveBenchmark(benchmark, http2) {
|
|
if (http2) {
|
|
return createHTTP2Server(benchmark);
|
|
} else {
|
|
return createHTTPServer();
|
|
}
|
|
}
|
|
|
|
// if run directly via CLI
|
|
if (require.main === module) {
|
|
const benchmarkInput = argv._[0];
|
|
|
|
if (benchmarkInput) {
|
|
serveBenchmark(benchmarkInput);
|
|
} else {
|
|
console.error('Please specify a benchmark directory to serve!');
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
module.exports = serveBenchmark;
|