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
95
96
97
98
99
100
101
102
|
const http = require('http');
const Router = require('trouter');
const { parse } = require('querystring');
const parser = require('@polka/url');
function lead(x) {
return x.charCodeAt(0) === 47 ? x : ('/' + x);
}
function value(x) {
let y = x.indexOf('/', 1);
return y > 1 ? x.substring(0, y) : x;
}
function mutate(str, req) {
req.url = req.url.substring(str.length) || '/';
req.path = req.path.substring(str.length) || '/';
}
function onError(err, req, res, next) {
let code = (res.statusCode = err.code || err.status || 500);
res.end(err.length && err || err.message || http.STATUS_CODES[code]);
}
class Polka extends Router {
constructor(opts={}) {
super(opts);
this.apps = {};
this.wares = [];
this.bwares = {};
this.parse = parser;
this.server = opts.server;
this.handler = this.handler.bind(this);
this.onError = opts.onError || onError; // catch-all handler
this.onNoMatch = opts.onNoMatch || this.onError.bind(null, { code:404 });
}
add(method, pattern, ...fns) {
let base = lead(value(pattern));
if (this.apps[base] !== void 0) throw new Error(`Cannot mount ".${method.toLowerCase()}('${lead(pattern)}')" because a Polka application at ".use('${base}')" already exists! You should move this handler into your Polka application instead.`);
return super.add(method, pattern, ...fns);
}
use(base, ...fns) {
if (typeof base === 'function') {
this.wares = this.wares.concat(base, fns);
} else if (base === '/') {
this.wares = this.wares.concat(fns);
} else {
base = lead(base);
fns.forEach(fn => {
if (fn instanceof Polka) {
this.apps[base] = fn;
} else {
let arr = this.bwares[base] || [];
arr.length > 0 || arr.push((r, _, nxt) => (mutate(base, r),nxt()));
this.bwares[base] = arr.concat(fn);
}
});
}
return this; // chainable
}
listen() {
(this.server = this.server || http.createServer()).on('request', this.handler);
this.server.listen.apply(this.server, arguments);
return this;
}
handler(req, res, info) {
info = info || this.parse(req);
let fns=[], arr=this.wares, obj=this.find(req.method, info.pathname);
req.originalUrl = req.originalUrl || req.url;
let base = value(req.path = info.pathname);
if (this.bwares[base] !== void 0) {
arr = arr.concat(this.bwares[base]);
}
if (obj) {
fns = obj.handlers;
req.params = obj.params;
} else if (this.apps[base] !== void 0) {
mutate(base, req); info.pathname=req.path; //=> updates
fns.push(this.apps[base].handler.bind(null, req, res, info));
} else if (fns.length === 0) {
fns.push(this.onNoMatch);
}
// Grab addl values from `info`
req.search = info.search;
req.query = parse(info.query);
// Exit if only a single function
let i=0, len=arr.length, num=fns.length;
if (len === i && num === 1) return fns[0](req, res);
// Otherwise loop thru all middlware
let next = err => err ? this.onError(err, req, res, next) : loop();
let loop = _ => res.finished || (i < len) && arr[i++](req, res, next);
arr = arr.concat(fns);
len += num;
loop(); // init
}
}
module.exports = opts => new Polka(opts);
|
