API to shut down the server

In some CI environment it might be useful to shut down the server with a network request.

This is achieved with the request 'node-coverage-please-exit'

Fix #22

README.md

@@ -99,7 +99,7 @@ It's also possible to specify a report name from the `submit` function
 * `--session`, `--no-session` Enable or disable session storage for information not strictly needed by the browser. By default it's enabled. Disabling this means that more code is sent to and from the client.
 * `-i` or `--ignore` Ignore file or folder. This file/folder won't be instrumented. Path is relative to document root.
 * `--proxy` Proxy mode. You can use node-coverage to instrument files on a differnt host.
-* `--exit-on-submit` The default behavior is to keep the server running in order to collect multiple reports. By enabling this options the server will automatically shut down when a coverage report is received. This is useful for some continous integration environment.
+* `--exit-on-submit` The default behavior is to keep the server running in order to collect multiple reports. By enabling this options the server will automatically shut down when a coverage report is received. This is useful for some continous integration environment. If you want to collect more coverage reports but still be able to shut down the server when tests are done you can submit a request to '/node-coverage-please-exit'.
 * `-v` or `--verbose` Enable more verbose logging information. Default `false`.
 
 By default function coverage is disabled, to enable it you can run

lib/server/instrumentation.js

@@ -28,11 +28,9 @@ exports.start = function (docRoot, port, adminRoot, adminPort, coverageOptions,
 		}
 
 		var headers = {
-			"Content-Type" : "text/javascript"
+			"Content-Type": "text/javascript",
+			"Connection": "close"
 		};
-		if (coverageOptions["exit-on-submit"]) {
-			headers["Connection"] = "close";
-		}
 
 		if (instrumentedCode) {
 			res.send(instrumentedCode.clientCode, headers);
@@ -58,14 +56,17 @@ exports.start = function (docRoot, port, adminRoot, adminPort, coverageOptions,
 		}
 	});
 
+	var whileStoringCoverage = 0;
 	app.post("/node-coverage-store", function (req, res) {
+		whileStoringCoverage += 1;
 		if (req.headers["content-type"] === "application/x-www-form-urlencoded") {
 			// Using a real form submit the coverage reports happens to be a string
 			req.body = JSON.parse(req.body.coverage);
 		}
 		mergeCodeFromMemory(req.body.code, sourceCodeCache.highlight);
 
 		common.saveCoverage(req.body, adminRoot, function (error) {
+			whileStoringCoverage -= 1;
 			if (error) {
 				res.send(error, 500);
 			} else {
@@ -80,6 +81,20 @@ exports.start = function (docRoot, port, adminRoot, adminPort, coverageOptions,
 		});
 	});
 
+	app.all("/node-coverage-please-exit", function (req, res) {
+		function exitWhenDone () {
+			if (whileStoringCoverage) {
+				setTimeout(exitWhenDone, 100);
+			} else {
+				res.send(200, {"Connection": "close"});
+				process.nextTick(function () {
+					server.close(onClose);
+				});
+			}
+		}
+		exitWhenDone();
+	});
+
 	app.post("/*", function (req, res, next) {
 		res.sendfile(docRoot + req.path);
 	});