@agent-native/core 0.133.2 → 0.133.3

package/corpus/templates/content/actions/preview-content-database-source-attach.ts

@@ -0,0 +1,92 @@
+import { defineAction } from "@agent-native/core";
+import { assertAccess } from "@agent-native/core/sharing";
+import { z } from "zod";
+
+import type {
+  BuilderCmsAttachPreviewResponse,
+  ContentDatabaseItem,
+} from "../shared/api.js";
+import { readBuilderCmsContentEntries } from "./_builder-cms-read-client.js";
+import {
+  builderCmsImportIds,
+  resolveDatabaseForSourceMutation,
+} from "./_database-source-utils.js";
+import { getContentDatabaseResponse } from "./_database-utils.js";
+
+export default defineAction({
+  description:
+    "Preview the first projected Builder rows for a database source attachment without writing local or provider data.",
+  schema: z.object({
+    databaseId: z.string().optional(),
+    documentId: z.string().optional(),
+    sourceTable: z.string().min(1).max(500),
+    fieldPaths: z.array(z.string().max(500)).max(200).optional(),
+  }),
+  http: { method: "GET" },
+  readOnly: true,
+  run: async (args): Promise<BuilderCmsAttachPreviewResponse> => {
+    const database = await resolveDatabaseForSourceMutation(args);
+    if (!database) throw new Error("Database not found.");
+    await assertAccess("document", database.documentId, "editor");
+
+    const [read, base] = await Promise.all([
+      readBuilderCmsContentEntries({
+        model: args.sourceTable,
+        fieldPaths: args.fieldPaths,
+        allowCached: true,
+        maxPages: 1,
+      }),
+      getContentDatabaseResponse(database.id, { limit: 100, offset: 0 }),
+    ]);
+    if (read.state !== "live") {
+      throw new Error(
+        read.message ?? "Builder rows are not available for preview.",
+      );
+    }
+
+    const items: ContentDatabaseItem[] = read.entries.map((entry, index) => {
+      const ids = builderCmsImportIds({
+        ownerEmail: database.ownerEmail,
+        databaseId: database.id,
+        sourceTable: args.sourceTable,
+        entryId: entry.id,
+      });
+      return {
+        id: ids.itemId,
+        databaseId: database.id,
+        document: {
+          id: ids.documentId,
+          parentId: database.documentId,
+          title: entry.title.trim() || entry.id,
+          content: "",
+          icon: null,
+          position: index,
+          isFavorite: false,
+          hideFromSearch: false,
+          accessRole: "viewer",
+          canEdit: false,
+          canManage: false,
+          createdAt: entry.updatedAt,
+          updatedAt: entry.updatedAt,
+        },
+        position: index,
+        properties: [],
+        bodyHydration: {
+          status: "pending",
+          attemptedAt: null,
+          error: null,
+          version: null,
+        },
+      };
+    });
+    return {
+      databaseId: database.id,
+      documentId: database.documentId,
+      sourceTable: args.sourceTable,
+      base,
+      items,
+      fetchedAt: read.fetchedAt,
+      hasMore: read.progress.hasMore,
+    };
+  },
+});

package/corpus/templates/content/actions/process-builder-body-hydration.ts

@@ -15,7 +15,7 @@ export default defineAction({
   schema: z.object({
     sourceId: z.string(),
     documentId: z.string().optional(),
-    limit: z.number().int().positive().max(50).optional(),
+    limit: z.number().int().positive().max(600).optional(),
   }),
   agentTool: false,
   run: async (
@@ -42,6 +42,7 @@ export default defineAction({
       sourceId: args.sourceId,
       documentId: args.documentId,
       limit: args.limit,
+      preloadBodies: !args.documentId,
     });
   },
 });

package/corpus/templates/content/actions/query-content-database-items.ts

