@blinkk/root-cms 2.0.1 → 2.0.3

package/dist/plugin.js

@@ -1,12 +1,42 @@
+var __create = Object.create;
 var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __esm = (fn, res) => function __init() {
   return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
 };
+var __commonJS = (cb, mod) => function __require() {
+  return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
+};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
 };
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+
+// shared/ai/prompts/edit.txt
+var require_edit = __commonJS({
+  "shared/ai/prompts/edit.txt"(exports, module) {
+    module.exports = 'Role: You are a highly accurate AI assistant specializing in converting unstructured documents into JSON files compatible with Root CMS. Root CMS is a web-based CMS for building websites and landing pages, storing its content as JSON files in a Firebase Firestore database.\n\nObjective: To edit and/or generate highly accurate JSON files compatible with Root CMS, based on user input (screenshots, chat messages, etc.) and, when editing JSON files, the original JSON file you are editing. The resulting content should contain all of the marketing content provided by the user input (and none of the annotations or superfluous content). The JSON files should be perfectly compatible with Root CMS.\n\nInput Requirements & Handling:\n\n- You will receive unstructured documents and you must convert them into JSON files compatible with Root CMS by either editing the existing JSON files or generating new ones from scratch. The JSON files you produce will typically represent a single section of a page of the website, such as a hero section, a feature section, or a footer.\n\n- When editing an existing JSON file, it will be annotated with a key `_type` that indicates the type of section it is, such as `Hero`, `Feature`, or `Footer`. You must ensure that the resulting JSON file is compatible with Root CMS and contains all of the marketing content provided by the user input. You will also be given a TypeScript interface in `root-cms.d.ts` that describes the structure of the JSON file you are editing. You must ensure that the resulting JSON file is compatible with this interface.\n\n- The input you receive will be broken into three parts. (1) `root-cms-d.ts` (the TypeScript interface), (2) the existing JSON file you are editing (or an empty JSON file when generating new content), and (3) the user input (screenshots, chat messages, instructions, etc.). You must ensure that the resulting JSON file is compatible with the structure defined by the TypeScript interface and contains all of the marketing content provided by the user input.\n\n- When you provide your response, if the user instructs you to only edit one part of the original JSON, make sure to leave the rest of the JSON file untouched. Your response, however, should always contain the _entire_ JSON file combining the original contents with your new edits (think of it like you\'re applying a PATCH operation and the response is the entire file).\n\nRoot CMS JSON File Structure:\n\n- Root CMS JSON files are standard JSON files described by the TypeScript interface in `root-cms.d.ts`. However, there is one critical syntax rule that you must understand and always follow regarding array values. Array values are expressed as hash maps. The order of the resulting data when Root CMS unmarshals the data is provided by a special key called `_array`. For example, to represent an array [1, 2, 3] in the CMS, the JSON data may look like this. You may generate your own UIDs for the keys and the items within `_array`.\n\n```json\n{\n  "_array": ["2HGwoV", "LjFHWk", "OLakGA", "UaGHJh"],\n  "2HGwoV": 1,\n  "LjFHWk": 2,\n  "OLakGA": 3,\n  "UaGHJh": 4\n}\n```\n\nWhen creating arrays in your JSON output, you must _never_ forget to add an `_array` key. If you forget to add an `_array` key, the output will NOT work with Root CMS. When adding new array items, you can create your own array keys.\n\n2. Images and file fields have system-provided metadata, that looks like this example:\n\n```\n{\n    "src": "https://storage.googleapis.com/rootjs-dev.appspot.com/www/uploads/2cad5d65bb7690bad5dcd4a041cf2fef46361678.svg",\n    "filename": "600x400.svg",\n    "uploadedAt": "1754023634564",\n    "alt": "",\n    "width": "600",\n    "uploadedBy": "user@example.com",\n    "height": "400"\n}\n```\n\n3. Rich Text fields are stored as Editor.js documents, which look like this. You may generate your own UIDs for the `id` fields.\n\n```\n{\n  "version": "2.28.2",\n  "time": 1754024041637,\n  "blocks": [\n    {\n      "type": "paragraph",\n      "data": {\n        "text": "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. <a href=\\"https://example.com\\">This accepts HTML</a>."\n      },\n      "id": "_8PEf3kBx2"\n    },\n    {\n      "data": {\n        "style": "unordered",\n        "items": [\n          {\n            "content": "This is list item 1.",\n            "items": []\n          },\n          {\n            "items": [],\n            "content": "This is list item 2."\n          }\n        ]\n      },\n      "id": "G5zMDP1ytM",\n      "type": "unorderedList"\n    }\n  ]\n}\n```\n\n4. Root CMS supports "oneOf" fields, where values can be "one of" another schema. These are typically used as array values or object values, where the user can from one of other schema types to structure the data value. For example, each module in a page\'s module list is "one of" the modules available in the library. One of values have a system-provided `_type` key that describes the type of the value. For example, this is a hero module, whose type is "TemplateHeadline", one of the available templates from the module library:\n\n```\n{\n  "eyebrow": "Lorem ipsum",\n  "_type": "TemplateHeadline",\n  "body": {\n    "time": 1754023554170,\n    "version": "2.28.2",\n    "blocks": [\n      {\n        "type": "paragraph",\n        "data": {\n          "text": "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat."\n        },\n        "id": "IVmxEVsxmt"\n      },\n      {\n        "id": "aMiHiywkQD",\n        "type": "paragraph",\n        "data": {\n          "text": "Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum."\n        }\n      }\n    ]\n  },\n  "title": "Welcome to our website"\n}\n```\n\nWhen adding new items to arrays that use `oneOf` fields, try your best to figure out which content type would best satisfy the user\'s request. You must never _invent_ new content types, for example, if "TemplateFoo" doesn\'t exist in `root-cms.d.ts`, you must never add a value with `{"_type": "TemplateFoo"}`. If none of the available content types within `root-cms.d.ts` satisfy the user\'s request, avoid creating the new array item and inform the user that there is no existing template compatible.\n\nAdditionally, when adding new items to existing JSON data, you must carefully check the structure of those items as defined by `root-cms.d.ts`. Avoid inventing new fields for content types, use only those defined by the `root-cms.d.ts`. For example, if you are adding a button to a module, carefully check the button structure defined in `root-cms.d.ts` and only populate fields that are defined there.\n\n5. Here is the `root-cms.d.ts` file for this project:\n\n```\n{{ROOT_CMS_DEFS}}\n```\n\nFinally, when you provide your response, it MUST be structured in the following way for handling by the chat client.\n\n{\n  "data": <The JSON generated by you via the instructions above>,\n  "message": <A brief message that describes what you did.>\n}\n\nYou must always ONLY reply in this exact format. The brief message that describes what you did should be written succinctly, tersely, in a friendly but not cute tone, and be no more than a sentence or two.\n';
+  }
+});
 
 // cli/generate-types.ts
 var generate_types_exports = {};
