63 lines
1.7 KiB
TypeScript
63 lines
1.7 KiB
TypeScript
import {
|
|
findNodeOfType,
|
|
type ParseTree,
|
|
renderToText,
|
|
replaceNodesMatchingAsync,
|
|
} from "./tree.ts";
|
|
|
|
import { cleanupJSON } from "@silverbulletmd/silverbullet/lib/json";
|
|
|
|
import { system, YAML } from "../syscalls.ts";
|
|
|
|
/**
|
|
* Extracts attributes from a tree, optionally cleaning them out of the tree.
|
|
* @param tree tree to extract attributes from
|
|
* @param clean whether or not to clean out the attributes from the tree
|
|
* @returns mapping from attribute name to attribute value
|
|
*/
|
|
export async function extractAttributes(
|
|
tags: string[],
|
|
tree: ParseTree,
|
|
clean: boolean,
|
|
): Promise<Record<string, any>> {
|
|
let attributes: Record<string, any> = {};
|
|
await replaceNodesMatchingAsync(tree, async (n) => {
|
|
if (n.type === "ListItem") {
|
|
// Find top-level only, no nested lists
|
|
return n;
|
|
}
|
|
if (n.type === "Attribute") {
|
|
const nameNode = findNodeOfType(n, "AttributeName");
|
|
const valueNode = findNodeOfType(n, "AttributeValue");
|
|
if (nameNode && valueNode) {
|
|
const name = nameNode.children![0].text!;
|
|
const val = valueNode.children![0].text!;
|
|
try {
|
|
attributes[name] = cleanupJSON(await YAML.parse(val));
|
|
} catch (e: any) {
|
|
console.error("Error parsing attribute value as YAML", val, e);
|
|
}
|
|
}
|
|
// Remove from tree
|
|
if (clean) {
|
|
return null;
|
|
} else {
|
|
return n;
|
|
}
|
|
}
|
|
// Go on...
|
|
return undefined;
|
|
});
|
|
const text = renderToText(tree);
|
|
const spaceScriptAttributes = await system.applyAttributeExtractors(
|
|
tags,
|
|
text,
|
|
tree,
|
|
);
|
|
attributes = {
|
|
...attributes,
|
|
...spaceScriptAttributes,
|
|
};
|
|
return attributes;
|
|
}
|