88 lines
2.3 KiB
TypeScript
88 lines
2.3 KiB
TypeScript
import type { IndexTreeEvent, QueryProviderEvent } from "$sb/app_event.ts";
|
|
|
|
import { index } from "$sb/silverbullet-syscall/mod.ts";
|
|
import { collectNodesOfType, ParseTree, renderToText } from "$sb/lib/tree.ts";
|
|
import { applyQuery, removeQueries } from "$sb/lib/query.ts";
|
|
import { extractAttributes } from "$sb/lib/attribute.ts";
|
|
|
|
export type Item = {
|
|
name: string;
|
|
nested?: string;
|
|
tags?: string[];
|
|
// Not stored in DB
|
|
page?: string;
|
|
pos?: number;
|
|
} & Record<string, any>;
|
|
|
|
export async function indexItems({ name, tree }: IndexTreeEvent) {
|
|
const items: { key: string; value: Item }[] = [];
|
|
removeQueries(tree);
|
|
|
|
// console.log("Indexing items", name);
|
|
|
|
const coll = collectNodesOfType(tree, "ListItem");
|
|
|
|
coll.forEach((n) => {
|
|
if (!n.children) {
|
|
return;
|
|
}
|
|
if (collectNodesOfType(n, "Task").length > 0) {
|
|
// This is a task item, skip it
|
|
return;
|
|
}
|
|
|
|
const item: Item = {
|
|
name: "", // to be replaced
|
|
};
|
|
|
|
const textNodes: ParseTree[] = [];
|
|
let nested: string | undefined;
|
|
for (const child of n.children!.slice(1)) {
|
|
if (child.type === "OrderedList" || child.type === "BulletList") {
|
|
nested = renderToText(child);
|
|
break;
|
|
}
|
|
// Extract attributes and remove from tree
|
|
const extractedAttributes = extractAttributes(child, true);
|
|
for (const [key, value] of Object.entries(extractedAttributes)) {
|
|
item[key] = value;
|
|
}
|
|
textNodes.push(child);
|
|
}
|
|
|
|
item.name = textNodes.map(renderToText).join("").trim();
|
|
if (nested) {
|
|
item.nested = nested;
|
|
}
|
|
collectNodesOfType(n, "Hashtag").forEach((h) => {
|
|
if (!item.tags) {
|
|
item.tags = [];
|
|
}
|
|
// Push tag to the list, removinn the initial #
|
|
item.tags.push(h.children![0].text!.substring(1));
|
|
});
|
|
|
|
items.push({
|
|
key: `it:${n.from}`,
|
|
value: item,
|
|
});
|
|
});
|
|
// console.log("Found", items.length, "item(s)");
|
|
await index.batchSet(name, items);
|
|
}
|
|
|
|
export async function queryProvider({
|
|
query,
|
|
}: QueryProviderEvent): Promise<any[]> {
|
|
const allItems: Item[] = [];
|
|
for (const { key, page, value } of await index.queryPrefix("it:")) {
|
|
const [, pos] = key.split(":");
|
|
allItems.push({
|
|
...value,
|
|
page: page,
|
|
pos: +pos,
|
|
});
|
|
}
|
|
return applyQuery(query, allItems);
|
|
}
|