86 lines
2.7 KiB
TypeScript
86 lines
2.7 KiB
TypeScript
import { editor, markdown, YAML } from "$sb/syscalls.ts";
|
|
import { CodeWidgetContent } from "$sb/types.ts";
|
|
import { renderToText, traverseTree } from "$sb/lib/tree.ts";
|
|
|
|
const defaultHeaderThreshold = 0;
|
|
|
|
type Header = {
|
|
name: string;
|
|
pos: number;
|
|
level: number;
|
|
};
|
|
|
|
type TocConfig = {
|
|
minHeaders?: number;
|
|
header?: boolean;
|
|
};
|
|
|
|
export async function widget(
|
|
bodyText: string,
|
|
): Promise<CodeWidgetContent | null> {
|
|
let config: TocConfig = {};
|
|
if (bodyText.trim() !== "") {
|
|
config = await YAML.parse(bodyText);
|
|
}
|
|
|
|
const page = await editor.getCurrentPage();
|
|
const text = await editor.getText();
|
|
const tree = await markdown.parseMarkdown(text);
|
|
const headers: Header[] = [];
|
|
traverseTree(tree, (n) => {
|
|
if (n.type?.startsWith("ATXHeading")) {
|
|
headers.push({
|
|
name: n.children!.slice(1).map(renderToText).join("").trim(),
|
|
pos: n.from!,
|
|
level: +n.type[n.type.length - 1],
|
|
});
|
|
|
|
return true;
|
|
}
|
|
return false;
|
|
});
|
|
|
|
let headerThreshold = defaultHeaderThreshold;
|
|
if (config.minHeaders) {
|
|
headerThreshold = config.minHeaders;
|
|
}
|
|
if (headers.length < headerThreshold) {
|
|
// Not enough headers, not showing TOC
|
|
return null;
|
|
}
|
|
let headerText = "# Table of Contents\n";
|
|
if (config.header === false) {
|
|
headerText = "";
|
|
}
|
|
// console.log("Headers", headers);
|
|
// Adjust level down if only sub-headers are used
|
|
const minLevel = headers.reduce(
|
|
(min, header) => Math.min(min, header.level),
|
|
6,
|
|
);
|
|
const renderedMd = headerText +
|
|
headers.map((header) =>
|
|
`${
|
|
" ".repeat((header.level - minLevel) * 2)
|
|
}* [[${page}@${header.pos}|${header.name}]]`
|
|
).join("\n");
|
|
// console.log("Markdown", renderedMd);
|
|
return {
|
|
markdown: renderedMd,
|
|
buttons: [
|
|
{
|
|
description: "Edit",
|
|
svg:
|
|
`<svg xmlns="http://www.w3.org/2000/svg" width="15" height="15" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-edit"><path d="M11 4H4a2 2 0 0 0-2 2v14a2 2 0 0 0 2 2h14a2 2 0 0 0 2-2v-7"></path><path d="M18.5 2.5a2.121 2.121 0 0 1 3 3L12 15l-4 1 1-4 9.5-9.5z"></path></svg>`,
|
|
invokeFunction: "query.editButton",
|
|
},
|
|
{
|
|
description: "Reload",
|
|
svg:
|
|
`<svg xmlns="http://www.w3.org/2000/svg" width="15" height="15" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-refresh-cw"><polyline points="23 4 23 10 17 10"></polyline><polyline points="1 20 1 14 7 14"></polyline><path d="M3.51 9a9 9 0 0 1 14.85-3.36L23 10M1 14l4.64 4.36A9 9 0 0 0 20.49 15"></path></svg>`,
|
|
invokeFunction: "index.refreshWidgets",
|
|
},
|
|
],
|
|
};
|
|
}
|