@liveblocks/react-tiptap 2.16.0 → 2.16.1-ai1

package/dist/ai/AiExtension.js

@@ -0,0 +1,299 @@
+'use strict';
+
+var core$1 = require('@liveblocks/core');
+var core = require('@tiptap/core');
+var model = require('@tiptap/pm/model');
+var state = require('@tiptap/pm/state');
+var view = require('@tiptap/pm/view');
+var yProsemirror = require('y-prosemirror');
+var yjs = require('yjs');
+var types = require('../types.js');
+
+const DEFAULT_AI_NAME = "AI";
+const DEFAULT_STATE = { phase: "closed" };
+const RESOLVE_AI_PROMPT_RETRY_ATTEMPTS = 3;
+const RESOLVE_AI_PROMPT_RETRY_DELAYS = [1e3, 2e3, 4e3];
+function getYjsBinding(editor) {
+  return yProsemirror.ySyncPluginKey.getState(editor.view.state).binding;
+}
+function getLiveblocksYjsProvider(editor) {
+  return editor.extensionStorage.liveblocksExtension?.provider;
+}
+const AiExtension = core.Extension.create({
+  name: "liveblocksAi",
+  addOptions() {
+    return {
+      doc: void 0,
+      pud: void 0,
+      resolveAiPrompt: () => Promise.reject(),
+      name: DEFAULT_AI_NAME
+    };
+  },
+  addStorage() {
+    return {
+      state: DEFAULT_STATE,
+      name: this.options.name
+    };
+  },
+  addCommands() {
+    return {
+      askAi: (prompt) => () => {
+        if (typeof prompt === "string") {
+          this.editor.commands.$startAiToolbarThinking(prompt);
+        } else {
+          this.editor.commands.$openAiToolbarAsking();
+        }
+        return true;
+      },
+      $acceptAiToolbarOutput: () => ({ tr }) => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "reviewing") {
+          return false;
+        }
+        const binding = getYjsBinding(this.editor);
+        if (!binding) {
+          return false;
+        }
+        const fragmentContent = yProsemirror.yXmlFragmentToProseMirrorFragment(
+          binding.type,
+          this.editor.state.schema
+        );
+        tr.setMeta("addToHistory", false);
+        tr.replace(
+          0,
+          this.editor.state.doc.content.size,
+          new model.Slice(model.Fragment.from(fragmentContent), 0, 0)
+        );
+        tr.setMeta(yProsemirror.ySyncPluginKey, {
+          snapshot: null,
+          prevSnapshot: null
+        });
+        getLiveblocksYjsProvider(this.editor)?.unpause();
+        this.editor.setEditable(true);
+        this.storage.snapshot = void 0;
+        this.storage.state = { phase: "closed" };
+        return true;
+      },
+      $closeAiToolbar: () => ({ tr }) => {
+        const currentState = this.storage.state;
+        if (currentState.phase === "thinking") {
+          currentState.abortController.abort();
+        }
+        if (currentState.phase === "thinking" || currentState.phase === "reviewing") {
+          if (this.storage.snapshot) {
+            const binding = getYjsBinding(this.editor);
+            if (binding) {
+              binding.mapping.clear();
+              const docFromSnapshot = yjs.createDocFromSnapshot(
+                binding.doc,
+                this.storage.snapshot
+              );
+              const type = docFromSnapshot.getXmlFragment("default");
+              const fragmentContent = yProsemirror.yXmlFragmentToProseMirrorFragment(
+                type,
+                this.editor.state.schema
+              );
+              tr.setMeta("addToHistory", false);
+              tr.replace(
+                0,
+                this.editor.state.doc.content.size,
+                new model.Slice(model.Fragment.from(fragmentContent), 0, 0)
+              );
+              tr.setMeta(yProsemirror.ySyncPluginKey, {
+                snapshot: null,
+                prevSnapshot: null
+              });
+              getLiveblocksYjsProvider(this.editor)?.unpause();
+              if (this.options.doc) {
+                this.options.doc.gc = true;
+              }
+              this.storage.snapshot = void 0;
+            }
+          }
+          this.editor.setEditable(true);
+        }
+        this.storage.state = { phase: "closed" };
+        return true;
+      },
+      $openAiToolbarAsking: () => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "closed") {
+          return false;
+        }
+        if (this.editor.isFocused) {
+          this.editor.commands.blur();
+        }
+        this.storage.state = {
+          phase: "asking",
+          customPrompt: ""
+        };
+        return true;
+      },
+      $startAiToolbarThinking: (prompt) => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase === "thinking") {
+          return false;
+        }
+        if (currentState.phase === "reviewing") {
+          return this.editor.commands.$closeAiToolbar();
+        }
+        if (this.editor.isFocused) {
+          this.editor.commands.blur();
+        }
+        const abortController = new AbortController();
+        const provider = getLiveblocksYjsProvider(this.editor);
+        this.storage.state = {
+          phase: "thinking",
+          customPrompt: currentState.customPrompt ?? "",
+          prompt,
+          abortController
+        };
+        this.editor.setEditable(false);
+        core$1.autoRetry(
+          async () => {
+            await provider?.pause();
+            const { from, to } = this.editor.state.selection.empty ? {
+              from: Math.max(this.editor.state.selection.to - 30, 0),
+              to: this.editor.state.selection.to
+            } : this.editor.state.selection;
+            return this.options.resolveAiPrompt({
+              prompt,
+              selectionText: this.editor.state.doc.textBetween(from, to, " "),
+              context: this.editor.getText().slice(0, 3e3),
+              signal: abortController.signal
+            });
+          },
+          RESOLVE_AI_PROMPT_RETRY_ATTEMPTS,
+          RESOLVE_AI_PROMPT_RETRY_DELAYS,
+          (error) => {
+            return abortController.signal.aborted || error instanceof core$1.HttpError && error.status >= 400 && error.status < 500;
+          }
+        ).then((output) => {
+          if (abortController.signal.aborted) {
+            return;
+          }
+          this.editor.commands._handleAiToolbarThinkingSuccess({
+            type: output.type,
+            text: output.content
+          });
+        }).catch((error) => {
+          if (abortController.signal.aborted) {
+            return;
+          }
+          this.editor.commands._handleAiToolbarThinkingError(error);
+        });
+        return true;
+      },
+      $cancelAiToolbarThinking: () => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "thinking") {
+          return false;
+        }
+        currentState.abortController.abort();
+        this.editor.setEditable(true);
+        this.storage.state = {
+          phase: "asking",
+          customPrompt: currentState.prompt === currentState.customPrompt ? currentState.customPrompt : ""
+        };
+        return true;
+      },
+      _handleAiToolbarThinkingSuccess: (output) => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "thinking" || !this.options.doc) {
+          return false;
+        }
+        let selection = this.editor.state.selection;
+        if (["modification", "insert"].includes(output.type)) {
+          this.options.doc.gc = false;
+          this.storage.snapshot = yjs.snapshot(this.options.doc);
+          const { empty, from, to } = this.editor.state.selection;
+          const contentTarget = empty || output.type === "insert" ? to : {
+            from,
+            to
+          };
+          const targetTo = output.type === "insert" ? to : from;
+          this.editor.commands.insertContentAt(contentTarget, output.text);
+          setTimeout(() => {
+            if (this.storage.snapshot) {
+              this.editor.commands._renderAiToolbarDiffInEditor(this.storage.snapshot);
+            }
+          }, 100);
+          selection = {
+            from,
+            to: targetTo + output.text.length
+          };
+        }
+        this.storage.state = {
+          phase: "reviewing",
+          customPrompt: "",
+          prompt: currentState.prompt,
+          output,
+          selection
+        };
+        return true;
+      },
+      _handleAiToolbarThinkingError: (error) => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "thinking") {
+          return false;
+        }
+        this.editor.setEditable(true);
+        this.storage.state = {
+          phase: "asking",
+          customPrompt: currentState.prompt === currentState.customPrompt ? currentState.customPrompt : "",
+          error
+        };
+        return true;
+      },
+      _updateAiToolbarCustomPrompt: (customPrompt) => () => {
+        const currentState = this.storage.state;
+        if (typeof currentState.customPrompt !== "string") {
+          return false;
+        }
+        this.storage.state.customPrompt = typeof customPrompt === "function" ? customPrompt(currentState.customPrompt) : customPrompt;
+        return true;
+      },
+      _renderAiToolbarDiffInEditor: (previous) => () => {
+        if (!this.options.doc) {
+          return false;
+        }
+        const previousSnapshot = previous ?? yjs.emptySnapshot;
+        const currentSnapshot = yjs.snapshot(this.options.doc);
+        if (yjs.equalSnapshots(previousSnapshot, currentSnapshot)) {
+          return true;
+        }
+        const binding = getYjsBinding(this.editor);
+        if (binding) {
+          binding.renderSnapshot(currentSnapshot, previousSnapshot);
+          return true;
+        }
+        return false;
+      }
+    };
+  },
+  addProseMirrorPlugins() {
+    return [
+      new state.Plugin({
+        key: types.AI_TOOLBAR_SELECTION_PLUGIN,
+        props: {
+          decorations: ({ doc, selection }) => {
+            if (this.storage.state.phase === "closed" || this.storage.state.phase === "reviewing" && this.storage.state.output.type !== "other") {
+              return view.DecorationSet.create(doc, []);
+            }
+            const { from, to } = this.storage.state.selection ?? selection;
+            const decorations = [
+              view.Decoration.inline(from, to, {
+                class: "lb-root lb-selection lb-tiptap-active-selection"
+              })
+            ];
+            return view.DecorationSet.create(doc, decorations);
+          }
+        }
+      })
+    ];
+  }
+});
+
+exports.AiExtension = AiExtension;
+exports.DEFAULT_STATE = DEFAULT_STATE;
+//# sourceMappingURL=AiExtension.js.map

