summaryrefslogtreecommitdiffstats
path: root/packages/gitbook-html/lib/dom.js
blob: 96d370ea8e5c2899c6f15791b6eed04ec2c6da68 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
'use strict';

var cheerio = require('cheerio');

/**
 * Parse an HTML string and return its content.
 * @param {String}
 * @return {cheerio.DOM}
 */
function parse(html) {
    var $ = cheerio.load(html);
    var $el = $('html, body').first();

    return $el.length > 0 ? $el : $;
}

/**
 * Return main element for a DOM.
 * @param {cheerio.DOM}
 * @return {cheerio.Node}
 */
function root($) {
    var $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(function (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 () {
        var $div = $(this);
        cleanup($div, $);

        $div.replaceWith($div.html());
    });

    return $el;
}

module.exports = {
    parse: parse,
    textNode: textNode,
    root: root,
    cleanup: cleanup
};
//# sourceMappingURL=dom.js.map