TheThing
/
koa-lite
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
koa-lite/lib/application.js

241 lines
4.7 KiB
JavaScript
Raw Normal View History

Initial commit
2013-08-17 07:15:57 +00:00
/**
* Module dependencies.
*/
var debug = require('debug')('koa:app');
var Emitter = require('events').EventEmitter;
var compose = require('koa-compose');
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
var context = require('./context');
add Context#cookies support. Closes #7 need to add signing support in there as well
2013-08-21 04:51:14 +00:00
var Cookies = require('cookies');
Initial commit
2013-08-17 07:15:57 +00:00
var Stream = require('stream');
var http = require('http');
var co = require('co');
/**
* Application prototype.
*/
var app = Application.prototype;
/**
* Expose `Application`.
*/
exports = module.exports = Application;
/**
* Initialize a new `Application`.
*
* @api public
*/
function Application() {
if (!(this instanceof Application)) return new Application;
this.env = process.env.NODE_ENV || 'development';
change app to emit "error" events instead of app.error(fn)
2013-08-28 03:24:04 +00:00
this.on('error', this.onerror.bind(this));
change .outputErrors to false for "test" env only good to know in prod ;)
2013-08-22 02:48:47 +00:00
this.outputErrors = 'test' != this.env;
Initial commit
2013-08-17 07:15:57 +00:00
this.subdomainOffset = 2;
this.poweredBy = true;
this.jsonSpaces = 2;
this.middleware = [];
add createContext()
2013-08-18 18:14:01 +00:00
this.Context = createContext();
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
this.context(context);
Initial commit
2013-08-17 07:15:57 +00:00
}
/**
* Inherit from `Emitter.prototype`.
*/
Application.prototype.__proto__ = Emitter.prototype;
/**
* Shorthand for:
*
* http.createServer(app.callback()).listen(...)
*
* @param {Mixed} ...
* @return {Server}
* @api public
*/
app.listen = function(){
var server = http.createServer(this.callback());
return server.listen.apply(server, arguments);
};
/**
* Use the given middleware `fn`.
*
* @param {Function} fn
* @return {Application} self
* @api public
*/
app.use = function(fn){
debug('use %s', fn.name || 'unnamed');
this.middleware.push(fn);
return this;
};
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
/**
docs
2013-08-18 18:15:22 +00:00
* Mixin `obj` to this app's context.
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
*
docs
2013-08-18 18:15:22 +00:00
* app.context({
* get something(){
* return 'hi';
* },
*
* set something(val){
* this._something = val;
* },
*
* render: function(){
* this.body = '<html></html>';
* }
* });
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
*
* @param {Object} obj
* @return {Application} self
* @api public
*/
app.context = function(obj){
add createContext()
2013-08-18 18:14:01 +00:00
var ctx = this.Context.prototype;
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
Object.getOwnPropertyNames(obj).forEach(function(name){
var descriptor = Object.getOwnPropertyDescriptor(obj, name);
Object.defineProperty(ctx, name, descriptor);
});
return this;
};
Initial commit
2013-08-17 07:15:57 +00:00
/**
* Return a request handler callback
* for node's native htto server.
*
* @return {Function}
* @api public
*/
app.callback = function(){
var mw = [respond].concat(this.middleware);
var fn = compose(mw)(downstream);
var self = this;
return function(req, res){
add createContext()
2013-08-18 18:14:01 +00:00
var ctx = new self.Context(self, req, res);
Initial commit
2013-08-17 07:15:57 +00:00
change socket error listener to just use length
2013-08-22 18:53:14 +00:00
if (!ctx.socket.listeners('error').length) {
change app to emit "error" events instead of app.error(fn)
2013-08-28 03:24:04 +00:00
ctx.socket.on('error', function(err){
self.emit('error', err);
});
Initial commit
2013-08-17 07:15:57 +00:00
}
co.call(ctx, function *(){
yield fn;
change app to emit "error" events instead of app.error(fn)
2013-08-28 03:24:04 +00:00
}, function(err){
if (err) ctx.onerror(err);
});
Initial commit
2013-08-17 07:15:57 +00:00
}
};
add err.status support. Closes #20
2013-08-22 02:47:56 +00:00
/**
* Default error handler.
*
* @param {Error} err
* @api private
*/
app.onerror = function(err){
if (this.outputErrors) console.error(err.stack);
};
Initial commit
2013-08-17 07:15:57 +00:00
/**
* Response middleware.
*/
function respond(next){
return function *(){
yield next;
change app to emit "error" events instead of app.error(fn)
2013-08-28 03:24:04 +00:00
var app = this.app;
Initial commit
2013-08-17 07:15:57 +00:00
var res = this.res;
var body = this.body;
var head = 'HEAD' == this.method;
var ignore = 204 == this.status || 304 == this.status;
// 404
if (null == body && 200 == this.status) {
this.status = 404;
}
// ignore body
if (ignore) return res.end();
// status body
if (null == body) {
this.set('Content-Type', 'text/plain');
body = http.STATUS_CODES[this.status];
}
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
Initial commit
2013-08-17 07:15:57 +00:00
// Buffer body
if (Buffer.isBuffer(body)) {
var ct = this.responseHeader['content-type'];
if (!ct) this.set('Content-Type', 'application/octet-stream');
this.set('Content-Length', body.length);
if (head) return res.end();
return res.end(body);
}
// string body
if ('string' == typeof body) {
var ct = this.responseHeader['content-type'];
if (!ct) this.set('Content-Type', 'text/plain; charset=utf-8');
this.set('Content-Length', Buffer.byteLength(body));
if (head) return res.end();
return res.end(body);
}
// Stream body
if (body instanceof Stream) {
fix .onerror bind for stream "error" handling
2013-08-22 02:50:32 +00:00
body.on('error', this.onerror.bind(this));
Initial commit
2013-08-17 07:15:57 +00:00
if (head) return res.end();
return body.pipe(res);
}
app.context: extend the context with your own properties
2013-08-17 22:36:51 +00:00
Initial commit
2013-08-17 07:15:57 +00:00
// body: json
body = JSON.stringify(body, null, this.app.jsonSpaces);
this.set('Content-Length', body.length);
this.set('Content-Type', 'application/json');
if (head) return res.end();
res.end(body);
}
}
/**
* Default downstream middleware.
add createContext()
2013-08-18 18:14:01 +00:00
*
* @api private
Initial commit
2013-08-17 07:15:57 +00:00
*/
function *downstream() {
this.status = 200;
if (this.app.poweredBy) this.set('X-Powered-By', 'koa');
}
add createContext()
2013-08-18 18:14:01 +00:00
/**
* Create a new `Context` constructor.
*
* @return {Function}
* @api private
*/
function createContext() {
return function Context(app, req, res){
this.app = app;
this.req = req;
this.res = res;
add Context#cookies support. Closes #7 need to add signing support in there as well
2013-08-21 04:51:14 +00:00
this.cookies = new Cookies(req, res);
add createContext()
2013-08-18 18:14:01 +00:00
}
}