ember-introjs
Version:
An Ember Component for intro.js
95 lines (74 loc) • 2.19 kB
JavaScript
;
function fastbootExpressMiddleware(distPath, options) {
let opts = options;
if (arguments.length === 1) {
if (typeof distPath === 'string') {
opts = { distPath: distPath };
} else {
opts = distPath;
}
}
opts = opts || {};
let log = opts.log !== false ? _log : function() {};
let fastboot = opts.fastboot;
if (!fastboot) {
let FastBoot = require('fastboot');
fastboot = new FastBoot({
distPath: opts.distPath,
resilient: opts.resilient
});
}
return function(req, res, next) {
let path = req.url;
fastboot.visit(path, { request: req, response: res })
.then(success, failure);
function success(result) {
let responseBody = opts.chunkedResponse ? result.chunks() : result.html();
responseBody.then(body => {
let headers = result.headers;
let statusMessage = result.error ? 'NOT OK ' : 'OK ';
for (var pair of headers.entries()) {
res.set(pair[0], pair[1]);
}
if (result.error) {
log("RESILIENT MODE CAUGHT:", result.error.stack);
next(result.error);
}
log(result.statusCode, statusMessage + path);
res.status(result.statusCode);
if (typeof body === 'string') {
res.send(body);
} else if (result.error) {
res.send(body[0]);
} else {
body.forEach(chunk => res.write(chunk));
res.end();
}
})
.catch(error => {
res.status(500);
next(error);
});
}
function failure(error) {
if (error.name === "UnrecognizedURLError") {
next();
} else {
res.status(500);
next(error);
}
}
};
}
let chalk;
function _log(statusCode, message, startTime) {
chalk = chalk || require('chalk');
let color = statusCode === 200 ? 'green' : 'red';
let now = new Date();
if (startTime) {
let diff = Date.now() - startTime;
message = message + chalk.blue(" " + diff + "ms");
}
console.log(chalk.blue(now.toISOString()) + " " + chalk[color](statusCode) + " " + message);
}
module.exports = fastbootExpressMiddleware;