@agent-native/core 0.86.0 → 0.87.0

package/corpus/templates/analytics/server/plugins/db.ts

@@ -671,6 +671,41 @@ const runAnalyticsMigrations = runMigrations(
       name: "analytics-alert-rules-enabled-eval-idx",
       sql: `CREATE INDEX IF NOT EXISTS analytics_alert_rules_enabled_eval_idx ON analytics_alert_rules (enabled, last_status, last_evaluated_at, created_at)`,
     },
+    {
+      version: 81,
+      name: "analytics-db-admin-connections-table",
+      sql: {
+        postgres: `CREATE TABLE IF NOT EXISTS analytics_db_admin_connections (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL,
+      app_id TEXT,
+      app_url TEXT,
+      database_url_secret_key TEXT NOT NULL,
+      database_auth_token_secret_key TEXT,
+      created_by TEXT NOT NULL,
+      created_at TEXT NOT NULL DEFAULT (now()::text),
+      updated_at TEXT NOT NULL DEFAULT (now()::text),
+      org_id TEXT NOT NULL
+    )`,
+        sqlite: `CREATE TABLE IF NOT EXISTS analytics_db_admin_connections (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL,
+      app_id TEXT,
+      app_url TEXT,
+      database_url_secret_key TEXT NOT NULL,
+      database_auth_token_secret_key TEXT,
+      created_by TEXT NOT NULL,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now')),
+      org_id TEXT NOT NULL
+    )`,
+      },
+    },
+    {
+      version: 82,
+      name: "analytics-db-admin-connections-org-updated-idx",
+      sql: `CREATE INDEX IF NOT EXISTS analytics_db_admin_connections_org_updated_idx ON analytics_db_admin_connections (org_id, updated_at)`,
+    },
   ],
   { table: "analytics_migrations" },
 );

package/corpus/templates/analytics/server/routes/[base]/_agent-native/analytics-db-admin/[...path].ts

@@ -0,0 +1 @@
+export { default } from "../../../_agent-native/analytics-db-admin/[...path].js";

package/corpus/templates/analytics/server/routes/_agent-native/analytics-db-admin/[...path].ts

