@lofcz/platejs-ai 52.3.2 → 52.3.6

package/dist/getEditorPrompt-axDm9Fg4.js

@@ -0,0 +1,235 @@
+import { ElementApi, KEYS, TextApi, bindFirst, createTSlatePlugin, getPluginType } from "platejs";
+import cloneDeep from "lodash/cloneDeep.js";
+import { getTransientSuggestionKey } from "@platejs/suggestion";
+import { isSelecting } from "@platejs/selection";
+
+//#region src/lib/transforms/aiStreamSnapshot.ts
+const AI_PREVIEW_KEY = "aiPreview";
+const AI_STREAM_SNAPSHOT = /* @__PURE__ */ new WeakMap();
+const clearAIPreview = (editor) => {
+	AI_STREAM_SNAPSHOT.delete(editor);
+};
+const getAIPreview = (editor) => AI_STREAM_SNAPSHOT.get(editor);
+const getAIPreviewRange = (editor) => {
+	let closed = false;
+	let end = -1;
+	let invalid = false;
+	let start = -1;
+	editor.children.forEach((node, index) => {
+		if (!node?.[AI_PREVIEW_KEY]) {
+			if (start !== -1) closed = true;
+			return;
+		}
+		if (closed) {
+			invalid = true;
+			return;
+		}
+		if (start === -1) start = index;
+		end = index;
+	});
+	if (invalid) return { kind: "invalid" };
+	if (start === -1 && end === -1) return { kind: "none" };
+	return {
+		end,
+		kind: "range",
+		start
+	};
+};
+const removeAIPreviewAnchor = (editor) => {
+	const aiChatType = getPluginType(editor, KEYS.aiChat);
+	editor.tf.removeNodes({
+		at: [],
+		match: (node) => ElementApi.isElement(node) && node.type === aiChatType
+	});
+};
+const restoreAIPreviewSelection = (editor, selection) => {
+	if (selection) {
+		editor.tf.select(cloneDeep(selection));
+		return;
+	}
+	editor.tf.deselect();
+};
+const removePreviewRange = (editor, range) => {
+	for (let index = range.end; index >= range.start; index--) editor.tf.removeNodes({ at: [index] });
+};
+const replacePreviewRange = (editor, range, blocks) => {
+	removePreviewRange(editor, range);
+	if (blocks.length === 0) return;
+	editor.tf.insertNodes(cloneDeep(blocks), { at: [range.start] });
+};
+const cloneAcceptedPreviewBlocks = (editor, range) => {
+	const aiType = getPluginType(editor, KEYS.ai);
+	const blocks = cloneDeep(editor.children.slice(range.start, range.end + 1));
+	const stripNode = (node) => {
+		if (ElementApi.isElement(node)) {
+			const { [AI_PREVIEW_KEY]: _preview, children, ...rest$1 } = node;
+			return {
+				...rest$1,
+				children: children.map(stripNode)
+			};
+		}
+		const { [aiType]: _ai, ...rest } = node;
+		return rest;
+	};
+	return blocks.map(stripNode);
+};
+const beginAIPreview = (editor, { originalBlocks = [] } = {}) => {
+	if (getAIPreview(editor)) return false;
+	AI_STREAM_SNAPSHOT.set(editor, {
+		originalBlocks: cloneDeep(originalBlocks),
+		selectionBefore: cloneDeep(editor.selection)
+	});
+	return true;
+};
+const hasAIPreview = (editor) => !!getAIPreview(editor);
+const cancelAIPreview = (editor) => {
+	const preview = getAIPreview(editor);
+	if (!preview) return false;
+	const range = getAIPreviewRange(editor);
+	if (range.kind === "invalid") return false;
+	editor.tf.withoutSaving(() => {
+		if (range.kind === "range") replacePreviewRange(editor, range, preview.originalBlocks);
+		removeAIPreviewAnchor(editor);
+		restoreAIPreviewSelection(editor, preview.selectionBefore);
+	});
+	clearAIPreview(editor);
+	return true;
+};
+const discardAIPreview = (editor) => {
+	if (!getAIPreview(editor)) return false;
+	clearAIPreview(editor);
+	return true;
+};
+const acceptAIPreview = (editor, _value) => {
+	const preview = getAIPreview(editor);
+	if (!preview) return false;
+	const range = getAIPreviewRange(editor);
+	if (range.kind === "invalid") return false;
+	if (range.kind === "range") {
+		const acceptedBlocks = cloneAcceptedPreviewBlocks(editor, range);
+		editor.tf.withoutSaving(() => {
+			replacePreviewRange(editor, range, preview.originalBlocks);
+			removeAIPreviewAnchor(editor);
+			restoreAIPreviewSelection(editor, preview.selectionBefore);
+		});
+		editor.tf.withNewBatch(() => {
+			if (preview.originalBlocks.length > 0) for (let index = preview.originalBlocks.length - 1; index >= 0; index--) editor.tf.removeNodes({ at: [range.start + index] });
+			if (acceptedBlocks.length > 0) editor.tf.insertNodes(acceptedBlocks, { at: [range.start] });
+		});
+		const lastBatch = editor.history?.undos.at(-1);
+		if (lastBatch) lastBatch.selectionBefore = cloneDeep(preview.selectionBefore);
+	} else editor.tf.withoutSaving(() => {
+		removeAIPreviewAnchor(editor);
+	});
+	clearAIPreview(editor);
+	return true;
+};
+
+//#endregion
+//#region src/lib/transforms/insertAINodes.ts
+const insertAINodes = (editor, nodes, { target } = {}) => {
+	if (!target && !editor.selection?.focus.path) return;
+	const aiNodes = nodes.map((node) => ({
+		...node,
+		ai: true
+	}));
+	editor.tf.withoutNormalizing(() => {
+		editor.tf.insertNodes(aiNodes, {
+			at: editor.api.end(target || editor.selection.focus.path),
+			select: true
+		});
+		editor.tf.collapse({ edge: "end" });
+	});
+};
+
+//#endregion
+//#region src/lib/transforms/removeAIMarks.ts
+const removeAIMarks = (editor, { at = [] } = {}) => {
+	const nodeType = getPluginType(editor, KEYS.ai);
+	editor.tf.unsetNodes(nodeType, {
+		at,
+		match: (n) => n[nodeType]
+	});
+};
+
+//#endregion
+//#region src/lib/transforms/removeAINodes.ts
+const removeAINodes = (editor, { at = [] } = {}) => {
+	editor.tf.removeNodes({
+		at,
+		match: (n) => TextApi.isText(n) && !!n.ai
+	});
+};
+
+//#endregion
+//#region src/lib/transforms/undoAI.ts
+const undoAI = (editor) => {
+	if (hasAIPreview(editor) && cancelAIPreview(editor)) return;
+	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();
+		return;
+	}
+	if (hasAINodeOrAISuggestion) cancelAIPreview(editor);
+};
+
+//#endregion
+//#region src/lib/transforms/withAIBatch.ts
+const withAIBatch = (editor, fn, { split } = {}) => {
+	if (split) editor.tf.withNewBatch(fn);
+	else editor.tf.withMerging(fn);
+	const lastBatch = editor.history.undos?.at(-1);
+	if (lastBatch) lastBatch.ai = true;
+};
+
+//#endregion
+//#region src/lib/BaseAIPlugin.ts
+const getAITransforms = (editor) => ({
+	acceptPreview: bindFirst(acceptAIPreview, editor),
+	beginPreview: bindFirst(beginAIPreview, editor),
+	cancelPreview: bindFirst(cancelAIPreview, editor),
+	discardPreview: bindFirst(discardAIPreview, editor),
+	hasPreview: bindFirst(hasAIPreview, editor),
+	insertNodes: bindFirst(insertAINodes, editor),
+	removeMarks: bindFirst(removeAIMarks, editor),
+	removeNodes: bindFirst(removeAINodes, editor),
+	undo: bindFirst(undoAI, editor)
+});
+const BaseAIPlugin = createTSlatePlugin({
+	key: KEYS.ai,
+	node: {
+		isDecoration: false,
+		isLeaf: true
+	}
+}).extendTransforms(({ editor }) => getAITransforms(editor)).extendEditorTransforms(({ editor }) => ({ ai: getAITransforms(editor) }));
+
+//#endregion
+//#region src/lib/utils/getEditorPrompt.ts
+const createPromptFromConfig = (config, params) => {
+	const { isBlockSelecting, isSelecting: isSelecting$1 } = params;
+	if (isBlockSelecting && config.blockSelecting) return config.blockSelecting ?? config.default;
+	if (isSelecting$1 && config.selecting) return config.selecting ?? config.default;
+	return config.default;
+};
+const getEditorPrompt = (editor, { prompt = "" }) => {
+	const params = {
+		editor,
+		isBlockSelecting: editor.getOption({ key: KEYS.blockSelection }, "isSelectingSome"),
+		isSelecting: isSelecting(editor)
+	};
+	let promptText = "";
+	if (typeof prompt === "function") promptText = prompt(params);
+	else if (typeof prompt === "object") promptText = createPromptFromConfig(prompt, params);
+	else promptText = prompt;
+	return promptText;
+};
+
+//#endregion
+export { removeAINodes as a, AI_PREVIEW_KEY as c, cancelAIPreview as d, discardAIPreview as f, undoAI as i, acceptAIPreview as l, BaseAIPlugin as n, removeAIMarks as o, hasAIPreview as p, withAIBatch as r, insertAINodes as s, getEditorPrompt as t, beginAIPreview as u };