@@ -14,21 +44,21 @@ __export(generate_types_exports, {
   generateSchemaDts: () => generateSchemaDts,
   generateTypes: () => generateTypes
 });
-import { promises as fs } from "node:fs";
-import path2 from "node:path";
+import { promises as fs2 } from "node:fs";
+import path3 from "node:path";
 import { fileURLToPath } from "node:url";
 import { loadRootConfig, viteSsrLoadModule } from "@blinkk/root/node";
 import * as dom from "dts-dom";
 async function generateTypes() {
   const rootDir = process.cwd();
   const rootConfig = await loadRootConfig(rootDir, { command: "root-cms" });
-  const modulePath = path2.resolve(__dirname, "./project.js");
+  const modulePath = path3.resolve(__dirname, "./project.js");
   const project = await viteSsrLoadModule(
     rootConfig,
     modulePath
   );
   const schemas = project.getProjectSchemas();
-  const outputPath = path2.resolve(rootDir, "root-cms.d.ts");
+  const outputPath = path3.resolve(rootDir, "root-cms.d.ts");
   await generateSchemaDts(outputPath, schemas);
   console.log("saved root-cms.d.ts!");
 }
@@ -39,7 +69,7 @@ async function generateSchemaDts(outputPath, schemas) {
     dtsFormatter.addSchemaFile(fileId, schema);
   }
   const output = dtsFormatter.toString();
