1 files changed, 62 insertions, 0 deletions
diff --git a/packages/gitbook-html/src/dom.js b/packages/gitbook-html/src/dom.js
new file mode 100644
index 0000000..9c5e070
--- /dev/null
+++ b/packages/gitbook-html/src/dom.js
@@ -0,0 +1,62 @@
+const cheerio = require('cheerio');
+
+/**
+ * Parse an HTML string and return its content.
+ * @param {String}
+ * @return {cheerio.DOM}
+ */
+function parse(html) {
+    const $ = cheerio.load(html);
+    const $el = $('html, body').first();
+
+    return $el.length > 0 ? $el : $;
+}
+
+/**
+ * Return main element for a DOM.
+ * @param {cheerio.DOM}
+ * @return {cheerio.Node}
+ */
+function root($) {
+    const $el = $('html, body, > div').first();
+    return $el.length > 0 ? $el : $.root();
+}
+
+/**
+ * Return text node of an element.
+ * @param {cheerio.Node}
+ * @return {String}
+ */
+function textNode($el) {
+    return $el.children.reduce(
+        (text, e) => {
+            if (e.type == 'text') text += e.data;
+            return text;
+        },
+        ''
+    );
+}
+
+/**
+ * Cleanup a DOM by removing all useless divs.
+ * @param {cheerio.Node}
+ * @param {cheerio.DOM}
+ * @return {cheerio.Node}
+ */
+function cleanup($el, $) {
+    $el.find('div').each(function() {
+        const $div = $(this);
+        cleanup($div, $);
+
+        $div.replaceWith($div.html());
+    });
+
+    return $el;
+}
+
+module.exports = {
+    parse,
+    textNode,
+    root,
+    cleanup
+};