@jsdevtools/rehype-toc/lib/create-toc.js

A rehype plugin that adds a table of contents (TOC) to the page

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.createTOC = void 0;
const get_inner_text_1 = require("./get-inner-text");
const options_1 = require("./options");
/**
 * Creates a `<nav>` and/or `<ol>` element containing the table of contents.
 */
function createTOC(headings, options) {
    let list = createTocList(headings, options);
    if (options.nav) {
        return {
            type: "element",
            tagName: "nav",
            properties: {
                className: options.cssClasses.toc || undefined,
            },
            children: [list],
        };
    }
    else {
        list.properties.className =
            [options.cssClasses.toc, list.properties.className].filter(Boolean).join(" ") || undefined;
        return list;
    }
}
exports.createTOC = createTOC;
/**
 * Creates an `<ol>` element containing the table of contents.
 */
function createTocList(headings, options) {
    let levels = [];
    let currentLevel = {
        depth: 0,
        headingNumber: 0,
        list: undefined,
    };
    for (let heading of headings) {
        let headingNumber = parseInt(heading.tagName.slice(-1), 10);
        if (headingNumber > currentLevel.headingNumber) {
            // This is a higher heading number, so start a new level
            let depth = currentLevel.depth + 1;
            let level = {
                depth,
                headingNumber,
                list: createList(heading, depth, options),
            };
            // Add the new list to the previous level's list
            if (currentLevel.list) {
                let lastItem = currentLevel.list.children.slice(-1)[0];
                lastItem.children.push(level.list);
            }
            levels.push(level);
            currentLevel = level;
        }
        else {
            if (headingNumber < currentLevel.headingNumber) {
                // This is a lower heading number, so we need to go up to a previous level
                for (let i = levels.length - 2; i >= 0; i--) {
                    let level = levels[i];
                    if (level.headingNumber === headingNumber) {
                        // We found the previous level that matches this heading
                        levels = levels.slice(0, i + 1);
                        currentLevel = level;
                        break;
                    }
                }
                // If headings are in an incorrect order, then we may need to adjust the headingNumber
                currentLevel.headingNumber = Math.min(currentLevel.headingNumber, headingNumber);
            }
            // This heading is the same level as the previous heading,
            // so just add another <li> to the same <ol>
            let listItem = createListItem(heading, options);
            currentLevel.list.children.push(listItem);
        }
    }
    if (levels.length === 0) {
        return createList(undefined, 1, options);
    }
    else {
        return levels[0].list;
    }
}
/**
 * Creates an `<ol>` and `<li>` element for the given heading
 */
function createList(heading, depth, options) {
    let list = {
        type: "element",
        tagName: "ol",
        properties: {
            className: options_1.buildClass(options.cssClasses.list, depth),
        },
        children: [],
    };
    if (heading) {
        let listItem = createListItem(heading, options);
        list.children.push(listItem);
    }
    return list;
}
/**
 * Creates an `<li>` element for the given heading
 */
function createListItem(heading, options) {
    return {
        type: "element",
        tagName: "li",
        data: {
            hookArgs: [heading],
        },
        properties: {
            className: options_1.buildClass(options.cssClasses.listItem, heading.tagName),
        },
        children: [
            {
                type: "element",
                tagName: "a",
                properties: {
                    className: options_1.buildClass(options.cssClasses.link, heading.tagName),
                    href: `#${heading.properties.id || ""}`,
                },
                children: [
                    {
                        type: "text",
                        value: get_inner_text_1.getInnerText(heading),
                    }
                ]
            }
        ],
    };
}
//# sourceMappingURL=create-toc.js.map