silverbullet/plugs/directive/query_directive.ts

import { events } from "$sb/syscalls.ts";

import { replaceTemplateVars } from "../template/template.ts";
import { renderTemplate } from "./util.ts";
import { jsonToMDTable } from "./util.ts";
import { ParseTree, parseTreeToAST } from "$sb/lib/tree.ts";
import { astToKvQuery } from "$sb/lib/parse-query.ts";
import { PageMeta, Query } from "$sb/types.ts";

export async function queryDirectiveRenderer(
  _directive: string,
  pageMeta: PageMeta,
  query: string | ParseTree,
): Promise<string> {
  if (typeof query === "string") {
    throw new Error("Argument must be a ParseTree");
  }
  const parsedQuery: Query = astToKvQuery(
    parseTreeToAST(
      JSON.parse(await replaceTemplateVars(JSON.stringify(query), pageMeta)),
    ),
  );
  // console.log("QUERY", parsedQuery);

  const eventName = `query:${parsedQuery.querySource}`;

  // console.log("Parsed query", parsedQuery);
  // Let's dispatch an event and see what happens
  const results = await events.dispatchEvent(
    eventName,
    { query: parsedQuery, pageName: pageMeta.name },
    30 * 1000,
  );
  if (results.length === 0) {
    // This means there was no handler for the event which means it's unsupported
    return `**Error:** Unsupported query source '${parsedQuery.querySource}'`;
  } else {
    // console.log("Parsed query", parsedQuery);
    const allResults = results.flat();
    if (parsedQuery.render) {
      const rendered = await renderTemplate(
        pageMeta,
        parsedQuery.render,
        allResults,
        parsedQuery.renderAll!,
      );
      return rendered.trim();
    } else {
      if (allResults.length === 0) {
        return "No results";
      } else {
        return jsonToMDTable(allResults);
      }
    }
  }
}