package/dist/index-D2hJDomP.d.ts

@@ -0,0 +1,137 @@
+import * as platejs0 from "platejs";
+import { Descendant, History, OmitFirst, Path, PluginConfig, SlateEditor, TLocation, Value } from "platejs";
+
+//#region src/lib/transforms/aiStreamSnapshot.d.ts
+type BeginAIPreviewOptions = {
+  originalBlocks?: Value;
+};
+declare const AI_PREVIEW_KEY = "aiPreview";
+declare const beginAIPreview: (editor: SlateEditor, {
+  originalBlocks
+}?: BeginAIPreviewOptions) => boolean;
+declare const hasAIPreview: (editor: SlateEditor) => boolean;
+declare const cancelAIPreview: (editor: SlateEditor) => boolean;
+declare const discardAIPreview: (editor: SlateEditor) => boolean;
+declare const acceptAIPreview: (editor: SlateEditor, _value?: Value) => boolean;
+//#endregion
+//#region src/lib/transforms/insertAINodes.d.ts
+declare const insertAINodes: (editor: SlateEditor, nodes: Descendant[], {
+  target
+}?: {
+  target?: Path;
+}) => void;
+//#endregion
+//#region src/lib/transforms/removeAIMarks.d.ts
+declare const removeAIMarks: (editor: SlateEditor, {
+  at
+}?: {
+  at?: TLocation;
+}) => void;
+//#endregion
+//#region src/lib/transforms/removeAINodes.d.ts
+declare const removeAINodes: (editor: SlateEditor, {
+  at
+}?: {
+  at?: Path;
+}) => void;
+//#endregion
+//#region src/lib/transforms/undoAI.d.ts
+declare const undoAI: (editor: SlateEditor) => void;
+//#endregion
+//#region src/lib/transforms/withAIBatch.d.ts
+type AIBatch = History['undos'][number] & {
+  ai?: boolean;
+};
+declare const withAIBatch: (editor: SlateEditor, fn: () => void, {
+  split
+}?: {
+  split?: boolean;
+}) => void;
+//#endregion
+//#region src/lib/BaseAIPlugin.d.ts
+type BaseAIPluginConfig = PluginConfig<'ai', {}, {}, {
+  ai: {
+    /** Commit the active preview as one fresh undoable batch. */
+    acceptPreview: OmitFirst<typeof acceptAIPreview>;
+    /** Capture the rollback slice and selection for AI preview. */
+    beginPreview: OmitFirst<typeof beginAIPreview>;
+    /** Restore the rollback point and clear active preview state. */
+    cancelPreview: OmitFirst<typeof cancelAIPreview>;
+    /** Clear active preview bookkeeping without restoring content. */
+    discardPreview: OmitFirst<typeof discardAIPreview>;
+    /** Report whether an AI preview rollback point is active. */
+    hasPreview: OmitFirst<typeof hasAIPreview>;
+    insertNodes: OmitFirst<typeof insertAINodes>;
+    removeMarks: OmitFirst<typeof removeAIMarks>;
+    removeNodes: OmitFirst<typeof removeAINodes>;
+    undo: OmitFirst<typeof undoAI>;
+  };
+}>;
+declare const BaseAIPlugin: platejs0.SlatePlugin<PluginConfig<"ai", {}, {}, {
+  ai: {
+    acceptPreview: ((_value?: platejs0.Value | undefined) => boolean) & ((_value?: platejs0.Value | undefined) => boolean);
+    beginPreview: ((args_0?: {
+      originalBlocks?: platejs0.Value;
+    } | undefined) => boolean) & ((args_0?: {
+      originalBlocks?: platejs0.Value;
+    } | undefined) => boolean);
+    cancelPreview: (() => boolean) & (() => boolean);
+    discardPreview: (() => boolean) & (() => boolean);
+    hasPreview: (() => boolean) & (() => boolean);
+    insertNodes: ((nodes: platejs0.Descendant[], args_1?: {
+      target?: platejs0.Path;
+    } | undefined) => void) & ((nodes: platejs0.Descendant[], args_1?: {
+      target?: platejs0.Path;
+    } | undefined) => void);
+    removeMarks: ((args_0?: {
+      at?: platejs0.TLocation;
+    } | undefined) => void) & ((args_0?: {
+      at?: platejs0.TLocation;
+    } | undefined) => void);
+    removeNodes: ((args_0?: {
+      at?: platejs0.Path;
+    } | undefined) => void) & ((args_0?: {
+      at?: platejs0.Path;
+    } | undefined) => void);
+    undo: (() => void) & (() => void);
+  };
+}, {}>>;
+//#endregion
+//#region src/lib/types.d.ts
+type AIToolName = 'comment' | 'edit' | 'generate' | null;
+type AIMode = 'chat' | 'insert';
+//#endregion
+//#region src/lib/utils/getEditorPrompt.d.ts
+type EditorPrompt = ((params: EditorPromptParams) => string) | PromptConfig | string;
+type EditorPromptParams = {
+  editor: SlateEditor;
+  isBlockSelecting: boolean;
+  isSelecting: boolean;
+};
+type PromptConfig = {
+  default: string;
+  blockSelecting?: string;
+  selecting?: string;
+};
+declare const getEditorPrompt: (editor: SlateEditor, {
+  prompt
+}: {
+  prompt?: EditorPrompt;
+}) => string;
+//#endregion
+//#region src/lib/utils/replacePlaceholders.d.ts
+type MarkdownType = 'block' | 'blockSelection' | 'blockSelectionWithBlockId' | 'blockWithBlockId' | 'editor' | 'editorWithBlockId' | 'tableCellWithId';
+declare const replacePlaceholders: (editor: SlateEditor, text: string, {
+  prompt
+}?: {
+  prompt?: string;
+}) => string;
+//#endregion
+//#region src/lib/utils/getMarkdown.d.ts
+declare const getMarkdown: (editor: SlateEditor, {
+  type
+}: {
+  type: MarkdownType;
+}) => string;
+//#endregion
+export { hasAIPreview as C, discardAIPreview as S, insertAINodes as _, EditorPromptParams as a, beginAIPreview as b, AIMode as c, BaseAIPluginConfig as d, AIBatch as f, removeAIMarks as g, removeAINodes as h, EditorPrompt as i, AIToolName as l, undoAI as m, MarkdownType as n, PromptConfig as o, withAIBatch as p, replacePlaceholders as r, getEditorPrompt as s, getMarkdown as t, BaseAIPlugin as u, AI_PREVIEW_KEY as v, cancelAIPreview as x, acceptAIPreview as y };

