1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 | 'use strict';
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 | var debug = require('debug')('connect:dispatcher');
|
17 | var EventEmitter = require('events').EventEmitter;
|
18 | var finalhandler = require('finalhandler');
|
19 | var http = require('http');
|
20 | var merge = require('utils-merge');
|
21 | var parseUrl = require('parseurl');
|
22 |
|
23 |
|
24 |
|
25 |
|
26 |
|
27 |
|
28 | module.exports = createServer;
|
29 |
|
30 |
|
31 |
|
32 |
|
33 |
|
34 |
|
35 | var env = process.env.NODE_ENV || 'development';
|
36 | var proto = {};
|
37 |
|
38 |
|
39 | var defer = typeof setImmediate === 'function'
|
40 | ? setImmediate
|
41 | : function(fn){ process.nextTick(fn.bind.apply(fn, arguments)) }
|
42 |
|
43 |
|
44 |
|
45 |
|
46 |
|
47 |
|
48 |
|
49 |
|
50 | function createServer() {
|
51 | function app(req, res, next){ app.handle(req, res, next); }
|
52 | merge(app, proto);
|
53 | merge(app, EventEmitter.prototype);
|
54 | app.route = '/';
|
55 | app.stack = [];
|
56 | return app;
|
57 | }
|
58 |
|
59 |
|
60 |
|
61 |
|
62 |
|
63 |
|
64 |
|
65 |
|
66 |
|
67 |
|
68 |
|
69 |
|
70 |
|
71 |
|
72 |
|
73 |
|
74 |
|
75 |
|
76 | proto.use = function use(route, fn) {
|
77 | var handle = fn;
|
78 | var path = route;
|
79 |
|
80 |
|
81 | if (typeof route !== 'string') {
|
82 | handle = route;
|
83 | path = '/';
|
84 | }
|
85 |
|
86 |
|
87 | if (typeof handle.handle === 'function') {
|
88 | var server = handle;
|
89 | server.route = path;
|
90 | handle = function (req, res, next) {
|
91 | server.handle(req, res, next);
|
92 | };
|
93 | }
|
94 |
|
95 |
|
96 | if (handle instanceof http.Server) {
|
97 | handle = handle.listeners('request')[0];
|
98 | }
|
99 |
|
100 |
|
101 | if (path[path.length - 1] === '/') {
|
102 | path = path.slice(0, -1);
|
103 | }
|
104 |
|
105 |
|
106 | debug('use %s %s', path || '/', handle.name || 'anonymous');
|
107 | this.stack.push({ route: path, handle: handle });
|
108 |
|
109 | return this;
|
110 | };
|
111 |
|
112 |
|
113 |
|
114 |
|
115 |
|
116 |
|
117 |
|
118 |
|
119 | proto.handle = function handle(req, res, out) {
|
120 | var index = 0;
|
121 | var protohost = getProtohost(req.url) || '';
|
122 | var removed = '';
|
123 | var slashAdded = false;
|
124 | var stack = this.stack;
|
125 |
|
126 |
|
127 | var done = out || finalhandler(req, res, {
|
128 | env: env,
|
129 | onerror: logerror
|
130 | });
|
131 |
|
132 |
|
133 | req.originalUrl = req.originalUrl || req.url;
|
134 |
|
135 | function next(err) {
|
136 | if (slashAdded) {
|
137 | req.url = req.url.substr(1);
|
138 | slashAdded = false;
|
139 | }
|
140 |
|
141 | if (removed.length !== 0) {
|
142 | req.url = protohost + removed + req.url.substr(protohost.length);
|
143 | removed = '';
|
144 | }
|
145 |
|
146 |
|
147 | var layer = stack[index++];
|
148 |
|
149 |
|
150 | if (!layer) {
|
151 | defer(done, err);
|
152 | return;
|
153 | }
|
154 |
|
155 |
|
156 | var path = parseUrl(req).pathname || '/';
|
157 | var route = layer.route;
|
158 |
|
159 |
|
160 | if (path.toLowerCase().substr(0, route.length) !== route.toLowerCase()) {
|
161 | return next(err);
|
162 | }
|
163 |
|
164 |
|
165 | var c = path.length > route.length && path[route.length];
|
166 | if (c && c !== '/' && c !== '.') {
|
167 | return next(err);
|
168 | }
|
169 |
|
170 |
|
171 | if (route.length !== 0 && route !== '/') {
|
172 | removed = route;
|
173 | req.url = protohost + req.url.substr(protohost.length + removed.length);
|
174 |
|
175 |
|
176 | if (!protohost && req.url[0] !== '/') {
|
177 | req.url = '/' + req.url;
|
178 | slashAdded = true;
|
179 | }
|
180 | }
|
181 |
|
182 |
|
183 | call(layer.handle, route, err, req, res, next);
|
184 | }
|
185 |
|
186 | next();
|
187 | };
|
188 |
|
189 |
|
190 |
|
191 |
|
192 |
|
193 |
|
194 |
|
195 |
|
196 |
|
197 |
|
198 |
|
199 |
|
200 |
|
201 |
|
202 |
|
203 |
|
204 |
|
205 |
|
206 |
|
207 |
|
208 |
|
209 |
|
210 |
|
211 |
|
212 |
|
213 |
|
214 |
|
215 | proto.listen = function listen() {
|
216 | var server = http.createServer(this);
|
217 | return server.listen.apply(server, arguments);
|
218 | };
|
219 |
|
220 |
|
221 |
|
222 |
|
223 |
|
224 |
|
225 | function call(handle, route, err, req, res, next) {
|
226 | var arity = handle.length;
|
227 | var error = err;
|
228 | var hasError = Boolean(err);
|
229 |
|
230 | debug('%s %s : %s', handle.name || '<anonymous>', route, req.originalUrl);
|
231 |
|
232 | try {
|
233 | if (hasError && arity === 4) {
|
234 |
|
235 | handle(err, req, res, next);
|
236 | return;
|
237 | } else if (!hasError && arity < 4) {
|
238 |
|
239 | handle(req, res, next);
|
240 | return;
|
241 | }
|
242 | } catch (e) {
|
243 |
|
244 | error = e;
|
245 | }
|
246 |
|
247 |
|
248 | next(error);
|
249 | }
|
250 |
|
251 |
|
252 |
|
253 |
|
254 |
|
255 |
|
256 |
|
257 |
|
258 | function logerror(err) {
|
259 | if (env !== 'test') console.error(err.stack || err.toString());
|
260 | }
|
261 |
|
262 |
|
263 |
|
264 |
|
265 |
|
266 |
|
267 |
|
268 |
|
269 | function getProtohost(url) {
|
270 | if (url.length === 0 || url[0] === '/') {
|
271 | return undefined;
|
272 | }
|
273 |
|
274 | var fqdnIndex = url.indexOf('://')
|
275 |
|
276 | return fqdnIndex !== -1 && url.lastIndexOf('?', fqdnIndex) === -1
|
277 | ? url.substr(0, url.indexOf('/', 3 + fqdnIndex))
|
278 | : undefined;
|
279 | }
|