gruntjs · pconerly · Apr 10, 2013 · Apr 10, 2013 · Apr 10, 2013 · Apr 10, 2013
diff --git a/Gruntfile.js b/Gruntfile.js
@@ -42,6 +42,21 @@ module.exports = function(grunt) {
           }
         }
       },
+      headers: {
+        options: {
+          url: 'http://localhost:8000',
+          server: './test/fixtures/headers_server.js',
+          page: {
+            customHeaders: {
+              'X-CUSTOM': 'custom_header_567'
+            }
+          },
+          expected: 'custom_header_567',
+          test: function test(msg) {
+            test.actual = msg;
+          }
+        }
+      },
     },
   });
 
@@ -50,20 +65,33 @@ module.exports = function(grunt) {
     var options = this.options();
     var phantomjs = require('./lib/phantomjs').init(grunt);
 
+    var server = null;
+    if (options.server) { server = require(options.server); }
+
     // Do something.
     phantomjs.on('test', options.test);
-    phantomjs.on('done', phantomjs.halt);
+
+    phantomjs.on('done', function() {
+      phantomjs.halt();
+      if (options.server) { server = require(options.server); }
+    });
+
+    phantomjs.on('debug', function(msg) {
+        grunt.log.writeln('debug:' + msg);
+    });
 
     // Built-in error handlers.
     phantomjs.on('fail.load', function(url) {
       phantomjs.halt();
+      if (options.server) { server = require(options.server); }
       grunt.verbose.write('Running PhantomJS...').or.write('...');
       grunt.log.error();
       grunt.warn('PhantomJS unable to load "' + url + '" URI.');
     });
 
     phantomjs.on('fail.timeout', function() {
       phantomjs.halt();
+      if (options.server) { server = require(options.server); }
       grunt.log.writeln();
       grunt.warn('PhantomJS timed out.');
     });

diff --git a/README.md b/README.md
@@ -8,6 +8,14 @@ The best way to understand how this lib should be used is by looking at the [gru
 
 Also, in the case of the grunt-contrib-qunit plugin, it's important to know that the page being loaded into PhantomJS *doesn't* know it will be loaded into PhantomJS, and as such doesn't have any PhantomJS->Grunt code in it. That communication code, aka. the ["bridge"](https://github.com/gruntjs/grunt-contrib-qunit/blob/master/phantomjs/bridge.js), is dynamically [injected into the html page](https://github.com/gruntjs/grunt-contrib-qunit/blob/master/tasks/qunit.js#L136).
 
+### Options
+
+This lib has a few options that can be added to it:
+
+* `timeout`: phantomjs' timeout, in milliseconds.
+* `inject`: Javascript to inject into the page.
+* `page`: an object of options for the phantomjs `page` object.
+
 ## An inline example
 
 If a Grunt task looked something like this:

diff --git a/package.json b/package.json
@@ -35,7 +35,9 @@
   "devDependencies": {
     "grunt-contrib-jshint": "~0.1.1",
     "grunt": "~0.4.0",
-    "difflet": "~0.2.3"
+    "difflet": "~0.2.3",
+    "express": "~3.1.1",
+    "underscore": "~1.4.4"
   },
   "main": "lib/phantomjs"
 }
diff --git a/phantomjs/main.js b/phantomjs/main.js
@@ -56,6 +56,13 @@ var inject = function() {
   injected = true;
 };
 
+// Add options to the page.
+if (options.page) {
+  for (var prop in options.page) {
+    page[prop] = options.page[prop];
+  }
+}
+
 // Keep track if the client-side helper script already has been injected.
 page.onUrlChanged = function(newUrl) {
   injected = false;

diff --git a/test/fixtures/headers.html b/test/fixtures/headers.html
@@ -0,0 +1,27 @@
+<!doctype html>
+<html>
+<head>
+<script>
+
+// Send messages to the parent PhantomJS process via alert! Good times!!
+function sendMessage() {
+  var args = [].slice.call(arguments);
+  alert(JSON.stringify(args));
+}
+
+// Template the headers out
+var headers = {
+  <% _.each(_.pairs(headers), function(pair) { %> "<%= pair[0] %>":"<%= pair[1] %>", <% }); %>
+};
+
+for (var prop in headers) {
+  sendMessage('debug', prop + ":" + headers[prop]);
+}
+sendMessage('test', headers['x-custom']);
+sendMessage('done');
+
+</script>
+</head>
+<body>
+</body>
+</html>
diff --git a/test/fixtures/headers_server.js b/test/fixtures/headers_server.js
@@ -0,0 +1,18 @@
+var _ = require('underscore');
+var fs = require('fs');
+var express = require('express'); 
+var port = 8000;
+var site = express();
+
+site.get('*', function(req, res) {
+  fs.readFile('./test/fixtures/headers.html', 'utf8', function (err, data) {  
+    if (err) throw err;
+    var h_tmpl = _.template(data);
+    res.write(h_tmpl({headers: req.headers}));
+    res.end();
+  });
+});
+
+site.listen(port);
+
+console.log("Listening on port " + port);