1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
/**
* Module dependencies.
*/
const debug = require('debug')('koa-mount');
const compose = require('koa-compose');
const assert = require('assert');
/**
* Expose `mount()`.
*/
module.exports = mount;
/**
* Mount `app` with `prefix`, `app`
* may be a Koa application or
* middleware function.
*
* @param {String|Application|Function} prefix, app, or function
* @param {Application|Function} [app or function]
* @return {Function}
* @api public
*/
function mount(prefix, app) {
if ('string' != typeof prefix) {
app = prefix;
prefix = '/';
}
assert('/' == prefix[0], 'mount path must begin with "/"');
// compose
const downstream = app.middleware
? compose(app.middleware)
: app;
// don't need to do mounting here
if ('/' == prefix) return downstream;
const trailingSlash = '/' == prefix.slice(-1);
const name = app.name || 'unnamed';
debug('mount %s %s', prefix, name);
return async function (ctx, upstream){
const prev = ctx.path;
const newPath = match(prev);
debug('mount %s %s -> %s', prefix, name, newPath);
if (!newPath) return await upstream();
ctx.mountPath = prefix;
ctx.path = newPath;
debug('enter %s -> %s', prev, ctx.path);
await downstream(ctx, async () => {
ctx.path = prev;
await upstream();
ctx.path = newPath;
});
debug('leave %s -> %s', prev, ctx.path);
ctx.path = prev;
};
/**
* Check if `prefix` satisfies a `path`.
* Returns the new path.
*
* match('/images/', '/lkajsldkjf') => false
* match('/images', '/images') => /
* match('/images/', '/images') => false
* match('/images/', '/images/asdf') => /asdf
*
* @param {String} prefix
* @param {String} path
* @return {String|Boolean}
* @api private
*/
function match(path) {
// does not match prefix at all
if (0 != path.indexOf(prefix)) return false;
const newPath = path.replace(prefix, '') || '/';
if (trailingSlash) return newPath;
// `/mount` does not match `/mountlkjalskjdf`
if ('/' != newPath[0]) return false;
return newPath;
}
}
|
