feat: add a fetch function for autocomplete

frectonz · Oct 17, 2024 · b9a0228 · b9a0228
1 parent bab013d
commit b9a0228
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/ui/src/api.ts b/ui/src/api.ts
@@ -68,6 +68,15 @@ const metadata = z.object({
   can_shutdown: z.boolean(),
 });
 
+const autocomplete = z.object({
+  tables: z
+    .object({
+      columns: z.string().array(),
+      table_name: z.string(),
+    })
+    .array(),
+});
+
 const $fetch = createZodFetcher();
 
 export const fetchOverview = () => $fetch(overview, `${BASE_URL}/`);
@@ -86,6 +95,8 @@ export const fetchQuery = (value: string) =>
     body: JSON.stringify({ query: value }),
   });
 export const fetchMetadata = () => $fetch(metadata, `${BASE_URL}/metadata`);
+export const fetchAutocomplete = () =>
+  $fetch(autocomplete, `${BASE_URL}/autocomplete`);
 
 export const sendShutdown = () =>
   fetch(`${BASE_URL}/shutdown`, { method: "POST" });
diff --git a/ui/src/routes/query.lazy.tsx b/ui/src/routes/query.lazy.tsx
@@ -18,7 +18,7 @@ import {
 import { createFileRoute } from "@tanstack/react-router";
 
 import { cn } from "@/lib/utils";
-import { fetchQuery } from "@/api";
+import { fetchAutocomplete, fetchQuery } from "@/api";
 import {
   useQueries,
   QueriesProvider,
@@ -177,6 +177,13 @@ function Query({ sql, onChange, onSave, onDelete, onUpdate }: QueryProps) {
     retry: false,
   });
 
+  const { data: autocompleteData } = useQuery({
+    queryKey: ["autocomplete"],
+    queryFn: () => fetchAutocomplete(),
+  });
+
+  console.log(autocompleteData);
+
   const grid = !data ? (
     !autoExecute && code && error ? (
       <Card>