Lloyd Brookes
8 years ago
12 changed files with 105 additions and 436 deletions
-
23example/stack/cache-control.js
-
4lib/cli-data.js
-
12lib/debug.js
-
309lib/default-stack.js
-
109lib/local-web-server.js
-
59lib/middleware.js
-
3package.json
-
18test/fixture/ajax.html
-
1test/fixture/file.txt
-
1test/fixture/one/file.txt
-
1test/fixture/spa/one.txt
-
1test/fixture/spa/two.txt
@ -0,0 +1,23 @@ |
|||
'use strict' |
|||
const LocalWebServer = require('../../') |
|||
const cacheControl = require('koa-cache-control') |
|||
const DefaultStack = require('local-web-server-default-stack') |
|||
|
|||
class CacheControl extends DefaultStack { |
|||
addAll () { |
|||
this.addLogging('dev') |
|||
.add({ |
|||
optionDefinitions: { |
|||
name: 'maxage', type: Number, |
|||
description: 'The maxage to set on each response.' |
|||
}, |
|||
middleware: function (options) { |
|||
return cacheControl({ maxAge: options.maxage }) |
|||
} |
|||
}) |
|||
.addStatic() |
|||
.addIndex() |
|||
} |
|||
} |
|||
|
|||
module.exports = CacheControl |
|||
@ -1,12 +0,0 @@ |
|||
'use strict' |
|||
module.exports = debug |
|||
|
|||
let level = 0 |
|||
|
|||
function debug () { |
|||
if (level) console.error.apply(console.error, arguments) |
|||
} |
|||
|
|||
debug.setLevel = function () { |
|||
level = 1 |
|||
} |
|||
@ -1,309 +0,0 @@ |
|||
'use strict' |
|||
const arrayify = require('array-back') |
|||
const path = require('path') |
|||
const url = require('url') |
|||
const debug = require('./debug') |
|||
const mw = require('./middleware') |
|||
const t = require('typical') |
|||
const compose = require('koa-compose') |
|||
const flatten = require('reduce-flatten') |
|||
const MiddlewareStack = require('local-web-server-stack') |
|||
const mockResponses = require('koa-mock-response') |
|||
|
|||
class DefaultStack extends MiddlewareStack { |
|||
addAll () { |
|||
this |
|||
.addCors() |
|||
.addJson() |
|||
.addRewrite() |
|||
.addBodyParser() |
|||
.addBlacklist() |
|||
.addCache() |
|||
.addMimeOverride() |
|||
.addCompression() |
|||
.addLogging() |
|||
.addMockResponses() |
|||
.addSpa() |
|||
.addStatic() |
|||
.addIndex() |
|||
return this |
|||
} |
|||
/** |
|||
* allow from any origin |
|||
*/ |
|||
addCors () { |
|||
this.push({ middleware: require('kcors') }) |
|||
return this |
|||
} |
|||
|
|||
/* pretty print JSON */ |
|||
addJson () { |
|||
this.push({ middleware: require('koa-json') }) |
|||
return this |
|||
} |
|||
|
|||
/* rewrite rules */ |
|||
addRewrite (rewriteRules) { |
|||
this.push({ |
|||
optionDefinitions: { |
|||
name: 'rewrite', alias: 'r', type: String, multiple: true, |
|||
typeLabel: '[underline]{expression} ...', |
|||
description: "A list of URL rewrite rules. For each rule, separate the 'from' and 'to' routes with '->'. Whitespace surrounded the routes is ignored. E.g. '/from -> /to'." |
|||
}, |
|||
middleware: function (cliOptions) { |
|||
const options = parseRewriteRules(arrayify(cliOptions.rewrite || rewriteRules)) |
|||
if (options.length) { |
|||
return options.map(route => { |
|||
if (route.to) { |
|||
/* `to` address is remote if the url specifies a host */ |
|||
if (url.parse(route.to).host) { |
|||
const _ = require('koa-route') |
|||
debug('proxy rewrite', `${route.from} -> ${route.to}`) |
|||
return _.all(route.from, mw.proxyRequest(route)) |
|||
} else { |
|||
const rewrite = require('koa-rewrite') |
|||
const rmw = rewrite(route.from, route.to) |
|||
rmw._name = 'rewrite' |
|||
return rmw |
|||
} |
|||
} |
|||
}) |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* must come after rewrite. |
|||
See https://github.com/nodejitsu/node-http-proxy/issues/180. */
|
|||
addBodyParser () { |
|||
this.push({ middleware: require('koa-bodyparser') }) |
|||
return this |
|||
} |
|||
|
|||
/* path blacklist */ |
|||
addBlacklist (forbidList) { |
|||
this.push({ |
|||
optionDefinitions: { |
|||
name: 'forbid', alias: 'b', type: String, |
|||
multiple: true, typeLabel: '[underline]{path} ...', |
|||
description: 'A list of forbidden routes.' |
|||
}, |
|||
middleware: function (cliOptions) { |
|||
forbidList = arrayify(cliOptions.forbid || forbidList) |
|||
if (forbidList.length) { |
|||
const pathToRegexp = require('path-to-regexp') |
|||
debug('forbid', forbidList.join(', ')) |
|||
return function blacklist (ctx, next) { |
|||
if (forbidList.some(expression => pathToRegexp(expression).test(ctx.path))) { |
|||
ctx.status = 403 |
|||
} else { |
|||
return next() |
|||
} |
|||
} |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* cache */ |
|||
addCache () { |
|||
this.push({ |
|||
optionDefinitions: { |
|||
name: 'no-cache', alias: 'n', type: Boolean, |
|||
description: 'Disable etag-based caching - forces loading from disk each request.' |
|||
}, |
|||
middleware: function (cliOptions) { |
|||
const noCache = cliOptions['no-cache'] |
|||
if (!noCache) { |
|||
return [ |
|||
require('koa-conditional-get')(), |
|||
require('koa-etag')() |
|||
] |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* mime-type overrides */ |
|||
addMimeOverride (mime) { |
|||
this.push({ |
|||
middleware: function (cliOptions) { |
|||
mime = cliOptions.mime || mime |
|||
if (mime) { |
|||
debug('mime override', JSON.stringify(mime)) |
|||
return mw.mime(mime) |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* compress response */ |
|||
addCompression (compress) { |
|||
this.push({ |
|||
optionDefinitions: { |
|||
name: 'compress', alias: 'c', type: Boolean, |
|||
description: 'Serve gzip-compressed resources, where applicable.' |
|||
}, |
|||
middleware: function (cliOptions) { |
|||
compress = t.isDefined(cliOptions.compress) |
|||
? cliOptions.compress |
|||
: compress |
|||
if (compress) { |
|||
debug('compression', 'enabled') |
|||
return require('koa-compress')() |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* Logging */ |
|||
addLogging (format, options) { |
|||
options = options || {} |
|||
this.push({ |
|||
optionDefinitions: { |
|||
name: 'log-format', |
|||
alias: 'f', |
|||
type: String, |
|||
description: "If a format is supplied an access log is written to stdout. If not, a dynamic statistics view is displayed. Use a preset ('none', 'dev','combined', 'short', 'tiny' or 'logstalgia') or supply a custom format (e.g. ':method -> :url')." |
|||
}, |
|||
middleware: function (cliOptions) { |
|||
format = cliOptions['log-format'] || format |
|||
|
|||
if (cliOptions.verbose && !format) { |
|||
format = 'none' |
|||
} |
|||
|
|||
if (format !== 'none') { |
|||
const morgan = require('koa-morgan') |
|||
|
|||
if (!format) { |
|||
const streamLogStats = require('stream-log-stats') |
|||
options.stream = streamLogStats({ refreshRate: 500 }) |
|||
return morgan('common', options) |
|||
} else if (format === 'logstalgia') { |
|||
morgan.token('date', () => { |
|||
var d = new Date() |
|||
return (`${d.getDate()}/${d.getUTCMonth()}/${d.getFullYear()}:${d.toTimeString()}`).replace('GMT', '').replace(' (BST)', '') |
|||
}) |
|||
return morgan('combined', options) |
|||
} else { |
|||
return morgan(format, options) |
|||
} |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* Mock Responses */ |
|||
addMockResponses (mocks) { |
|||
this.push({ |
|||
middleware: function (cliOptions) { |
|||
mocks = arrayify(cliOptions.mocks || mocks) |
|||
return mocks.map(mock => { |
|||
if (mock.module) { |
|||
const modulePath = path.resolve(path.join(cliOptions.directory, mock.module)) |
|||
mock.responses = require(modulePath) |
|||
} |
|||
|
|||
if (mock.responses) { |
|||
return mockResponses(mock.route, mock.responses) |
|||
} else if (mock.response) { |
|||
mock.target = { |
|||
request: mock.request, |
|||
response: mock.response |
|||
} |
|||
return mockResponses(mock.route, mock.target) |
|||
} |
|||
}) |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* for any URL not matched by static (e.g. `/search`), serve the SPA */ |
|||
addSpa (spa, assetTest) { |
|||
this.push({ |
|||
optionDefinitions: { |
|||
name: 'spa', alias: 's', type: String, typeLabel: '[underline]{file}', |
|||
description: 'Path to a Single Page App, e.g. app.html.' |
|||
}, |
|||
middleware: function (cliOptions) { |
|||
spa = cliOptions.spa || spa || 'index.html' |
|||
assetTest = new RegExp(cliOptions['spa-asset-test'] || assetTest || '\\.') |
|||
if (spa) { |
|||
const send = require('koa-send') |
|||
const _ = require('koa-route') |
|||
debug('SPA', spa) |
|||
return _.get('*', function spaMw (ctx, route, next) { |
|||
const root = path.resolve(cliOptions.directory || process.cwd()) |
|||
if (ctx.accepts('text/html') && !assetTest.test(route)) { |
|||
debug(`SPA request. Route: ${route}, isAsset: ${assetTest.test(route)}`) |
|||
return send(ctx, spa, { root: root }).then(next) |
|||
} else { |
|||
return send(ctx, route, { root: root }).then(next) |
|||
} |
|||
}) |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* serve static files */ |
|||
addStatic (root, options) { |
|||
this.push({ |
|||
optionDefinitions: { |
|||
name: 'directory', alias: 'd', type: String, typeLabel: '[underline]{path}', |
|||
description: 'Root directory, defaults to the current directory.' |
|||
}, |
|||
middleware: function (cliOptions) { |
|||
/* update global cliOptions */ |
|||
cliOptions.directory = cliOptions.directory || root || process.cwd() |
|||
options = Object.assign({ hidden: true }, options) |
|||
if (cliOptions.directory) { |
|||
const serve = require('koa-static') |
|||
return serve(cliOptions.directory, options) |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
|
|||
/* serve directory index */ |
|||
addIndex (path, options) { |
|||
this.push({ |
|||
middleware: function (cliOptions) { |
|||
path = cliOptions.directory || path || process.cwd() |
|||
options = Object.assign({ icons: true, hidden: true }, options) |
|||
if (path) { |
|||
const serveIndex = require('koa-serve-index') |
|||
return serveIndex(path, options) |
|||
} |
|||
} |
|||
}) |
|||
return this |
|||
} |
|||
} |
|||
|
|||
module.exports = DefaultStack |
|||
|
|||
function parseRewriteRules (rules) { |
|||
return rules && rules.map(rule => { |
|||
if (t.isString(rule)) { |
|||
const matches = rule.match(/(\S*)\s*->\s*(\S*)/) |
|||
if (!(matches && matches.length >= 3)) throw new Error('Invalid rule: ' + rule) |
|||
return { |
|||
from: matches[1], |
|||
to: matches[2] |
|||
} |
|||
} else { |
|||
return rule |
|||
} |
|||
}) |
|||
} |
|||
@ -1,59 +0,0 @@ |
|||
'use strict' |
|||
const path = require('path') |
|||
const url = require('url') |
|||
const arrayify = require('array-back') |
|||
const t = require('typical') |
|||
const pathToRegexp = require('path-to-regexp') |
|||
const debug = require('./debug') |
|||
|
|||
/** |
|||
* @module middleware |
|||
*/ |
|||
exports.proxyRequest = proxyRequest |
|||
exports.mime = mime |
|||
|
|||
function proxyRequest (route) { |
|||
const httpProxy = require('http-proxy') |
|||
const proxy = httpProxy.createProxyServer({ |
|||
changeOrigin: true, |
|||
secure: false |
|||
}) |
|||
|
|||
return function proxyMiddleware () { |
|||
const keys = [] |
|||
route.re = pathToRegexp(route.from, keys) |
|||
route.new = this.url.replace(route.re, route.to) |
|||
|
|||
keys.forEach((key, index) => { |
|||
const re = RegExp(`:${key.name}`, 'g') |
|||
route.new = route.new |
|||
.replace(re, arguments[index + 1] || '') |
|||
}) |
|||
|
|||
debug('proxy request', `from: ${this.path}, to: ${url.parse(route.new).href}`) |
|||
|
|||
return new Promise((resolve, reject) => { |
|||
proxy.once('error', err => { |
|||
err.message = `[PROXY] Error: ${err.message} Target: ${route.new}` |
|||
reject(err) |
|||
}) |
|||
proxy.once('proxyReq', function (proxyReq) { |
|||
proxyReq.path = url.parse(route.new).path |
|||
}) |
|||
proxy.once('close', resolve) |
|||
proxy.web(this.req, this.res, { target: route.new }) |
|||
}) |
|||
} |
|||
} |
|||
|
|||
function mime (mimeTypes) { |
|||
return function mime (ctx, next) { |
|||
return next().then(() => { |
|||
const reqPathExtension = path.extname(ctx.path).slice(1) |
|||
Object.keys(mimeTypes).forEach(mimeType => { |
|||
const extsToOverride = mimeTypes[mimeType] |
|||
if (extsToOverride.indexOf(reqPathExtension) > -1) ctx.type = mimeType |
|||
}) |
|||
}) |
|||
} |
|||
} |
|||
@ -1,18 +0,0 @@ |
|||
<!DOCTYPE html> |
|||
<head> |
|||
<title>Ajax test</title> |
|||
</head> |
|||
<body> |
|||
<h1>README</h1> |
|||
<h2>loaded in the "Ajax" style</h2> |
|||
<pre id="readme"></pre> |
|||
<script> |
|||
var $ = document.querySelector.bind(document); |
|||
var req = new XMLHttpRequest(); |
|||
req.open("get", "http://localhost:8000/big-file.txt", true); |
|||
req.onload = function(){ |
|||
$("#readme").textContent = this.responseText; |
|||
} |
|||
req.send() |
|||
</script> |
|||
</body> |
|||
@ -1 +0,0 @@ |
|||
test |
|||
@ -1 +0,0 @@ |
|||
one |
|||
@ -1 +0,0 @@ |
|||
one |
|||
@ -1 +0,0 @@ |
|||
two |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue