hexo-util/dist/toc_obj.js

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
const htmlparser2_1 = require("htmlparser2");
const escape_html_1 = __importDefault(require("./escape_html"));
const nonWord = /^\s*[^a-zA-Z0-9]\s*$/;
const parseHtml = (html) => {
    const handler = new htmlparser2_1.DomHandler(null, {});
    new htmlparser2_1.Parser(handler, {}).end(html);
    return handler.dom;
};
const getId = ({ attribs = {}, parent }) => {
    return attribs.id || (!parent ? '' : getId(parent));
};
/**
 * Identify a heading that to be unnumbered or not.
 */
const isUnnumbered = ({ attribs = {} }) => {
    return attribs['data-toc-unnumbered'] === 'true';
};
function tocObj(str, options = {}) {
    const { min_depth, max_depth } = Object.assign({
        min_depth: 1,
        max_depth: 6
    }, options);
    const headingsSelector = ['h1', 'h2', 'h3', 'h4', 'h5', 'h6'].slice(min_depth - 1, max_depth);
    const headings = htmlparser2_1.DomUtils.find(element => 'tagName' in element && headingsSelector.includes(element.tagName), parseHtml(str), true, Infinity);
    const headingsLen = headings.length;
    if (!headingsLen)
        return [];
    const result = [];
    for (let i = 0; i < headingsLen; i++) {
        const el = headings[i];
        const level = +el.name[1];
        const id = getId(el);
        const unnumbered = isUnnumbered(el);
        let text = '';
        for (const element of el.children) {
            const elText = htmlparser2_1.DomUtils.textContent(element);
            // Skip permalink symbol wrapped in <a>
            // permalink is a single non-word character, word = [a-Z0-9]
            // permalink may be wrapped in whitespace(s)
            if (!('name' in element) || element.name !== 'a' || !nonWord.test(elText)) {
                text += (0, escape_html_1.default)(elText);
            }
        }
        if (!text)
            text = (0, escape_html_1.default)(htmlparser2_1.DomUtils.textContent(el));
        const res = { text, id, level };
        if (unnumbered)
            res.unnumbered = true;
        result.push(res);
    }
    return result;
}
module.exports = tocObj;
//# sourceMappingURL=toc_obj.js.map