package/dist/index.d.ts

@@ -1,2 +1,2 @@
-import { _ as insertAINodes, a as EditorPromptParams, c as AIMode, d as BaseAIPluginConfig, f as AIBatch, g as removeAIMarks, h as removeAINodes, i as EditorPrompt, l as AIToolName, m as undoAI, n as MarkdownType, o as PromptConfig, p as withAIBatch, r as replacePlaceholders, s as getEditorPrompt, t as getMarkdown, u as BaseAIPlugin } from "./index-B_bqJoKL";
-export { AIBatch, AIMode, AIToolName, BaseAIPlugin, BaseAIPluginConfig, EditorPrompt, EditorPromptParams, MarkdownType, PromptConfig, getEditorPrompt, getMarkdown, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };
+import { C as hasAIPreview, S as discardAIPreview, _ as insertAINodes, a as EditorPromptParams, b as beginAIPreview, c as AIMode, d as BaseAIPluginConfig, f as AIBatch, g as removeAIMarks, h as removeAINodes, i as EditorPrompt, l as AIToolName, m as undoAI, n as MarkdownType, o as PromptConfig, p as withAIBatch, r as replacePlaceholders, s as getEditorPrompt, t as getMarkdown, u as BaseAIPlugin, v as AI_PREVIEW_KEY, x as cancelAIPreview, y as acceptAIPreview } from "./index-D2hJDomP";
+export { AIBatch, AIMode, AIToolName, AI_PREVIEW_KEY, BaseAIPlugin, BaseAIPluginConfig, EditorPrompt, EditorPromptParams, MarkdownType, PromptConfig, acceptAIPreview, beginAIPreview, cancelAIPreview, discardAIPreview, getEditorPrompt, getMarkdown, hasAIPreview, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };

