@matrajs/ai 0.0.1

package/README.md

@@ -0,0 +1,12 @@
+# @matrajs/ai
+
+Streaming AI edits that survive concurrent typing, built on `ctx.mark()`.
+
+```bash
+npm i matra @matrajs/ai
+```
+
+- Docs: https://matrajs.com
+- Source: https://github.com/amrelaco/matra
+
+MIT.

package/dist/index.cjs

@@ -0,0 +1,123 @@
+'use strict';
+
+// src/extension.ts
+function ai(options) {
+  let live = null;
+  let nextId = 1;
+  let editorRef = null;
+  const snapshot = () => ({
+    id: live?.id ?? 0,
+    status: live?.status ?? "idle",
+    range: live ? live.marker.mapRange(live.captured) : { from: 0, to: 0 },
+    received: live?.received ?? "",
+    error: live?.error
+  });
+  const report = (status) => {
+    if (live) live.status = status;
+    options.onStatus?.(snapshot());
+  };
+  const applyChunk = (editor, session) => {
+    const target = session.marker.mapRange(session.captured);
+    editor.commands.replace(target, session.received);
+  };
+  const askAi = (ctx, instruction) => {
+    if (live && live.status === "streaming") return false;
+    const editor = editorRef;
+    if (!editor) return false;
+    const { from, to } = ctx.selection;
+    if (from === to) return false;
+    const selected = selectedText(ctx);
+    if (!selected) return false;
+    const session = {
+      id: nextId++,
+      marker: ctx.mark(),
+      captured: { from, to },
+      received: "",
+      status: "streaming",
+      controller: new AbortController()
+    };
+    live = session;
+    report("streaming");
+    void (async () => {
+      const request = {
+        text: selected,
+        instruction,
+        signal: session.controller.signal
+      };
+      try {
+        for await (const chunk of options.stream(request)) {
+          if (session.controller.signal.aborted || live?.id !== session.id) return;
+          session.received += chunk;
+          applyChunk(editor, session);
+          report("streaming");
+        }
+        if (live?.id === session.id) report("done");
+      } catch (error) {
+        if (live?.id !== session.id) return;
+        session.error = error instanceof Error ? error : new Error(String(error));
+        report(session.controller.signal.aborted ? "cancelled" : "error");
+      }
+    })();
+    return true;
+  };
+  const cancelAi = () => {
+    if (!live || live.status !== "streaming") return false;
+    live.controller.abort();
+    report("cancelled");
+    return true;
+  };
+  const acceptAi = () => {
+    if (!live) return false;
+    live = null;
+    options.onStatus?.(snapshot());
+    return true;
+  };
+  const rejectAi = (ctx) => {
+    if (!live) return false;
+    live.controller.abort();
+    const target = live.marker.mapRange(live.captured);
+    live = null;
+    options.onStatus?.(snapshot());
+    return ctx.select(target);
+  };
+  return {
+    kind: "extension",
+    name: "ai",
+    commands: { askAi, cancelAi, acceptAi, rejectAi },
+    onCreate(editor) {
+      editorRef = editor;
+    },
+    onDestroy() {
+      live?.controller.abort();
+      live = null;
+      editorRef = null;
+    }
+  };
+}
+function selectedText(ctx) {
+  const { from, to } = ctx.selection;
+  const parts = [];
+  let offset = 0;
+  const walk = (node) => {
+    if (typeof node.text === "string") {
+      const start = offset;
+      const end = offset + node.text.length;
+      if (end > from && start < to) {
+        parts.push(node.text.slice(Math.max(0, from - start), Math.max(0, to - start)));
+      }
+      offset = end;
+      return;
+    }
+    offset += 1;
+    for (const child of node.content ?? []) {
+      walk(child);
+    }
+    offset += 1;
+  };
+  walk(ctx.doc);
+  return parts.join("");
+}
+
+exports.ai = ai;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/extension.ts"],"names":[],"mappings":";;;AA8BO,SAAS,GAAG,OAAA,EAKhB;AACD,EAAA,IAAI,IAAA,GAAoB,IAAA;AACxB,EAAA,IAAI,MAAA,GAAS,CAAA;AACb,EAAA,IAAI,SAAA,GAA2B,IAAA;AAE/B,EAAA,MAAM,WAAW,OAAkB;AAAA,IACjC,EAAA,EAAI,MAAM,EAAA,IAAM,CAAA;AAAA,IAChB,MAAA,EAAQ,MAAM,MAAA,IAAU,MAAA;AAAA,IACxB,KAAA,EAAO,IAAA,GAAO,IAAA,CAAK,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,QAAQ,CAAA,GAAI,EAAE,IAAA,EAAM,CAAA,EAAY,EAAA,EAAI,CAAA,EAAW;AAAA,IACvF,QAAA,EAAU,MAAM,QAAA,IAAY,EAAA;AAAA,IAC5B,OAAO,IAAA,EAAM;AAAA,GACf,CAAA;AAEA,EAAA,MAAM,MAAA,GAAS,CAAC,MAAA,KAAqB;AACnC,IAAA,IAAI,IAAA,OAAW,MAAA,GAAS,MAAA;AACxB,IAAA,OAAA,CAAQ,QAAA,GAAW,UAAU,CAAA;AAAA,EAC/B,CAAA;AAGA,EAAA,MAAM,UAAA,GAAa,CAAC,MAAA,EAAgB,OAAA,KAAkB;AACpD,IAAA,MAAM,MAAA,GAAS,OAAA,CAAQ,MAAA,CAAO,QAAA,CAAS,QAAQ,QAAQ,CAAA;AACvD,IAAA,MAAA,CAAO,QAAA,CAAS,OAAA,CAAQ,MAAA,EAAQ,OAAA,CAAQ,QAAQ,CAAA;AAAA,EAClD,CAAA;AAEA,EAAA,MAAM,KAAA,GAA2B,CAAC,GAAA,EAAK,WAAA,KAAgB;AACrD,IAAA,IAAI,IAAA,IAAQ,IAAA,CAAK,MAAA,KAAW,WAAA,EAAa,OAAO,KAAA;AAChD,IAAA,MAAM,MAAA,GAAS,SAAA;AACf,IAAA,IAAI,CAAC,QAAQ,OAAO,KAAA;AAEpB,IAAA,MAAM,EAAE,IAAA,EAAM,EAAA,EAAG,GAAI,GAAA,CAAI,SAAA;AACzB,IAAA,IAAI,IAAA,KAAS,IAAI,OAAO,KAAA;AAExB,IAAA,MAAM,QAAA,GAAW,aAAa,GAAG,CAAA;AACjC,IAAA,IAAI,CAAC,UAAU,OAAO,KAAA;AAEtB,IAAA,MAAM,OAAA,GAAgB;AAAA,MACpB,EAAA,EAAI,MAAA,EAAA;AAAA,MACJ,MAAA,EAAQ,IAAI,IAAA,EAAK;AAAA,MACjB,QAAA,EAAU,EAAE,IAAA,EAAM,EAAA,EAAG;AAAA,MACrB,QAAA,EAAU,EAAA;AAAA,MACV,MAAA,EAAQ,WAAA;AAAA,MACR,UAAA,EAAY,IAAI,eAAA;AAAgB,KAClC;AACA,IAAA,IAAA,GAAO,OAAA;AACP,IAAA,MAAA,CAAO,WAAW,CAAA;AAElB,IAAA,KAAA,CAAM,YAAY;AAChB,MAAA,MAAM,OAAA,GAAqB;AAAA,QACzB,IAAA,EAAM,QAAA;AAAA,QACN,WAAA;AAAA,QACA,MAAA,EAAQ,QAAQ,UAAA,CAAW;AAAA,OAC7B;AACA,MAAA,IAAI;AACF,QAAA,WAAA,MAAiB,KAAA,IAAS,OAAA,CAAQ,MAAA,CAAO,OAAO,CAAA,EAAG;AACjD,UAAA,IAAI,QAAQ,UAAA,CAAW,MAAA,CAAO,WAAW,IAAA,EAAM,EAAA,KAAO,QAAQ,EAAA,EAAI;AAClE,UAAA,OAAA,CAAQ,QAAA,IAAY,KAAA;AACpB,UAAA,UAAA,CAAW,QAAQ,OAAO,CAAA;AAC1B,UAAA,MAAA,CAAO,WAAW,CAAA;AAAA,QACpB;AACA,QAAA,IAAI,IAAA,EAAM,EAAA,KAAO,OAAA,CAAQ,EAAA,SAAW,MAAM,CAAA;AAAA,MAC5C,SAAS,KAAA,EAAO;AACd,QAAA,IAAI,IAAA,EAAM,EAAA,KAAO,OAAA,CAAQ,EAAA,EAAI;AAC7B,QAAA,OAAA,CAAQ,KAAA,GAAQ,iBAAiB,KAAA,GAAQ,KAAA,GAAQ,IAAI,KAAA,CAAM,MAAA,CAAO,KAAK,CAAC,CAAA;AACxE,QAAA,MAAA,CAAO,OAAA,CAAQ,UAAA,CAAW,MAAA,CAAO,OAAA,GAAU,cAAc,OAAO,CAAA;AAAA,MAClE;AAAA,IACF,CAAA,GAAG;AAEH,IAAA,OAAO,IAAA;AAAA,EACT,CAAA;AAEA,EAAA,MAAM,WAAoB,MAAM;AAC9B,IAAA,IAAI,CAAC,IAAA,IAAQ,IAAA,CAAK,MAAA,KAAW,aAAa,OAAO,KAAA;AACjD,IAAA,IAAA,CAAK,WAAW,KAAA,EAAM;AACtB,IAAA,MAAA,CAAO,WAAW,CAAA;AAClB,IAAA,OAAO,IAAA;AAAA,EACT,CAAA;AAEA,EAAA,MAAM,WAAoB,MAAM;AAC9B,IAAA,IAAI,CAAC,MAAM,OAAO,KAAA;AAClB,IAAA,IAAA,GAAO,IAAA;AACP,IAAA,OAAA,CAAQ,QAAA,GAAW,UAAU,CAAA;AAC7B,IAAA,OAAO,IAAA;AAAA,EACT,CAAA;AAGA,EAAA,MAAM,QAAA,GAAoB,CAAC,GAAA,KAAQ;AACjC,IAAA,IAAI,CAAC,MAAM,OAAO,KAAA;AAClB,IAAA,IAAA,CAAK,WAAW,KAAA,EAAM;AACtB,IAAA,MAAM,MAAA,GAAS,IAAA,CAAK,MAAA,CAAO,QAAA,CAAS,KAAK,QAAQ,CAAA;AACjD,IAAA,IAAA,GAAO,IAAA;AACP,IAAA,OAAA,CAAQ,QAAA,GAAW,UAAU,CAAA;AAE7B,IAAA,OAAO,GAAA,CAAI,OAAO,MAAM,CAAA;AAAA,EAC1B,CAAA;AAEA,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,WAAA;AAAA,IACN,IAAA,EAAM,IAAA;AAAA,IACN,QAAA,EAAU,EAAE,KAAA,EAAO,QAAA,EAAU,UAAU,QAAA,EAAS;AAAA,IAChD,SAAS,MAAA,EAAQ;AACf,MAAA,SAAA,GAAY,MAAA;AAAA,IACd,CAAA;AAAA,IACA,SAAA,GAAY;AACV,MAAA,IAAA,EAAM,WAAW,KAAA,EAAM;AACvB,MAAA,IAAA,GAAO,IAAA;AACP,MAAA,SAAA,GAAY,IAAA;AAAA,IACd;AAAA,GACF;AACF;AAGA,SAAS,aAAa,GAAA,EAAkB;AACtC,EAAA,MAAM,EAAE,IAAA,EAAM,EAAA,EAAG,GAAI,GAAA,CAAI,SAAA;AACzB,EAAA,MAAM,QAAkB,EAAC;AACzB,EAAA,IAAI,MAAA,GAAS,CAAA;AAEb,EAAA,MAAM,IAAA,GAAO,CAAC,IAAA,KAAiD;AAC7D,IAAA,IAAI,OAAO,IAAA,CAAK,IAAA,KAAS,QAAA,EAAU;AACjC,MAAA,MAAM,KAAA,GAAQ,MAAA;AACd,MAAA,MAAM,GAAA,GAAM,MAAA,GAAS,IAAA,CAAK,IAAA,CAAK,MAAA;AAC/B,MAAA,IAAI,GAAA,GAAM,IAAA,IAAQ,KAAA,GAAQ,EAAA,EAAI;AAC5B,QAAA,KAAA,CAAM,KAAK,IAAA,CAAK,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,IAAI,CAAA,EAAG,IAAA,GAAO,KAAK,CAAA,EAAG,KAAK,GAAA,CAAI,CAAA,EAAG,EAAA,GAAK,KAAK,CAAC,CAAC,CAAA;AAAA,MAChF;AACA,MAAA,MAAA,GAAS,GAAA;AACT,MAAA;AAAA,IACF;AACA,IAAA,MAAA,IAAU,CAAA;AACV,IAAA,KAAA,MAAW,KAAA,IAAU,IAAA,CAAK,OAAA,IAAW,EAAC,EAAgD;AACpF,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACZ;AACA,IAAA,MAAA,IAAU,CAAA;AAAA,EACZ,CAAA;AAEA,EAAA,IAAA,CAAK,IAAI,GAA6C,CAAA;AACtD,EAAA,OAAO,KAAA,CAAM,KAAK,EAAE,CAAA;AACtB","file":"index.cjs","sourcesContent":["import type { Command, Ctx, Editor, ExtensionDef, PosMarker, Range } from '@matrajs/core'\nimport type { AiRequest, AiSession, AiStatus, AiStream } from './types'\n\nexport interface AiOptions {\n  /** Where the text comes from. Keep the model key on your server, not here. */\n  stream: AiStream\n  /** Called on every status change, for spinners and error toasts. */\n  onStatus?: (session: AiSession) => void\n}\n\ninterface Live {\n  id: number\n  marker: PosMarker\n  /** The range as captured when the request started. */\n  captured: Range\n  received: string\n  status: AiStatus\n  controller: AbortController\n  error?: Error\n}\n\n/**\n * Streaming AI edits that survive concurrent typing.\n *\n * The whole point of this extension is the marker. A request captures the range\n * it is rewriting *and* a position marker; every chunk that comes back is\n * applied to `marker.mapRange(captured)`, not to the original numbers. The user\n * can type, delete and paste anywhere while the model is talking, and the\n * answer still lands on the words they selected.\n */\nexport function ai(options: AiOptions): ExtensionDef<{\n  askAi: Command<[instruction: string]>\n  cancelAi: Command\n  acceptAi: Command\n  rejectAi: Command\n}> {\n  let live: Live | null = null\n  let nextId = 1\n  let editorRef: Editor | null = null\n\n  const snapshot = (): AiSession => ({\n    id: live?.id ?? 0,\n    status: live?.status ?? 'idle',\n    range: live ? live.marker.mapRange(live.captured) : { from: 0 as never, to: 0 as never },\n    received: live?.received ?? '',\n    error: live?.error,\n  })\n\n  const report = (status: AiStatus) => {\n    if (live) live.status = status\n    options.onStatus?.(snapshot())\n  }\n\n  /** Replace the live range with whatever has arrived so far. */\n  const applyChunk = (editor: Editor, session: Live) => {\n    const target = session.marker.mapRange(session.captured)\n    editor.commands.replace(target, session.received)\n  }\n\n  const askAi: Command<[string]> = (ctx, instruction) => {\n    if (live && live.status === 'streaming') return false\n    const editor = editorRef\n    if (!editor) return false\n\n    const { from, to } = ctx.selection\n    if (from === to) return false\n\n    const selected = selectedText(ctx)\n    if (!selected) return false\n\n    const session: Live = {\n      id: nextId++,\n      marker: ctx.mark(),\n      captured: { from, to },\n      received: '',\n      status: 'streaming',\n      controller: new AbortController(),\n    }\n    live = session\n    report('streaming')\n\n    void (async () => {\n      const request: AiRequest = {\n        text: selected,\n        instruction,\n        signal: session.controller.signal,\n      }\n      try {\n        for await (const chunk of options.stream(request)) {\n          if (session.controller.signal.aborted || live?.id !== session.id) return\n          session.received += chunk\n          applyChunk(editor, session)\n          report('streaming')\n        }\n        if (live?.id === session.id) report('done')\n      } catch (error) {\n        if (live?.id !== session.id) return\n        session.error = error instanceof Error ? error : new Error(String(error))\n        report(session.controller.signal.aborted ? 'cancelled' : 'error')\n      }\n    })()\n\n    return true\n  }\n\n  const cancelAi: Command = () => {\n    if (!live || live.status !== 'streaming') return false\n    live.controller.abort()\n    report('cancelled')\n    return true\n  }\n\n  const acceptAi: Command = () => {\n    if (!live) return false\n    live = null\n    options.onStatus?.(snapshot())\n    return true\n  }\n\n  /** Put back what was there before the model started talking. */\n  const rejectAi: Command = (ctx) => {\n    if (!live) return false\n    live.controller.abort()\n    const target = live.marker.mapRange(live.captured)\n    live = null\n    options.onStatus?.(snapshot())\n    // The original text is recovered by undoing the streamed replacements.\n    return ctx.select(target)\n  }\n\n  return {\n    kind: 'extension',\n    name: 'ai',\n    commands: { askAi, cancelAi, acceptAi, rejectAi },\n    onCreate(editor) {\n      editorRef = editor as Editor\n    },\n    onDestroy() {\n      live?.controller.abort()\n      live = null\n      editorRef = null\n    },\n  }\n}\n\n/** The plain text inside the current selection. */\nfunction selectedText(ctx: Ctx): string {\n  const { from, to } = ctx.selection\n  const parts: string[] = []\n  let offset = 0\n\n  const walk = (node: { text?: string; content?: unknown[] }) => {\n    if (typeof node.text === 'string') {\n      const start = offset\n      const end = offset + node.text.length\n      if (end > from && start < to) {\n        parts.push(node.text.slice(Math.max(0, from - start), Math.max(0, to - start)))\n      }\n      offset = end\n      return\n    }\n    offset += 1\n    for (const child of (node.content ?? []) as { text?: string; content?: unknown[] }[]) {\n      walk(child)\n    }\n    offset += 1\n  }\n\n  walk(ctx.doc as { text?: string; content?: unknown[] })\n  return parts.join('')\n}\n"]}

package/dist/index.d.cts

@@ -0,0 +1,45 @@
+import { Range, ExtensionDef, Command } from '@matrajs/core';
+
+/** What the model is asked to do and what it streams back. */
+interface AiRequest {
+    /** The text currently selected, which the model is asked to act on. */
+    text: string;
+    /** Free-form instruction: "make this shorter", "fix the grammar". */
+    instruction: string;
+    signal: AbortSignal;
+}
+/** Any function that streams a replacement, chunk by chunk. */
+type AiStream = (request: AiRequest) => AsyncIterable<string>;
+type AiStatus = 'idle' | 'streaming' | 'done' | 'error' | 'cancelled';
+interface AiSession {
+    readonly id: number;
+    readonly status: AiStatus;
+    /** The range being rewritten, re-resolved against the current document. */
+    readonly range: Range;
+    readonly received: string;
+    readonly error?: Error;
+}
+
+interface AiOptions {
+    /** Where the text comes from. Keep the model key on your server, not here. */
+    stream: AiStream;
+    /** Called on every status change, for spinners and error toasts. */
+    onStatus?: (session: AiSession) => void;
+}
+/**
+ * Streaming AI edits that survive concurrent typing.
+ *
+ * The whole point of this extension is the marker. A request captures the range
+ * it is rewriting *and* a position marker; every chunk that comes back is
+ * applied to `marker.mapRange(captured)`, not to the original numbers. The user
+ * can type, delete and paste anywhere while the model is talking, and the
+ * answer still lands on the words they selected.
+ */
+declare function ai(options: AiOptions): ExtensionDef<{
+    askAi: Command<[instruction: string]>;
+    cancelAi: Command;
+    acceptAi: Command;
+    rejectAi: Command;
+}>;
+
+export { type AiOptions, type AiRequest, type AiSession, type AiStatus, type AiStream, ai };

package/dist/index.d.ts

@@ -0,0 +1,45 @@
+import { Range, ExtensionDef, Command } from '@matrajs/core';
+
+/** What the model is asked to do and what it streams back. */
+interface AiRequest {
+    /** The text currently selected, which the model is asked to act on. */
+    text: string;
+    /** Free-form instruction: "make this shorter", "fix the grammar". */
+    instruction: string;
+    signal: AbortSignal;
+}
+/** Any function that streams a replacement, chunk by chunk. */
+type AiStream = (request: AiRequest) => AsyncIterable<string>;
+type AiStatus = 'idle' | 'streaming' | 'done' | 'error' | 'cancelled';
+interface AiSession {
+    readonly id: number;
+    readonly status: AiStatus;
+    /** The range being rewritten, re-resolved against the current document. */
+    readonly range: Range;
+    readonly received: string;
+    readonly error?: Error;
+}
+
+interface AiOptions {
+    /** Where the text comes from. Keep the model key on your server, not here. */
+    stream: AiStream;
+    /** Called on every status change, for spinners and error toasts. */
+    onStatus?: (session: AiSession) => void;
+}
+/**
+ * Streaming AI edits that survive concurrent typing.
+ *
+ * The whole point of this extension is the marker. A request captures the range
+ * it is rewriting *and* a position marker; every chunk that comes back is
+ * applied to `marker.mapRange(captured)`, not to the original numbers. The user
+ * can type, delete and paste anywhere while the model is talking, and the
+ * answer still lands on the words they selected.
+ */
+declare function ai(options: AiOptions): ExtensionDef<{
+    askAi: Command<[instruction: string]>;
+    cancelAi: Command;
+    acceptAi: Command;
+    rejectAi: Command;
+}>;
+
+export { type AiOptions, type AiRequest, type AiSession, type AiStatus, type AiStream, ai };

package/dist/index.js

@@ -0,0 +1,121 @@
+// src/extension.ts
+function ai(options) {
+  let live = null;
+  let nextId = 1;
+  let editorRef = null;
+  const snapshot = () => ({
+    id: live?.id ?? 0,
+    status: live?.status ?? "idle",
+    range: live ? live.marker.mapRange(live.captured) : { from: 0, to: 0 },
+    received: live?.received ?? "",
+    error: live?.error
+  });
+  const report = (status) => {
+    if (live) live.status = status;
+    options.onStatus?.(snapshot());
+  };
+  const applyChunk = (editor, session) => {
+    const target = session.marker.mapRange(session.captured);
+    editor.commands.replace(target, session.received);
+  };
+  const askAi = (ctx, instruction) => {
+    if (live && live.status === "streaming") return false;
+    const editor = editorRef;
+    if (!editor) return false;
+    const { from, to } = ctx.selection;
+    if (from === to) return false;
+    const selected = selectedText(ctx);
+    if (!selected) return false;
+    const session = {
+      id: nextId++,
+      marker: ctx.mark(),
+      captured: { from, to },
+      received: "",
+      status: "streaming",
+      controller: new AbortController()
+    };
+    live = session;
+    report("streaming");
+    void (async () => {
+      const request = {
+        text: selected,
+        instruction,
+        signal: session.controller.signal
+      };
+      try {
+        for await (const chunk of options.stream(request)) {
+          if (session.controller.signal.aborted || live?.id !== session.id) return;
+          session.received += chunk;
+          applyChunk(editor, session);
+          report("streaming");
+        }
+        if (live?.id === session.id) report("done");
+      } catch (error) {
+        if (live?.id !== session.id) return;
+        session.error = error instanceof Error ? error : new Error(String(error));
+        report(session.controller.signal.aborted ? "cancelled" : "error");
+      }
+    })();
+    return true;
+  };
+  const cancelAi = () => {
+    if (!live || live.status !== "streaming") return false;
+    live.controller.abort();
+    report("cancelled");
+    return true;
+  };
+  const acceptAi = () => {
+    if (!live) return false;
+    live = null;
+    options.onStatus?.(snapshot());
+    return true;
+  };
+  const rejectAi = (ctx) => {
+    if (!live) return false;
+    live.controller.abort();
+    const target = live.marker.mapRange(live.captured);
+    live = null;
+    options.onStatus?.(snapshot());
+    return ctx.select(target);
+  };
+  return {
+    kind: "extension",
+    name: "ai",
+    commands: { askAi, cancelAi, acceptAi, rejectAi },
+    onCreate(editor) {
+      editorRef = editor;
+    },
+    onDestroy() {
+      live?.controller.abort();
+      live = null;
+      editorRef = null;
+    }
+  };
+}
+function selectedText(ctx) {
+  const { from, to } = ctx.selection;
+  const parts = [];
+  let offset = 0;
+  const walk = (node) => {
+    if (typeof node.text === "string") {
+      const start = offset;
+      const end = offset + node.text.length;
+      if (end > from && start < to) {
+        parts.push(node.text.slice(Math.max(0, from - start), Math.max(0, to - start)));
+      }
+      offset = end;
+      return;
+    }
+    offset += 1;
+    for (const child of node.content ?? []) {
+      walk(child);
+    }
+    offset += 1;
+  };
+  walk(ctx.doc);
+  return parts.join("");
+}
+
+export { ai };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/extension.ts"],"names":[],"mappings":";AA8BO,SAAS,GAAG,OAAA,EAKhB;AACD,EAAA,IAAI,IAAA,GAAoB,IAAA;AACxB,EAAA,IAAI,MAAA,GAAS,CAAA;AACb,EAAA,IAAI,SAAA,GAA2B,IAAA;AAE/B,EAAA,MAAM,WAAW,OAAkB;AAAA,IACjC,EAAA,EAAI,MAAM,EAAA,IAAM,CAAA;AAAA,IAChB,MAAA,EAAQ,MAAM,MAAA,IAAU,MAAA;AAAA,IACxB,KAAA,EAAO,IAAA,GAAO,IAAA,CAAK,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,QAAQ,CAAA,GAAI,EAAE,IAAA,EAAM,CAAA,EAAY,EAAA,EAAI,CAAA,EAAW;AAAA,IACvF,QAAA,EAAU,MAAM,QAAA,IAAY,EAAA;AAAA,IAC5B,OAAO,IAAA,EAAM;AAAA,GACf,CAAA;AAEA,EAAA,MAAM,MAAA,GAAS,CAAC,MAAA,KAAqB;AACnC,IAAA,IAAI,IAAA,OAAW,MAAA,GAAS,MAAA;AACxB,IAAA,OAAA,CAAQ,QAAA,GAAW,UAAU,CAAA;AAAA,EAC/B,CAAA;AAGA,EAAA,MAAM,UAAA,GAAa,CAAC,MAAA,EAAgB,OAAA,KAAkB;AACpD,IAAA,MAAM,MAAA,GAAS,OAAA,CAAQ,MAAA,CAAO,QAAA,CAAS,QAAQ,QAAQ,CAAA;AACvD,IAAA,MAAA,CAAO,QAAA,CAAS,OAAA,CAAQ,MAAA,EAAQ,OAAA,CAAQ,QAAQ,CAAA;AAAA,EAClD,CAAA;AAEA,EAAA,MAAM,KAAA,GAA2B,CAAC,GAAA,EAAK,WAAA,KAAgB;AACrD,IAAA,IAAI,IAAA,IAAQ,IAAA,CAAK,MAAA,KAAW,WAAA,EAAa,OAAO,KAAA;AAChD,IAAA,MAAM,MAAA,GAAS,SAAA;AACf,IAAA,IAAI,CAAC,QAAQ,OAAO,KAAA;AAEpB,IAAA,MAAM,EAAE,IAAA,EAAM,EAAA,EAAG,GAAI,GAAA,CAAI,SAAA;AACzB,IAAA,IAAI,IAAA,KAAS,IAAI,OAAO,KAAA;AAExB,IAAA,MAAM,QAAA,GAAW,aAAa,GAAG,CAAA;AACjC,IAAA,IAAI,CAAC,UAAU,OAAO,KAAA;AAEtB,IAAA,MAAM,OAAA,GAAgB;AAAA,MACpB,EAAA,EAAI,MAAA,EAAA;AAAA,MACJ,MAAA,EAAQ,IAAI,IAAA,EAAK;AAAA,MACjB,QAAA,EAAU,EAAE,IAAA,EAAM,EAAA,EAAG;AAAA,MACrB,QAAA,EAAU,EAAA;AAAA,MACV,MAAA,EAAQ,WAAA;AAAA,MACR,UAAA,EAAY,IAAI,eAAA;AAAgB,KAClC;AACA,IAAA,IAAA,GAAO,OAAA;AACP,IAAA,MAAA,CAAO,WAAW,CAAA;AAElB,IAAA,KAAA,CAAM,YAAY;AAChB,MAAA,MAAM,OAAA,GAAqB;AAAA,QACzB,IAAA,EAAM,QAAA;AAAA,QACN,WAAA;AAAA,QACA,MAAA,EAAQ,QAAQ,UAAA,CAAW;AAAA,OAC7B;AACA,MAAA,IAAI;AACF,QAAA,WAAA,MAAiB,KAAA,IAAS,OAAA,CAAQ,MAAA,CAAO,OAAO,CAAA,EAAG;AACjD,UAAA,IAAI,QAAQ,UAAA,CAAW,MAAA,CAAO,WAAW,IAAA,EAAM,EAAA,KAAO,QAAQ,EAAA,EAAI;AAClE,UAAA,OAAA,CAAQ,QAAA,IAAY,KAAA;AACpB,UAAA,UAAA,CAAW,QAAQ,OAAO,CAAA;AAC1B,UAAA,MAAA,CAAO,WAAW,CAAA;AAAA,QACpB;AACA,QAAA,IAAI,IAAA,EAAM,EAAA,KAAO,OAAA,CAAQ,EAAA,SAAW,MAAM,CAAA;AAAA,MAC5C,SAAS,KAAA,EAAO;AACd,QAAA,IAAI,IAAA,EAAM,EAAA,KAAO,OAAA,CAAQ,EAAA,EAAI;AAC7B,QAAA,OAAA,CAAQ,KAAA,GAAQ,iBAAiB,KAAA,GAAQ,KAAA,GAAQ,IAAI,KAAA,CAAM,MAAA,CAAO,KAAK,CAAC,CAAA;AACxE,QAAA,MAAA,CAAO,OAAA,CAAQ,UAAA,CAAW,MAAA,CAAO,OAAA,GAAU,cAAc,OAAO,CAAA;AAAA,MAClE;AAAA,IACF,CAAA,GAAG;AAEH,IAAA,OAAO,IAAA;AAAA,EACT,CAAA;AAEA,EAAA,MAAM,WAAoB,MAAM;AAC9B,IAAA,IAAI,CAAC,IAAA,IAAQ,IAAA,CAAK,MAAA,KAAW,aAAa,OAAO,KAAA;AACjD,IAAA,IAAA,CAAK,WAAW,KAAA,EAAM;AACtB,IAAA,MAAA,CAAO,WAAW,CAAA;AAClB,IAAA,OAAO,IAAA;AAAA,EACT,CAAA;AAEA,EAAA,MAAM,WAAoB,MAAM;AAC9B,IAAA,IAAI,CAAC,MAAM,OAAO,KAAA;AAClB,IAAA,IAAA,GAAO,IAAA;AACP,IAAA,OAAA,CAAQ,QAAA,GAAW,UAAU,CAAA;AAC7B,IAAA,OAAO,IAAA;AAAA,EACT,CAAA;AAGA,EAAA,MAAM,QAAA,GAAoB,CAAC,GAAA,KAAQ;AACjC,IAAA,IAAI,CAAC,MAAM,OAAO,KAAA;AAClB,IAAA,IAAA,CAAK,WAAW,KAAA,EAAM;AACtB,IAAA,MAAM,MAAA,GAAS,IAAA,CAAK,MAAA,CAAO,QAAA,CAAS,KAAK,QAAQ,CAAA;AACjD,IAAA,IAAA,GAAO,IAAA;AACP,IAAA,OAAA,CAAQ,QAAA,GAAW,UAAU,CAAA;AAE7B,IAAA,OAAO,GAAA,CAAI,OAAO,MAAM,CAAA;AAAA,EAC1B,CAAA;AAEA,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,WAAA;AAAA,IACN,IAAA,EAAM,IAAA;AAAA,IACN,QAAA,EAAU,EAAE,KAAA,EAAO,QAAA,EAAU,UAAU,QAAA,EAAS;AAAA,IAChD,SAAS,MAAA,EAAQ;AACf,MAAA,SAAA,GAAY,MAAA;AAAA,IACd,CAAA;AAAA,IACA,SAAA,GAAY;AACV,MAAA,IAAA,EAAM,WAAW,KAAA,EAAM;AACvB,MAAA,IAAA,GAAO,IAAA;AACP,MAAA,SAAA,GAAY,IAAA;AAAA,IACd;AAAA,GACF;AACF;AAGA,SAAS,aAAa,GAAA,EAAkB;AACtC,EAAA,MAAM,EAAE,IAAA,EAAM,EAAA,EAAG,GAAI,GAAA,CAAI,SAAA;AACzB,EAAA,MAAM,QAAkB,EAAC;AACzB,EAAA,IAAI,MAAA,GAAS,CAAA;AAEb,EAAA,MAAM,IAAA,GAAO,CAAC,IAAA,KAAiD;AAC7D,IAAA,IAAI,OAAO,IAAA,CAAK,IAAA,KAAS,QAAA,EAAU;AACjC,MAAA,MAAM,KAAA,GAAQ,MAAA;AACd,MAAA,MAAM,GAAA,GAAM,MAAA,GAAS,IAAA,CAAK,IAAA,CAAK,MAAA;AAC/B,MAAA,IAAI,GAAA,GAAM,IAAA,IAAQ,KAAA,GAAQ,EAAA,EAAI;AAC5B,QAAA,KAAA,CAAM,KAAK,IAAA,CAAK,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,IAAI,CAAA,EAAG,IAAA,GAAO,KAAK,CAAA,EAAG,KAAK,GAAA,CAAI,CAAA,EAAG,EAAA,GAAK,KAAK,CAAC,CAAC,CAAA;AAAA,MAChF;AACA,MAAA,MAAA,GAAS,GAAA;AACT,MAAA;AAAA,IACF;AACA,IAAA,MAAA,IAAU,CAAA;AACV,IAAA,KAAA,MAAW,KAAA,IAAU,IAAA,CAAK,OAAA,IAAW,EAAC,EAAgD;AACpF,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACZ;AACA,IAAA,MAAA,IAAU,CAAA;AAAA,EACZ,CAAA;AAEA,EAAA,IAAA,CAAK,IAAI,GAA6C,CAAA;AACtD,EAAA,OAAO,KAAA,CAAM,KAAK,EAAE,CAAA;AACtB","file":"index.js","sourcesContent":["import type { Command, Ctx, Editor, ExtensionDef, PosMarker, Range } from '@matrajs/core'\nimport type { AiRequest, AiSession, AiStatus, AiStream } from './types'\n\nexport interface AiOptions {\n  /** Where the text comes from. Keep the model key on your server, not here. */\n  stream: AiStream\n  /** Called on every status change, for spinners and error toasts. */\n  onStatus?: (session: AiSession) => void\n}\n\ninterface Live {\n  id: number\n  marker: PosMarker\n  /** The range as captured when the request started. */\n  captured: Range\n  received: string\n  status: AiStatus\n  controller: AbortController\n  error?: Error\n}\n\n/**\n * Streaming AI edits that survive concurrent typing.\n *\n * The whole point of this extension is the marker. A request captures the range\n * it is rewriting *and* a position marker; every chunk that comes back is\n * applied to `marker.mapRange(captured)`, not to the original numbers. The user\n * can type, delete and paste anywhere while the model is talking, and the\n * answer still lands on the words they selected.\n */\nexport function ai(options: AiOptions): ExtensionDef<{\n  askAi: Command<[instruction: string]>\n  cancelAi: Command\n  acceptAi: Command\n  rejectAi: Command\n}> {\n  let live: Live | null = null\n  let nextId = 1\n  let editorRef: Editor | null = null\n\n  const snapshot = (): AiSession => ({\n    id: live?.id ?? 0,\n    status: live?.status ?? 'idle',\n    range: live ? live.marker.mapRange(live.captured) : { from: 0 as never, to: 0 as never },\n    received: live?.received ?? '',\n    error: live?.error,\n  })\n\n  const report = (status: AiStatus) => {\n    if (live) live.status = status\n    options.onStatus?.(snapshot())\n  }\n\n  /** Replace the live range with whatever has arrived so far. */\n  const applyChunk = (editor: Editor, session: Live) => {\n    const target = session.marker.mapRange(session.captured)\n    editor.commands.replace(target, session.received)\n  }\n\n  const askAi: Command<[string]> = (ctx, instruction) => {\n    if (live && live.status === 'streaming') return false\n    const editor = editorRef\n    if (!editor) return false\n\n    const { from, to } = ctx.selection\n    if (from === to) return false\n\n    const selected = selectedText(ctx)\n    if (!selected) return false\n\n    const session: Live = {\n      id: nextId++,\n      marker: ctx.mark(),\n      captured: { from, to },\n      received: '',\n      status: 'streaming',\n      controller: new AbortController(),\n    }\n    live = session\n    report('streaming')\n\n    void (async () => {\n      const request: AiRequest = {\n        text: selected,\n        instruction,\n        signal: session.controller.signal,\n      }\n      try {\n        for await (const chunk of options.stream(request)) {\n          if (session.controller.signal.aborted || live?.id !== session.id) return\n          session.received += chunk\n          applyChunk(editor, session)\n          report('streaming')\n        }\n        if (live?.id === session.id) report('done')\n      } catch (error) {\n        if (live?.id !== session.id) return\n        session.error = error instanceof Error ? error : new Error(String(error))\n        report(session.controller.signal.aborted ? 'cancelled' : 'error')\n      }\n    })()\n\n    return true\n  }\n\n  const cancelAi: Command = () => {\n    if (!live || live.status !== 'streaming') return false\n    live.controller.abort()\n    report('cancelled')\n    return true\n  }\n\n  const acceptAi: Command = () => {\n    if (!live) return false\n    live = null\n    options.onStatus?.(snapshot())\n    return true\n  }\n\n  /** Put back what was there before the model started talking. */\n  const rejectAi: Command = (ctx) => {\n    if (!live) return false\n    live.controller.abort()\n    const target = live.marker.mapRange(live.captured)\n    live = null\n    options.onStatus?.(snapshot())\n    // The original text is recovered by undoing the streamed replacements.\n    return ctx.select(target)\n  }\n\n  return {\n    kind: 'extension',\n    name: 'ai',\n    commands: { askAi, cancelAi, acceptAi, rejectAi },\n    onCreate(editor) {\n      editorRef = editor as Editor\n    },\n    onDestroy() {\n      live?.controller.abort()\n      live = null\n      editorRef = null\n    },\n  }\n}\n\n/** The plain text inside the current selection. */\nfunction selectedText(ctx: Ctx): string {\n  const { from, to } = ctx.selection\n  const parts: string[] = []\n  let offset = 0\n\n  const walk = (node: { text?: string; content?: unknown[] }) => {\n    if (typeof node.text === 'string') {\n      const start = offset\n      const end = offset + node.text.length\n      if (end > from && start < to) {\n        parts.push(node.text.slice(Math.max(0, from - start), Math.max(0, to - start)))\n      }\n      offset = end\n      return\n    }\n    offset += 1\n    for (const child of (node.content ?? []) as { text?: string; content?: unknown[] }[]) {\n      walk(child)\n    }\n    offset += 1\n  }\n\n  walk(ctx.doc as { text?: string; content?: unknown[] })\n  return parts.join('')\n}\n"]}

package/package.json

@@ -0,0 +1,50 @@
+{
+  "name": "@matrajs/ai",
+  "version": "0.0.1",
+  "description": "AI extension for Matra \u2014 streaming edits that survive concurrent typing.",
+  "license": "MIT",
+  "author": "Nahim Hossain Shohan",
+  "homepage": "https://matrajs.com",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/amrelaco/matra.git",
+    "directory": "packages/ai"
+  },
+  "bugs": {
+    "url": "https://github.com/amrelaco/matra/issues"
+  },
+  "keywords": [
+    "editor",
+    "rich-text",
+    "ai",
+    "streaming",
+    "matra",
+    "prosemirror"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "type": "module",
+  "sideEffects": false,
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "tsup",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "@matrajs/core": "workspace:*"
+  }
+}