@blinkk/root-cms 2.4.2 → 2.4.4

package/dist/plugin.js

@@ -45,6 +45,287 @@ var require_altText = __commonJS({
   }
 });
 
+// core/ai.ts
+var ai_exports = {};
+__export(ai_exports, {
+  Chat: () => Chat,
+  ChatClient: () => ChatClient,
+  generateImage: () => generateImage,
+  summarizeDiff: () => summarizeDiff
+});
+import crypto from "crypto";
+import fs from "fs";
+import path from "path";
+import { vertexAI } from "@genkit-ai/vertexai";
+import { Timestamp } from "firebase-admin/firestore";
+import { genkit } from "genkit";
+import { logger } from "genkit/logging";
+async function summarizeDiff(cmsClient, options) {
+  const cmsPluginOptions = cmsClient.cmsPlugin.getConfig();
+  const firebaseConfig = cmsPluginOptions.firebaseConfig;
+  const model = (typeof cmsPluginOptions.experiments?.ai === "object" ? cmsPluginOptions.experiments.ai.model : void 0) || DEFAULT_MODEL;
+  const ai = genkit({
+    plugins: [
+      vertexAI({
+        projectId: firebaseConfig.projectId,
+        location: firebaseConfig.location || "us-central1"
+      })
+    ]
+  });
+  const beforeJson = JSON.stringify(options.before ?? null, null, 2);
+  const afterJson = JSON.stringify(options.after ?? null, null, 2);
+  const systemPrompt = [
+    "You are an assistant that summarizes changes made to CMS documents stored as JSON.",
+    "Provide a concise description of the most important updates using short bullet points.",
+    'If there are no meaningful differences, respond with "No significant changes."',
+    `Focus on just the content changes, ignore insignificant changes to richtext blocks and structure, such as updates to the richtext block's "timestamp" and "version" fields.`
+  ].join("\n");
+  const diffPrompt = [
+    "Previous version JSON:",
+    "```json",
+    beforeJson,
+    "```",
+    "",
+    "Updated version JSON:",
+    "```json",
+    afterJson,
+    "```",
+    "",
+    "Summarize the differences between the two payloads."
+  ].join("\n");
+  const res = await ai.generate({
+    model,
+    messages: [
+      {
+        role: "system",
+        content: [{ text: systemPrompt }]
+      }
+    ],
+    prompt: [{ text: diffPrompt }]
+  });
+  return res.text?.trim() || "";
+}
+async function generateImage(cmsClient, options) {
+  const cmsPluginOptions = cmsClient.cmsPlugin.getConfig();
+  const firebaseConfig = cmsPluginOptions.firebaseConfig;
+  const model = "vertexai/gemini-2.5-flash-image";
+  const ai = genkit({
+    plugins: [
+      vertexAI({
+        projectId: firebaseConfig.projectId,
+        location: firebaseConfig.location || "us-central1"
+      })
+    ]
+  });
+  const res = await ai.generate({
+    model,
+    prompt: [
+      {
+        text: options.prompt
+      }
+    ],
+    config: {
+      imageConfig: {
+        aspectRatio: options.aspectRatio
+      }
+    },
+    output: {
+      format: "media"
+    }
+  });
+  if (!res.media) {
+    throw new Error("No image generated");
+  }
+  return res.media.url;
+}
+var DEFAULT_MODEL, Chat, ChatClient;
+var init_ai = __esm({
+  "core/ai.ts"() {
+    "use strict";
+    logger.setLogLevel("warn");
+    DEFAULT_MODEL = "vertexai/gemini-2.5-flash";
+    Chat = class {
+      constructor(chatClient, id, options) {
+        this.chatClient = chatClient;
+        this.cmsClient = chatClient.cmsClient;
+        this.cmsPluginOptions = this.cmsClient.cmsPlugin.getConfig();
+        this.id = id;
+        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: [
+            vertexAI({
+              projectId: firebaseConfig.projectId,
+              location: firebaseConfig.location || "us-central1"
+            })
+          ]
+        });
+      }
+      /** Builds the messages for the AI request. */
+      async buildMessages(options) {
+        const messages = this.history;
+        const hasSystemPrompt = messages.some(
+          (msg) => msg.role === "system" && msg.content.length > 0
+        );
+        if (!hasSystemPrompt) {
+          messages.push({
+            role: "system",
+            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;
+      }
+      /** Builds the request sent to the AI based on the `ChatMode`. */
+      async buildGenerateRequest(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.buildGenerateRequest(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({
+          history: this.history,
+          modifiedAt: Timestamp.now()
+        });
+        if (options.mode === "edit") {
+          const aiResponse = res.output;
+          if (aiResponse?.data && !aiResponse.message) {
+            aiResponse.message = "";
+          }
+          return aiResponse;
+        }
+        return { message: res.text, data: null };
+      }
+      dbDoc() {
+        return this.chatClient.dbCollection().doc(this.id);
+      }
+      /**
+       * 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 rootCmsDefsPath = path.resolve(rootDir, "root-cms.d.ts");
+          const rootCmsDefs = fs.existsSync(rootCmsDefsPath) ? fs.readFileSync(rootCmsDefsPath, {
+            encoding: "utf8"
+          }) : null;
+          const text = [(await Promise.resolve().then(() => __toESM(require_edit(), 1))).default];
+          if (rootCmsDefs) {
+            text.push(
+              "Here is the `root-cms.d.ts` file for this project:",
+              "```",
+              rootCmsDefs,
+              "```"
+            );
+          }
+          return text.join("\n");
+        }
+        if (options.mode === "altText") {
+          return (await Promise.resolve().then(() => __toESM(require_altText(), 1))).default;
+        }
+        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:",
+          "```",
+          serializedRootConfig,
+          "```",
+          "",
+          "Here are the docs that exist in the system:"
+        ];
+        const pages = await this.cmsClient.listDocs("Pages", { mode: "draft" });
+        pages.docs.forEach((doc) => {
+          systemText.push(JSON.stringify(doc));
+        });
+        return systemText.join("\n");
+      }
+    };
+    ChatClient = class {
+      constructor(cmsClient, user) {
+        this.cmsClient = cmsClient;
+        this.user = user;
+      }
+      async createChat() {
+        const chatId = crypto.randomUUID();
+        const docRef = this.dbCollection().doc(chatId);
+        const chat = new Chat(this, chatId);
+        await docRef.set({
+          id: chatId,
+          createdBy: this.user,
+          createdAt: Timestamp.now(),
+          modifiedAt: Timestamp.now(),
+          model: chat.model
+        });
+        return chat;
+      }
+      async getOrCreateChat(chatId) {
+        return chatId ? this.getChat(chatId) : this.createChat();
+      }
+      async getChat(chatId) {
+        const docRef = this.dbCollection().doc(chatId);
+        const chatDoc = await docRef.get();
+        if (!chatDoc.exists) {
+          throw new Error(`${chatId} does not exist`);
+        }
+        const chatData = chatDoc.data() || {};
+        return new Chat(this, chatId, {
+          history: chatData.history,
+          model: chatData.model
+        });
+      }
+      async listChats(options) {
+        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());
+      }
+      dbCollection() {
+        return this.cmsClient.db.collection(
+          `Projects/${this.cmsClient.projectId}/Experiments/ai/Chat`
+        );
+      }
+    };
+  }
+});
+
 // cli/generate-types.ts
 var generate_types_exports = {};
 __export(generate_types_exports, {
@@ -383,245 +664,11 @@ var SSEEvent = {
 };
 
 // core/api.ts
+init_ai();
 import { promises as fs2 } from "fs";
 import path3 from "path";
 import { multipartMiddleware } from "@blinkk/root/middleware";
 
-// core/ai.ts
-import crypto from "crypto";
-import fs from "fs";
-import path from "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";
-async function summarizeDiff(cmsClient, options) {
-  const cmsPluginOptions = cmsClient.cmsPlugin.getConfig();
-  const firebaseConfig = cmsPluginOptions.firebaseConfig;
-  const model = (typeof cmsPluginOptions.experiments?.ai === "object" ? cmsPluginOptions.experiments.ai.model : void 0) || DEFAULT_MODEL;
-  const ai = genkit({
-    plugins: [
-      vertexAI({
-        projectId: firebaseConfig.projectId,
-        location: firebaseConfig.location || "us-central1"
-      })
-    ]
-  });
-  const beforeJson = JSON.stringify(options.before ?? null, null, 2);
-  const afterJson = JSON.stringify(options.after ?? null, null, 2);
-  const systemPrompt = [
-    "You are an assistant that summarizes changes made to CMS documents stored as JSON.",
-    "Provide a concise description of the most important updates using short bullet points.",
-    'If there are no meaningful differences, respond with "No significant changes."',
-    `Focus on just the content changes, ignore insignificant changes to richtext blocks and structure, such as updates to the richtext block's "timestamp" and "version" fields.`
-  ].join("\n");
-  const diffPrompt = [
-    "Previous version JSON:",
-    "```json",
-    beforeJson,
-    "```",
-    "",
-    "Updated version JSON:",
-    "```json",
-    afterJson,
-    "```",
-    "",
-    "Summarize the differences between the two payloads."
-  ].join("\n");
-  const res = await ai.generate({
-    model,
-    messages: [
-      {
-        role: "system",
-        content: [{ text: systemPrompt }]
-      }
-    ],
-    prompt: [{ text: diffPrompt }]
-  });
-  return res.text?.trim() || "";
-}
-var Chat = class {
-  constructor(chatClient, id, options) {
-    this.chatClient = chatClient;
-    this.cmsClient = chatClient.cmsClient;
-    this.cmsPluginOptions = this.cmsClient.cmsPlugin.getConfig();
-    this.id = id;
-    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: [
-        vertexAI({
-          projectId: firebaseConfig.projectId,
-          location: firebaseConfig.location || "us-central1"
-        })
-      ]
-    });
-  }
-  /** Builds the messages for the AI request. */
-  async buildMessages(options) {
-    const messages = this.history;
-    const hasSystemPrompt = messages.some(
-      (msg) => msg.role === "system" && msg.content.length > 0
-    );
-    if (!hasSystemPrompt) {
-      messages.push({
-        role: "system",
-        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;
-  }
-  /** Builds the request sent to the AI based on the `ChatMode`. */
-  async buildGenerateRequest(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.buildGenerateRequest(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({
-      history: this.history,
-      modifiedAt: Timestamp.now()
-    });
-    if (options.mode === "edit") {
-      const aiResponse = res.output;
-      if (aiResponse?.data && !aiResponse.message) {
-        aiResponse.message = "";
-      }
-      return aiResponse;
-    }
-    return { message: res.text, data: null };
-  }
-  dbDoc() {
-    return this.chatClient.dbCollection().doc(this.id);
-  }
-  /**
-   * 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 rootCmsDefsPath = path.resolve(rootDir, "root-cms.d.ts");
-      const rootCmsDefs = fs.existsSync(rootCmsDefsPath) ? fs.readFileSync(rootCmsDefsPath, {
-        encoding: "utf8"
-      }) : null;
-      const text = [(await Promise.resolve().then(() => __toESM(require_edit(), 1))).default];
-      if (rootCmsDefs) {
-        text.push(
-          "Here is the `root-cms.d.ts` file for this project:",
-          "```",
-          rootCmsDefs,
-          "```"
-        );
-      }
-      return text.join("\n");
-    }
-    if (options.mode === "altText") {
-      return (await Promise.resolve().then(() => __toESM(require_altText(), 1))).default;
-    }
-    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:",
-      "```",
-      serializedRootConfig,
-      "```",
-      "",
-      "Here are the docs that exist in the system:"
-    ];
-    const pages = await this.cmsClient.listDocs("Pages", { mode: "draft" });
-    pages.docs.forEach((doc) => {
-      systemText.push(JSON.stringify(doc));
-    });
-    return systemText.join("\n");
-  }
-};
-var ChatClient = class {
-  constructor(cmsClient, user) {
-    this.cmsClient = cmsClient;
-    this.user = user;
-  }
-  async createChat() {
-    const chatId = crypto.randomUUID();
-    const docRef = this.dbCollection().doc(chatId);
-    const chat = new Chat(this, chatId);
-    await docRef.set({
-      id: chatId,
-      createdBy: this.user,
-      createdAt: Timestamp.now(),
-      modifiedAt: Timestamp.now(),
-      model: chat.model
-    });
-    return chat;
-  }
-  async getOrCreateChat(chatId) {
-    return chatId ? this.getChat(chatId) : this.createChat();
-  }
-  async getChat(chatId) {
-    const docRef = this.dbCollection().doc(chatId);
-    const chatDoc = await docRef.get();
-    if (!chatDoc.exists) {
-      throw new Error(`${chatId} does not exist`);
-    }
-    const chatData = chatDoc.data() || {};
-    return new Chat(this, chatId, {
-      history: chatData.history,
-      model: chatData.model
-    });
-  }
-  async listChats(options) {
-    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());
-  }
-  dbCollection() {
-    return this.cmsClient.db.collection(
-      `Projects/${this.cmsClient.projectId}/Experiments/ai/Chat`
-    );
-  }
-};
-
 // core/client.ts
 import crypto2 from "crypto";
 import {
@@ -1760,6 +1807,9 @@ var RootCMSClient = class {
       by: options?.by || "system",
       metadata: options?.metadata || {}
     };
+    if (options?.links) {
+      data.links = options.links;
+    }
     const colRef = this.db.collection(`Projects/${this.projectId}/ActionLogs`);
     await colRef.add(data);
     const metaStr = options?.metadata ? stringifyObj(options.metadata) : "";
@@ -2448,7 +2498,8 @@ function api(server, options) {
     try {
       await cmsClient.logAction(action, {
         by: req.user?.email,
-        metadata
+        metadata,
+        links: reqBody.links
       });
       res.status(200).json({ success: true });
     } catch (err) {
@@ -2551,12 +2602,44 @@ function api(server, options) {
       res.status(500).json({ success: false, error: "UNKNOWN" });
     }
   });
+  server.use(
+    "/cms/api/ai.generate_image",
+    async (req, res) => {
+      if (req.method !== "POST" || !String(req.get("content-type")).startsWith("application/json")) {
+        res.status(400).json({ success: false, error: "BAD_REQUEST" });
+        return;
+      }
+      if (!req.user?.email) {
+        res.status(401).json({ success: false, error: "UNAUTHORIZED" });
+        return;
+      }
+      const reqBody = req.body || {};
+      const prompt = reqBody.prompt;
+      const aspectRatio = reqBody.aspectRatio;
+      if (!prompt || !aspectRatio) {
+        res.status(400).json({ success: false, error: "MISSING_REQUIRED_FIELD" });
+        return;
+      }
+      try {
+        const cmsClient = new RootCMSClient(req.rootConfig);
+        const { generateImage: generateImage2 } = await Promise.resolve().then(() => (init_ai(), ai_exports));
+        const imageUrl = await generateImage2(cmsClient, {
+          prompt,
+          aspectRatio
+        });
+        res.status(200).json({ success: true, image: imageUrl });
+      } catch (err) {
+        console.error(err.stack || err);
+        res.status(500).json({ success: false, error: err.message || "UNKNOWN" });
+      }
+    }
+  );
 }
 
 // package.json
 var package_default = {
   name: "@blinkk/root-cms",
-  version: "2.4.2",
+  version: "2.4.4",
   author: "s@blinkk.com",
   license: "MIT",
   engines: {
@@ -2616,10 +2699,15 @@ var package_default = {
     "dev:core": "pnpm build:core --watch",
     "dev:signin": "pnpm build:signin --watch",
     "dev:ui": "pnpm build:ui --watch",
-    test: "pnpm build && firebase emulators:exec 'vitest run'",
+    test: "pnpm build && firebase emulators:exec 'vitest run --exclude=**/*.visual.test.tsx'",
+    "test:visual": "vitest run --config=vitest.config.visual.ts",
     "test:watch": "pnpm build && firebase emulators:exec 'vitest'"
   },
   dependencies: {
+    "@ag-grid-community/client-side-row-model": "32.3.9",
+    "@ag-grid-community/core": "32.3.9",
+    "@ag-grid-community/react": "32.3.9",
+    "@ag-grid-community/styles": "32.3.9",
     "@genkit-ai/ai": "1.18.0",
     "@genkit-ai/core": "1.18.0",
     "@genkit-ai/vertexai": "1.18.0",
@@ -2663,6 +2751,7 @@ var package_default = {
     "@lexical/react": "0.33.1",
     "@lexical/rich-text": "0.33.1",
     "@lexical/selection": "0.33.1",
+    "@lexical/table": "0.33.1",
     "@lexical/utils": "0.33.1",
     "@mantine/core": "4.2.12",
     "@mantine/hooks": "4.2.12",
@@ -2671,6 +2760,7 @@ var package_default = {
     "@mantine/spotlight": "4.2.12",
     "@preact/compat": "18.3.1",
     "@tabler/icons-preact": "3.35.0",
+    "@testing-library/preact": "3.2.4",
     "@types/body-parser": "1.19.3",
     "@types/fnv-plus": "1.3.2",
     "@types/gapi": "0.0.47",
@@ -2679,6 +2769,8 @@ var package_default = {
     "@types/google.accounts": "0.0.14",
     "@types/jsonwebtoken": "9.0.1",
     "@types/node": "24.3.1",
+    "@vitest/browser": "4.0.10",
+    "@vitest/browser-playwright": "4.0.10",
     concurrently: "7.6.0",
     esbuild: "0.25.9",
     firebase: "12.2.1",
@@ -2692,6 +2784,7 @@ var package_default = {
     "mdast-util-from-markdown": "2.0.1",
     "mdast-util-gfm": "3.0.0",
     "micromark-extension-gfm": "3.0.0",
+    playwright: "1.56.1",
     preact: "10.27.1",
     "preact-render-to-string": "6.6.1",
     "preact-router": "4.1.2",
@@ -2701,11 +2794,11 @@ var package_default = {
     tsup: "8.5.0",
     typescript: "5.9.2",
     vite: "7.1.4",
-    vitest: "3.2.4",
+    vitest: "4.0.10",
     yjs: "13.6.27"
   },
   peerDependencies: {
-    "@blinkk/root": "2.4.2",
+    "@blinkk/root": "2.4.4",
     "firebase-admin": ">=11",
     "firebase-functions": ">=4",
     preact: ">=10",
@@ -3129,6 +3222,22 @@ function cmsPlugin(options) {
           res.status(500).send("UNKNOWN SERVER ERROR");
         }
       });
+      process.on("unhandledRejection", (reason) => {
+        const err = reason;
+        if (err?.message?.includes("reauth related error") || err?.message?.includes("invalid_grant")) {
+          console.log("\n");
+          console.log("==================================================");
+          console.log("Google Cloud Reauthentication Required");
+          console.log("==================================================");
+          console.log(
+            "Your Google Cloud credentials have expired or require reauthentication."
+          );
+          console.log("Please run the following command to reauthenticate:");
+          console.log("\n    gcloud auth application-default login\n");
+          console.log("==================================================");
+          console.log("\n");
+        }
+      });
     }
   };
   if (process.env.NODE_ENV === "development" && options.watch !== false) {

package/dist/project.d.ts

@@ -1,4 +1,4 @@
-import { S as Schema, C as Collection } from './schema-sKTRRRSq.js';
+import { S as Schema, C as Collection } from './schema-BxPmMeRc.js';
 
 /**
  * Loads various files or configurations from the project.