package/dist/ai/AiExtension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AiExtension.js","sources":["../../src/ai/AiExtension.ts"],"sourcesContent":["import { autoRetry, HttpError } from \"@liveblocks/core\";\nimport type { LiveblocksYjsProvider } from \"@liveblocks/yjs\";\nimport type { CommandProps, Editor } from \"@tiptap/core\";\nimport { Extension } from \"@tiptap/core\";\nimport { Fragment, Slice } from \"@tiptap/pm/model\";\nimport { Plugin } from \"@tiptap/pm/state\";\nimport { Decoration, DecorationSet } from \"@tiptap/pm/view\";\nimport {\n  ySyncPluginKey,\n  yXmlFragmentToProseMirrorFragment,\n} from \"y-prosemirror\";\nimport type { Snapshot } from \"yjs\";\nimport {\n  createDocFromSnapshot,\n  emptySnapshot,\n  equalSnapshots,\n  snapshot,\n} from \"yjs\";\n\nimport {\n  AI_TOOLBAR_SELECTION_PLUGIN,\n  type AiCommands,\n  type AiExtensionOptions,\n  type AiExtensionStorage,\n  type AiToolbarOutput,\n  type AiToolbarState,\n  type LiveblocksExtensionStorage,\n  type YSyncPluginState,\n} from \"../types\";\n\nconst DEFAULT_AI_NAME = \"AI\";\nexport const DEFAULT_STATE: AiToolbarState = { phase: \"closed\" };\n\nconst RESOLVE_AI_PROMPT_RETRY_ATTEMPTS = 3;\nconst RESOLVE_AI_PROMPT_RETRY_DELAYS = [1000, 2000, 4000];\n\nfunction getYjsBinding(editor: Editor) {\n  return (ySyncPluginKey.getState(editor.view.state) as YSyncPluginState)\n    .binding;\n}\n\nfunction getLiveblocksYjsProvider(editor: Editor) {\n  return (\n    editor.extensionStorage.liveblocksExtension as\n      | LiveblocksExtensionStorage\n      | undefined\n  )?.provider as LiveblocksYjsProvider | undefined;\n}\n\nexport const AiExtension = Extension.create<\n  AiExtensionOptions,\n  AiExtensionStorage\n>({\n  name: \"liveblocksAi\",\n  addOptions() {\n    return {\n      doc: undefined,\n      pud: undefined,\n\n      // The actual default resolver is set in LiveblocksExtension via AiExtension.configure()\n      resolveAiPrompt: () => Promise.reject(),\n      name: DEFAULT_AI_NAME,\n    };\n  },\n  addStorage() {\n    return {\n      state: DEFAULT_STATE,\n      name: this.options.name,\n    };\n  },\n  addCommands() {\n    return {\n      askAi: (prompt) => () => {\n        if (typeof prompt === \"string\") {\n          (\n            this.editor.commands as unknown as AiCommands\n          ).$startAiToolbarThinking(prompt);\n        } else {\n          (\n            this.editor.commands as unknown as AiCommands\n          ).$openAiToolbarAsking();\n        }\n\n        return true;\n      },\n\n      $acceptAiToolbarOutput:\n        () =>\n        ({ tr }: CommandProps) => {\n          const currentState = this.storage.state;\n          if (currentState.phase !== \"reviewing\") {\n            return false;\n          }\n          const binding = getYjsBinding(this.editor);\n          if (!binding) {\n            return false;\n          }\n\n          const fragmentContent = yXmlFragmentToProseMirrorFragment(\n            binding.type,\n            this.editor.state.schema\n          );\n          tr.setMeta(\"addToHistory\", false);\n          tr.replace(\n            0,\n            this.editor.state.doc.content.size,\n            new Slice(Fragment.from(fragmentContent), 0, 0)\n          );\n          tr.setMeta(ySyncPluginKey, {\n            snapshot: null,\n            prevSnapshot: null,\n          });\n\n          // TODO: move this cleanup to somewhere that closeAIToolbar can share\n          getLiveblocksYjsProvider(this.editor)?.unpause();\n          this.editor.setEditable(true);\n          this.storage.snapshot = undefined;\n          this.storage.state = { phase: \"closed\" };\n          return true;\n        },\n\n      $closeAiToolbar:\n        () =>\n        ({ tr }: CommandProps) => {\n          const currentState = this.storage.state;\n\n          // 1. If in \"thinking\" phase, cancel the current AI request\n          if (currentState.phase === \"thinking\") {\n            currentState.abortController.abort();\n          }\n\n          // 2. If in \"thinking\" or \"reviewing\" phase, revert the editor if possible and unblock it\n          if (\n            currentState.phase === \"thinking\" ||\n            currentState.phase === \"reviewing\"\n          ) {\n            if (this.storage.snapshot) {\n              const binding = getYjsBinding(this.editor);\n\n              if (binding) {\n                binding.mapping.clear();\n\n                const docFromSnapshot = createDocFromSnapshot(\n                  binding.doc,\n                  this.storage.snapshot\n                );\n                const type = docFromSnapshot.getXmlFragment(\"default\"); // TODO: field\n                const fragmentContent = yXmlFragmentToProseMirrorFragment(\n                  type,\n                  this.editor.state.schema\n                );\n\n                tr.setMeta(\"addToHistory\", false);\n                tr.replace(\n                  0,\n                  this.editor.state.doc.content.size,\n                  new Slice(Fragment.from(fragmentContent), 0, 0)\n                );\n                tr.setMeta(ySyncPluginKey, {\n                  snapshot: null,\n                  prevSnapshot: null,\n                });\n\n                getLiveblocksYjsProvider(this.editor)?.unpause();\n\n                if (this.options.doc) {\n                  this.options.doc.gc = true;\n                }\n\n                this.storage.snapshot = undefined;\n              }\n            }\n\n            this.editor.setEditable(true);\n          }\n\n          // 4. Set to \"closed\" phase\n          this.storage.state = { phase: \"closed\" };\n\n          return true;\n        },\n\n      $openAiToolbarAsking: () => () => {\n        const currentState = this.storage.state;\n\n        // 1. If NOT in \"closed\" phase, do nothing\n        if (currentState.phase !== \"closed\") {\n          return false;\n        }\n\n        // 2. Blur the editor if needed\n        if (this.editor.isFocused) {\n          this.editor.commands.blur();\n        }\n\n        // 3. Set to \"asking\" phase\n        this.storage.state = {\n          phase: \"asking\",\n\n          // Initialize the custom prompt as empty\n          customPrompt: \"\",\n        };\n\n        return true;\n      },\n\n      $startAiToolbarThinking: (prompt: string) => () => {\n        const currentState = this.storage.state;\n\n        // 1. If in \"thinking\" phase already, do nothing\n        if (currentState.phase === \"thinking\") {\n          return false;\n        }\n\n        if (currentState.phase === \"reviewing\") {\n          // TODO: this is a retry, we should actually retry and start thinking again\n          //       Maybe not, it could be a refinement prompt\n          return (\n            this.editor.commands as unknown as AiCommands\n          ).$closeAiToolbar();\n        }\n\n        // 2. Blur the editor if needed\n        if (this.editor.isFocused) {\n          this.editor.commands.blur();\n        }\n\n        const abortController = new AbortController();\n        const provider = getLiveblocksYjsProvider(this.editor);\n\n        // 3. Set to \"thinking\" phase\n        this.storage.state = {\n          phase: \"thinking\",\n          customPrompt: currentState.customPrompt ?? \"\",\n          prompt,\n          abortController,\n        };\n\n        // 4. Block the editor\n        this.editor.setEditable(false);\n\n        // 5. Start the AI request\n        autoRetry(\n          async () => {\n            await provider?.pause();\n            const { from, to } = this.editor.state.selection.empty\n              ? {\n                  // TODO: this is a hack to get the context around the selection, we need to improve this\n                  from: Math.max(this.editor.state.selection.to - 30, 0),\n                  to: this.editor.state.selection.to,\n                }\n              : this.editor.state.selection;\n\n            return this.options.resolveAiPrompt({\n              prompt,\n              selectionText: this.editor.state.doc.textBetween(from, to, \" \"),\n              /*\n               TODO: This needs a maximum to avoid overloading context, for now I've arbitrailiry chosen 3000\n               characters but this will need to be improved, probably using word boundary of some sort (languages can make that tricky)\n               as well as choosing text around the selection, so before/after.\n            */\n              context: this.editor.getText().slice(0, 3_000),\n              signal: abortController.signal,\n            });\n          },\n          RESOLVE_AI_PROMPT_RETRY_ATTEMPTS,\n          RESOLVE_AI_PROMPT_RETRY_DELAYS,\n\n          (error) => {\n            // Don't retry on 4xx errors or if the request was aborted\n            return (\n              abortController.signal.aborted ||\n              (error instanceof HttpError &&\n                error.status >= 400 &&\n                error.status < 500)\n            );\n          }\n        )\n          .then((output) => {\n            if (abortController.signal.aborted) {\n              return;\n            }\n\n            // 5.a. If the AI request succeeds, set to \"reviewing\" phase with the output\n            (\n              this.editor.commands as unknown as AiCommands\n            )._handleAiToolbarThinkingSuccess({\n              type: output.type,\n              text: output.content,\n            });\n          })\n          .catch((error) => {\n            if (abortController.signal.aborted) {\n              return;\n            }\n\n            // 5.b. If the AI request fails, set to \"asking\" phase with error\n            (\n              this.editor.commands as unknown as AiCommands\n            )._handleAiToolbarThinkingError(error as Error);\n          });\n\n        return true;\n      },\n\n      $cancelAiToolbarThinking: () => () => {\n        const currentState = this.storage.state;\n\n        // 1. If NOT in \"thinking\" phase, do nothing\n        if (currentState.phase !== \"thinking\") {\n          return false;\n        }\n\n        // 2. Cancel the current AI request\n        currentState.abortController.abort();\n\n        // 3. Unblock the editor\n        this.editor.setEditable(true);\n\n        // 4. Set to \"asking\" phase\n        this.storage.state = {\n          phase: \"asking\",\n          // If the custom prompt is different than the prompt, reset it\n          customPrompt:\n            currentState.prompt === currentState.customPrompt\n              ? currentState.customPrompt\n              : \"\",\n        };\n\n        return true;\n      },\n\n      _handleAiToolbarThinkingSuccess: (output: AiToolbarOutput) => () => {\n        const currentState = this.storage.state;\n\n        // 1. If NOT in \"thinking\" phase, do nothing\n        if (currentState.phase !== \"thinking\" || !this.options.doc) {\n          return false;\n        }\n\n        let selection: { from: number; to: number } =\n          this.editor.state.selection;\n\n        // 2. If the output is a diff, apply it to the editor\n        if ([\"modification\", \"insert\"].includes(output.type)) {\n          this.options.doc.gc = false;\n          this.storage.snapshot = snapshot(this.options.doc);\n\n          const { empty, from, to } = this.editor.state.selection;\n\n          // if the selection is empty, insert at the end of the selection\n          const contentTarget =\n            empty || output.type === \"insert\"\n              ? to\n              : {\n                  from,\n                  to,\n                };\n\n          // when inserting, use the \"to\" position, otherwise when modifing, use the \"from\" position\n          const targetTo = output.type === \"insert\" ? to : from;\n\n          // 2.a. Insert the output.\n          this.editor.commands.insertContentAt(contentTarget, output.text);\n\n          // 2.b. Diff the output in the editor\n          // TODO: setTimeout will make this execute after the current transaction is committed, which is returned by the insert content command.\n          setTimeout(() => {\n            if (this.storage.snapshot) {\n              (\n                this.editor.commands as unknown as AiCommands\n              )._renderAiToolbarDiffInEditor(this.storage.snapshot);\n            }\n          }, 100);\n\n          selection = {\n            from,\n            to: targetTo + output.text.length, // take into account the new length with output\n          };\n        }\n\n        // 3. Set to \"reviewing\" phase with the output\n        this.storage.state = {\n          phase: \"reviewing\",\n          customPrompt: \"\",\n          prompt: currentState.prompt,\n          output,\n          selection,\n        };\n\n        return true;\n      },\n\n      _handleAiToolbarThinkingError: (error: Error) => () => {\n        const currentState = this.storage.state;\n\n        // 1. If NOT in \"thinking\" phase, do nothing\n        if (currentState.phase !== \"thinking\") {\n          return false;\n        }\n\n        // 2. Unblock the editor\n        this.editor.setEditable(true);\n\n        // 3. Set to \"asking\" phase with error\n        this.storage.state = {\n          phase: \"asking\",\n          // If the custom prompt is different than the prompt, reset it\n          customPrompt:\n            currentState.prompt === currentState.customPrompt\n              ? currentState.customPrompt\n              : \"\",\n          // TODO: Improve error handling\n          error,\n        };\n\n        return true;\n      },\n\n      _updateAiToolbarCustomPrompt:\n        (customPrompt: string | ((currentCustomPrompt: string) => string)) =>\n        () => {\n          const currentState = this.storage.state;\n\n          // 1. If NOT in a phase with a custom prompt, do nothing\n          if (typeof currentState.customPrompt !== \"string\") {\n            return false;\n          }\n\n          // 2. Update the custom prompt\n          this.storage.state.customPrompt =\n            typeof customPrompt === \"function\"\n              ? customPrompt(currentState.customPrompt)\n              : customPrompt;\n\n          return true;\n        },\n\n      _renderAiToolbarDiffInEditor: (previous?: Snapshot) => () => {\n        if (!this.options.doc) {\n          return false;\n        }\n\n        const previousSnapshot: Snapshot = previous ?? emptySnapshot;\n        const currentSnapshot = snapshot(this.options.doc);\n\n        if (equalSnapshots(previousSnapshot, currentSnapshot)) {\n          return true;\n        }\n\n        const binding = getYjsBinding(this.editor);\n\n        if (binding) {\n          binding.renderSnapshot(currentSnapshot, previousSnapshot);\n          return true;\n        }\n\n        return false;\n      },\n    };\n  },\n  addProseMirrorPlugins() {\n    return [\n      new Plugin({\n        key: AI_TOOLBAR_SELECTION_PLUGIN,\n        props: {\n          decorations: ({ doc, selection }) => {\n            if (\n              this.storage.state.phase === \"closed\" ||\n              // TODO: Don't show the selection when reviewing diff outputs\n              (this.storage.state.phase === \"reviewing\" &&\n                this.storage.state.output.type !== \"other\")\n            ) {\n              return DecorationSet.create(doc, []);\n            }\n            const { from, to } = this.storage.state.selection ?? selection;\n            const decorations: Decoration[] = [\n              Decoration.inline(from, to, {\n                class: \"lb-root lb-selection lb-tiptap-active-selection\",\n              }),\n            ];\n            return DecorationSet.create(doc, decorations);\n          },\n        },\n      }),\n    ];\n  },\n});\n"],"names":["ySyncPluginKey","Extension","yXmlFragmentToProseMirrorFragment","Slice","Fragment","createDocFromSnapshot","autoRetry","HttpError","snapshot","emptySnapshot","equalSnapshots","Plugin","AI_TOOLBAR_SELECTION_PLUGIN","DecorationSet","Decoration"],"mappings":";;;;;;;;;;;AA8BA,MAAM,eAAkB,GAAA,IAAA,CAAA;AACX,MAAA,aAAA,GAAgC,EAAE,KAAA,EAAO,QAAS,GAAA;AAE/D,MAAM,gCAAmC,GAAA,CAAA,CAAA;AACzC,MAAM,8BAAiC,GAAA,CAAC,GAAM,EAAA,GAAA,EAAM,GAAI,CAAA,CAAA;AAExD,SAAS,cAAc,MAAgB,EAAA;AACrC,EAAA,OAAQA,2BAAe,CAAA,QAAA,CAAS,MAAO,CAAA,IAAA,CAAK,KAAK,CAC9C,CAAA,OAAA,CAAA;AACL,CAAA;AAEA,SAAS,yBAAyB,MAAgB,EAAA;AAChD,EACE,OAAA,MAAA,CAAO,iBAAiB,mBAGvB,EAAA,QAAA,CAAA;AACL,CAAA;AAEa,MAAA,WAAA,GAAcC,eAAU,MAGnC,CAAA;AAAA,EACA,IAAM,EAAA,cAAA;AAAA,EACN,UAAa,GAAA;AACX,IAAO,OAAA;AAAA,MACL,GAAK,EAAA,KAAA,CAAA;AAAA,MACL,GAAK,EAAA,KAAA,CAAA;AAAA,MAGL,eAAA,EAAiB,MAAM,OAAA,CAAQ,MAAO,EAAA;AAAA,MACtC,IAAM,EAAA,eAAA;AAAA,KACR,CAAA;AAAA,GACF;AAAA,EACA,UAAa,GAAA;AACX,IAAO,OAAA;AAAA,MACL,KAAO,EAAA,aAAA;AAAA,MACP,IAAA,EAAM,KAAK,OAAQ,CAAA,IAAA;AAAA,KACrB,CAAA;AAAA,GACF;AAAA,EACA,WAAc,GAAA;AACZ,IAAO,OAAA;AAAA,MACL,KAAA,EAAO,CAAC,MAAA,KAAW,MAAM;AACvB,QAAI,IAAA,OAAO,WAAW,QAAU,EAAA;AAC9B,UACE,IAAK,CAAA,MAAA,CAAO,QACZ,CAAA,uBAAA,CAAwB,MAAM,CAAA,CAAA;AAAA,SAC3B,MAAA;AACL,UACE,IAAA,CAAK,MAAO,CAAA,QAAA,CACZ,oBAAqB,EAAA,CAAA;AAAA,SACzB;AAEA,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEA,sBACE,EAAA,MACA,CAAC,EAAE,IAAuB,KAAA;AACxB,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAClC,QAAI,IAAA,YAAA,CAAa,UAAU,WAAa,EAAA;AACtC,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AACA,QAAM,MAAA,OAAA,GAAU,aAAc,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AACzC,QAAA,IAAI,CAAC,OAAS,EAAA;AACZ,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAEA,QAAA,MAAM,eAAkB,GAAAC,8CAAA;AAAA,UACtB,OAAQ,CAAA,IAAA;AAAA,UACR,IAAA,CAAK,OAAO,KAAM,CAAA,MAAA;AAAA,SACpB,CAAA;AACA,QAAG,EAAA,CAAA,OAAA,CAAQ,gBAAgB,KAAK,CAAA,CAAA;AAChC,QAAG,EAAA,CAAA,OAAA;AAAA,UACD,CAAA;AAAA,UACA,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,GAAA,CAAI,OAAQ,CAAA,IAAA;AAAA,UAC9B,IAAIC,WAAM,CAAAC,cAAA,CAAS,KAAK,eAAe,CAAA,EAAG,GAAG,CAAC,CAAA;AAAA,SAChD,CAAA;AACA,QAAA,EAAA,CAAG,QAAQJ,2BAAgB,EAAA;AAAA,UACzB,QAAU,EAAA,IAAA;AAAA,UACV,YAAc,EAAA,IAAA;AAAA,SACf,CAAA,CAAA;AAGD,QAAyB,wBAAA,CAAA,IAAA,CAAK,MAAM,CAAA,EAAG,OAAQ,EAAA,CAAA;AAC/C,QAAK,IAAA,CAAA,MAAA,CAAO,YAAY,IAAI,CAAA,CAAA;AAC5B,QAAA,IAAA,CAAK,QAAQ,QAAW,GAAA,KAAA,CAAA,CAAA;AACxB,QAAA,IAAA,CAAK,OAAQ,CAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,QAAS,EAAA,CAAA;AACvC,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEF,eACE,EAAA,MACA,CAAC,EAAE,IAAuB,KAAA;AACxB,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAGlC,QAAI,IAAA,YAAA,CAAa,UAAU,UAAY,EAAA;AACrC,UAAA,YAAA,CAAa,gBAAgB,KAAM,EAAA,CAAA;AAAA,SACrC;AAGA,QAAA,IACE,YAAa,CAAA,KAAA,KAAU,UACvB,IAAA,YAAA,CAAa,UAAU,WACvB,EAAA;AACA,UAAI,IAAA,IAAA,CAAK,QAAQ,QAAU,EAAA;AACzB,YAAM,MAAA,OAAA,GAAU,aAAc,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAEzC,YAAA,IAAI,OAAS,EAAA;AACX,cAAA,OAAA,CAAQ,QAAQ,KAAM,EAAA,CAAA;AAEtB,cAAA,MAAM,eAAkB,GAAAK,yBAAA;AAAA,gBACtB,OAAQ,CAAA,GAAA;AAAA,gBACR,KAAK,OAAQ,CAAA,QAAA;AAAA,eACf,CAAA;AACA,cAAM,MAAA,IAAA,GAAO,eAAgB,CAAA,cAAA,CAAe,SAAS,CAAA,CAAA;AACrD,cAAA,MAAM,eAAkB,GAAAH,8CAAA;AAAA,gBACtB,IAAA;AAAA,gBACA,IAAA,CAAK,OAAO,KAAM,CAAA,MAAA;AAAA,eACpB,CAAA;AAEA,cAAG,EAAA,CAAA,OAAA,CAAQ,gBAAgB,KAAK,CAAA,CAAA;AAChC,cAAG,EAAA,CAAA,OAAA;AAAA,gBACD,CAAA;AAAA,gBACA,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,GAAA,CAAI,OAAQ,CAAA,IAAA;AAAA,gBAC9B,IAAIC,WAAM,CAAAC,cAAA,CAAS,KAAK,eAAe,CAAA,EAAG,GAAG,CAAC,CAAA;AAAA,eAChD,CAAA;AACA,cAAA,EAAA,CAAG,QAAQJ,2BAAgB,EAAA;AAAA,gBACzB,QAAU,EAAA,IAAA;AAAA,gBACV,YAAc,EAAA,IAAA;AAAA,eACf,CAAA,CAAA;AAED,cAAyB,wBAAA,CAAA,IAAA,CAAK,MAAM,CAAA,EAAG,OAAQ,EAAA,CAAA;AAE/C,cAAI,IAAA,IAAA,CAAK,QAAQ,GAAK,EAAA;AACpB,gBAAK,IAAA,CAAA,OAAA,CAAQ,IAAI,EAAK,GAAA,IAAA,CAAA;AAAA,eACxB;AAEA,cAAA,IAAA,CAAK,QAAQ,QAAW,GAAA,KAAA,CAAA,CAAA;AAAA,aAC1B;AAAA,WACF;AAEA,UAAK,IAAA,CAAA,MAAA,CAAO,YAAY,IAAI,CAAA,CAAA;AAAA,SAC9B;AAGA,QAAA,IAAA,CAAK,OAAQ,CAAA,KAAA,GAAQ,EAAE,KAAA,EAAO,QAAS,EAAA,CAAA;AAEvC,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEF,oBAAA,EAAsB,MAAM,MAAM;AAChC,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAGlC,QAAI,IAAA,YAAA,CAAa,UAAU,QAAU,EAAA;AACnC,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAGA,QAAI,IAAA,IAAA,CAAK,OAAO,SAAW,EAAA;AACzB,UAAK,IAAA,CAAA,MAAA,CAAO,SAAS,IAAK,EAAA,CAAA;AAAA,SAC5B;AAGA,QAAA,IAAA,CAAK,QAAQ,KAAQ,GAAA;AAAA,UACnB,KAAO,EAAA,QAAA;AAAA,UAGP,YAAc,EAAA,EAAA;AAAA,SAChB,CAAA;AAEA,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEA,uBAAA,EAAyB,CAAC,MAAA,KAAmB,MAAM;AACjD,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAGlC,QAAI,IAAA,YAAA,CAAa,UAAU,UAAY,EAAA;AACrC,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAEA,QAAI,IAAA,YAAA,CAAa,UAAU,WAAa,EAAA;AAGtC,UACE,OAAA,IAAA,CAAK,MAAO,CAAA,QAAA,CACZ,eAAgB,EAAA,CAAA;AAAA,SACpB;AAGA,QAAI,IAAA,IAAA,CAAK,OAAO,SAAW,EAAA;AACzB,UAAK,IAAA,CAAA,MAAA,CAAO,SAAS,IAAK,EAAA,CAAA;AAAA,SAC5B;AAEA,QAAM,MAAA,eAAA,GAAkB,IAAI,eAAgB,EAAA,CAAA;AAC5C,QAAM,MAAA,QAAA,GAAW,wBAAyB,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAGrD,QAAA,IAAA,CAAK,QAAQ,KAAQ,GAAA;AAAA,UACnB,KAAO,EAAA,UAAA;AAAA,UACP,YAAA,EAAc,aAAa,YAAgB,IAAA,EAAA;AAAA,UAC3C,MAAA;AAAA,UACA,eAAA;AAAA,SACF,CAAA;AAGA,QAAK,IAAA,CAAA,MAAA,CAAO,YAAY,KAAK,CAAA,CAAA;AAG7B,QAAAM,gBAAA;AAAA,UACE,YAAY;AACV,YAAA,MAAM,UAAU,KAAM,EAAA,CAAA;AACtB,YAAM,MAAA,EAAE,MAAM,EAAG,EAAA,GAAI,KAAK,MAAO,CAAA,KAAA,CAAM,UAAU,KAC7C,GAAA;AAAA,cAEE,IAAA,EAAM,KAAK,GAAI,CAAA,IAAA,CAAK,OAAO,KAAM,CAAA,SAAA,CAAU,EAAK,GAAA,EAAA,EAAI,CAAC,CAAA;AAAA,cACrD,EAAI,EAAA,IAAA,CAAK,MAAO,CAAA,KAAA,CAAM,SAAU,CAAA,EAAA;AAAA,aAClC,GACA,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,SAAA,CAAA;AAEtB,YAAO,OAAA,IAAA,CAAK,QAAQ,eAAgB,CAAA;AAAA,cAClC,MAAA;AAAA,cACA,aAAA,EAAe,KAAK,MAAO,CAAA,KAAA,CAAM,IAAI,WAAY,CAAA,IAAA,EAAM,IAAI,GAAG,CAAA;AAAA,cAM9D,SAAS,IAAK,CAAA,MAAA,CAAO,SAAU,CAAA,KAAA,CAAM,GAAG,GAAK,CAAA;AAAA,cAC7C,QAAQ,eAAgB,CAAA,MAAA;AAAA,aACzB,CAAA,CAAA;AAAA,WACH;AAAA,UACA,gCAAA;AAAA,UACA,8BAAA;AAAA,UAEA,CAAC,KAAU,KAAA;AAET,YACE,OAAA,eAAA,CAAgB,OAAO,OACtB,IAAA,KAAA,YAAiBC,oBAChB,KAAM,CAAA,MAAA,IAAU,GAChB,IAAA,KAAA,CAAM,MAAS,GAAA,GAAA,CAAA;AAAA,WAErB;AAAA,SACF,CACG,IAAK,CAAA,CAAC,MAAW,KAAA;AAChB,UAAI,IAAA,eAAA,CAAgB,OAAO,OAAS,EAAA;AAClC,YAAA,OAAA;AAAA,WACF;AAGA,UACE,IAAA,CAAK,MAAO,CAAA,QAAA,CACZ,+BAAgC,CAAA;AAAA,YAChC,MAAM,MAAO,CAAA,IAAA;AAAA,YACb,MAAM,MAAO,CAAA,OAAA;AAAA,WACd,CAAA,CAAA;AAAA,SACF,CAAA,CACA,KAAM,CAAA,CAAC,KAAU,KAAA;AAChB,UAAI,IAAA,eAAA,CAAgB,OAAO,OAAS,EAAA;AAClC,YAAA,OAAA;AAAA,WACF;AAGA,UACE,IAAK,CAAA,MAAA,CAAO,QACZ,CAAA,6BAAA,CAA8B,KAAc,CAAA,CAAA;AAAA,SAC/C,CAAA,CAAA;AAEH,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEA,wBAAA,EAA0B,MAAM,MAAM;AACpC,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAGlC,QAAI,IAAA,YAAA,CAAa,UAAU,UAAY,EAAA;AACrC,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAGA,QAAA,YAAA,CAAa,gBAAgB,KAAM,EAAA,CAAA;AAGnC,QAAK,IAAA,CAAA,MAAA,CAAO,YAAY,IAAI,CAAA,CAAA;AAG5B,QAAA,IAAA,CAAK,QAAQ,KAAQ,GAAA;AAAA,UACnB,KAAO,EAAA,QAAA;AAAA,UAEP,cACE,YAAa,CAAA,MAAA,KAAW,YAAa,CAAA,YAAA,GACjC,aAAa,YACb,GAAA,EAAA;AAAA,SACR,CAAA;AAEA,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEA,+BAAA,EAAiC,CAAC,MAAA,KAA4B,MAAM;AAClE,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAGlC,QAAA,IAAI,aAAa,KAAU,KAAA,UAAA,IAAc,CAAC,IAAA,CAAK,QAAQ,GAAK,EAAA;AAC1D,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAEA,QAAI,IAAA,SAAA,GACF,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,SAAA,CAAA;AAGpB,QAAA,IAAI,CAAC,cAAgB,EAAA,QAAQ,EAAE,QAAS,CAAA,MAAA,CAAO,IAAI,CAAG,EAAA;AACpD,UAAK,IAAA,CAAA,OAAA,CAAQ,IAAI,EAAK,GAAA,KAAA,CAAA;AACtB,UAAA,IAAA,CAAK,OAAQ,CAAA,QAAA,GAAWC,YAAS,CAAA,IAAA,CAAK,QAAQ,GAAG,CAAA,CAAA;AAEjD,UAAA,MAAM,EAAE,KAAO,EAAA,IAAA,EAAM,IAAO,GAAA,IAAA,CAAK,OAAO,KAAM,CAAA,SAAA,CAAA;AAG9C,UAAA,MAAM,aACJ,GAAA,KAAA,IAAS,MAAO,CAAA,IAAA,KAAS,WACrB,EACA,GAAA;AAAA,YACE,IAAA;AAAA,YACA,EAAA;AAAA,WACF,CAAA;AAGN,UAAA,MAAM,QAAW,GAAA,MAAA,CAAO,IAAS,KAAA,QAAA,GAAW,EAAK,GAAA,IAAA,CAAA;AAGjD,UAAA,IAAA,CAAK,MAAO,CAAA,QAAA,CAAS,eAAgB,CAAA,aAAA,EAAe,OAAO,IAAI,CAAA,CAAA;AAI/D,UAAA,UAAA,CAAW,MAAM;AACf,YAAI,IAAA,IAAA,CAAK,QAAQ,QAAU,EAAA;AACzB,cACE,KAAK,MAAO,CAAA,QAAA,CACZ,4BAA6B,CAAA,IAAA,CAAK,QAAQ,QAAQ,CAAA,CAAA;AAAA,aACtD;AAAA,aACC,GAAG,CAAA,CAAA;AAEN,UAAY,SAAA,GAAA;AAAA,YACV,IAAA;AAAA,YACA,EAAA,EAAI,QAAW,GAAA,MAAA,CAAO,IAAK,CAAA,MAAA;AAAA,WAC7B,CAAA;AAAA,SACF;AAGA,QAAA,IAAA,CAAK,QAAQ,KAAQ,GAAA;AAAA,UACnB,KAAO,EAAA,WAAA;AAAA,UACP,YAAc,EAAA,EAAA;AAAA,UACd,QAAQ,YAAa,CAAA,MAAA;AAAA,UACrB,MAAA;AAAA,UACA,SAAA;AAAA,SACF,CAAA;AAEA,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEA,6BAAA,EAA+B,CAAC,KAAA,KAAiB,MAAM;AACrD,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAGlC,QAAI,IAAA,YAAA,CAAa,UAAU,UAAY,EAAA;AACrC,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAGA,QAAK,IAAA,CAAA,MAAA,CAAO,YAAY,IAAI,CAAA,CAAA;AAG5B,QAAA,IAAA,CAAK,QAAQ,KAAQ,GAAA;AAAA,UACnB,KAAO,EAAA,QAAA;AAAA,UAEP,cACE,YAAa,CAAA,MAAA,KAAW,YAAa,CAAA,YAAA,GACjC,aAAa,YACb,GAAA,EAAA;AAAA,UAEN,KAAA;AAAA,SACF,CAAA;AAEA,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEA,4BAAA,EACE,CAAC,YAAA,KACD,MAAM;AACJ,QAAM,MAAA,YAAA,GAAe,KAAK,OAAQ,CAAA,KAAA,CAAA;AAGlC,QAAI,IAAA,OAAO,YAAa,CAAA,YAAA,KAAiB,QAAU,EAAA;AACjD,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAGA,QAAK,IAAA,CAAA,OAAA,CAAQ,MAAM,YACjB,GAAA,OAAO,iBAAiB,UACpB,GAAA,YAAA,CAAa,YAAa,CAAA,YAAY,CACtC,GAAA,YAAA,CAAA;AAEN,QAAO,OAAA,IAAA,CAAA;AAAA,OACT;AAAA,MAEF,4BAAA,EAA8B,CAAC,QAAA,KAAwB,MAAM;AAC3D,QAAI,IAAA,CAAC,IAAK,CAAA,OAAA,CAAQ,GAAK,EAAA;AACrB,UAAO,OAAA,KAAA,CAAA;AAAA,SACT;AAEA,QAAA,MAAM,mBAA6B,QAAY,IAAAC,iBAAA,CAAA;AAC/C,QAAA,MAAM,eAAkB,GAAAD,YAAA,CAAS,IAAK,CAAA,OAAA,CAAQ,GAAG,CAAA,CAAA;AAEjD,QAAI,IAAAE,kBAAA,CAAe,gBAAkB,EAAA,eAAe,CAAG,EAAA;AACrD,UAAO,OAAA,IAAA,CAAA;AAAA,SACT;AAEA,QAAM,MAAA,OAAA,GAAU,aAAc,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAEzC,QAAA,IAAI,OAAS,EAAA;AACX,UAAQ,OAAA,CAAA,cAAA,CAAe,iBAAiB,gBAAgB,CAAA,CAAA;AACxD,UAAO,OAAA,IAAA,CAAA;AAAA,SACT;AAEA,QAAO,OAAA,KAAA,CAAA;AAAA,OACT;AAAA,KACF,CAAA;AAAA,GACF;AAAA,EACA,qBAAwB,GAAA;AACtB,IAAO,OAAA;AAAA,MACL,IAAIC,YAAO,CAAA;AAAA,QACT,GAAK,EAAAC,iCAAA;AAAA,QACL,KAAO,EAAA;AAAA,UACL,WAAa,EAAA,CAAC,EAAE,GAAA,EAAK,WAAgB,KAAA;AACnC,YAAA,IACE,IAAK,CAAA,OAAA,CAAQ,KAAM,CAAA,KAAA,KAAU,YAE5B,IAAK,CAAA,OAAA,CAAQ,KAAM,CAAA,KAAA,KAAU,eAC5B,IAAK,CAAA,OAAA,CAAQ,KAAM,CAAA,MAAA,CAAO,SAAS,OACrC,EAAA;AACA,cAAA,OAAOC,kBAAc,CAAA,MAAA,CAAO,GAAK,EAAA,EAAE,CAAA,CAAA;AAAA,aACrC;AACA,YAAA,MAAM,EAAE,IAAM,EAAA,EAAA,KAAO,IAAK,CAAA,OAAA,CAAQ,MAAM,SAAa,IAAA,SAAA,CAAA;AACrD,YAAA,MAAM,WAA4B,GAAA;AAAA,cAChCC,eAAA,CAAW,MAAO,CAAA,IAAA,EAAM,EAAI,EAAA;AAAA,gBAC1B,KAAO,EAAA,iDAAA;AAAA,eACR,CAAA;AAAA,aACH,CAAA;AACA,YAAO,OAAAD,kBAAA,CAAc,MAAO,CAAA,GAAA,EAAK,WAAW,CAAA,CAAA;AAAA,WAC9C;AAAA,SACF;AAAA,OACD,CAAA;AAAA,KACH,CAAA;AAAA,GACF;AACF,CAAC;;;;;"}