@@ -0,0 +1,64 @@
+import { defineAction } from "@agent-native/core";
+import { z } from "zod";
+
+import type {
+  ContentDatabaseItemsPageResponse,
+  ContentDatabaseUnavailableResponse,
+} from "../shared/api.js";
+import {
+  CONTENT_DATABASE_MAX_READ_LIMIT,
+  contentDatabaseTableQuerySchema,
+  getContentDatabasePageResponse,
+  resolveContentDatabaseRead,
+} from "./_database-utils.js";
+
+export default defineAction({
+  description:
+    "Query one ordered and filtered page of a content database without reopening its metadata.",
+  schema: z.object({
+    databaseId: z.string().optional().describe("Database ID"),
+    documentId: z.string().optional().describe("Database document/page ID"),
+    limit: z.coerce
+      .number()
+      .int()
+      .min(1)
+      .max(CONTENT_DATABASE_MAX_READ_LIMIT)
+      .optional(),
+    offset: z.coerce.number().int().min(0).optional(),
+    tableQuery: contentDatabaseTableQuerySchema,
+  }),
+  http: { method: "GET" },
+  readOnly: true,
+  agentTool: false,
+  run: async ({
+    databaseId,
+    documentId,
+    limit,
+    offset,
+    tableQuery,
+  }): Promise<
+    ContentDatabaseItemsPageResponse | ContentDatabaseUnavailableResponse
+  > => {
+    const resolved = await resolveContentDatabaseRead({
+      databaseId,
+      documentId,
+    });
+    if (!resolved.available) return resolved;
+
+    const page = await getContentDatabasePageResponse(resolved.database.id, {
+      // This action is the bounded table replacement path; unlike the legacy
+      // database response, an omitted limit must not turn it into a full read.
+      limit: limit ?? 100,
+      offset,
+      tableQuery,
+      database: resolved.database,
+    });
+    return {
+      items: page.items,
+      source: page.source,
+      sources: page.sources,
+      pagination: page.pagination,
+      tableQueryMode: page.tableQueryMode,
+    };
+  },
+});

package/corpus/templates/content/actions/refresh-content-database-source.ts

