1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2024-12-24 16:19:12 -05:00
denoland-deno/website/showdown_toc.js

143 lines
3.7 KiB
JavaScript

// Extension loading compatible with AMD and CommonJs
(function(extension) {
"use strict";
if (typeof showdown === "object") {
// global (browser or nodejs global)
showdown.extension("toc", extension());
} else if (typeof define === "function" && define.amd) {
// AMD
define("toc", extension());
} else if (typeof exports === "object") {
// Node, CommonJS-like
module.exports = extension();
} else {
// showdown was not found so we throw
throw Error("Could not find showdown library");
}
})(function() {
function getHeaderEntries(sourceHtml) {
if (typeof window === "undefined") {
return getHeaderEntriesInNodeJs(sourceHtml);
} else {
return getHeaderEntriesInBrowser(sourceHtml);
}
}
function getHeaderEntriesInNodeJs(sourceHtml) {
// eslint-disable-next-line @typescript-eslint/no-var-requires
const cheerio = require("cheerio");
const $ = cheerio.load(sourceHtml);
const headers = $("h1, h2, h3, h4, h5, h6");
const headerList = [];
for (let i = 0; i < headers.length; i++) {
const el = headers[i];
headerList.push(new TocEntry(el.name, $(el).text(), $(el).attr("id")));
}
return headerList;
}
function getHeaderEntriesInBrowser(sourceHtml) {
// Generate dummy element
const source = document.createElement("div");
source.innerHTML = sourceHtml;
// Find headers
const headers = source.querySelectorAll("h1, h2, h3, h4, h5, h6");
const headerList = [];
for (let i = 0; i < headers.length; i++) {
const el = headers[i];
headerList.push(new TocEntry(el.tagName, el.textContent, el.id));
}
return headerList;
}
function TocEntry(tagName, text, anchor) {
this.tagName = tagName;
this.text = text;
this.anchor = anchor;
this.children = [];
}
TocEntry.prototype.childrenToString = function() {
if (this.children.length === 0) {
return "";
}
let result = "<ul>\n";
for (let i = 0; i < this.children.length; i++) {
result += this.children[i].toString();
}
result += "</ul>\n";
return result;
};
TocEntry.prototype.toString = function() {
let result = "<li>";
if (this.text) {
result += '<a href="#' + this.anchor + '">' + this.text + "</a>";
}
result += this.childrenToString();
result += "</li>\n";
return result;
};
function sortHeader(tocEntries, level) {
level = level || 1;
const tagName = "H" + level;
const result = [];
function push(tocEntry) {
if (tocEntry !== undefined) {
if (tocEntry.children.length > 0) {
tocEntry.children = sortHeader(tocEntry.children, level + 1);
}
result.push(tocEntry);
}
}
let currentTocEntry;
for (let i = 0; i < tocEntries.length; i++) {
const tocEntry = tocEntries[i];
if (tocEntry.tagName.toUpperCase() !== tagName) {
if (currentTocEntry === undefined) {
currentTocEntry = new TocEntry();
}
currentTocEntry.children.push(tocEntry);
} else {
push(currentTocEntry);
currentTocEntry = tocEntry;
}
}
push(currentTocEntry);
return result;
}
return {
type: "output",
filter: function(sourceHtml) {
let headerList = getHeaderEntries(sourceHtml);
// No header found
if (headerList.length === 0) {
return sourceHtml;
}
// Sort header
headerList = sortHeader(headerList);
// Skip the title.
if (headerList.length == 1) {
headerList = headerList[0].children;
}
// Build result and replace all [toc]
const result =
'<div class="toc">\n<ul>\n' + headerList.join("") + "</ul>\n</div>\n";
return sourceHtml.replace(/\[toc\]/gi, result);
}
};
});