@puckeditor/plugin-ai 0.3.0-canary.41e65b36 → 0.3.0-canary.78ff547c

package/dist/index.d.mts

@@ -1,6 +1,6 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { LanguageModelUsage, UIMessage, DataUIPart, ChatStatus } from 'ai';
-import { PuckAction } from '@measured/puck';
+import { PuckAction, Data } from '@measured/puck';
 
 type _JSONSchema = boolean | JSONSchema;
 type JSONSchema = {
@@ -160,6 +160,7 @@ type PuckDataParts = {
             input: any;
         }[];
     };
+    page: Data;
 };
 type PuckProviderMetadata = {
     tokenUsage?: LanguageModelUsage;

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { LanguageModelUsage, UIMessage, DataUIPart, ChatStatus } from 'ai';
-import { PuckAction } from '@measured/puck';
+import { PuckAction, Data } from '@measured/puck';
 
 type _JSONSchema = boolean | JSONSchema;
 type JSONSchema = {
@@ -160,6 +160,7 @@ type PuckDataParts = {
             input: any;
         }[];
     };
+    page: Data;
 };
 type PuckProviderMetadata = {
     tokenUsage?: LanguageModelUsage;

package/dist/index.js

@@ -1365,6 +1365,147 @@ function ExamplePrompt({
 
 // src/components/Chat/index.tsx
 var import_qler = __toESM(require("qler"));
+
+// ../reducer/index.ts
+init_react_import();
+var getSelectorForId = (state, id) => {
+  const node = state.indexes.nodes[id];
+  if (!node) return;
+  const zoneCompound = `${node.parentId}:${node.zone}`;
+  const index = state.indexes.zones[zoneCompound].contentIds.indexOf(id);
+  return { zone: zoneCompound, index };
+};
+var getItemById = (state, id) => state.indexes.nodes[id].data;
+var dispatchOp = (operation, {
+  getState,
+  dispatchAction,
+  config
+}) => {
+  const state = getState();
+  try {
+    if (operation.op === "add") {
+      if (operation.zone) {
+        dispatchAction({
+          type: "insert",
+          destinationIndex: operation.index,
+          destinationZone: operation.zone,
+          componentType: operation.type,
+          id: operation.id,
+          recordHistory: false
+        });
+        const existing = getItemById(getState(), operation.id);
+        if (!existing) {
+          throw new Error(
+            `Tried to update an item that doesn't exist: ${operation.id}`
+          );
+        }
+        dispatchAction({
+          type: "replace",
+          destinationIndex: operation.index,
+          destinationZone: operation.zone,
+          data: {
+            ...existing,
+            props: {
+              ...existing.props,
+              ...operation.props
+            }
+          },
+          recordHistory: false
+        });
+      }
+    } else if (operation.op === "update") {
+      const selector = getSelectorForId(state, operation.id);
+      const existing = getItemById(state, operation.id);
+      if (!selector || !existing) {
+        throw new Error(
+          `Tried to update an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "replace",
+        destinationIndex: selector.index,
+        destinationZone: selector.zone,
+        data: {
+          ...existing,
+          props: {
+            ...existing.props,
+            ...operation.props
+          }
+        },
+        recordHistory: false
+      });
+    } else if (operation.op === "updateRoot") {
+      const existing = state.data.root;
+      const defaultProps = config.root?.defaultProps ?? {};
+      dispatchAction({
+        type: "replaceRoot",
+        root: {
+          ...existing,
+          props: {
+            ...defaultProps,
+            ...existing.props,
+            ...operation.props
+          }
+        },
+        recordHistory: false
+      });
+    } else if (operation.op === "delete") {
+      const selector = getSelectorForId(state, operation.id);
+      if (!selector) {
+        throw new Error(
+          `Tried to delete an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "remove",
+        zone: selector.zone,
+        index: selector.index,
+        recordHistory: false
+      });
+    } else if (operation.op === "duplicate") {
+      const selector = getSelectorForId(state, operation.id);
+      if (!selector) {
+        throw new Error(
+          `Tried to duplicate an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "duplicate",
+        sourceZone: selector.zone,
+        sourceIndex: selector.index,
+        recordHistory: false
+      });
+    } else if (operation.op === "move") {
+      const selector = getSelectorForId(state, operation.id);
+      if (!selector) {
+        throw new Error(
+          `Tried to move an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "move",
+        sourceZone: selector.zone,
+        sourceIndex: selector.index,
+        destinationIndex: operation.index,
+        destinationZone: operation.zone,
+        recordHistory: false
+      });
+    } else if (operation.op === "reset") {
+      const defaultRootProps = config.root?.defaultProps ?? {};
+      dispatchAction({
+        type: "setData",
+        data: { content: [], root: defaultRootProps },
+        recordHistory: false
+      });
+    } else {
+      throw new Error(`Unknown operation: ${operation.op}`);
+    }
+  } catch (e) {
+    console.error("Error applying operation, skipping...", operation, e);
+  }
+};
+
+// src/components/Chat/index.tsx
 var import_jsx_runtime27 = require("react/jsx-runtime");
 var q = (0, import_qler.default)();
 var BENCHMARK = false;
@@ -1401,130 +1542,15 @@ function Chat2({ chat, host = "/api/puck/chat" }) {
         case "data-build-op": {
           const data = dataPart.data;
           q.queue(() => {
-            const puckDispatch2 = getPuck().dispatch;
-            try {
-              if (data.op === "add") {
-                if (data.zone) {
-                  puckDispatch2({
-                    type: "insert",
-                    destinationIndex: data.index,
-                    destinationZone: data.zone,
-                    componentType: data.type,
-                    id: data.id,
-                    recordHistory: false
-                  });
-                  const existing = getPuck().getItemById(data.id);
-                  if (!existing) {
-                    throw new Error(
-                      `Tried to update an item that doesn't exist: ${data.id}`
-                    );
-                  }
-                  puckDispatch2({
-                    type: "replace",
-                    destinationIndex: data.index,
-                    destinationZone: data.zone,
-                    data: {
-                      ...existing,
-                      props: {
-                        ...existing.props,
-                        ...data.props
-                      }
-                    },
-                    recordHistory: false
-                  });
-                }
-              } else if (data.op === "update") {
-                const selector = getPuck().getSelectorForId(data.id);
-                const existing = getPuck().getItemById(data.id);
-                if (!selector || !existing) {
-                  throw new Error(
-                    `Tried to update an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "replace",
-                  destinationIndex: selector.index,
-                  destinationZone: selector.zone,
-                  data: {
-                    ...existing,
-                    props: {
-                      ...existing.props,
-                      ...data.props
-                    }
-                  },
-                  recordHistory: false
-                });
-              } else if (data.op === "updateRoot") {
-                const { appState, config } = getPuck();
-                const existing = appState.data.root;
-                const defaultProps = config.root?.defaultProps ?? {};
-                puckDispatch2({
-                  type: "replaceRoot",
-                  root: {
-                    ...existing,
-                    props: {
-                      ...defaultProps,
-                      ...existing.props,
-                      ...data.props
-                    }
-                  },
-                  recordHistory: false
-                });
-              } else if (data.op === "delete") {
-                const selector = getPuck().getSelectorForId(data.id);
-                if (!selector) {
-                  throw new Error(
-                    `Tried to delete an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "remove",
-                  zone: selector.zone,
-                  index: selector.index,
-                  recordHistory: false
-                });
-              } else if (data.op === "duplicate") {
-                const selector = getPuck().getSelectorForId(data.id);
-                if (!selector) {
-                  throw new Error(
-                    `Tried to duplicate an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "duplicate",
-                  sourceZone: selector.zone,
-                  sourceIndex: selector.index,
-                  recordHistory: false
-                });
-              } else if (data.op === "move") {
-                const selector = getPuck().getSelectorForId(data.id);
-                if (!selector) {
-                  throw new Error(
-                    `Tried to move an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "move",
-                  sourceZone: selector.zone,
-                  sourceIndex: selector.index,
-                  destinationIndex: data.index,
-                  destinationZone: data.zone,
-                  recordHistory: false
-                });
-              } else if (data.op === "reset") {
-                const { config } = getPuck();
-                const defaultRootProps = config.root?.defaultProps ?? {};
-                puckDispatch2({
-                  type: "setData",
-                  data: { content: [], root: defaultRootProps },
-                  recordHistory: false
-                });
-              } else {
-                throw new Error(`Unknown operation: ${data.op}`);
-              }
-            } catch (e) {
-              console.error("Error applying operation, skipping...", data, e);
-            }
+            const { dispatch, config } = getPuck();
+            dispatchOp(data, {
+              getState: () => {
+                const { __private } = getPuck();
+                return __private.appState;
+              },
+              dispatchAction: dispatch,
+              config
+            });
           });
           return;
         }

package/dist/index.mjs

@@ -1363,6 +1363,147 @@ function ExamplePrompt({
 
 // src/components/Chat/index.tsx
 import qler from "qler";
+
+// ../reducer/index.ts
+init_react_import();
+var getSelectorForId = (state, id) => {
+  const node = state.indexes.nodes[id];
+  if (!node) return;
+  const zoneCompound = `${node.parentId}:${node.zone}`;
+  const index = state.indexes.zones[zoneCompound].contentIds.indexOf(id);
+  return { zone: zoneCompound, index };
+};
+var getItemById = (state, id) => state.indexes.nodes[id].data;
+var dispatchOp = (operation, {
+  getState,
+  dispatchAction,
+  config
+}) => {
+  const state = getState();
+  try {
+    if (operation.op === "add") {
+      if (operation.zone) {
+        dispatchAction({
+          type: "insert",
+          destinationIndex: operation.index,
+          destinationZone: operation.zone,
+          componentType: operation.type,
+          id: operation.id,
+          recordHistory: false
+        });
+        const existing = getItemById(getState(), operation.id);
+        if (!existing) {
+          throw new Error(
+            `Tried to update an item that doesn't exist: ${operation.id}`
+          );
+        }
+        dispatchAction({
+          type: "replace",
+          destinationIndex: operation.index,
+          destinationZone: operation.zone,
+          data: {
+            ...existing,
+            props: {
+              ...existing.props,
+              ...operation.props
+            }
+          },
+          recordHistory: false
+        });
+      }
+    } else if (operation.op === "update") {
+      const selector = getSelectorForId(state, operation.id);
+      const existing = getItemById(state, operation.id);
+      if (!selector || !existing) {
+        throw new Error(
+          `Tried to update an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "replace",
+        destinationIndex: selector.index,
+        destinationZone: selector.zone,
+        data: {
+          ...existing,
+          props: {
+            ...existing.props,
+            ...operation.props
+          }
+        },
+        recordHistory: false
+      });
+    } else if (operation.op === "updateRoot") {
+      const existing = state.data.root;
+      const defaultProps = config.root?.defaultProps ?? {};
+      dispatchAction({
+        type: "replaceRoot",
+        root: {
+          ...existing,
+          props: {
+            ...defaultProps,
+            ...existing.props,
+            ...operation.props
+          }
+        },
+        recordHistory: false
+      });
+    } else if (operation.op === "delete") {
+      const selector = getSelectorForId(state, operation.id);
+      if (!selector) {
+        throw new Error(
+          `Tried to delete an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "remove",
+        zone: selector.zone,
+        index: selector.index,
+        recordHistory: false
+      });
+    } else if (operation.op === "duplicate") {
+      const selector = getSelectorForId(state, operation.id);
+      if (!selector) {
+        throw new Error(
+          `Tried to duplicate an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "duplicate",
+        sourceZone: selector.zone,
+        sourceIndex: selector.index,
+        recordHistory: false
+      });
+    } else if (operation.op === "move") {
+      const selector = getSelectorForId(state, operation.id);
+      if (!selector) {
+        throw new Error(
+          `Tried to move an item that doesn't exist: ${operation.id}`
+        );
+      }
+      dispatchAction({
+        type: "move",
+        sourceZone: selector.zone,
+        sourceIndex: selector.index,
+        destinationIndex: operation.index,
+        destinationZone: operation.zone,
+        recordHistory: false
+      });
+    } else if (operation.op === "reset") {
+      const defaultRootProps = config.root?.defaultProps ?? {};
+      dispatchAction({
+        type: "setData",
+        data: { content: [], root: defaultRootProps },
+        recordHistory: false
+      });
+    } else {
+      throw new Error(`Unknown operation: ${operation.op}`);
+    }
+  } catch (e) {
+    console.error("Error applying operation, skipping...", operation, e);
+  }
+};
+
+// src/components/Chat/index.tsx
 import { jsx as jsx27, jsxs as jsxs14 } from "react/jsx-runtime";
 var q = qler();
 var BENCHMARK = false;
@@ -1399,130 +1540,15 @@ function Chat2({ chat, host = "/api/puck/chat" }) {
         case "data-build-op": {
           const data = dataPart.data;
           q.queue(() => {
-            const puckDispatch2 = getPuck().dispatch;
-            try {
-              if (data.op === "add") {
-                if (data.zone) {
-                  puckDispatch2({
-                    type: "insert",
-                    destinationIndex: data.index,
-                    destinationZone: data.zone,
-                    componentType: data.type,
-                    id: data.id,
-                    recordHistory: false
-                  });
-                  const existing = getPuck().getItemById(data.id);
-                  if (!existing) {
-                    throw new Error(
-                      `Tried to update an item that doesn't exist: ${data.id}`
-                    );
-                  }
-                  puckDispatch2({
-                    type: "replace",
-                    destinationIndex: data.index,
-                    destinationZone: data.zone,
-                    data: {
-                      ...existing,
-                      props: {
-                        ...existing.props,
-                        ...data.props
-                      }
-                    },
-                    recordHistory: false
-                  });
-                }
-              } else if (data.op === "update") {
-                const selector = getPuck().getSelectorForId(data.id);
-                const existing = getPuck().getItemById(data.id);
-                if (!selector || !existing) {
-                  throw new Error(
-                    `Tried to update an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "replace",
-                  destinationIndex: selector.index,
-                  destinationZone: selector.zone,
-                  data: {
-                    ...existing,
-                    props: {
-                      ...existing.props,
-                      ...data.props
-                    }
-                  },
-                  recordHistory: false
-                });
-              } else if (data.op === "updateRoot") {
-                const { appState, config } = getPuck();
-                const existing = appState.data.root;
-                const defaultProps = config.root?.defaultProps ?? {};
-                puckDispatch2({
-                  type: "replaceRoot",
-                  root: {
-                    ...existing,
-                    props: {
-                      ...defaultProps,
-                      ...existing.props,
-                      ...data.props
-                    }
-                  },
-                  recordHistory: false
-                });
-              } else if (data.op === "delete") {
-                const selector = getPuck().getSelectorForId(data.id);
-                if (!selector) {
-                  throw new Error(
-                    `Tried to delete an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "remove",
-                  zone: selector.zone,
-                  index: selector.index,
-                  recordHistory: false
-                });
-              } else if (data.op === "duplicate") {
-                const selector = getPuck().getSelectorForId(data.id);
-                if (!selector) {
-                  throw new Error(
-                    `Tried to duplicate an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "duplicate",
-                  sourceZone: selector.zone,
-                  sourceIndex: selector.index,
-                  recordHistory: false
-                });
-              } else if (data.op === "move") {
-                const selector = getPuck().getSelectorForId(data.id);
-                if (!selector) {
-                  throw new Error(
-                    `Tried to move an item that doesn't exist: ${data.id}`
-                  );
-                }
-                puckDispatch2({
-                  type: "move",
-                  sourceZone: selector.zone,
-                  sourceIndex: selector.index,
-                  destinationIndex: data.index,
-                  destinationZone: data.zone,
-                  recordHistory: false
-                });
-              } else if (data.op === "reset") {
-                const { config } = getPuck();
-                const defaultRootProps = config.root?.defaultProps ?? {};
-                puckDispatch2({
-                  type: "setData",
-                  data: { content: [], root: defaultRootProps },
-                  recordHistory: false
-                });
-              } else {
-                throw new Error(`Unknown operation: ${data.op}`);
-              }
-            } catch (e) {
-              console.error("Error applying operation, skipping...", data, e);
-            }
+            const { dispatch, config } = getPuck();
+            dispatchOp(data, {
+              getState: () => {
+                const { __private } = getPuck();
+                return __private.appState;
+              },
+              dispatchAction: dispatch,
+              config
+            });
           });
           return;
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@puckeditor/plugin-ai",
-  "version": "0.3.0-canary.41e65b36",
+  "version": "0.3.0-canary.78ff547c",
   "author": "Chris Villa <chris@puckeditor.com>",
   "repository": "puckeditor/puck",
   "bugs": "https://github.com/puckeditor/puck/issues",
@@ -36,7 +36,7 @@
   },
   "devDependencies": {
     "@ai-sdk/react": "^2.0.29",
-    "@measured/puck": "0.21.0-canary.8416d520",
+    "@measured/puck": "0.21.0-canary.c0db75c1",
     "@puckeditor/ai-types": "workspace:*",
     "@puckeditor/platform-types": "workspace:*",
     "@types/jest": "^30.0.0",
@@ -53,7 +53,7 @@
     "typescript": "^5.5.4"
   },
   "peerDependencies": {
-    "@measured/puck": "0.21.0-canary.8416d520",
+    "@measured/puck": "0.21.0-canary.c0db75c1",
     "react": "^18.0.0 || ^19.0.0"
   }
 }