@platejs/ai 50.1.2 → 50.2.1

package/dist/react/index.mjs

@@ -46,11 +46,16 @@ var removeAINodes = (editor, { at = [] } = {}) => {
 };
 
 // src/lib/transforms/undoAI.ts
+import { getTransientSuggestionKey } from "@platejs/suggestion";
 var undoAI = (editor) => {
-  if (editor.history.undos.at(-1)?.ai && editor.api.some({
+  const hasAINodeOrAISuggestion = editor.api.some({
     at: [],
     match: (n) => !!n.ai
-  })) {
+  }) || editor.api.some({
+    at: [],
+    match: (n) => !!n[getTransientSuggestionKey()]
+  });
+  if (editor.history.undos.at(-1)?.ai && hasAINodeOrAISuggestion) {
     editor.undo();
     editor.history.redos.pop();
   }
@@ -2925,22 +2930,59 @@ import {
 import { createTPlatePlugin } from "platejs/react";
 
 // src/react/ai-chat/transforms/acceptAIChat.ts
+import {
+  acceptSuggestion,
+  getSuggestionKey,
+  getTransientSuggestionKey as getTransientSuggestionKey2
+} from "@platejs/suggestion";
+import { SuggestionPlugin } from "@platejs/suggestion/react";
 import { KEYS as KEYS4 } from "platejs";
 import { getEditorPlugin } from "platejs/react";
 var acceptAIChat = (editor) => {
-  const { tf } = getEditorPlugin(editor, AIPlugin);
-  const api = editor.getApi({ key: KEYS4.ai });
-  const lastAINodePath = api.aiChat.node({ at: [], reverse: true })[1];
-  withAIBatch(editor, () => {
-    tf.ai.removeMarks();
-    editor.getTransforms(AIChatPlugin).aiChat.removeAnchor();
+  const mode = editor.getOption(AIChatPlugin, "mode");
+  if (mode === "insert") {
+    const { tf } = getEditorPlugin(editor, AIPlugin);
+    const api = editor.getApi({ key: KEYS4.ai });
+    const lastAINodePath = api.aiChat.node({ at: [], reverse: true })[1];
+    withAIBatch(editor, () => {
+      tf.ai.removeMarks();
+      editor.getTransforms(AIChatPlugin).aiChat.removeAnchor();
+    });
+    api.aiChat.hide();
+    editor.tf.focus();
+    const focusPoint = editor.api.end(lastAINodePath);
+    editor.tf.setSelection({
+      anchor: focusPoint,
+      focus: focusPoint
+    });
+  }
+  if (mode === "chat") {
+    withAIBatch(editor, () => {
+      acceptSuggestions(editor);
+    });
+    editor.getApi(AIChatPlugin).aiChat.hide();
+  }
+};
+var acceptSuggestions = (editor) => {
+  const suggestions = editor.getApi(SuggestionPlugin).suggestion.nodes({
+    transient: true
+  });
+  suggestions.forEach(([suggestionNode]) => {
+    const suggestionData = editor.getApi(SuggestionPlugin).suggestion.suggestionData(suggestionNode);
+    if (!suggestionData) return;
+    const description = {
+      createdAt: new Date(suggestionData.createdAt),
+      keyId: getSuggestionKey(suggestionData.id),
+      suggestionId: suggestionData.id,
+      type: suggestionData.type,
+      userId: suggestionData.userId
+    };
+    acceptSuggestion(editor, description);
   });
-  api.aiChat.hide();
-  editor.tf.focus();
-  const focusPoint = editor.api.end(lastAINodePath);
-  editor.tf.setSelection({
-    anchor: focusPoint,
-    focus: focusPoint
+  editor.tf.unsetNodes([getTransientSuggestionKey2()], {
+    at: [],
+    mode: "all",
+    match: (n) => !!n[getTransientSuggestionKey2()]
   });
 };
 
@@ -3062,6 +3104,29 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
 
 // src/react/ai-chat/transforms/insertBelowAIChat.ts
 var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
+  const { toolName } = editor.getOptions(AIChatPlugin);
+  if (toolName === "generate")
+    return insertBelowGenerate(editor, sourceEditor, { format });
+  const selectedBlocks = editor.getApi(BlockSelectionPlugin2).blockSelection.getNodes();
+  const selectedIds = editor.getOptions(BlockSelectionPlugin2).selectedIds;
+  editor.getTransforms(AIPlugin).ai.undo();
+  const insertBlocksAndSelect = editor.getTransforms(BlockSelectionPlugin2).blockSelection.insertBlocksAndSelect;
+  if (!selectedIds || selectedIds.size === 0) return;
+  const lastBlock = selectedBlocks.at(-1);
+  if (!lastBlock) return;
+  const nextPath = PathApi.next(lastBlock[1]);
+  const nodes = selectedBlocks.map((block) => block[0]);
+  insertBlocksAndSelect(nodes, {
+    at: nextPath,
+    insertedCallback: () => {
+      withAIBatch(editor, () => {
+        acceptSuggestions(editor);
+      });
+    }
+  });
+  editor.getApi(AIChatPlugin).aiChat.hide({ focus: false });
+};
+var insertBelowGenerate = (editor, sourceEditor, { format = "single" } = {}) => {
   if (!sourceEditor || sourceEditor.api.isEmpty()) return;
   const isBlockSelecting = editor.getOption(
     BlockSelectionPlugin2,
@@ -3138,14 +3203,23 @@ var removeAnchorAIChat = (editor, options) => {
 import { KEYS as KEYS8 } from "platejs";
 import { getEditorPlugin as getEditorPlugin2 } from "platejs/react";
 var resetAIChat = (editor) => {
-  const { api, getOptions } = getEditorPlugin2(editor, {
-    key: KEYS8.aiChat
-  });
+  const { api, getOptions, setOptions } = getEditorPlugin2(
+    editor,
+    {
+      key: KEYS8.aiChat
+    }
+  );
   api.aiChat.stop();
   const chat = getOptions().chat;
   if (chat.messages && chat.messages.length > 0) {
     chat.setMessages?.([]);
   }
+  setOptions({
+    _replaceIds: [],
+    chatNodes: [],
+    mode: "insert",
+    toolName: "generate"
+  });
   editor.getTransforms(AIPlugin).ai.undo();
 };
 
@@ -3182,10 +3256,14 @@ var submitAIChat = (editor, input, {
   setOption("mode", mode);
   setOption("toolName", toolName ?? null);
   const blocks = editor.getApi(BlockSelectionPlugin3).blockSelection.getNodes();
+  const blocksRange = editor.api.nodesRange(blocks);
   const promptText = getEditorPrompt(editor, {
     prompt
   });
-  const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
+  const selection = blocks.length > 0 ? blocksRange : editor.selection;
+  const chatNodes = blocks.length > 0 ? blocks.map((block) => block[0]) : editor.api.fragment();
+  setOption("chatNodes", chatNodes);
+  setOption("chatSelection", blocks.length > 0 ? null : editor.selection);
   const ctx = {
     children: editor.children,
     selection: selection ?? null,
@@ -3210,7 +3288,7 @@ var withAIChat = ({
   api,
   editor,
   getOptions,
-  tf: { insertText, normalizeNode, setSelection },
+  tf: { apply, insertText, normalizeNode },
   type
 }) => {
   const tf = editor.getTransforms(AIPlugin);
@@ -3272,8 +3350,11 @@ var AIChatPlugin = createTPlatePlugin({
     _blockChunks: "",
     _blockPath: null,
     _mdxName: null,
+    _replaceIds: [],
     aiEditor: null,
     chat: { messages: [] },
+    chatNodes: [],
+    chatSelection: null,
     experimental_lastTextId: null,
     mode: "insert",
     open: false,
@@ -3313,11 +3394,25 @@ var AIChatPlugin = createTPlatePlugin({
       });
     },
     reload: () => {
-      const { chat, mode } = getOptions();
-      if (mode === "insert") {
-        editor.getTransforms(AIPlugin).ai.undo();
+      const { chat, chatNodes, chatSelection } = getOptions();
+      editor.getTransforms(AIPlugin).ai.undo();
+      if (chatSelection) {
+        editor.tf.setSelection(chatSelection);
+      } else {
+        editor.getApi(BlockSelectionPlugin4).blockSelection.set(chatNodes.map((node) => node.id));
       }
-      void chat.regenerate?.();
+      const blocks = editor.getApi(BlockSelectionPlugin4).blockSelection.getNodes();
+      const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
+      const ctx = {
+        children: editor.children,
+        selection: selection ?? null,
+        toolName: getOption("toolName")
+      };
+      void chat.regenerate?.({
+        body: {
+          ctx
+        }
+      });
     },
     stop: () => {
       setOption("streaming", false);
@@ -3325,12 +3420,15 @@ var AIChatPlugin = createTPlatePlugin({
     }
   };
 }).extendApi(({ api, editor, getOptions, setOption, tf, type }) => ({
-  hide: () => {
+  hide: ({ focus = true } = {}) => {
     api.aiChat.reset();
     setOption("open", false);
-    if (editor.getOption(BlockSelectionPlugin4, "isSelectingSome")) {
-    } else {
-      editor.tf.focus();
+    if (focus) {
+      if (editor.getOption(BlockSelectionPlugin4, "isSelectingSome")) {
+        editor.getApi(BlockSelectionPlugin4).blockSelection.focus();
+      } else {
+        editor.tf.focus();
+      }
     }
     const lastBatch = editor.history.undos.at(-1);
     if (lastBatch?.ai) {
@@ -3340,6 +3438,7 @@ var AIChatPlugin = createTPlatePlugin({
   },
   show: () => {
     api.aiChat.reset();
+    setOption("toolName", null);
     getOptions().chat.setMessages?.([]);
     setOption("open", true);
   }
@@ -3884,25 +3983,161 @@ var getCurrentBlockPath = (editor) => {
   return path;
 };
 
+// src/react/ai-chat/utils/applyAISuggestions.ts
+import { deserializeMd as deserializeMd2 } from "@platejs/markdown";
+import { BlockSelectionPlugin as BlockSelectionPlugin6 } from "@platejs/selection/react";
+import {
+  diffToSuggestions,
+  getTransientSuggestionKey as getTransientSuggestionKey3,
+  SkipSuggestionDeletes
+} from "@platejs/suggestion";
+import {
+  ElementApi as ElementApi6,
+  KEYS as KEYS17,
+  nanoid,
+  TextApi as TextApi5
+} from "platejs";
+var applyAISuggestions = (editor, content) => {
+  const isBlockSelecting = editor.getOption(
+    BlockSelectionPlugin6,
+    "isSelectingSome"
+  );
+  const { chatNodes } = editor.getOptions(AIChatPlugin);
+  if (isBlockSelecting) {
+    const setReplaceIds = (ids) => {
+      editor.setOption(AIChatPlugin, "_replaceIds", ids);
+    };
+    if (editor.getOption(AIChatPlugin, "_replaceIds").length === 0) {
+      setReplaceIds(chatNodes.map((node) => node.id));
+    }
+    const diffNodes = getDiffNodes(editor, content);
+    const replaceNodes = Array.from(
+      editor.api.nodes({
+        at: [],
+        match: (n) => ElementApi6.isElement(n) && editor.getOption(AIChatPlugin, "_replaceIds").includes(n.id)
+      })
+    );
+    replaceNodes.forEach(([node, path], index) => {
+      const replaceNode = node;
+      const diffNode = diffNodes[index];
+      const isSameString = SkipSuggestionDeletes(editor, replaceNode) === SkipSuggestionDeletes(editor, diffNode);
+      const isSameSuggestion = replaceNode.suggestion?.type === diffNode.suggestion?.type;
+      if (isSameString && isSameSuggestion && node.id == diffNode.id) {
+        return;
+      }
+      if (index === replaceNodes.length - 1 && diffNodes.length > replaceNodes.length) {
+        editor.tf.replaceNodes(diffNodes.slice(index), {
+          at: path
+        });
+      } else {
+        editor.tf.replaceNodes(diffNode, {
+          at: path
+        });
+      }
+    });
+    editor.getApi(BlockSelectionPlugin6).blockSelection.set(diffNodes.map((node) => node.id));
+    setReplaceIds(diffNodes.map((node) => node.id));
+  } else {
+    const diffNodes = getDiffNodes(editor, content);
+    editor.tf.insertFragment(diffNodes);
+    const nodes = editor.api.nodes({
+      at: [],
+      mode: "lowest",
+      match: (n) => TextApi5.isText(n) && !!n[getTransientSuggestionKey3()]
+    });
+    const range = editor.api.nodesRange(Array.from(nodes));
+    editor.tf.setSelection(range);
+    return;
+  }
+};
+var withProps = (diffNodes, chatNodes) => {
+  return diffNodes.map((node, index) => {
+    if (!ElementApi6.isElement(node)) return node;
+    const originalNode = chatNodes?.[index];
+    return {
+      ...node,
+      ...originalNode ?? { id: nanoid() },
+      children: node.children
+    };
+  });
+};
+var withTransient = (diffNodes) => {
+  return diffNodes.map((node) => {
+    if (TextApi5.isText(node)) {
+      return {
+        ...node,
+        [getTransientSuggestionKey3()]: true
+      };
+    } else {
+      return {
+        ...node,
+        children: withTransient(node.children),
+        [getTransientSuggestionKey3()]: true
+      };
+    }
+  });
+};
+var withoutSuggestionAndComments = (nodes) => {
+  return nodes.map((node) => {
+    if (TextApi5.isText(node)) {
+      if (node[KEYS17.suggestion] || node[KEYS17.comment]) {
+        return {
+          text: node.text
+        };
+      }
+      return node;
+    }
+    if (ElementApi6.isElement(node)) {
+      if (node[KEYS17.suggestion]) {
+        const nodeWithoutSuggestion = {};
+        Object.keys(node).forEach((key) => {
+          if (key !== KEYS17.suggestion && !key.startsWith(KEYS17.suggestion)) {
+            nodeWithoutSuggestion[key] = node[key];
+          }
+        });
+        return {
+          ...nodeWithoutSuggestion,
+          children: withoutSuggestionAndComments(node.children)
+        };
+      }
+      return {
+        ...node,
+        children: withoutSuggestionAndComments(node.children)
+      };
+    }
+    return node;
+  });
+};
+var getDiffNodes = (editor, aiContent) => {
+  const chatNodes = withoutSuggestionAndComments(
+    editor.getOption(AIChatPlugin, "chatNodes")
+  );
+  const aiNodes = withProps(deserializeMd2(editor, aiContent), chatNodes);
+  const diffNodes = withTransient(
+    diffToSuggestions(editor, chatNodes, aiNodes, { ignoreProps: ["id"] })
+  );
+  return diffNodes;
+};
+
 // src/react/copilot/CopilotPlugin.tsx
 import { serializeMd } from "@platejs/markdown";
 import debounce from "lodash/debounce.js";
 import {
   bindFirst as bindFirst3,
-  KEYS as KEYS21,
+  KEYS as KEYS22,
   NodeApi as NodeApi4
 } from "platejs";
 import { createTPlatePlugin as createTPlatePlugin2 } from "platejs/react";
 
 // src/react/copilot/renderCopilotBelowNodes.tsx
 import React from "react";
-import { KEYS as KEYS17 } from "platejs";
+import { KEYS as KEYS18 } from "platejs";
 import { getEditorPlugin as getEditorPlugin4 } from "platejs/react";
 var renderCopilotBelowNodes = ({
   editor
 }) => {
   const copilot = getEditorPlugin4(editor, {
-    key: KEYS17.copilot
+    key: KEYS18.copilot
   });
   const { renderGhostText: GhostText } = copilot.getOptions();
   if (!GhostText) return;
@@ -3913,10 +4148,10 @@ var renderCopilotBelowNodes = ({
 
 // src/react/copilot/transforms/acceptCopilot.ts
 import { deserializeInlineMd } from "@platejs/markdown";
-import { KEYS as KEYS18 } from "platejs";
+import { KEYS as KEYS19 } from "platejs";
 var acceptCopilot = (editor) => {
   const { suggestionText } = editor.getOptions({
-    key: KEYS18.copilot
+    key: KEYS19.copilot
   });
   if (!suggestionText?.length) return false;
   editor.tf.insertFragment(deserializeInlineMd(editor, suggestionText));
@@ -3924,7 +4159,7 @@ var acceptCopilot = (editor) => {
 
 // src/react/copilot/transforms/acceptCopilotNextWord.ts
 import { deserializeInlineMd as deserializeInlineMd2 } from "@platejs/markdown";
-import { KEYS as KEYS20 } from "platejs";
+import { KEYS as KEYS21 } from "platejs";
 import { getEditorPlugin as getEditorPlugin6 } from "platejs/react";
 
 // src/react/copilot/utils/callCompletionApi.ts
@@ -4042,17 +4277,17 @@ var getNextWord = ({ text }) => {
 };
 
 // src/react/copilot/utils/triggerCopilotSuggestion.ts
-import { KEYS as KEYS19 } from "platejs";
+import { KEYS as KEYS20 } from "platejs";
 import { getEditorPlugin as getEditorPlugin5 } from "platejs/react";
 var triggerCopilotSuggestion = async (editor) => {
   const { api, getOptions, setOption } = getEditorPlugin5(
     editor,
     {
-      key: KEYS19.copilot
+      key: KEYS20.copilot
     }
   );
   const { completeOptions, getPrompt, isLoading, triggerQuery } = getOptions();
-  if (isLoading || editor.getOptions({ key: KEYS19.aiChat }).chat?.isLoading) {
+  if (isLoading || editor.getOptions({ key: KEYS20.aiChat }).chat?.isLoading) {
     return false;
   }
   if (!triggerQuery({ editor })) return false;
@@ -4086,7 +4321,7 @@ var withoutAbort = (editor, fn) => {
 // src/react/copilot/transforms/acceptCopilotNextWord.ts
 var acceptCopilotNextWord = (editor) => {
   const { api, getOptions } = getEditorPlugin6(editor, {
-    key: KEYS20.copilot
+    key: KEYS21.copilot
   });
   const { getNextWord: getNextWord2, suggestionText } = getOptions();
   if (!suggestionText?.length) {
@@ -4202,7 +4437,7 @@ var withCopilot = ({
 
 // src/react/copilot/CopilotPlugin.tsx
 var CopilotPlugin = createTPlatePlugin2({
-  key: KEYS21.copilot,
+  key: KEYS22.copilot,
   handlers: {
     onBlur: ({ api }) => {
       api.copilot.reject();
@@ -4224,7 +4459,7 @@ var CopilotPlugin = createTPlatePlugin2({
     suggestionNodeId: null,
     suggestionText: null,
     autoTriggerQuery: ({ editor }) => {
-      if (editor.getOptions({ key: KEYS21.copilot }).suggestionText) {
+      if (editor.getOptions({ key: KEYS22.copilot }).suggestionText) {
         return false;
       }
       const isEmpty = editor.api.isEmpty(editor.selection, { block: true });
@@ -4314,7 +4549,9 @@ export {
   acceptAIChat,
   acceptCopilot,
   acceptCopilotNextWord,
+  acceptSuggestions,
   aiCommentToRange,
+  applyAISuggestions,
   callCompletionApi,
   createFormattedBlocks,
   escapeInput,
@@ -4325,6 +4562,7 @@ export {
   getListNode,
   getNextWord,
   insertBelowAIChat,
+  insertBelowGenerate,
   isCompleteCodeBlock,
   isCompleteMath,
   isSameNode,