Add support for HAST@2.0.0 doctype node

wooorm · wooorm · commit 15463b524160 · 2016-07-25T22:32:13.000+02:00
diff --git a/lib/doctype.js b/lib/doctype.js
@@ -0,0 +1,50 @@
+/**
+ * @author Titus Wormer
+ * @copyright 2016 Titus Wormer
+ * @license MIT
+ * @module hast:to-html:doctype
+ * @fileoverview Stringify doctypes.
+ */
+
+'use strict';
+
+/* eslint-env commonjs */
+
+/* Expose. */
+module.exports = doctype;
+
+/**
+ * Stringify a doctype `node`.
+ *
+ * @param {Object} ctx - Stringify configuration.
+ * @param {HastDoctype} node - Node.
+ * @return {string} - Stringified `node`.
+ */
+function doctype(ctx, node) {
+  var pub = node.public;
+  var sys = node.system;
+  var val = '<!DOCTYPE';
+
+  if (!node.name) {
+    return val + '>';
+  }
+
+  val += ' ' + node.name;
+
+  if (pub != null) {
+    val += ' PUBLIC ' + smart(pub);
+  } else if (sys != null) {
+    val += ' SYSTEM';
+  }
+
+  if (sys != null) {
+    val += ' ' + smart(sys);
+  }
+
+  return val + '>';
+}
+
+function smart(value) {
+  var quote = value.indexOf('"') === -1 ? '"' : '\'';
+  return quote + value + quote;
+}
diff --git a/lib/one.js b/lib/one.js
@@ -22,11 +22,14 @@ var handlers = {};
 handlers.root = require('./all');
 handlers.text = require('./text');
 handlers.element = require('./element');
-handlers.directive = require('./directive');
+handlers.doctype = require('./doctype');
 handlers.comment = require('./comment');
-handlers.characterData = require('./character-data');
 handlers.raw = require('./raw');
 
+/* Deprecated */
+handlers.directive = require('./directive');
+handlers.characterData = require('./character-data');
+
 /**
  * Stringify `node`.
  *
diff --git a/test/doctype.js b/test/doctype.js
@@ -0,0 +1,70 @@
+/**
+ * @author Titus Wormer
+ * @copyright 2016 Titus Wormer
+ * @license MIT
+ * @module hast-util-to-html
+ * @fileoverview Test suite for `hast-util-to-html`.
+ */
+
+'use strict';
+
+/* eslint-env node */
+
+/* Dependencies. */
+var test = require('tape');
+var u = require('unist-builder');
+var to = require('..');
+
+/* Tests. */
+test('`text`', function (t) {
+  t.deepEqual(
+    to(u('doctype')),
+    '<!DOCTYPE>',
+    'should stringify doctypes without `name`'
+  );
+
+  t.deepEqual(
+    to(u('doctype', {name: 'html'})),
+    '<!DOCTYPE html>',
+    'should stringify doctypes with `name`'
+  );
+
+  t.deepEqual(
+    to(u('doctype', {
+      name: 'html',
+      public: '-//W3C//DTD XHTML 1.0 Transitional//EN'
+    })),
+    '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN">',
+    'should stringify doctypes with a public identifier'
+  );
+
+  t.deepEqual(
+    to(u('doctype', {
+      name: 'html',
+      system: 'about:legacy-compat'
+    })),
+    '<!DOCTYPE html SYSTEM "about:legacy-compat">',
+    'should stringify doctypes with a system identifier'
+  );
+
+  t.deepEqual(
+    to(u('doctype', {
+      name: 'html',
+      public: '-//W3C//DTD HTML 4.01//',
+      system: 'http://www.w3.org/TR/html4/strict.dtd'
+    })),
+    '<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01//" "http://www.w3.org/TR/html4/strict.dtd">',
+    'should stringify doctypes with both identifiers'
+  );
+
+  t.deepEqual(
+    to(u('doctype', {
+      name: 'html',
+      system: 'taco"'
+    })),
+    '<!DOCTYPE html SYSTEM \'taco"\'>',
+    'should quote smartly'
+  );
+
+  t.end();
+});
diff --git a/test/index.js b/test/index.js
@@ -14,12 +14,15 @@
 require('./core');
 require('./root');
 require('./comment');
-require('./character-data');
-require('./directive');
+require('./doctype');
 require('./text');
 require('./raw');
 require('./element');
 require('./attribute');
 require('./omission');
 require('./omission-opening');
 require('./omission-closing');
+
+/* Legacy. */
+require('./character-data');
+require('./directive');