@@ -0,0 +1,160 @@
+import {
+  applyMutations,
+  DbAdminConfirmRequiredError,
+  getRows,
+  getTableSchema,
+  listTables,
+  runSql,
+  type DbAdminMutation,
+  type DbAdminRowsRequest,
+} from "@agent-native/core/db-admin";
+import { readBody } from "@agent-native/core/server";
+import {
+  defineEventHandler,
+  getMethod,
+  getRequestURL,
+  setResponseHeader,
+  setResponseStatus,
+  type H3Event,
+} from "h3";
+
+import {
+  DbAdminConnectionError,
+  redactDbAdminError,
+  runWithDbAdminEventContext,
+  withDbAdminConnectionRuntime,
+} from "../../../lib/db-admin-connections";
+
+function decodeSegment(value: string | undefined): string | undefined {
+  if (!value) return undefined;
+  try {
+    return decodeURIComponent(value);
+  } catch {
+    return value;
+  }
+}
+
+function pathSegments(event: H3Event): string[] {
+  const url = getRequestURL(event);
+  const marker = "/_agent-native/analytics-db-admin";
+  const markerIndex = url.pathname.indexOf(marker);
+  const suffix =
+    markerIndex === -1 ? "" : url.pathname.slice(markerIndex + marker.length);
+  return suffix
+    .replace(/^\/+/, "")
+    .split("/")
+    .filter(Boolean)
+    .map(decodeSegment)
+    .filter((segment): segment is string => Boolean(segment));
+}
+
+function methodNotAllowed(event: H3Event) {
+  setResponseStatus(event, 405);
+  return { ok: false, error: "Method not allowed" };
+}
+
+function badRequest(event: H3Event, error: string) {
+  setResponseStatus(event, 400);
+  return { ok: false, error };
+}
+
+function notFound(event: H3Event, error: string) {
+  setResponseStatus(event, 404);
+  return { ok: false, error };
+}
+
+export default defineEventHandler(async (event) => {
+  setResponseHeader(event, "Cache-Control", "no-store");
+
+  const method = getMethod(event);
+  const [connectionId, ...segments] = pathSegments(event);
+  if (!connectionId) {
+    return badRequest(event, "Connection id is required");
+  }
+
+  try {
+    return await runWithDbAdminEventContext(event, async (admin) =>
+      withDbAdminConnectionRuntime(admin, connectionId, async (runtime) => {
+        if (segments[0] === "overview" && segments.length === 1) {
+          if (method !== "GET") return methodNotAllowed(event);
+          const result = await listTables(runtime);
+          return { ok: true, ...result };
+        }
+
+        if (segments[0] === "table") {
+          const name = segments[1];
+          if (!name) return badRequest(event, "Table name is required");
+          const sub = segments[2];
+
+          if (sub === "schema" && segments.length === 3) {
+            if (method !== "GET") return methodNotAllowed(event);
+            const table = await getTableSchema(name, runtime);
+            return { ok: true, table };
+          }
+
+          if (sub === "rows" && segments.length === 3) {
+            if (method !== "POST") return methodNotAllowed(event);
+            const body = await readBody<Partial<DbAdminRowsRequest>>(event);
+            const result = await getRows(
+              name,
+              {
+                page: Number(body.page) || 1,
+                pageSize: Number(body.pageSize) || 50,
+                sort: body.sort,
+                filters: body.filters,
+              },
+              runtime,
+            );
+            return { ok: true, ...result };
+          }
+
+          if (sub === "mutate" && segments.length === 3) {
+            if (method !== "POST") return methodNotAllowed(event);
+            const body = await readBody<DbAdminMutation>(event);
+            const result = await applyMutations(name, body ?? {}, runtime);
+            return { ok: true, ...result };
+          }
+
+          return notFound(event, "Unknown analytics db-admin table route");
+        }
+
+        if (segments[0] === "query" && segments.length === 1) {
+          if (method !== "POST") return methodNotAllowed(event);
+          const body = await readBody<{
+            sql?: string;
+            params?: unknown[];
+            confirmDestructive?: boolean;
+          }>(event);
+          try {
+            const result = await runSql(
+              String(body.sql ?? ""),
+              Array.isArray(body.params) ? body.params : undefined,
+              { confirmDestructive: body.confirmDestructive === true },
+              runtime,
+            );
+            return { ok: true, ...result };
+          } catch (err) {
+            if (err instanceof DbAdminConfirmRequiredError) {
+              setResponseStatus(event, 400);
+              return {
+                ok: false,
+                error: err.message,
+                needsConfirm: true,
+              };
+            }
+            throw err;
+          }
+        }
+
+        return notFound(event, "Unknown analytics db-admin route");
+      }),
+    );
+  } catch (err) {
+    if (err instanceof DbAdminConnectionError) {
+      setResponseStatus(event, err.statusCode);
+      return { ok: false, error: err.message };
+    }
+    setResponseStatus(event, 500);
+    return { ok: false, error: redactDbAdminError(err) };
+  }
+});

package/corpus/templates/analytics/shared/session-replay-agent-access.ts

@@ -1,10 +1,21 @@
-export const SESSION_REPLAY_AGENT_ACCESS_PARAM = "agent_access";
+import {
+  AGENT_ACCESS_PARAM,
+  scopedAgentAccessResourceId,
+} from "@agent-native/core/shared";
 
-const SESSION_REPLAY_AGENT_ACCESS_TOKEN_PREFIX =
+export const SESSION_REPLAY_AGENT_ACCESS_PARAM =
+  AGENT_ACCESS_PARAM || "agent_access";
+export const SESSION_REPLAY_AGENT_ACCESS_TOKEN_PREFIX =
   "analytics-session-replay-agent-context";
 
 export function sessionReplayAgentAccessTokenResourceId(
   recordingId: string,
 ): string {
-  return `${SESSION_REPLAY_AGENT_ACCESS_TOKEN_PREFIX}:${recordingId}`;
+  if (typeof scopedAgentAccessResourceId !== "function") {
+    return `${SESSION_REPLAY_AGENT_ACCESS_TOKEN_PREFIX}:${recordingId}`;
+  }
+  return scopedAgentAccessResourceId(
+    SESSION_REPLAY_AGENT_ACCESS_TOKEN_PREFIX,
+    recordingId,
+  );
 }

package/corpus/templates/assets/app/components/create/RecentDraftsSection.tsx

@@ -1,10 +1,9 @@
 import { useActionQuery, useT } from "@agent-native/core/client";
+import { Button } from "@agent-native/toolkit/ui/button";
+import { Skeleton } from "@agent-native/toolkit/ui/skeleton";
 import { IconArrowUpRight, IconPhoto } from "@tabler/icons-react";
 import { Link } from "react-router";
 
