@pixerate/editor 0.58.0 → 0.60.0

package/dist/agent/index.js

@@ -51,19 +51,26 @@ import {
   validateJsonSchema
 } from "../chunk-4FX5ACVG.js";
 import {
+  DEFAULT_AGENT_CARET_COLOR,
+  DEFAULT_AGENT_CARET_NAME,
   MAX_SPARKLINE_VALUES,
   SPARKLINE_KINDS,
   TABLE_ACTIONS,
   applyReplacements,
+  clearAgentCaret,
   getDocumentCharts,
   getTableActionState,
   getTableState,
   isTableEnabled,
+  markAgentChanges,
   normalizeSparklineValues,
+  replaceWithTextTransaction,
   runTableAction,
   searchDocument,
-  sparklineLabel
-} from "../chunk-WLVZ235X.js";
+  setAgentCaret,
+  sparklineLabel,
+  typeAgentText
+} from "../chunk-OK7T2LHK.js";
 import {
   describeChart
 } from "../chunk-ZOH6X3VR.js";
@@ -74,8 +81,7 @@ import {
 } from "../chunk-633QC34G.js";
 import {
   getEditorMarkdown,
-  getEditorText,
-  plainTextToSlice
+  getEditorText
 } from "../chunk-UKI64KUD.js";
 import "../chunk-O5RYH2PG.js";
 import "../chunk-SWE3NPOU.js";
@@ -611,11 +617,6 @@ function createDocumentChartTools(editor) {
 
 // src/agent/providers.ts
 var toEditor = (target) => "commands" in target ? target : target.editor;
-function replaceWithText(editor, from, to, text) {
-  const { state, view } = editor;
-  const tr = /[\r\n]/.test(text) ? state.tr.replace(from, to, plainTextToSlice(text, state.schema)) : text ? state.tr.insertText(text, from, to) : state.tr.delete(from, to);
-  view.dispatch(tr.scrollIntoView());
-}
 function selectionInfo(editor) {
   const { from, to } = editor.state.selection;
   return { from, to, text: editor.state.doc.textBetween(from, to, "\n") };
@@ -634,6 +635,71 @@ function textBlocks(editor) {
   });
   return blocks;
 }
