draft for koa supports: needs to remove dup, fix results response

goenning · goenning · commit f00c995c6ecc · 2016-05-24T12:39:17.000-03:00
diff --git a/examples/koa.js b/examples/koa.js
@@ -0,0 +1,11 @@
+var koa = require('koa');
+var app = koa();
+var miniprofiler = require('../lib/miniprofiler.js')
+
+app.use(miniprofiler.koa(() => {return false;}));
+
+app.use(function *(){
+  this.body = '<html><body>hello' + this.req.miniprofiler.include() + '</body></html>';
+});
+
+app.listen(8080);
diff --git a/lib/middlewares/koa.js b/lib/middlewares/koa.js
@@ -0,0 +1,35 @@
+module.exports = function(f, profiler) {
+	if(!f) f = function() { return true; };
+	return function *(next) {
+		profiler.configure();
+		var enabled = f(this.req, this.res);
+
+		if (this.path.startsWith(profiler.resourcePath)) {
+			if (!enabled) {
+        this.type = 'text';
+        this.status = 404;
+        this.body = 'MiniProfiler is disabled';
+        return;
+			}
+
+			var sp = this.path.split('/');
+			var reqPath = sp[sp.length - 1];
+			if(reqPath == 'results')
+				profiler.results(this.req, this.res);
+			else
+				profiler.static(reqPath, this.res);
+			return;
+		}
+		var id = profiler.startProfiling(this.req, enabled);
+
+		if (enabled) {
+      var request = this.req
+			this.res.on('finish', function() {
+				profiler.stopProfiling(request);
+			});
+			this.res.setHeader('X-MiniProfiler-Ids', `["${id}"]`);
+		}
+
+    yield next;
+	};
+};
diff --git a/lib/miniprofiler.js b/lib/miniprofiler.js
@@ -37,6 +37,9 @@ exports.configure = configure;
 exports.express = function(f) {
 	return require('./middlewares/express.js')(f, shared());
 };
+exports.koa = function(f) {
+	return require('./middlewares/koa.js')(f, shared());
+};
 
 exports.for = {
 	pg: require('./providers/miniprofiler.pg.js'),
diff --git a/package.json b/package.json
@@ -30,6 +30,7 @@
     "eslint": "^2.10.2",
     "express": "^4.13.4",
     "istanbul": "^0.4.3",
+    "koa": "^1.2.0",
     "mocha": "^2.4.5",
     "pg": "^4.5.5",
     "pug": "^2.0.0-alpha7",