@@ -35,6 +35,12 @@ export default defineAction({
       .describe(
         "For paginated Builder CMS or Notion sources, read a bounded multi-page snapshot in this refresh.",
       ),
+    finishBuilderPagination: z
+      .boolean()
+      .optional()
+      .describe(
+        "Internal Builder continuation mode. Read every remaining page from the persisted continuation offset.",
+      ),
     expectedBuilderContinuationOffset: z
       .number()
       .int()
@@ -75,6 +81,7 @@ export default defineAction({
           source: claimedSource.source,
           now,
           runFullRefresh: args.fullRefresh === true,
+          finishPagination: args.finishBuilderPagination === true,
           refreshClaimId: claimedSource.claimId,
         }).catch(async (error: unknown) => {
           await releaseBuilderCmsSourceRefreshClaim({

package/corpus/templates/content/actions/review-content-database-source-change-set.ts

@@ -116,6 +116,6 @@ export default defineAction({
       .set({ updatedAt: now })
       .where(eq(schema.contentDatabaseSources.id, source.id));
 
-    return getContentDatabaseResponse(database.id);
+    return getContentDatabaseResponse(database.id, { limit: 100, offset: 0 });
   },
 });

package/corpus/templates/content/actions/set-content-database-source-write-mode.ts

@@ -121,6 +121,6 @@ export default defineAction({
         "Builder source settings changed repeatedly while saving write mode.",
     });
 
-    return getContentDatabaseResponse(database.id);
+    return getContentDatabaseResponse(database.id, { limit: 100, offset: 0 });
   },
 });

package/corpus/templates/content/actions/stage-builder-revision.ts

@@ -114,6 +114,6 @@ export default defineAction({
       .set({ updatedAt: now })
       .where(eq(schema.contentDatabaseSources.id, source.id));
 
-    return getContentDatabaseResponse(database.id);
+    return getContentDatabaseResponse(database.id, { limit: 100, offset: 0 });
   },
 });

package/corpus/templates/content/actions/update-content-database-personal-view.ts

@@ -1,14 +1,26 @@
 import { defineAction } from "@agent-native/core";
 import { deleteUserSetting, putUserSetting } from "@agent-native/core/settings";
+import { and, eq, inArray } from "drizzle-orm";
 import { z } from "zod";
 
+import { getDb, schema } from "../server/db/index.js";
+import { bulkChunkSizeForColumnCount, chunks } from "./_batch-utils.js";
 import {
   assertContentDatabaseViewerAccess,
   normalizePersonalDatabaseViewOverrides,
   personalDatabaseViewSettingKey,
   personalViewOverridesSchema,
 } from "./_content-database-personal-view.js";
-import { getContentDatabaseResponse } from "./_database-utils.js";
+
+export function personalSidebarOrderItemIds(
+  overrides: z.infer<typeof personalViewOverridesSchema>,
+) {
+  return [
+    ...new Set(
+      overrides.views.flatMap((view) => view.sidebarOrder?.itemIds ?? []),
+    ),
+  ];
+}
 
 export default defineAction({
   description:
@@ -23,10 +35,24 @@ export default defineAction({
 
     const key = personalDatabaseViewSettingKey(databaseId);
     if (overrides) {
-      const visibleDatabase = await getContentDatabaseResponse(databaseId);
+      const requestedItemIds = personalSidebarOrderItemIds(overrides);
+      const validItemIds = new Set<string>();
+      const itemIdChunkSize = Math.max(1, bulkChunkSizeForColumnCount(1) - 1);
+      for (const itemIds of chunks(requestedItemIds, itemIdChunkSize)) {
+        const rows = await getDb()
+          .select({ id: schema.contentDatabaseItems.id })
+          .from(schema.contentDatabaseItems)
+          .where(
+            and(
+              eq(schema.contentDatabaseItems.databaseId, databaseId),
+              inArray(schema.contentDatabaseItems.id, itemIds),
+            ),
+          );
+        for (const row of rows) validItemIds.add(row.id);
+      }
       overrides = normalizePersonalDatabaseViewOverrides(
         overrides,
-        new Set(visibleDatabase.items.map((item) => item.id)),
+        validItemIds,
       );
       await putUserSetting(ctx.userEmail, key, overrides);
     } else {

package/corpus/templates/content/actions/update-content-database-view.ts

@@ -136,6 +136,6 @@ export default defineAction({
 
     await writeAppState("refresh-signal", { ts: Date.now() });
 
-    return getContentDatabaseResponse(databaseId);
+    return getContentDatabaseResponse(databaseId, { limit: 100, offset: 0 });
   },
 });

package/corpus/templates/content/actions/validate-builder-source-execution.ts

@@ -153,6 +153,6 @@ export default defineAction({
       .set({ updatedAt: now })
       .where(eq(schema.contentDatabaseSources.id, source.id));
 
-    return getContentDatabaseResponse(database.id);
+    return getContentDatabaseResponse(database.id, { limit: 100, offset: 0 });
   },
 });

package/corpus/templates/content/app/components/editor/DocumentProperties.tsx

@@ -19,10 +19,8 @@ import {
 } from "@dnd-kit/sortable";
 import { CSS } from "@dnd-kit/utilities";
 import type {
-  AddContentDatabaseSourceFieldPropertyRequest,
   BindContentDatabaseSourceFieldRequest,
   ContentDatabaseResponse,
-  ContentDatabaseSourceFieldPropertyResponse,
   ContentDatabaseSource,
   DocumentProperty,
 } from "@shared/api";
@@ -130,7 +128,7 @@ import {
   TooltipContent,
   TooltipTrigger,
 } from "@/components/ui/tooltip";
-import { applySourceFieldPropertyToDatabaseResponse } from "@/hooks/use-content-database";
+import { useAddContentDatabaseSourceFieldProperty } from "@/hooks/use-content-database";
 import {
   documentPropertiesResponseMatchesScope,
   useConfigureDocumentProperty,
@@ -3148,25 +3146,8 @@ export function AddProperty({
 }) {
   const t = useT();
   const configure = useConfigureDocumentProperty(documentId, databaseId);
-  const queryClient = useQueryClient();
-  const addSourceFieldProperty = useActionMutation<
-    ContentDatabaseSourceFieldPropertyResponse,
-    AddContentDatabaseSourceFieldPropertyRequest
-  >("add-content-database-source-field-property", {
-    onSuccess: (data) => {
-      queryClient.setQueriesData<ContentDatabaseResponse>(
-        { queryKey: ["action", "get-content-database"] },
-        (current) => applySourceFieldPropertyToDatabaseResponse(current, data),
-      );
-      queryClient.invalidateQueries({
-        queryKey: [
-          "action",
-          "list-document-properties",
-          { documentId, databaseId },
-        ],
-      });
-    },
-  });
+  const addSourceFieldProperty =
+    useAddContentDatabaseSourceFieldProperty(documentId);
   const [open, setOpen] = useState(false);
   const [typeQuery, setTypeQuery] = useState("");
   const filteredPropertyTypes = filterDocumentPropertyTypes(typeQuery);