-import { Button } from "@/components/ui/button";
-import { Skeleton } from "@/components/ui/skeleton";
-
 type DraftAsset = {
   id: string;
   title?: string | null;

package/corpus/templates/assets/app/components/generation/GenerationResults.tsx

@@ -5,11 +5,6 @@ import {
   useActionQuery,
   useT,
 } from "@agent-native/core/client";
-import { IconMessageCircle, IconPhoto, IconTrash } from "@tabler/icons-react";
-import { useQuery, useQueryClient } from "@tanstack/react-query";
-import { useEffect, useMemo, useRef, useState } from "react";
-import { toast } from "sonner";
-
 import {
   AlertDialog,
   AlertDialogAction,
@@ -19,10 +14,15 @@ import {
   AlertDialogFooter,
   AlertDialogHeader,
   AlertDialogTitle,
-} from "@/components/ui/alert-dialog";
-import { Badge } from "@/components/ui/badge";
-import { Button } from "@/components/ui/button";
-import { Spinner } from "@/components/ui/spinner";
+} from "@agent-native/toolkit/ui/alert-dialog";
+import { Badge } from "@agent-native/toolkit/ui/badge";
+import { Button } from "@agent-native/toolkit/ui/button";
+import { Spinner } from "@agent-native/toolkit/ui/spinner";
+import { IconMessageCircle, IconPhoto, IconTrash } from "@tabler/icons-react";
+import { useQuery, useQueryClient } from "@tanstack/react-query";
+import { useEffect, useMemo, useRef, useState } from "react";
+import { toast } from "sonner";
+
 import { assetPreviewSources } from "@/lib/asset-preview-sources";
 
 import type {

package/corpus/templates/assets/app/components/layout/Header.tsx

@@ -3,10 +3,12 @@ import {
   useActionQuery,
   useT,
 } from "@agent-native/core/client";
+import {
+  useHeaderTitle,
+  useHeaderActions,
+} from "@agent-native/toolkit/app-shell";
 import { useLocation } from "react-router";
 
-import { useHeaderTitle, useHeaderActions } from "./HeaderActions";
-
 const pageTitleKeys: Record<string, string> = {
   "/": "navigation.create",
   "/library": "navigation.library",

package/corpus/templates/assets/app/components/layout/Layout.tsx

@@ -13,6 +13,7 @@ import {
   EMBED_MODE_QUERY_PARAM,
   EMBED_TOKEN_QUERY_PARAM,
 } from "@agent-native/core/shared";
+import { HeaderActionsProvider } from "@agent-native/toolkit/app-shell";
 import { IconMenu2 } from "@tabler/icons-react";
 import { useState, useEffect } from "react";
 import { useLocation, useNavigate } from "react-router";
@@ -24,7 +25,6 @@ import { ASSETS_CHAT_STORAGE_KEY } from "@/lib/chat";
 import { cn } from "@/lib/utils";
 
 import { Header } from "./Header";
-import { HeaderActionsProvider } from "./HeaderActions";
 import { Sidebar } from "./Sidebar";
 
 interface LayoutProps {

package/corpus/templates/assets/app/components/layout/PageShell.tsx

@@ -1,14 +1,13 @@
-import { IconInfoCircle } from "@tabler/icons-react";
-import { type ReactNode } from "react";
-
+import { useSetPageTitle } from "@agent-native/toolkit/app-shell";
 import {
   Tooltip,
   TooltipContent,
   TooltipTrigger,
-} from "@/components/ui/tooltip";
-import { cn } from "@/lib/utils";
+} from "@agent-native/toolkit/ui/tooltip";
+import { IconInfoCircle } from "@tabler/icons-react";
+import { type ReactNode } from "react";
 
-import { useSetPageTitle } from "./HeaderActions";
+import { cn } from "@/lib/utils";
 
 export function PageShell({
   title,

package/corpus/templates/assets/app/components/layout/Sidebar.tsx

@@ -11,6 +11,18 @@ import {
 } from "@agent-native/core/client";
 import { ExtensionsSidebarSection } from "@agent-native/core/client/extensions";
 import { OrgSwitcher } from "@agent-native/core/client/org";
+import {
+  DropdownMenu,
+  DropdownMenuContent,
+  DropdownMenuItem,
+  DropdownMenuTrigger,
+} from "@agent-native/toolkit/ui/dropdown-menu";
+import { Input } from "@agent-native/toolkit/ui/input";
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipTrigger,
+} from "@agent-native/toolkit/ui/tooltip";
 import {
   IconArchive,
   IconClipboardList,
@@ -29,18 +41,6 @@ import { useEffect, useMemo, useRef, useState, type FormEvent } from "react";
 import { Link, useLocation, useNavigate } from "react-router";
 import { toast } from "sonner";
 
-import {
-  DropdownMenu,
-  DropdownMenuContent,
-  DropdownMenuItem,
-  DropdownMenuTrigger,
-} from "@/components/ui/dropdown-menu";
-import { Input } from "@/components/ui/input";
-import {
-  Tooltip,
-  TooltipContent,
-  TooltipTrigger,
-} from "@/components/ui/tooltip";
 import { ASSETS_CHAT_STORAGE_KEY } from "@/lib/chat";
 import { cn } from "@/lib/utils";
 

package/corpus/templates/assets/app/components/library/CreateLibraryDialog.tsx

@@ -3,10 +3,7 @@ import {
   useActionQuery,
   useT,
 } from "@agent-native/core/client";
-import { useState } from "react";
-import { toast } from "sonner";
-
-import { Button } from "@/components/ui/button";
+import { Button } from "@agent-native/toolkit/ui/button";
 import {
   Dialog,
   DialogContent,
@@ -14,10 +11,13 @@ import {
   DialogFooter,
   DialogHeader,
   DialogTitle,
-} from "@/components/ui/dialog";
-import { Input } from "@/components/ui/input";
-import { Label } from "@/components/ui/label";
-import { Separator } from "@/components/ui/separator";
+} from "@agent-native/toolkit/ui/dialog";
+import { Input } from "@agent-native/toolkit/ui/input";
+import { Label } from "@agent-native/toolkit/ui/label";
+import { Separator } from "@agent-native/toolkit/ui/separator";
+import { useState } from "react";
+import { toast } from "sonner";
+
 import { Textarea } from "@/components/ui/textarea";
 
 import type { LibraryPreset } from "../../../shared/library-presets";

package/corpus/templates/assets/app/components/library/EditLibraryDialog.tsx

@@ -1,17 +1,17 @@
 import { useActionMutation, useT } from "@agent-native/core/client";
-import { useEffect, useState } from "react";
-import { toast } from "sonner";
-
-import { Button } from "@/components/ui/button";
+import { Button } from "@agent-native/toolkit/ui/button";
 import {
   Dialog,
   DialogContent,
   DialogFooter,
   DialogHeader,
   DialogTitle,
-} from "@/components/ui/dialog";
-import { Input } from "@/components/ui/input";
-import { Label } from "@/components/ui/label";
+} from "@agent-native/toolkit/ui/dialog";
+import { Input } from "@agent-native/toolkit/ui/input";
+import { Label } from "@agent-native/toolkit/ui/label";
+import { useEffect, useState } from "react";
+import { toast } from "sonner";
+
 import { Textarea } from "@/components/ui/textarea";
 
 type EditableLibrary = {

package/corpus/templates/assets/app/components/library/LibraryCard.tsx

@@ -1,4 +1,12 @@
 import { useT } from "@agent-native/core/client";
+import { Badge } from "@agent-native/toolkit/ui/badge";
+import { Button } from "@agent-native/toolkit/ui/button";
+import {
+  DropdownMenu,
+  DropdownMenuContent,
+  DropdownMenuItem,
+  DropdownMenuTrigger,
+} from "@agent-native/toolkit/ui/dropdown-menu";
 import {
   IconCopy,
   IconDots,
@@ -7,14 +15,6 @@ import {
 } from "@tabler/icons-react";
 import { Link } from "react-router";
 
-import { Badge } from "@/components/ui/badge";
-import { Button } from "@/components/ui/button";
-import {
-  DropdownMenu,
-  DropdownMenuContent,
-  DropdownMenuItem,
-  DropdownMenuTrigger,
-} from "@/components/ui/dropdown-menu";
 import { assetMediaUrl } from "@/lib/asset-urls";
 import {
   getLibraryCustomInstructions,

package/corpus/templates/assets/app/components/library/LibraryPresetGrid.tsx

@@ -1,9 +1,8 @@
 import { appPath } from "@agent-native/core/client";
+import { Badge } from "@agent-native/toolkit/ui/badge";
+import { Button } from "@agent-native/toolkit/ui/button";
 import { IconBox, IconBrush, IconLayersIntersect } from "@tabler/icons-react";
 
-import { Badge } from "@/components/ui/badge";
-import { Button } from "@/components/ui/button";
-
 import type { LibraryPreset } from "../../../shared/library-presets";
 
 const PRESET_ICONS = [IconBox, IconBrush, IconLayersIntersect];

package/corpus/templates/assets/app/root.tsx

@@ -11,6 +11,7 @@ import {
   useCommandMenuShortcut,
   useT,
 } from "@agent-native/core/client";
+import { Toaster } from "@agent-native/toolkit/ui/sonner";
 import { IconSun, IconMoon } from "@tabler/icons-react";
 import { useQueryClient } from "@tanstack/react-query";
 import { useTheme } from "next-themes";
@@ -19,7 +20,6 @@ import { Links, Meta, Outlet, Scripts, ScrollRestoration } from "react-router";
 import type { LinksFunction } from "react-router";
 
 import { Layout as AppLayout } from "@/components/layout/Layout";
-import { Toaster } from "@/components/ui/sonner";
 
 import changelog from "../CHANGELOG.md?raw";
 import { i18nCatalog } from "./i18n";

package/corpus/templates/assets/app/routes/asset.$id.tsx

@@ -4,6 +4,25 @@ import {
   useActionQuery,
   useT,
 } from "@agent-native/core/client";
+import {
+  AlertDialog,
+  AlertDialogAction,
+  AlertDialogCancel,
+  AlertDialogContent,
+  AlertDialogDescription,
+  AlertDialogFooter,
+  AlertDialogHeader,
+  AlertDialogTitle,
+  AlertDialogTrigger,
+} from "@agent-native/toolkit/ui/alert-dialog";
+import { Badge } from "@agent-native/toolkit/ui/badge";
+import { Button } from "@agent-native/toolkit/ui/button";
+import { Separator } from "@agent-native/toolkit/ui/separator";
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipTrigger,
+} from "@agent-native/toolkit/ui/tooltip";
 import {
   IconArrowLeft,
   IconClipboard,
@@ -21,25 +40,6 @@ import {
 import { Link, useNavigate, useParams } from "react-router";
 import { toast } from "sonner";
 
-import {
-  AlertDialog,
-  AlertDialogAction,
-  AlertDialogCancel,
-  AlertDialogContent,
-  AlertDialogDescription,
-  AlertDialogFooter,
-  AlertDialogHeader,
-  AlertDialogTitle,
-  AlertDialogTrigger,
-} from "@/components/ui/alert-dialog";
-import { Badge } from "@/components/ui/badge";
-import { Button } from "@/components/ui/button";
-import { Separator } from "@/components/ui/separator";
-import {
-  Tooltip,
-  TooltipContent,
-  TooltipTrigger,
-} from "@/components/ui/tooltip";
 import { assetPreviewSources } from "@/lib/asset-preview-sources";
 import { assetMediaUrl } from "@/lib/asset-urls";
 import { cn } from "@/lib/utils";

package/corpus/templates/assets/app/routes/audit.tsx

@@ -3,6 +3,20 @@ import {
   useActionQuery,
   useT,
 } from "@agent-native/core/client";
+import { Badge } from "@agent-native/toolkit/ui/badge";
+import { Button } from "@agent-native/toolkit/ui/button";
+import { Input } from "@agent-native/toolkit/ui/input";
+import { Label } from "@agent-native/toolkit/ui/label";
+import {
+  Select,
+  SelectContent,
+  SelectItem,
+  SelectTrigger,
+  SelectValue,
+} from "@agent-native/toolkit/ui/select";
+import { Separator } from "@agent-native/toolkit/ui/separator";
+import { Sheet, SheetContent } from "@agent-native/toolkit/ui/sheet";
+import { Skeleton } from "@agent-native/toolkit/ui/skeleton";
 import {
   IconAlertTriangle,
   IconClipboardList,
@@ -13,21 +27,6 @@ import {
 } from "@tabler/icons-react";
 import { useMemo, useState } from "react";
 
-import { Badge } from "@/components/ui/badge";
-import { Button } from "@/components/ui/button";
-import { Input } from "@/components/ui/input";
-import { Label } from "@/components/ui/label";
-import {
-  Select,
-  SelectContent,
-  SelectItem,
-  SelectTrigger,
-  SelectValue,
-} from "@/components/ui/select";
-import { Separator } from "@/components/ui/separator";
-import { Sheet, SheetContent } from "@/components/ui/sheet";
-import { Skeleton } from "@/components/ui/skeleton";
-
 import {
   IMAGE_MODELS,
   VIDEO_MODELS,