Allow routes to be defined before calling app.listen()

examples/simple.js

@@ -1,22 +1,16 @@
 #!/usr/bin/env node
 
 var express = require('express');
-
 var expressWs = require('..')
 
-
 var app = expressWs(express());
 
-app.listen(3000)
-
-
 app.use(function (req, res, next) {
   console.log('middleware');
   req.testing = 'testing';
   return next();
 });
 
-
 app.get('/', function(req, res, next){
   console.log('get route', req.testing);
   res.end();
@@ -28,3 +22,5 @@ app.ws('/', function(ws, req) {
   });
   console.log('socket', req.testing);
 });
+
+app.listen(3000)

index.js

@@ -1,5 +1,6 @@
 var path = require('path');
-var ServerResponse = require('http').ServerResponse;
+var http = require('http');
+var ServerResponse = http.ServerResponse;
 var WebSocketServer = require('ws').Server;
 
 /**
@@ -8,13 +9,11 @@ var WebSocketServer = require('ws').Server;
  */
 module.exports = function (app, server) {
   if(!server) {
-    var app_listen = app.listen
+    var server = http.createServer(app);
 
     app.listen = function()
     {
-      server = app_listen.apply(app, arguments)
-
-      return server
+      return server.listen.apply(server, arguments)
     }
   }