Inject OKServer views as dependencies

1 files changed, 11 insertions, 6 deletions

diff --git a/app/node_modules/okserver/index.js b/app/node_modules/okserver/index.js
index adef189..230a53c 100644
--- a/app/node_modules/okserver/index.js
+++ b/app/node_modules/okserver/index.js
@@ -1,16 +1,21 @@
+var specificity = require('route-order')();
 var express = require('express');
 
 function OKServer(options) {
   if (!(this instanceof OKServer)) return new OKServer(options);
   options = options || {};
-  this._app = express();
+  if (!options.views)
+    throw new Error('No views provided to OKServer!');
+  var views = options.views;
+  var app = this._app = express();
+  // Add views
+  Object.keys(views)
+    .sort(specificity)
+    .forEach(function(route) {
+      app.use(route, views[route].middleware());
+    });
 }
 
-OKServer.prototype.addView = function addView(route, view) {
-  this._app.use(route, view.middleware());
-  return this;
-};
-
 OKServer.prototype.listen = function listen(port) {
   this._app.listen(port || 1337);
   return this;