-  await fs.writeFile(outputPath, output, "utf-8");
+  await fs2.writeFile(outputPath, output, "utf-8");
 }
 function fieldProperty(field, options) {
   const prop = dom.create.property(
@@ -153,7 +183,7 @@ var __dirname, TEMPLATE, DtsFormatter;
 var init_generate_types = __esm({
   "cli/generate-types.ts"() {
     "use strict";
-    __dirname = path2.dirname(fileURLToPath(import.meta.url));
+    __dirname = path3.dirname(fileURLToPath(import.meta.url));
     TEMPLATE = `/* eslint-disable */
 /** Root.js CMS types. This file is autogenerated. */
 
@@ -227,7 +257,7 @@ export interface RootCMSDoc<Fields extends {}> {
       /** Adds the types for a `.schema.ts` file to the `.d.ts` file. */
       addSchemaFile(fileId, schema) {
         const jsdoc = `Generated from \`${fileId}\`.`;
-        const typeId = alphanumeric(path2.parse(fileId).name.split(".")[0]);
+        const typeId = alphanumeric(path3.parse(fileId).name.split(".")[0]);
         const oneOfTypes = {};
         const fieldsTypeId = `${typeId}Fields`;
         const fieldsType = dom.create.interface(
@@ -307,8 +337,8 @@ export interface RootCMSDoc<Fields extends {}> {
 });
 
 // core/plugin.ts
-import { promises as fs2 } from "node:fs";
-import path3 from "node:path";
+import { promises as fs3 } from "node:fs";
+import path4 from "node:path";
 import { fileURLToPath as fileURLToPath2 } from "node:url";
 import bodyParser from "body-parser";
 import {
@@ -326,19 +356,22 @@ import { multipartMiddleware } from "@blinkk/root/middleware";
 
 // core/ai.ts
 import crypto from "node:crypto";
+import fs from "node:fs";
+import path from "node:path";
 import { vertexAI } from "@genkit-ai/vertexai";
 import { Timestamp } from "firebase-admin/firestore";
 import { genkit } from "genkit";
+import { logger } from "genkit/logging";
+logger.setLogLevel("warn");
 var DEFAULT_MODEL = "vertexai/gemini-2.5-flash";
 var Chat = class {
   constructor(chatClient, id, options) {
-    var _a;
     this.chatClient = chatClient;
     this.cmsClient = chatClient.cmsClient;
     this.cmsPluginOptions = this.cmsClient.cmsPlugin.getConfig();
     this.id = id;
-    this.history = (options == null ? void 0 : options.history) ?? [];
-    this.model = (options == null ? void 0 : options.model) || (typeof ((_a = this.cmsPluginOptions.experiments) == null ? void 0 : _a.ai) === "object" ? this.cmsPluginOptions.experiments.ai.model : void 0) || DEFAULT_MODEL;
+    this.history = options?.history ?? [];
+    this.model = options?.model || (typeof this.cmsPluginOptions.experiments?.ai === "object" ? this.cmsPluginOptions.experiments.ai.model : void 0) || DEFAULT_MODEL;
     const firebaseConfig = this.cmsPluginOptions.firebaseConfig;
     this.ai = genkit({
       plugins: [
@@ -350,21 +383,52 @@ var Chat = class {
     });
   }
   /** Builds the messages for the AI request. */
-  async buildMessages() {
+  async buildMessages(options) {
     const messages = this.history;
     if (messages.length === 0) {
       messages.push({
         role: "system",
-        content: [{ text: await this.buildSystemPrompt() }]
+        content: [{ text: await this.buildSystemPrompt(options) }]
+      });
+    }
+    if (options.mode === "edit") {
+      messages.push({
+        role: "user",
+        content: [
+          {
+            text: [
+              "The JSON you must edit is:",
+              "",
+              JSON.stringify(options.editData || {}, null, 2)
+            ].join("")
+          }
+        ]
       });
     }
     return messages;
   }
-  async sendPrompt(prompt) {
-    const res = await this.ai.generate({
-      messages: await this.buildMessages(),
+  /** Builds the request sent to the AI based on the `ChatMode`. */
+  async buildChatRequest(prompt, options) {
+    if (options.mode === "edit") {
+      return {
+        messages: await this.buildMessages(options),
+        model: this.model,
+        prompt
+      };
+    }
+    return {
+      messages: await this.buildMessages(options),
       model: this.model,
       prompt
+    };
+  }
+  /** Sends the request to the AI and stores the history in the session and the database. */
+  async sendPrompt(prompt, options = {}) {
+    const chatRequest = await this.buildChatRequest(prompt, options);
+    const res = await this.ai.generate({
+      model: chatRequest.model,
+      messages: chatRequest.messages,
+      prompt: Array.isArray(prompt) ? prompt.flat() : prompt
     });
     this.history = res.messages;
     await this.dbDoc().update({
@@ -376,13 +440,35 @@ var Chat = class {
   dbDoc() {
     return this.chatClient.dbCollection().doc(this.id);
   }
-  async buildSystemPrompt() {
+  /**
+   * Builds the system prompt sent to the AI, based on the `ChatMode` and
+   * supplied `SendPromptOptions`. `SendPromptOptions` may contain data or
+   * references to information needed to construct the prompt.
+   */
+  async buildSystemPrompt(options) {
+    const serializedRootConfig = JSON.stringify(
+      this.cmsClient.rootConfig,
+      null,
+      2
+    );
+    if (options.mode === "edit") {
+      const rootDir = process.cwd();
+      const rootCmsDefs = fs.readFileSync(
+        path.resolve(rootDir, "root-cms.d.ts"),
+        {
+          encoding: "utf8"
+        }
+      );
+      const text = (await Promise.resolve().then(() => __toESM(require_edit(), 1))).default;
+      text.replace("{{ROOT_CMS_DEFS}}", rootCmsDefs);
+      return text;
+    }
     const systemText = [
       `You are an assistant for a headless CMS called Root CMS which is used on a website called ${this.cmsPluginOptions.name || this.cmsPluginOptions.id}. Your job is to answer questions about the docs in the system, and if requested, help suggest changes to the JSON data in the docs. If you don't know the answer, just say that you don't know, don't try to make up an answer. Be friendly and playful with your messaging.`,
       "",
       "Here is the root.config.ts file for the site:",
       "```",
-      JSON.stringify(this.cmsClient.rootConfig, null, 2),
+      serializedRootConfig,
       "```",
       "",
       "Here are the docs that exist in the system:"
@@ -425,7 +511,7 @@ var ChatClient = class {
     });
   }
   async listChats(options) {
-    const limit = (options == null ? void 0 : options.limit) || 20;
+    const limit = options?.limit || 20;
     const query = this.dbCollection().where("createdBy", "==", this.user).limit(limit).orderBy("createdAt", "desc");
     const res = await query.get();
     return res.docs.map((doc) => doc.data());
@@ -503,11 +589,11 @@ var TranslationsManager = class {
         locale,
         sys: {
           modifiedAt: Timestamp2.now(),
-          modifiedBy: (options == null ? void 0 : options.modifiedBy) || "root-cms-client"
+          modifiedBy: options?.modifiedBy || "root-cms-client"
         },
         strings: {}
       };
-      if ((options == null ? void 0 : options.tags) && options.tags.length > 0) {
+      if (options?.tags && options.tags.length > 0) {
         updates.tags = FieldValue.arrayUnion(...options.tags);
       }
       let numUpdates = 0;
@@ -547,13 +633,13 @@ var TranslationsManager = class {
       console.warn(`no translations to publish for ${id}`);
       return;
     }
-    const batch = (options == null ? void 0 : options.batch) || db.batch();
+    const batch = options?.batch || db.batch();
     res.docs.forEach((doc) => {
       const translationsLocaleDoc = doc.data();
       const sys = {
         ...translationsLocaleDoc.sys,
         publishedAt: Timestamp2.now(),
-        publishedBy: (options == null ? void 0 : options.publishedBy) || "root-cms-client"
+        publishedBy: options?.publishedBy || "root-cms-client"
       };
       batch.update(doc.ref, { sys });
       const publishedDocPath = buildTranslationsLocaleDocDbPath({
@@ -565,7 +651,7 @@ var TranslationsManager = class {
       const publishedDocRef = db.doc(publishedDocPath);
       batch.set(publishedDocRef, { ...translationsLocaleDoc, sys });
     });
-    const shouldCommitBatch = !(options == null ? void 0 : options.batch);
+    const shouldCommitBatch = !options?.batch;
     if (shouldCommitBatch) {
       await batch.commit();
     }
@@ -606,19 +692,19 @@ var TranslationsManager = class {
    * ```
    */
   async loadTranslations(options) {
-    const mode = (options == null ? void 0 : options.mode) || "published";
+    const mode = options?.mode || "published";
     const dbPath = buildTranslationsDbPath({
       project: this.cmsClient.projectId,
       mode
     });
     let query = this.cmsClient.db.collection(dbPath);
-    if ((options == null ? void 0 : options.ids) && options.ids.length > 0) {
+    if (options?.ids && options.ids.length > 0) {
       query = query.where("id", "in", options.ids);
     }
-    if ((options == null ? void 0 : options.tags) && options.tags.length > 0) {
+    if (options?.tags && options.tags.length > 0) {
       query = query.where("tags", "array-contains", options.tags);
     }
-    if ((options == null ? void 0 : options.locales) && options.locales.length > 0) {
+    if (options?.locales && options.locales.length > 0) {
       query = query.where("locale", "in", options.locales);
     }
     const results = await query.get();
@@ -649,11 +735,11 @@ var TranslationsManager = class {
   async loadTranslationsForLocale(locale, options) {
     const localeSet = /* @__PURE__ */ new Set([
       locale,
-      ...(options == null ? void 0 : options.fallbackLocales) || []
+      ...options?.fallbackLocales || []
     ]);
     const fallbackLocales = Array.from(localeSet);
     const multiLocaleStrings = await this.loadTranslations({
-      mode: options == null ? void 0 : options.mode,
+      mode: options?.mode,
       locales: fallbackLocales
     });
     return this.toSingleLocaleMap(multiLocaleStrings, fallbackLocales);
@@ -724,7 +810,6 @@ var TranslationsManager = class {
    * Import translations from the v1 system to the TranslationsManager.
    */
   async importTranslationsFromV1() {
-    var _a;
     const projectId = this.cmsClient.projectId;
     const db = this.cmsClient.db;
     const dbPath = `Projects/${projectId}/Translations`;
@@ -765,7 +850,7 @@ var TranslationsManager = class {
         const doc = await this.cmsClient.getDoc(collection, slug, {
           mode: "draft"
         });
-        const linkedSheet = (_a = doc == null ? void 0 : doc.sys) == null ? void 0 : _a.l10nSheet;
+        const linkedSheet = doc?.sys?.l10nSheet;
         if (linkedSheet) {
           translationsDocs[docId].sys.linkedSheet = linkedSheet;
         }
@@ -868,7 +953,7 @@ var RootCMSClient = class {
     const { collection, slug } = parseDocId(docId);
     const draftDoc = await this.getRawDoc(collection, slug, { mode: "draft" }) || {};
     const draftSys = draftDoc.sys || {};
-    const modifiedBy = (options == null ? void 0 : options.modifiedBy) || "root-cms-client";
+    const modifiedBy = options?.modifiedBy || "root-cms-client";
     const fields = marshalData(fieldsData || {});
     const data = {
       id: docId,
@@ -880,7 +965,7 @@ var RootCMSClient = class {
         createdBy: draftSys.createdBy ?? modifiedBy,
         modifiedAt: Timestamp3.now(),
         modifiedBy,
-        locales: (options == null ? void 0 : options.locales) ?? draftSys.locales ?? ["en"]
+        locales: options?.locales ?? draftSys.locales ?? ["en"]
       },
       fields
     };
@@ -950,7 +1035,7 @@ var RootCMSClient = class {
    */
   async publishDocs(docIds, options) {
     const projectCollectionsPath = `Projects/${this.projectId}/Collections`;
-    const publishedBy = (options == null ? void 0 : options.publishedBy) || "root-cms-client";
+    const publishedBy = options?.publishedBy || "root-cms-client";
     const docRefs = docIds.map((docId) => {
       const [collection, slug] = docId.split("/");
       if (!collection || !slug) {
@@ -973,9 +1058,9 @@ var RootCMSClient = class {
       }
     }
     let batchCount = 0;
-    const batch = (options == null ? void 0 : options.batch) || this.db.batch();
+    const batch = options?.batch || this.db.batch();
     const versionTags = ["published"];
-    if (options == null ? void 0 : options.releaseId) {
+    if (options?.releaseId) {
       versionTags.push(`release:${options.releaseId}`);
     }
     const publishedDocs = [];
@@ -1058,12 +1143,11 @@ var RootCMSClient = class {
     const now = Math.ceil((/* @__PURE__ */ new Date()).getTime());
     const snapshot = await this.db.collectionGroup("Scheduled").get();
     const docs = snapshot.docs.filter((d) => {
-      var _a;
       if (!d.ref.path.startsWith(projectCollectionsPath)) {
         return false;
       }
       const data = d.data() || {};
-      const scheduledAt = (_a = data.sys) == null ? void 0 : _a.scheduledAt;
+      const scheduledAt = data.sys?.scheduledAt;
       return scheduledAt && scheduledAt.toMillis && scheduledAt.toMillis() <= now;
     }).map((d) => {
       const dbPath = d.ref.path;
@@ -1194,8 +1278,7 @@ var RootCMSClient = class {
    * Checks if a doc is currently "locked" for publishing.
    */
   testPublishingLocked(doc) {
-    var _a;
-    if ((_a = doc.sys) == null ? void 0 : _a.publishingLocked) {
+    if (doc.sys?.publishingLocked) {
       if (doc.sys.publishingLocked.until) {
         const now = Timestamp3.now().toMillis();
         const until = doc.sys.publishingLocked.until.toMillis();
@@ -1220,9 +1303,8 @@ var RootCMSClient = class {
    * replace the v1 translations system.
    */
   getTranslationsManager() {
-    var _a;
     const cmsPluginOptions = this.cmsPlugin.getConfig();
-    if ((_a = cmsPluginOptions.experiments) == null ? void 0 : _a.v2TranslationsManager) {
+    if (cmsPluginOptions.experiments?.v2TranslationsManager) {
       throw new Error(
         "`v2TranslationsManager` is not enabled. update root.config.ts and add: `{experiments: {v2TranslationsManager: true}}`"
       );
@@ -1243,7 +1325,7 @@ var RootCMSClient = class {
   async loadTranslations(options) {
     const dbPath = `Projects/${this.projectId}/Translations`;
     let query = this.db.collection(dbPath);
-    if (options == null ? void 0 : options.tags) {
+    if (options?.tags) {
       query = query.where("tags", "array-contains-any", options.tags);
     }
     const querySnapshot = await query.get();
@@ -1361,7 +1443,7 @@ var RootCMSClient = class {
     const dataDocRef = this.db.doc(
       `Projects/${this.projectId}/DataSources/${dataSourceId}/Data/draft`
     );
-    const syncedBy = (options == null ? void 0 : options.syncedBy) || "root-cms-client";
+    const syncedBy = options?.syncedBy || "root-cms-client";
     const updatedDataSource = {
       ...dataSource,
       syncedAt: Timestamp3.now(),
@@ -1395,7 +1477,7 @@ var RootCMSClient = class {
       `Projects/${this.projectId}/DataSources/${dataSourceId}/published`
     );
     const dataRes = await this.getFromDataSource(dataSourceId, { mode: "draft" });
-    const publishedBy = (options == null ? void 0 : options.publishedBy) || "root-cms-client";
+    const publishedBy = options?.publishedBy || "root-cms-client";
     const updatedDataSource = {
       ...dataSource,
       publishedAt: Timestamp3.now(),
@@ -1404,8 +1486,8 @@ var RootCMSClient = class {
     const batch = this.db.batch();
     batch.set(dataDocRefPublished, {
       dataSource: updatedDataSource,
-      data: (dataRes == null ? void 0 : dataRes.data) || null,
-      ...(dataRes == null ? void 0 : dataRes.headers) ? { headers: dataRes.headers } : {}
+      data: dataRes?.data || null,
+      ...dataRes?.headers ? { headers: dataRes.headers } : {}
     });
     batch.update(dataDocRefDraft, {
       dataSource: updatedDataSource
@@ -1419,8 +1501,8 @@ var RootCMSClient = class {
     console.log(`published by: ${publishedBy}`);
   }
   async publishDataSources(dataSourceIds, options) {
-    const publishedBy = (options == null ? void 0 : options.publishedBy) || "root-cms-client";
-    const batch = (options == null ? void 0 : options.batch) || this.db.batch();
+    const publishedBy = options?.publishedBy || "root-cms-client";
+    const batch = options?.batch || this.db.batch();
     for (const id of dataSourceIds) {
       const dataSource = await this.getDataSource(id);
       if (!dataSource) {
@@ -1443,8 +1525,8 @@ var RootCMSClient = class {
       };
       batch.set(dataDocRefPublished, {
         dataSource: updatedDataSource,
-        data: (dataRes == null ? void 0 : dataRes.data) || null,
-        ...(dataRes == null ? void 0 : dataRes.headers) ? { headers: dataRes.headers } : {}
+        data: dataRes?.data || null,
+        ...dataRes?.headers ? { headers: dataRes.headers } : {}
       });
       batch.update(dataDocRefDraft, { dataSource: updatedDataSource });
       batch.update(dataSourceDocRef, {
@@ -1452,7 +1534,7 @@ var RootCMSClient = class {
         publishedBy
       });
     }
-    if (!(options == null ? void 0 : options.batch) || (options == null ? void 0 : options.commitBatch)) {
+    if (!options?.batch || options?.commitBatch) {
       await batch.commit();
     }
   }
@@ -1463,15 +1545,14 @@ var RootCMSClient = class {
     throw new Error(`unsupported data source: ${dataSource.type}`);
   }
   async fetchHttpData(dataSource) {
-    var _a, _b, _c;
     const url = dataSource.url || "";
     if (!url.startsWith("https://")) {
       throw new Error(`url not supported: ${url}`);
     }
     const res = await fetch(url, {
-      method: ((_a = dataSource.httpOptions) == null ? void 0 : _a.method) || "GET",
-      headers: ((_b = dataSource.httpOptions) == null ? void 0 : _b.headers) || [],
-      body: ((_c = dataSource.httpOptions) == null ? void 0 : _c.body) || void 0
+      method: dataSource.httpOptions?.method || "GET",
+      headers: dataSource.httpOptions?.headers || [],
+      body: dataSource.httpOptions?.body || void 0
     });
     if (res.status !== 200) {
       const err = await res.text();
@@ -1487,7 +1568,7 @@ var RootCMSClient = class {
    * Fetches data from a data source.
    */
   async getFromDataSource(dataSourceId, options) {
-    const mode = (options == null ? void 0 : options.mode) || "published";
+    const mode = options?.mode || "published";
     if (!(mode === "draft" || mode === "published")) {
       throw new Error(`invalid mode: ${mode}`);
     }
@@ -1577,12 +1658,12 @@ var RootCMSClient = class {
     const data = {
       action,
       timestamp: Timestamp3.now(),
-      by: (options == null ? void 0 : options.by) || "system",
-      metadata: (options == null ? void 0 : options.metadata) || {}
+      by: options?.by || "system",
+      metadata: options?.metadata || {}
     };
     const colRef = this.db.collection(`Projects/${this.projectId}/ActionLogs`);
     await colRef.add(data);
-    const metaStr = (options == null ? void 0 : options.metadata) ? stringifyObj(options.metadata) : "";
+    const metaStr = options?.metadata ? stringifyObj(options.metadata) : "";
     console.log(`[${data.timestamp.toMillis()}] action: ${action} ${metaStr}`);
     const cmsPluginConfig = this.cmsPlugin.getConfig();
     if (cmsPluginConfig.onAction) {
@@ -1940,7 +2021,7 @@ var BatchResponse = class {
 };
 
 // core/versions.ts
-import path from "node:path";
+import path2 from "node:path";
 import { Timestamp as Timestamp4 } from "firebase-admin/firestore";
 import glob from "tiny-glob";
 var DOCUMENT_SAVE_OFFSET = 5 * 60 * 1e3;
@@ -1973,8 +2054,7 @@ var VersionsService = class {
   async saveVersionsToFirestore(versions) {
     const batch = this.db.batch();
     versions.forEach((version) => {
-      var _a;
-      if (!version.collection || !version.slug || !((_a = version.sys) == null ? void 0 : _a.modifiedAt)) {
+      if (!version.collection || !version.slug || !version.sys?.modifiedAt) {
         return;
       }
       const modifiedAtMillis = version.sys.modifiedAt.toMillis();
@@ -2033,7 +2113,7 @@ var VersionsService = class {
   async listCollections() {
     const collectionIds = [];
     const collectionFileNames = await glob("*.schema.ts", {
-      cwd: path.join(this.rootConfig.rootDir, "collections")
+      cwd: path2.join(this.rootConfig.rootDir, "collections")
     });
     collectionFileNames.forEach((filename) => {
       collectionIds.push(filename.slice(0, -10));
@@ -2160,12 +2240,11 @@ function api(server, options) {
     }
   );
   server.use("/cms/api/data.sync", async (req, res) => {
-    var _a;
     if (req.method !== "POST" || !String(req.get("content-type")).startsWith("application/json")) {
       res.status(400).json({ success: false, error: "BAD_REQUEST" });
       return;
     }
-    if (!((_a = req.user) == null ? void 0 : _a.email)) {
+    if (!req.user?.email) {
       res.status(401).json({ success: false, error: "UNAUTHORIZED" });
     }
     const reqBody = req.body || {};
@@ -2184,12 +2263,11 @@ function api(server, options) {
     }
   });
   server.use("/cms/api/actions.log", async (req, res) => {
-    var _a, _b;
     if (req.method !== "POST" || !String(req.get("content-type")).startsWith("application/json")) {
       res.status(400).json({ success: false, error: "BAD_REQUEST" });
       return;
     }
-    if (!((_a = req.user) == null ? void 0 : _a.email)) {
+    if (!req.user?.email) {
       res.status(401).json({ success: false, error: "UNAUTHORIZED" });
     }
     const reqBody = req.body || {};
@@ -2205,7 +2283,7 @@ function api(server, options) {
     const cmsClient = new RootCMSClient(req.rootConfig);
     try {
       await cmsClient.logAction(action, {
-        by: (_b = req.user) == null ? void 0 : _b.email,
+        by: req.user?.email,
         metadata
       });
       res.status(200).json({ success: true });
@@ -2215,12 +2293,11 @@ function api(server, options) {
     }
   });
   server.use("/cms/api/ai.chat", async (req, res) => {
-    var _a;
     if (req.method !== "POST" || !String(req.get("content-type")).startsWith("application/json")) {
       res.status(400).json({ success: false, error: "BAD_REQUEST" });
       return;
     }
-    if (!((_a = req.user) == null ? void 0 : _a.email)) {
+    if (!req.user?.email) {
       res.status(401).json({ success: false, error: "UNAUTHORIZED" });
       return;
     }
@@ -2239,7 +2316,10 @@ function api(server, options) {
       } else {
         chat = await chatClient.createChat();
       }
-      const chatResponse = await chat.sendPrompt(prompt);
+      const chatResponse = await chat.sendPrompt(prompt, {
+        mode: reqBody.options?.mode || "chat",
+        editData: reqBody.options?.editData
+      });
       res.status(200).json({ success: true, chatId: chat.id, response: chatResponse });
     } catch (err) {
       console.error(err.stack || err);
@@ -2247,12 +2327,11 @@ function api(server, options) {
     }
   });
   server.use("/cms/api/ai.list_chats", async (req, res) => {
-    var _a;
     if (req.method !== "POST" || !String(req.get("content-type")).startsWith("application/json")) {
       res.status(400).json({ success: false, error: "BAD_REQUEST" });
       return;
     }
-    if (!((_a = req.user) == null ? void 0 : _a.email)) {
+    if (!req.user?.email) {
       res.status(401).json({ success: false, error: "UNAUTHORIZED" });
       return;
     }
@@ -2271,7 +2350,7 @@ function api(server, options) {
 }
 
 // core/plugin.ts
-var __dirname2 = path3.dirname(fileURLToPath2(import.meta.url));
+var __dirname2 = path4.dirname(fileURLToPath2(import.meta.url));
 var SESSION_COOKIE_AUTH = "root-cms-auth";
 var NONCONF_STATIC_PATHS = [
   "/cms/static/signin.css",
@@ -2330,8 +2409,7 @@ function cmsPlugin(options) {
     return false;
   }
   async function verifyUserLogin(req) {
-    var _a;
-    const idToken = (_a = req.body) == null ? void 0 : _a.idToken;
+    const idToken = req.body?.idToken;
     if (!idToken) {
       console.log("login failed: no id token");
       return { authorized: false, reason: "no token" };
@@ -2389,7 +2467,7 @@ function cmsPlugin(options) {
     }
     if (process.env.NODE_ENV === "development") {
       const jwt = jsonwebtoken.decode(sessionCookie);
-      if (!(jwt == null ? void 0 : jwt.email) || !(jwt == null ? void 0 : jwt.email_verified)) {
+      if (!jwt?.email || !jwt?.email_verified) {
         return null;
       }
       if (options.isUserAuthorized) {
@@ -2477,7 +2555,7 @@ function cmsPlugin(options) {
      * A map of ssr files to include when running `root build`.
      */
     ssrInput: () => ({
-      cms: path3.resolve(__dirname2, "./app.js")
+      cms: path4.resolve(__dirname2, "./app.js")
     }),
     /**
      * Attaches CMS-specific middleware to the Root.js server.
@@ -2486,13 +2564,13 @@ function cmsPlugin(options) {
       server.use(bodyParser.json());
       async function getRenderer(req) {
         if (serverOptions.type === "dev") {
-          const appFilePath = path3.resolve(__dirname2, "./app.js");
+          const appFilePath = path4.resolve(__dirname2, "./app.js");
           const app3 = await req.viteServer.ssrLoadModule(
             appFilePath
           );
           return app3;
         }
-        const appImportPath = path3.resolve(
+        const appImportPath = path4.resolve(
           req.rootConfig.rootDir,
           "dist/server/cms.js"
         );
@@ -2564,7 +2642,7 @@ function cmsPlugin(options) {
         }
         next();
       });
-      const staticDir = path3.resolve(__dirname2, "ui");
+      const staticDir = path4.resolve(__dirname2, "ui");
       server.use("/cms/static", sirv(staticDir, { dev: false }));
       api(server, { getRenderer });
       server.use("/cms", async (req, res) => {
@@ -2598,7 +2676,7 @@ function cmsPlugin(options) {
   return plugin;
 }
 function fileExists(filepath) {
-  return fs2.access(filepath).then(() => true).catch(() => false);
+  return fs3.access(filepath).then(() => true).catch(() => false);
 }
 export {
   cmsPlugin