Added configuration options to the server feature

Author: jamesryanbell

config/chrome.json

@@ -18,5 +18,9 @@
 		"/weather"
 	],
 	"outputDir": "shots/bbc",
-	"maxConnections": 20
+	"maxConnections": 20,
+	"server": {
+		"start": true,
+		"port": 23423
+	}
 }

lib/gallery.js

@@ -1,15 +1,15 @@
 'use strict';
 
-var fs       = require('fs');
-var path     = require('path');
-var mustache = require('mustache');
-var helpers  = require('./helpers');
-var log      = require('./logger');
-var connect  = require('connect');
+var fs          = require('fs');
+var path        = require('path');
+var mustache    = require('mustache');
+var helpers     = require('./helpers');
+var log         = require('./logger');
+var connect     = require('connect');
 var serveStatic = require('serve-static');
-var open     = require('open');
+var open        = require('open');
 
-module.exports.generate = function(dirs, compareList, outputDir, project, cb) {
+module.exports.generate = function(dirs, compareList, outputDir, project, server, cb) {
 	compareList = compareList.sort(helpers.sortByProp('sort'));
 	var template = path.join(__dirname, '/../gallery.html');
 	var view = {
@@ -40,10 +40,13 @@ module.exports.generate = function(dirs, compareList, outputDir, project, cb) {
 			} else {
 				log.success('Gallery generated');
 				cb();
-				connect().use(serveStatic(outputDir)).listen('9090', function(){
-					open('http://localhost:9090/gallery.html');
-					log.success('Server started on port 9090');
-				});
+
+				if(server.start) {
+					connect().use(serveStatic(outputDir)).listen(server.port, function(){
+						open('http://localhost:' + server.port + '/gallery.html');
+						log.success('Server started on port ' + server.port);
+					});
+				}
 			}
 		});
 	});

lib/wraith.js

@@ -51,6 +51,9 @@ Wraith.prototype.config = function(config) {
 	self.sizes          = [];
 	self.dirs           = [];
 	self.quiet          = config.quiet || false;
+	self.server         = {};
+	self.server.start   = config.server.start || false;
+	self.server.port    = config.server.port || 9090;
 
 	if(typeof config.paths !== 'undefined' && config.paths.length > 0 ) {
 		self.urls = config.paths;
@@ -298,5 +301,5 @@ Wraith.prototype.compareScreenshots = function() {
 
 Wraith.prototype.generateGallery = function() {
 	var self = this;
-	gallery.generate(self.dirs, self.compareQueue, self.outputFolder, self.config.project, self.cb);
+	gallery.generate(self.dirs, self.compareQueue, self.outputFolder, self.config.project, self.server, self.cb);
 };