Merge pull request #34 from nfarina/handle-doctype

Handle "doctype" elements.

lib/xmldoc.js

@@ -191,6 +191,9 @@ function XmlDocument(xml) {
   if (!xml)
     throw new Error("No XML to parse!");
 
+  // Stores doctype (if defined)
+  this.doctype = "";
+
   // Expose the parser to the other delegates while the parser is running
   this.parser = sax.parser(true); // strict
   addParserEvents(this.parser);
@@ -199,7 +202,7 @@ function XmlDocument(xml) {
   // parsing; they will push and pop off the stack as we get deeper into the XML hierarchy.
   // It's safe to use a global because JS is single-threaded.
   delegates = [this];
-  
+
   this.parser.write(xml);
 
   // Remove the parser as it is no longer needed and should not be exposed to clients
@@ -218,6 +221,10 @@ XmlDocument.prototype._opentag = function(tag) {
     XmlElement.prototype._opentag.apply(this,arguments);
 };
 
+XmlDocument.prototype._doctype = function(doctype) {
+  this.doctype += doctype;
+}
+
 // file-scoped global stack of delegates
 var delegates = null;
 
@@ -231,6 +238,7 @@ function addParserEvents(parser) {
   parser.ontext = parser_text;
   parser.oncdata = parser_cdata;
   parser.oncomment = parser_comment;
+  parser.ondoctype = parser_doctype;
   parser.onerror = parser_error;
 }
 
@@ -240,6 +248,7 @@ function parser_closetag() { delegates[0] && delegates[0]._closetag.apply(delega
 function parser_text() { delegates[0] && delegates[0]._text.apply(delegates[0],arguments) }
 function parser_cdata() { delegates[0] && delegates[0]._cdata.apply(delegates[0],arguments) }
 function parser_comment() { delegates[0] && delegates[0]._comment.apply(delegates[0],arguments) }
+function parser_doctype() { delegates[0] && delegates[0]._doctype.apply(delegates[0],arguments) }
 function parser_error() { delegates[0] && delegates[0]._error.apply(delegates[0],arguments) }
 
 // a relatively standard extend method

test/basic.js

@@ -58,6 +58,21 @@ t.test('cdata handling', function (t) {
   t.end();
 })
 
+t.test('doctype handling', function (t) {
+
+  var docWithType = new XmlDocument('<!DOCTYPE HelloWorld><hello>world</hello>');
+  t.equal(docWithType.doctype, " HelloWorld");
+
+  var docWithoutType = new XmlDocument('<hello>world</hello>');
+  t.equal(docWithoutType.doctype, "");
+
+  t.throws(function() {
+    new XmlDocument('<hello><!DOCTYPE HelloWorld>world</hello>');
+  });
+
+  t.end();
+})
+
 t.test('comment handling', function (t) {
 
   var xmlString = '<hello><!-- World --></hello>';