58 lines
2.2 KiB
JavaScript
58 lines
2.2 KiB
JavaScript
"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
|