2023-11-21 15:56:21 +00:00
|
|
|
import type { LintEvent, WidgetContent } from "$sb/app_event.ts";
|
|
|
|
import { events, language, space, system } from "$sb/syscalls.ts";
|
|
|
|
import {
|
|
|
|
findNodeOfType,
|
|
|
|
parseTreeToAST,
|
|
|
|
traverseTreeAsync,
|
|
|
|
} from "$sb/lib/tree.ts";
|
2023-10-03 12:16:33 +00:00
|
|
|
import { astToKvQuery } from "$sb/lib/parse-query.ts";
|
2023-11-09 08:26:44 +00:00
|
|
|
import { jsonToMDTable, renderQueryTemplate } from "../directive/util.ts";
|
2023-11-06 08:14:16 +00:00
|
|
|
import { loadPageObject, replaceTemplateVars } from "../template/template.ts";
|
2023-11-21 15:56:21 +00:00
|
|
|
import { cleanPageRef, resolvePath } from "$sb/lib/resolve.ts";
|
|
|
|
import { LintDiagnostic } from "$sb/types.ts";
|
2023-10-03 12:16:33 +00:00
|
|
|
|
2023-10-31 09:33:38 +00:00
|
|
|
export async function widget(
|
|
|
|
bodyText: string,
|
|
|
|
pageName: string,
|
|
|
|
): Promise<WidgetContent> {
|
2023-11-06 08:14:16 +00:00
|
|
|
const pageObject = await loadPageObject(pageName);
|
2023-10-03 12:16:33 +00:00
|
|
|
|
|
|
|
try {
|
|
|
|
const queryAST = parseTreeToAST(
|
2023-10-06 12:40:09 +00:00
|
|
|
await language.parseLanguage(
|
|
|
|
"query",
|
2023-11-06 08:14:16 +00:00
|
|
|
await replaceTemplateVars(bodyText, pageObject),
|
2023-10-03 12:16:33 +00:00
|
|
|
),
|
|
|
|
);
|
2023-10-06 12:40:09 +00:00
|
|
|
const parsedQuery = astToKvQuery(queryAST[1]);
|
2023-10-03 12:16:33 +00:00
|
|
|
|
2023-11-28 10:18:55 +00:00
|
|
|
if (!parsedQuery.limit) {
|
|
|
|
parsedQuery.limit = ["number", 1000];
|
|
|
|
}
|
|
|
|
|
2023-10-03 12:16:33 +00:00
|
|
|
const eventName = `query:${parsedQuery.querySource}`;
|
|
|
|
|
|
|
|
let resultMarkdown = "";
|
|
|
|
|
|
|
|
// console.log("Parsed query", parsedQuery);
|
|
|
|
// Let's dispatch an event and see what happens
|
|
|
|
const results = await events.dispatchEvent(
|
|
|
|
eventName,
|
2023-11-06 08:14:16 +00:00
|
|
|
{ query: parsedQuery, pageName: pageObject.name },
|
2023-10-03 12:16:33 +00:00
|
|
|
30 * 1000,
|
|
|
|
);
|
|
|
|
if (results.length === 0) {
|
|
|
|
// This means there was no handler for the event which means it's unsupported
|
|
|
|
return {
|
|
|
|
html:
|
|
|
|
`**Error:** Unsupported query source '${parsedQuery.querySource}'`,
|
|
|
|
};
|
|
|
|
} else {
|
|
|
|
const allResults = results.flat();
|
|
|
|
if (allResults.length === 0) {
|
|
|
|
resultMarkdown = "No results";
|
|
|
|
} else {
|
|
|
|
if (parsedQuery.render) {
|
|
|
|
// Configured a custom rendering template, let's use it!
|
2023-11-15 15:27:17 +00:00
|
|
|
const templatePage = resolvePath(pageName, parsedQuery.render);
|
2023-11-09 08:26:44 +00:00
|
|
|
const rendered = await renderQueryTemplate(
|
2023-11-06 08:14:16 +00:00
|
|
|
pageObject,
|
2023-11-15 15:27:17 +00:00
|
|
|
templatePage,
|
2023-10-03 12:16:33 +00:00
|
|
|
allResults,
|
2023-10-30 13:15:12 +00:00
|
|
|
parsedQuery.renderAll!,
|
2023-10-03 12:16:33 +00:00
|
|
|
);
|
|
|
|
resultMarkdown = rendered.trim();
|
|
|
|
} else {
|
|
|
|
// TODO: At this point it's a bit pointless to first render a markdown table, and then convert that to HTML
|
|
|
|
// We should just render the HTML table directly
|
|
|
|
resultMarkdown = jsonToMDTable(allResults);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-10-29 09:02:50 +00:00
|
|
|
return system.invokeFunction(
|
|
|
|
"markdown.markdownContentWidget",
|
|
|
|
resultMarkdown,
|
2023-11-03 11:01:33 +00:00
|
|
|
pageName,
|
2023-10-29 09:02:50 +00:00
|
|
|
);
|
2023-10-03 12:16:33 +00:00
|
|
|
} catch (e: any) {
|
2023-10-29 09:02:50 +00:00
|
|
|
return system.invokeFunction(
|
|
|
|
"markdown.markdownContentWidget",
|
|
|
|
`**Error:** ${e.message}`,
|
|
|
|
);
|
2023-10-03 12:16:33 +00:00
|
|
|
}
|
|
|
|
}
|
2023-11-21 15:56:21 +00:00
|
|
|
|
|
|
|
export async function lintQuery(
|
|
|
|
{ name, tree }: LintEvent,
|
|
|
|
): Promise<LintDiagnostic[]> {
|
|
|
|
const pageObject = await loadPageObject(name);
|
|
|
|
const diagnostics: LintDiagnostic[] = [];
|
|
|
|
await traverseTreeAsync(tree, async (node) => {
|
|
|
|
if (node.type === "FencedCode") {
|
|
|
|
const codeInfo = findNodeOfType(node, "CodeInfo")!;
|
|
|
|
if (!codeInfo) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
const codeLang = codeInfo.children![0].text!;
|
|
|
|
if (
|
|
|
|
codeLang !== "query"
|
|
|
|
) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
const codeText = findNodeOfType(node, "CodeText");
|
|
|
|
if (!codeText) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
const bodyText = codeText.children![0].text!;
|
|
|
|
try {
|
|
|
|
const queryAST = parseTreeToAST(
|
|
|
|
await language.parseLanguage(
|
|
|
|
"query",
|
|
|
|
await replaceTemplateVars(bodyText, pageObject),
|
|
|
|
),
|
|
|
|
);
|
|
|
|
const parsedQuery = astToKvQuery(queryAST[1]);
|
|
|
|
const allSources = await allQuerySources();
|
|
|
|
if (
|
|
|
|
parsedQuery.querySource &&
|
|
|
|
!allSources.includes(parsedQuery.querySource)
|
|
|
|
) {
|
|
|
|
diagnostics.push({
|
|
|
|
from: codeText.from!,
|
|
|
|
to: codeText.to!,
|
|
|
|
message: `Unknown query source '${parsedQuery.querySource}'`,
|
|
|
|
severity: "error",
|
|
|
|
});
|
|
|
|
}
|
|
|
|
if (parsedQuery.render) {
|
|
|
|
const templatePage = resolvePath(
|
|
|
|
name,
|
|
|
|
cleanPageRef(parsedQuery.render),
|
|
|
|
);
|
|
|
|
try {
|
|
|
|
await space.getPageMeta(templatePage);
|
|
|
|
} catch (e: any) {
|
|
|
|
diagnostics.push({
|
|
|
|
from: codeText.from!,
|
|
|
|
to: codeText.to!,
|
|
|
|
message: `Could not resolve template ${templatePage}`,
|
|
|
|
severity: "error",
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} catch (e: any) {
|
|
|
|
diagnostics.push({
|
|
|
|
from: codeText.from!,
|
|
|
|
to: codeText.to!,
|
|
|
|
message: e.message,
|
|
|
|
severity: "error",
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
});
|
|
|
|
return diagnostics;
|
|
|
|
}
|
|
|
|
|
|
|
|
async function allQuerySources(): Promise<string[]> {
|
|
|
|
const allEvents = await events.listEvents();
|
|
|
|
|
|
|
|
const allSources = allEvents
|
|
|
|
.filter((eventName) =>
|
|
|
|
eventName.startsWith("query:") && !eventName.includes("*")
|
|
|
|
)
|
|
|
|
.map((source) => source.substring("query:".length));
|
|
|
|
|
|
|
|
const allObjectTypes: string[] = (await events.dispatchEvent("query_", {}))
|
|
|
|
.flat();
|
|
|
|
|
2023-12-22 10:27:07 +00:00
|
|
|
// console.log("All object types", allObjectTypes);
|
|
|
|
|
2023-11-21 15:56:21 +00:00
|
|
|
return [...allSources, ...allObjectTypes];
|
|
|
|
}
|