Merge pull request http-party#8 from nodejitsu/404-pages

[api] On 404, attempt to serve file /404.html

Author: Marak

lib/http-server.js

@@ -9,6 +9,7 @@
 var colors = require('colors'),
     argv = require('optimist').argv,
     static = require('node-static'),
+    fs = require("fs");
     http = require('http');
 
 var HTTPServer = exports.HTTPServer = function (options) {
@@ -37,7 +38,19 @@ HTTPServer.prototype.start = function () {
   self.server = http.createServer(function(request, response) {
     request.on('end', function() {
       self.log('['.grey+'served'.yellow+'] '.grey + request.url);
-      return self.file.serve(request, response);
+      return self.file.serve(request, response, function (err, result) {
+        if (err && err.status == 404) {
+          // Catch case where there is no 404 file
+          fs.stat(self.root + "/404.html", function (err, stat) {
+            if (!err && stat.isFile()) {
+              self.file.serveFile("/404.html", err.status, {}, request, response);
+            } else {
+              response.writeHead(404);
+              response.end();
+            }
+          });
+        }
+      });
     });
   });
   self.server.listen(self.port);