+function createTextPresence(editor, providerId, option) {
+  const config = option === false ? null : option === true || option === void 0 ? {} : option;
+  const caret = {
+    id: config?.id ?? providerId,
+    name: config?.name ?? DEFAULT_AGENT_CARET_NAME,
+    color: config?.color ?? DEFAULT_AGENT_CARET_COLOR
+  };
+  const typing = !config?.typing ? null : config.typing === true ? {} : config.typing;
+  const idleMs = config?.idleMs ?? 12e3;
+  const changedMs = config?.changedMs ?? 1500;
+  let idleTimer;
+  let tintTimer;
+  const live = () => {
+    const ed = editor();
+    return config && ed && !ed.isDestroyed ? ed : null;
+  };
+  const touch = () => {
+    clearTimeout(idleTimer);
+    if (idleMs > 0) idleTimer = setTimeout(() => live() && clearAgentCaret(live(), caret.id), idleMs);
+  };
+  const tint = (ed, ranges) => {
+    clearTimeout(tintTimer);
+    markAgentChanges(ed, caret.id, ranges);
+    if (changedMs > 0 && ranges.length) tintTimer = setTimeout(() => live() && markAgentChanges(live(), caret.id, []), changedMs);
+  };
+  return {
+    /** Puts the caret at `head`, selecting from `anchor`. */
+    show(anchor, head = anchor) {
+      const ed = live();
+      if (!ed) return;
+      setAgentCaret(ed, { ...caret, from: anchor, to: head });
+      touch();
+    },
+    /** Shows the caret after ranges just changed by a tool, tinting them. */
+    changed(ranges) {
+      const ed = live();
+      if (!ed || !ranges.length) return;
+      const last = ranges[ranges.length - 1];
+      setAgentCaret(ed, { ...caret, from: last.to });
+      tint(ed, ranges);
+      touch();
+    },
+    /** Replaces `from..to` with literal text, typed out when typing is on. Returns the inserted range. */
+    async write(ed, from, to, text, signal) {
+      if (live() && typing) {
+        clearTimeout(tintTimer);
+        const range2 = await typeAgentText(ed, { ...typing, from, to, text, caret, signal });
+        if (changedMs > 0) tintTimer = setTimeout(() => live() && markAgentChanges(live(), caret.id, []), changedMs);
+        touch();
+        return range2;
+      }
+      const tr = replaceWithTextTransaction(ed.state, from, to, text);
+      const end = tr.mapping.map(to, 1);
+      const range = { from: Math.min(tr.mapping.map(from, -1), end), to: end };
+      ed.view.dispatch(tr.scrollIntoView());
+      if (range.from < range.to) this.changed([range]);
+      else this.show(range.to);
+      return range;
+    },
+    dispose() {
+      clearTimeout(idleTimer);
+      clearTimeout(tintTimer);
+    }
+  };
+}
 var FIND_SCHEMA_PROPERTIES = {
   matchCase: { type: "boolean", description: "Case-sensitive match (default false)." },
   wholeWord: { type: "boolean", description: "Match whole words only (default false)." },
@@ -656,6 +722,7 @@ function findMatches(editor, query, args, replacement) {
 function createRichTextToolProvider(target, options) {
   const editor = () => toEditor(target);
   const charts = createDocumentChartTools(editor);
+  const presence = createTextPresence(editor, options.id, options.presence);
   const getText = {
     name: "get_text",
     title: "Get document text",
@@ -697,7 +764,7 @@ function createRichTextToolProvider(target, options) {
   const insertText = {
     name: "insert_text",
     title: "Insert text",
-    description: "Inserts literal text (newlines start new paragraphs) at a document position, or at the cursor when `at` is omitted.",
+    description: "Inserts literal text (newlines start new paragraphs) at a document position, or at the cursor when `at` is omitted. Returns the inserted `range` {from, to}.",
     inputSchema: {
       type: "object",
       properties: {
@@ -714,15 +781,15 @@ function createRichTextToolProvider(target, options) {
       idempotentHint: false,
       openWorldHint: false
     },
-    handler: ({ text, at }) => {
+    handler: async ({ text, at }, { signal }) => {
       const ed = editor();
       const size = ed.state.doc.content.size;
       if (at !== void 0 && at > size) {
         throw new Error(`Position ${at} is outside the document (size ${size}).`);
       }
       const { from, to } = ed.state.selection;
-      replaceWithText(ed, at ?? from, at ?? to, text);
-      return { inserted: text.length, selection: selectionInfo(ed) };
+      const range = await presence.write(ed, at ?? from, at ?? to, text, signal);
+      return { inserted: text.length, range, selection: selectionInfo(ed) };
     }
   };
   const findText = {
@@ -750,6 +817,7 @@ function createRichTextToolProvider(target, options) {
       const ed = editor();
       const { doc } = ed.state;
       const { matches, truncated } = findMatches(ed, query, flags);
+      if (matches.length) presence.show(matches[0].from, matches[0].to);
       return {
         total: matches.length,
         truncated: truncated || matches.length > maxResults,
@@ -803,10 +871,12 @@ function createRichTextToolProvider(target, options) {
       const tr = ed.state.tr;
       const ends = applyReplacements(tr, chosen);
       ed.view.dispatch(tr.scrollIntoView());
+      const ranges = chosen.map((m, i) => ({ from: tr.mapping.map(m.from, -1), to: ends[i] }));
+      presence.changed(ranges.filter((r) => r.from < r.to));
       return {
         replaced: chosen.length,
         total: matches.length,
-        ranges: chosen.map((m, i) => ({ from: tr.mapping.map(m.from, -1), to: ends[i] })),
+        ranges,
         selection: selectionInfo(ed)
       };
     }
@@ -814,7 +884,7 @@ function createRichTextToolProvider(target, options) {
   const replaceRange = {
     name: "replace_range",
     title: "Replace a range",
-    description: "Replaces the content between two document positions (from get_text `blocks` or find_text) with literal text. from === to inserts; an empty text deletes.",
+    description: "Replaces the content between two document positions (from get_text `blocks` or find_text) with literal text. from === to inserts; an empty text deletes. Returns the new text's `range` {from, to}.",
     inputSchema: {
       type: "object",
       properties: {
@@ -832,11 +902,11 @@ function createRichTextToolProvider(target, options) {
       idempotentHint: false,
       openWorldHint: false
     },
-    handler: ({ from, to, text }) => {
+    handler: async ({ from, to, text }, { signal }) => {
       const ed = editor();
       checkRange(ed, from, to);
-      replaceWithText(ed, from, to, text);
-      return { replaced: { from, to }, selection: selectionInfo(ed), size: ed.state.doc.content.size };
+      const range = await presence.write(ed, from, to, text, signal);
+      return { replaced: { from, to }, range, selection: selectionInfo(ed), size: ed.state.doc.content.size };
     }
   };
   const setSelection = {
@@ -864,13 +934,15 @@ function createRichTextToolProvider(target, options) {
       checkRange(ed, from, to);
       const { doc } = ed.state;
       ed.view.dispatch(ed.state.tr.setSelection(TextSelection.between(doc.resolve(from), doc.resolve(to))));
-      return { selection: selectionInfo(ed) };
+      const selection = selectionInfo(ed);
+      presence.show(selection.from, selection.to);
+      return { selection };
     }
   };
   const replaceSelection = {
     name: "replace_selection",
     title: "Replace selection",
-    description: "Replaces the selected content with literal text (an empty string deletes the selection).",
+    description: "Replaces the selected content with literal text (an empty string deletes the selection). Returns the new text's `range` {from, to}.",
     inputSchema: {
       type: "object",
       properties: { text: { type: "string", description: "Replacement text." } },
@@ -884,11 +956,11 @@ function createRichTextToolProvider(target, options) {
       idempotentHint: false,
       openWorldHint: false
     },
-    handler: ({ text }) => {
+    handler: async ({ text }, { signal }) => {
       const ed = editor();
       const { from, to } = ed.state.selection;
-      replaceWithText(ed, from, to, text);
-      return { replaced: { from, to }, selection: selectionInfo(ed) };
+      const range = await presence.write(ed, from, to, text, signal);
+      return { replaced: { from, to }, range, selection: selectionInfo(ed) };
     }
   };
   const editTable = {
@@ -956,6 +1028,7 @@ function createRichTextToolProvider(target, options) {
       return () => {
         ed.off("update", notify);
         ed.off("destroy", notify);
+        presence.dispose();
       };
     }
   };