Add notFound function to set respondNotFound callback (#44)

README.md

@@ -218,6 +218,12 @@ var redbird = new require('redbird')({
 
 ##Reference
 
+[constructor](#redbird)
+[register](#register)
+[unregister](#unregister)
+[notFound](#notFound)
+[close](#close)
+
 <a name="redbird"/>
 ###Redbird(opts)
 
@@ -289,10 +295,34 @@ __Arguments__
 
 ---------------------------------------
 
+<a name="notFound"/>
+#### Redbird##notFound(callback)
+
+ Gives Redbird a callback function with two parameters, the HTTP request 
+ and response objects, respectively, which will be called when a proxy route is
+ not found. The default is
+```javascript
+    function(req, res){
+      res.statusCode = 404;
+      res.write('Not Found');
+      res.end();
+    };
+```
+.
+
+__Arguments__
+
+```javascript
+    src {Function(req, res)} The callback which will be called with the HTTP
+      request and response objects when a proxy route is not found.
+```
+
+---------------------------------------
+
 <a name="close"/>
 #### Redbird##close()
 
- Close the proxy stoping all the incoming connections.
+ Close the proxy, stopping all incoming connections.
 
 ---------------------------------------
 

lib/proxy.js

@@ -86,7 +86,7 @@ function ReverseProxy(opts){
           proxy.web(req, res, {target: target});
         }
       }else{
-        notFound(res);
+        respondNotFound(req, res);
       }
     });
 
@@ -138,7 +138,7 @@ function ReverseProxy(opts){
         if(target){
           proxy.web(req, res, {target: target});
         }else{
-          notFound(res);
+          respondNotFound(req, res);
         }
       });
 
@@ -174,7 +174,7 @@ function ReverseProxy(opts){
     if(target){
       proxy.ws(req, socket, head, {target: target});
     }else{
-      notFound(socket);
+      respondNotFound(req, socket);
     }
   }
 
@@ -417,11 +417,18 @@ function getSource(req){
   }
 */
 
-function notFound(res){
+var respondNotFound = function(req, res){
   res.statusCode = 404;
   res.write('Not Found');
   res.end();
-}
+};
+
+ReverseProxy.prototype.notFound = function(callback){
+  if(typeof callback == "function")
+    respondNotFound = callback;
+  else
+    throw Error('notFound callback is not a function');
+};
 
 //
 // Redirect to the HTTPS proxy