87 lines
2.4 KiB
TypeScript
87 lines
2.4 KiB
TypeScript
import YAML from "js-yaml";
|
|
import {
|
|
collectNodesOfType,
|
|
findNodeOfType,
|
|
type ParseTree,
|
|
} from "@silverbulletmd/silverbullet/lib/tree";
|
|
import type { TagObject } from "./tags.ts";
|
|
import type { FrontMatter } from "./frontmatter.ts";
|
|
import { updateITags } from "./tags.ts";
|
|
import type {
|
|
ObjectValue,
|
|
PageMeta,
|
|
} from "@silverbulletmd/silverbullet/type/index";
|
|
|
|
type DataObject = ObjectValue<
|
|
{
|
|
pos: number;
|
|
page: string;
|
|
} & Record<string, any>
|
|
>;
|
|
|
|
export function indexData(
|
|
pageMeta: PageMeta,
|
|
frontmatter: FrontMatter,
|
|
tree: ParseTree,
|
|
) {
|
|
const separator = "---";
|
|
const dataObjects: ObjectValue<DataObject>[] = [];
|
|
const tagObjects: Map<string, ObjectValue<TagObject>> = new Map();
|
|
|
|
collectNodesOfType(tree, "FencedCode").map((t) => {
|
|
const codeInfoNode = findNodeOfType(t, "CodeInfo");
|
|
if (!codeInfoNode) {
|
|
return;
|
|
}
|
|
const fenceType = codeInfoNode.children![0].text!;
|
|
if (fenceType !== "data" && !fenceType.startsWith("#")) {
|
|
return;
|
|
}
|
|
const codeTextNode = findNodeOfType(t, "CodeText");
|
|
if (!codeTextNode) {
|
|
// Honestly, this shouldn't happen
|
|
return;
|
|
}
|
|
const codeText = codeTextNode.children![0].text!;
|
|
const dataType = fenceType === "data" ? "data" : fenceType.substring(1);
|
|
try {
|
|
const docs = codeText.split(separator);
|
|
let cursor = 0;
|
|
// We support multiple YAML documents in one block
|
|
for (let i = 0; i < docs.length; i++) {
|
|
const docStart = t.from! + cursor;
|
|
const docEnd = docStart + docs[i].length;
|
|
const doc = YAML.load(docs[i]);
|
|
if (!doc) {
|
|
cursor += docs[i].length;
|
|
continue;
|
|
}
|
|
const dataObj = {
|
|
ref: `${pageMeta.name}@${docStart}`,
|
|
tag: dataType,
|
|
itags: ["data"],
|
|
pos: docStart,
|
|
range: [docStart, docEnd] as [number, number],
|
|
...doc,
|
|
page: pageMeta.name,
|
|
};
|
|
updateITags(dataObj, frontmatter);
|
|
dataObjects.push(dataObj);
|
|
cursor += docs[i].length;
|
|
}
|
|
tagObjects.set(dataType, {
|
|
ref: dataType,
|
|
tag: "tag",
|
|
name: dataType,
|
|
page: pageMeta.name,
|
|
parent: "data",
|
|
});
|
|
} catch (e) {
|
|
console.error("Could not parse data", codeText, "error:", e);
|
|
return;
|
|
}
|
|
});
|
|
|
|
return Promise.resolve([...dataObjects, ...tagObjects.values()]);
|
|
}
|