129 lines
4.3 KiB
TypeScript
129 lines
4.3 KiB
TypeScript
import { flashNotification, getCurrentPage, reloadPage, save } from "plugos-silverbullet-syscall/editor";
|
||
|
||
import { listPages, readPage, writePage } from "plugos-silverbullet-syscall/space";
|
||
import { invokeFunction } from "plugos-silverbullet-syscall/system";
|
||
import { scanPrefixGlobal } from "plugos-silverbullet-syscall";
|
||
|
||
export const queryRegex =
|
||
/(<!--\s*#query\s+(?<table>\w+)\s*(filter\s+["'“”‘’](?<filter>[^"'“”‘’]+)["'“”‘’])?\s*\s*(order by\s+(?<orderBy>\w+)(?<orderDesc>\s+desc)?)?(group by\s+(?<groupBy>\w+))?\s*(limit\s+(?<limit>\d+))?\s*-->)(.+?)(<!--\s*#end\s*-->)/gs;
|
||
|
||
export function whiteOutQueries(text: string): string {
|
||
return text.replaceAll(queryRegex, (match) =>
|
||
new Array(match.length + 1).join(" ")
|
||
);
|
||
}
|
||
|
||
async function replaceAsync(
|
||
str: string,
|
||
regex: RegExp,
|
||
asyncFn: (match: string, ...args: any[]) => Promise<string>
|
||
) {
|
||
const promises: Promise<string>[] = [];
|
||
str.replace(regex, (match: string, ...args: any[]): string => {
|
||
const promise = asyncFn(match, ...args);
|
||
promises.push(promise);
|
||
return "";
|
||
});
|
||
const data = await Promise.all(promises);
|
||
return str.replace(regex, () => data.shift()!);
|
||
}
|
||
|
||
export async function updateMaterializedQueriesCommand() {
|
||
const currentPage = await getCurrentPage();
|
||
await save();
|
||
await invokeFunction(
|
||
"server",
|
||
"updateMaterializedQueriesOnPage",
|
||
currentPage
|
||
);
|
||
await reloadPage();
|
||
await flashNotification("Updated materialized queries");
|
||
}
|
||
|
||
// Called from client, running on server
|
||
export async function updateMaterializedQueriesOnPage(pageName: string) {
|
||
let { text } = await readPage(pageName);
|
||
text = await replaceAsync(text, queryRegex, async (match, ...args) => {
|
||
let { table, filter, groupBy, limit, orderBy, orderDesc } =
|
||
args[args.length - 1];
|
||
const startQuery = args[0];
|
||
const endQuery = args[args.length - 4];
|
||
let results = [];
|
||
switch (table) {
|
||
case "page":
|
||
let pages = await listPages();
|
||
if (orderBy) {
|
||
pages = pages.sort((a: any, b: any) => {
|
||
console.log(a, orderBy, a[orderBy]);
|
||
if (a[orderBy] === b[orderBy]) {
|
||
return 0;
|
||
}
|
||
if (a[orderBy] < b[orderBy]) {
|
||
return !!orderDesc ? 1 : -1;
|
||
} else {
|
||
return !!orderDesc ? -1 : 1;
|
||
}
|
||
});
|
||
}
|
||
let matchCount = 0;
|
||
for (let pageMeta of pages) {
|
||
if (!filter || (filter && pageMeta.name.includes(filter))) {
|
||
matchCount++;
|
||
results.push(`* [[${pageMeta.name}]]`);
|
||
if (limit && matchCount === +limit) {
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
return `${startQuery}\n${results.join("\n")}\n${endQuery}`;
|
||
case "task":
|
||
for (let {
|
||
key,
|
||
page,
|
||
value: { task, complete, nested },
|
||
} of await scanPrefixGlobal("task:")) {
|
||
let [, pos] = key.split(":");
|
||
if (!filter || (filter && task.includes(filter))) {
|
||
results.push(
|
||
`* [${complete ? "x" : " "}] [[${page}@${pos}]] ${task}` +
|
||
(nested ? "\n " + nested : "")
|
||
);
|
||
}
|
||
}
|
||
return `${startQuery}\n${results.sort().join("\n")}\n${endQuery}`;
|
||
case "link":
|
||
let uniqueLinks = new Set<string>();
|
||
for (let { key, page, value: name } of await scanPrefixGlobal(
|
||
`pl:${pageName}:`
|
||
)) {
|
||
let [, pos] = key.split(":");
|
||
if (!filter || (filter && name.includes(filter))) {
|
||
uniqueLinks.add(name);
|
||
}
|
||
}
|
||
for (const uniqueResult of uniqueLinks) {
|
||
results.push(`* [[${uniqueResult}]]`);
|
||
}
|
||
return `${startQuery}\n${results.sort().join("\n")}\n${endQuery}`;
|
||
case "item":
|
||
for (let {
|
||
key,
|
||
page,
|
||
value: { item, nested },
|
||
} of await scanPrefixGlobal("it:")) {
|
||
let [, pos] = key.split(":");
|
||
if (!filter || (filter && item.includes(filter))) {
|
||
results.push(
|
||
`* [[${page}@${pos}]] ${item}` + (nested ? "\n " + nested : "")
|
||
);
|
||
}
|
||
}
|
||
return `${startQuery}\n${results.sort().join("\n")}\n${endQuery}`;
|
||
default:
|
||
return match;
|
||
}
|
||
});
|
||
// console.log("New text", text);
|
||
await writePage(pageName, text);
|
||
}
|