package/dist/ai/AiExtension.mjs

@@ -0,0 +1,296 @@
+import { autoRetry, HttpError } from '@liveblocks/core';
+import { Extension } from '@tiptap/core';
+import { Slice, Fragment } from '@tiptap/pm/model';
+import { Plugin } from '@tiptap/pm/state';
+import { DecorationSet, Decoration } from '@tiptap/pm/view';
+import { ySyncPluginKey, yXmlFragmentToProseMirrorFragment } from 'y-prosemirror';
+import { createDocFromSnapshot, snapshot, emptySnapshot, equalSnapshots } from 'yjs';
+import { AI_TOOLBAR_SELECTION_PLUGIN } from '../types.mjs';
+
+const DEFAULT_AI_NAME = "AI";
+const DEFAULT_STATE = { phase: "closed" };
+const RESOLVE_AI_PROMPT_RETRY_ATTEMPTS = 3;
+const RESOLVE_AI_PROMPT_RETRY_DELAYS = [1e3, 2e3, 4e3];
+function getYjsBinding(editor) {
+  return ySyncPluginKey.getState(editor.view.state).binding;
+}
+function getLiveblocksYjsProvider(editor) {
+  return editor.extensionStorage.liveblocksExtension?.provider;
+}
+const AiExtension = Extension.create({
+  name: "liveblocksAi",
+  addOptions() {
+    return {
+      doc: void 0,
+      pud: void 0,
+      resolveAiPrompt: () => Promise.reject(),
+      name: DEFAULT_AI_NAME
+    };
+  },
+  addStorage() {
+    return {
+      state: DEFAULT_STATE,
+      name: this.options.name
+    };
+  },
+  addCommands() {
+    return {
+      askAi: (prompt) => () => {
+        if (typeof prompt === "string") {
+          this.editor.commands.$startAiToolbarThinking(prompt);
+        } else {
+          this.editor.commands.$openAiToolbarAsking();
+        }
+        return true;
+      },
+      $acceptAiToolbarOutput: () => ({ tr }) => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "reviewing") {
+          return false;
+        }
+        const binding = getYjsBinding(this.editor);
+        if (!binding) {
+          return false;
+        }
+        const fragmentContent = yXmlFragmentToProseMirrorFragment(
+          binding.type,
+          this.editor.state.schema
+        );
+        tr.setMeta("addToHistory", false);
+        tr.replace(
+          0,
+          this.editor.state.doc.content.size,
+          new Slice(Fragment.from(fragmentContent), 0, 0)
+        );
+        tr.setMeta(ySyncPluginKey, {
+          snapshot: null,
+          prevSnapshot: null
+        });
+        getLiveblocksYjsProvider(this.editor)?.unpause();
+        this.editor.setEditable(true);
+        this.storage.snapshot = void 0;
+        this.storage.state = { phase: "closed" };
+        return true;
+      },
+      $closeAiToolbar: () => ({ tr }) => {
+        const currentState = this.storage.state;
+        if (currentState.phase === "thinking") {
+          currentState.abortController.abort();
+        }
+        if (currentState.phase === "thinking" || currentState.phase === "reviewing") {
+          if (this.storage.snapshot) {
+            const binding = getYjsBinding(this.editor);
+            if (binding) {
+              binding.mapping.clear();
+              const docFromSnapshot = createDocFromSnapshot(
+                binding.doc,
+                this.storage.snapshot
+              );
+              const type = docFromSnapshot.getXmlFragment("default");
+              const fragmentContent = yXmlFragmentToProseMirrorFragment(
+                type,
+                this.editor.state.schema
+              );
+              tr.setMeta("addToHistory", false);
+              tr.replace(
+                0,
+                this.editor.state.doc.content.size,
+                new Slice(Fragment.from(fragmentContent), 0, 0)
+              );
+              tr.setMeta(ySyncPluginKey, {
+                snapshot: null,
+                prevSnapshot: null
+              });
+              getLiveblocksYjsProvider(this.editor)?.unpause();
+              if (this.options.doc) {
+                this.options.doc.gc = true;
+              }
+              this.storage.snapshot = void 0;
+            }
+          }
+          this.editor.setEditable(true);
+        }
+        this.storage.state = { phase: "closed" };
+        return true;
+      },
+      $openAiToolbarAsking: () => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "closed") {
+          return false;
+        }
+        if (this.editor.isFocused) {
+          this.editor.commands.blur();
+        }
+        this.storage.state = {
+          phase: "asking",
+          customPrompt: ""
+        };
+        return true;
+      },
+      $startAiToolbarThinking: (prompt) => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase === "thinking") {
+          return false;
+        }
+        if (currentState.phase === "reviewing") {
+          return this.editor.commands.$closeAiToolbar();
+        }
+        if (this.editor.isFocused) {
+          this.editor.commands.blur();
+        }
+        const abortController = new AbortController();
+        const provider = getLiveblocksYjsProvider(this.editor);
+        this.storage.state = {
+          phase: "thinking",
+          customPrompt: currentState.customPrompt ?? "",
+          prompt,
+          abortController
+        };
+        this.editor.setEditable(false);
+        autoRetry(
+          async () => {
+            await provider?.pause();
+            const { from, to } = this.editor.state.selection.empty ? {
+              from: Math.max(this.editor.state.selection.to - 30, 0),
+              to: this.editor.state.selection.to
+            } : this.editor.state.selection;
+            return this.options.resolveAiPrompt({
+              prompt,
+              selectionText: this.editor.state.doc.textBetween(from, to, " "),
+              context: this.editor.getText().slice(0, 3e3),
+              signal: abortController.signal
+            });
+          },
+          RESOLVE_AI_PROMPT_RETRY_ATTEMPTS,
+          RESOLVE_AI_PROMPT_RETRY_DELAYS,
+          (error) => {
+            return abortController.signal.aborted || error instanceof HttpError && error.status >= 400 && error.status < 500;
+          }
+        ).then((output) => {
+          if (abortController.signal.aborted) {
+            return;
+          }
+          this.editor.commands._handleAiToolbarThinkingSuccess({
+            type: output.type,
+            text: output.content
+          });
+        }).catch((error) => {
+          if (abortController.signal.aborted) {
+            return;
+          }
+          this.editor.commands._handleAiToolbarThinkingError(error);
+        });
+        return true;
+      },
+      $cancelAiToolbarThinking: () => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "thinking") {
+          return false;
+        }
+        currentState.abortController.abort();
+        this.editor.setEditable(true);
+        this.storage.state = {
+          phase: "asking",
+          customPrompt: currentState.prompt === currentState.customPrompt ? currentState.customPrompt : ""
+        };
+        return true;
+      },
+      _handleAiToolbarThinkingSuccess: (output) => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "thinking" || !this.options.doc) {
+          return false;
+        }
+        let selection = this.editor.state.selection;
+        if (["modification", "insert"].includes(output.type)) {
+          this.options.doc.gc = false;
+          this.storage.snapshot = snapshot(this.options.doc);
+          const { empty, from, to } = this.editor.state.selection;
+          const contentTarget = empty || output.type === "insert" ? to : {
+            from,
+            to
+          };
+          const targetTo = output.type === "insert" ? to : from;
+          this.editor.commands.insertContentAt(contentTarget, output.text);
+          setTimeout(() => {
+            if (this.storage.snapshot) {
+              this.editor.commands._renderAiToolbarDiffInEditor(this.storage.snapshot);
+            }
+          }, 100);
+          selection = {
+            from,
+            to: targetTo + output.text.length
+          };
+        }
+        this.storage.state = {
+          phase: "reviewing",
+          customPrompt: "",
+          prompt: currentState.prompt,
+          output,
+          selection
+        };
+        return true;
+      },
+      _handleAiToolbarThinkingError: (error) => () => {
+        const currentState = this.storage.state;
+        if (currentState.phase !== "thinking") {
+          return false;
+        }
+        this.editor.setEditable(true);
+        this.storage.state = {
+          phase: "asking",
+          customPrompt: currentState.prompt === currentState.customPrompt ? currentState.customPrompt : "",
+          error
+        };
+        return true;
+      },
+      _updateAiToolbarCustomPrompt: (customPrompt) => () => {
+        const currentState = this.storage.state;
+        if (typeof currentState.customPrompt !== "string") {
+          return false;
+        }
+        this.storage.state.customPrompt = typeof customPrompt === "function" ? customPrompt(currentState.customPrompt) : customPrompt;
+        return true;
+      },
+      _renderAiToolbarDiffInEditor: (previous) => () => {
+        if (!this.options.doc) {
+          return false;
+        }
+        const previousSnapshot = previous ?? emptySnapshot;
+        const currentSnapshot = snapshot(this.options.doc);
+        if (equalSnapshots(previousSnapshot, currentSnapshot)) {
+          return true;
+        }
+        const binding = getYjsBinding(this.editor);
+        if (binding) {
+          binding.renderSnapshot(currentSnapshot, previousSnapshot);
+          return true;
+        }
+        return false;
+      }
+    };
+  },
+  addProseMirrorPlugins() {
+    return [
+      new Plugin({
+        key: AI_TOOLBAR_SELECTION_PLUGIN,
+        props: {
+          decorations: ({ doc, selection }) => {
+            if (this.storage.state.phase === "closed" || this.storage.state.phase === "reviewing" && this.storage.state.output.type !== "other") {
+              return DecorationSet.create(doc, []);
+            }
+            const { from, to } = this.storage.state.selection ?? selection;
+            const decorations = [
+              Decoration.inline(from, to, {
+                class: "lb-root lb-selection lb-tiptap-active-selection"
+              })
+            ];
+            return DecorationSet.create(doc, decorations);
+          }
+        }
+      })
+    ];
+  }
+});
+
+export { AiExtension, DEFAULT_STATE };
+//# sourceMappingURL=AiExtension.mjs.map