99 lines
2.8 KiB
TypeScript
99 lines
2.8 KiB
TypeScript
import { queryRegex } from "$sb/lib/query.ts";
|
|
import { ParseTree, renderToText } from "$sb/lib/tree.ts";
|
|
import { replaceAsync } from "$sb/lib/util.ts";
|
|
import { markdown, space } from "$sb/silverbullet-syscall/mod.ts";
|
|
import Handlebars from "handlebars";
|
|
|
|
import { replaceTemplateVars } from "../core/template.ts";
|
|
import { extractFrontmatter } from "$sb/lib/frontmatter.ts";
|
|
import { directiveRegex } from "./directives.ts";
|
|
import { updateDirectives } from "./command.ts";
|
|
import { registerHandlebarsHelpers } from "./util.ts";
|
|
|
|
const templateRegex = /\[\[([^\]]+)\]\]\s*(.*)\s*/;
|
|
|
|
export async function templateDirectiveRenderer(
|
|
directive: string,
|
|
pageName: string,
|
|
arg: string | ParseTree,
|
|
): Promise<string> {
|
|
if (typeof arg !== "string") {
|
|
throw new Error("Template directives must be a string");
|
|
}
|
|
const match = arg.match(templateRegex);
|
|
if (!match) {
|
|
throw new Error(`Invalid template directive: ${arg}`);
|
|
}
|
|
const template = match[1];
|
|
const args = match[2];
|
|
let parsedArgs = {};
|
|
if (args) {
|
|
try {
|
|
parsedArgs = JSON.parse(args);
|
|
} catch {
|
|
throw new Error(`Failed to parse template instantiation args: ${arg}`);
|
|
}
|
|
}
|
|
let templateText = "";
|
|
if (template.startsWith("http://") || template.startsWith("https://")) {
|
|
try {
|
|
const req = await fetch(template);
|
|
templateText = await req.text();
|
|
} catch (e: any) {
|
|
templateText = `ERROR: ${e.message}`;
|
|
}
|
|
} else {
|
|
templateText = await space.readPage(template);
|
|
}
|
|
let newBody = templateText;
|
|
// if it's a template injection (not a literal "include")
|
|
if (directive === "use") {
|
|
const tree = await markdown.parseMarkdown(templateText);
|
|
await extractFrontmatter(tree, ["$disableDirectives"]);
|
|
templateText = renderToText(tree);
|
|
registerHandlebarsHelpers();
|
|
const templateFn = Handlebars.compile(
|
|
replaceTemplateVars(templateText, pageName),
|
|
{ noEscape: true },
|
|
);
|
|
if (typeof parsedArgs !== "string") {
|
|
(parsedArgs as any).page = pageName;
|
|
}
|
|
newBody = templateFn(parsedArgs);
|
|
|
|
// Recursively render directives
|
|
newBody = await updateDirectives(pageName, newBody);
|
|
}
|
|
return newBody.trim();
|
|
}
|
|
|
|
export function cleanTemplateInstantiations(text: string): Promise<string> {
|
|
return replaceAsync(
|
|
text,
|
|
directiveRegex,
|
|
(
|
|
_fullMatch,
|
|
startInst,
|
|
type,
|
|
_args,
|
|
body,
|
|
endInst,
|
|
): Promise<string> => {
|
|
if (type === "use") {
|
|
body = body.replaceAll(
|
|
queryRegex,
|
|
(
|
|
_fullMatch: string,
|
|
_startQuery: string,
|
|
_query: string,
|
|
body: string,
|
|
) => {
|
|
return body.trim();
|
|
},
|
|
);
|
|
}
|
|
return Promise.resolve(`${startInst}${body}${endInst}`);
|
|
},
|
|
);
|
|
}
|