package/dist/index.js

@@ -1,4 +1,4 @@
-import { a as removeAINodes, i as undoAI, n as BaseAIPlugin, o as removeAIMarks, r as withAIBatch, s as insertAINodes, t as getEditorPrompt } from "./getEditorPrompt-BnVrgUl3.js";
+import { a as removeAINodes, c as AI_PREVIEW_KEY, d as cancelAIPreview, f as discardAIPreview, i as undoAI, l as acceptAIPreview, n as BaseAIPlugin, o as removeAIMarks, p as hasAIPreview, r as withAIBatch, s as insertAINodes, t as getEditorPrompt, u as beginAIPreview } from "./getEditorPrompt-axDm9Fg4.js";
 import { KEYS } from "platejs";
 import { serializeMd } from "@platejs/markdown";
 import { getTableGridAbove } from "@platejs/table";
@@ -105,7 +105,7 @@ const getMarkdown = (editor, { type }) => {
 //#endregion
 //#region src/lib/utils/replacePlaceholders.ts
 const replacePlaceholders = (editor, text, { prompt } = {}) => {
-	let result = text.replace("{prompt}", prompt || "");
+	let result = text.split("{prompt}").join(prompt || "");
 	Object.entries({
 		"{blockSelectionWithBlockId}": "blockSelectionWithBlockId",
 		"{blockSelection}": "blockSelection",
@@ -115,11 +115,10 @@ const replacePlaceholders = (editor, text, { prompt } = {}) => {
 		"{editor}": "editor",
 		"{tableCellWithId}": "tableCellWithId"
 	}).forEach(([placeholder, type]) => {
-		if (result.includes(placeholder)) result = result.replace(placeholder, getMarkdown(editor, { type }));
+		if (result.includes(placeholder)) result = result.split(placeholder).join(getMarkdown(editor, { type }));
 	});
 	return result;
 };
 
 //#endregion
-export { BaseAIPlugin, getEditorPrompt, getMarkdown, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };
-//# sourceMappingURL=index.js.map
+export { AI_PREVIEW_KEY, BaseAIPlugin, acceptAIPreview, beginAIPreview, cancelAIPreview, discardAIPreview, getEditorPrompt, getMarkdown, hasAIPreview, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };