Fixed a lot of bugs with new widget rendering

plugs/query/query.plug.yaml

@@ -26,11 +26,15 @@ functions:
     - editor:complete
 
   refreshAllWidgets:
-    path: widget.ts:refreshAll
+    path: widget.ts:refreshAllWidgets
     command:
       name: "Live Queries and Templates: Refresh All"
       key: "Alt-q"
 
+  # Query widget buttons
+  editButton:
+    path: widget.ts:editButton
+
   # Slash commands
   insertQuery:
     redirect: template.insertTemplateText

plugs/query/query.ts

@@ -9,12 +9,12 @@ import { astToKvQuery } from "$sb/lib/parse-query.ts";
 import { jsonToMDTable, renderQueryTemplate } from "../directive/util.ts";
 import { loadPageObject, replaceTemplateVars } from "../template/template.ts";
 import { cleanPageRef, resolvePath } from "$sb/lib/resolve.ts";
-import { LintDiagnostic } from "$sb/types.ts";
+import { CodeWidgetContent, LintDiagnostic } from "$sb/types.ts";
 
 export async function widget(
   bodyText: string,
   pageName: string,
-): Promise<WidgetContent> {
+): Promise<CodeWidgetContent> {
   const pageObject = await loadPageObject(pageName);
 
   try {
@@ -72,6 +72,20 @@ export async function widget(
 
     return {
       markdown: resultMarkdown,
+      buttons: [
+        {
+          description: "Edit",
+          svg:
+            `<svg xmlns="http://www.w3.org/2000/svg" width="15" height="15" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-edit"><path d="M11 4H4a2 2 0 0 0-2 2v14a2 2 0 0 0 2 2h14a2 2 0 0 0 2-2v-7"></path><path d="M18.5 2.5a2.121 2.121 0 0 1 3 3L12 15l-4 1 1-4 9.5-9.5z"></path></svg>`,
+          invokeFunction: "query.editButton",
+        },
+        {
+          description: "Reload",
+          svg:
+            `<svg xmlns="http://www.w3.org/2000/svg" width="15" height="15" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-refresh-cw"><polyline points="23 4 23 10 17 10"></polyline><polyline points="1 20 1 14 7 14"></polyline><path d="M3.51 9a9 9 0 0 1 14.85-3.36L23 10M1 14l4.64 4.36A9 9 0 0 0 20.49 15"></path></svg>`,
+          invokeFunction: "query.refreshAllWidgets",
+        },
+      ],
     };
   } catch (e: any) {
     return { markdown: `**Error:** ${e.message}` };

plugs/query/widget.ts

@@ -1,5 +1,9 @@
-import { codeWidget } from "$sb/syscalls.ts";
+import { codeWidget, editor } from "$sb/syscalls.ts";
 
-export function refreshAll() {
+export function refreshAllWidgets() {
   codeWidget.refreshAll();
 }
+
+export async function editButton(pos: number) {
+  await editor.moveCursor(pos);
+}