@liveblocks/react-tiptap 2.16.1-ai → 2.16.1-ai2

package/dist/LiveblocksExtension.js

@@ -96,7 +96,7 @@ const YChangeMark = core.Mark.create({
           }
           return {
             "data-ychange-type": attributes.type,
-            class: `lb-changed-${attributes.type}`
+            class: `lb-root lb-tiptap-change-${attributes.type}`
           };
         }
       },
@@ -122,10 +122,6 @@ const useLiveblocksExtension = (opts) => {
   };
   const [editor, setEditor] = react.useState(null);
   const room = react$1.useRoom();
-  react$1.useCommentsErrorListener((error) => {
-    if (error instanceof _private.CreateThreadError) {
-    }
-  });
   const isEditorReady = useIsEditorReady();
   const client = react$1.useClient();
   const store = _private.getUmbrellaStoreForClient(client);
@@ -293,8 +289,7 @@ const useLiveblocksExtension = (opts) => {
           prompt,
           selectionText,
           context,
-          signal,
-          retryCount = 0
+          signal
         }) => {
           const result = await room[core$1.kInternal].executeContextualPrompt({
             prompt,
@@ -302,28 +297,12 @@ const useLiveblocksExtension = (opts) => {
             context,
             signal
           });
-          if (retryCount > 3) {
-            throw new Error("Failed to resolve AI prompt");
-          }
-          const retry = () => {
-            return resolveAiPrompt({
-              prompt,
-              selectionText,
-              context,
-              signal,
-              retryCount: retryCount + 1
-            });
-          };
-          try {
-            const parsedResponse = JSON.parse(result);
-            const type = typeof parsedResponse.type === "string" ? parsedResponse.type : "";
-            if (typeof parsedResponse.content === "string" && ["insert", "modification", "other"].includes(type)) {
-              return parsedResponse;
-            }
-            return retry();
-          } catch (e) {
-            return retry();
+          const parsedResponse = JSON.parse(result);
+          const type = typeof parsedResponse.type === "string" ? parsedResponse.type : "";
+          if (typeof parsedResponse.content === "string" && ["insert", "modification", "other"].includes(type)) {
+            return parsedResponse;
           }
+          throw new Error("Failed to resolve AI prompt");
         };
         extensions.push(
           AiExtension.AiExtension.configure({

package/dist/LiveblocksExtension.js.map

@@ -1 +1 @@
-{"version":3,"file":"LiveblocksExtension.js","sources":["../src/LiveblocksExtension.ts"],"sourcesContent":["import type {\n  BaseUserMeta,\n  IUserInfo,\n  JsonObject,\n  User,\n} from \"@liveblocks/core\";\nimport { kInternal, TextEditorType } from \"@liveblocks/core\";\nimport {\n  useClient,\n  useCommentsErrorListener,\n  useRoom,\n} from \"@liveblocks/react\";\nimport {\n  CreateThreadError,\n  getUmbrellaStoreForClient,\n  useCreateTextMention,\n  useDeleteTextMention,\n  useReportTextEditor,\n  useYjsProvider,\n} from \"@liveblocks/react/_private\";\nimport { LiveblocksYjsProvider } from \"@liveblocks/yjs\";\nimport type { AnyExtension, Editor } from \"@tiptap/core\";\nimport { Extension, getMarkType, Mark } from \"@tiptap/core\";\nimport Collaboration from \"@tiptap/extension-collaboration\";\nimport CollaborationCursor from \"@tiptap/extension-collaboration-cursor\";\nimport type { Mark as PMMark } from \"@tiptap/pm/model\";\nimport { useCallback, useEffect, useState, useSyncExternalStore } from \"react\";\nimport { Doc, PermanentUserData } from \"yjs\";\n\nimport { AiExtension } from \"./ai/AiExtension\";\nimport { CommentsExtension } from \"./comments/CommentsExtension\";\nimport { MentionExtension } from \"./mentions/MentionExtension\";\nimport type {\n  AiResponse,\n  LiveblocksExtensionOptions,\n  LiveblocksExtensionStorage,\n  ResolveAiPromptArgs,\n} from \"./types\";\nimport { LIVEBLOCKS_COMMENT_MARK_TYPE } from \"./types\";\n\nconst providersMap = new Map<\n  string,\n  LiveblocksYjsProvider<any, any, any, any, any>\n>();\n\nconst docMap = new Map<string, Doc>();\n\ntype WithRequired<T, K extends keyof T> = T & { [P in K]-?: T[P] };\n\nconst pudMap = new Map<string, PermanentUserData>();\n\nconst DEFAULT_OPTIONS: WithRequired<LiveblocksExtensionOptions, \"field\"> = {\n  field: \"default\",\n  comments: true,\n  mentions: true,\n  offlineSupport_experimental: false,\n};\n\nconst LiveblocksCollab = Collaboration.extend({\n  // Override the onCreate method to warn users about potential misconfigurations\n  onCreate() {\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"doc\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap document extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      !this.editor.extensionManager.extensions.find(\n        (e) => e.name === \"paragraph\"\n      )\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap paragraph extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"text\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap text extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      this.editor.extensionManager.extensions.find((e) => e.name === \"history\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The history extension is enabled, Liveblocks extension provides its own. Please remove or disable the History plugin to prevent unwanted conflicts.\"\n      );\n    }\n  },\n});\n\n/**\n * Returns whether the editor has loaded the initial text contents from the\n * server and is ready to be used.\n *\n */\nexport function useIsEditorReady(): boolean {\n  const yjsProvider = useYjsProvider();\n\n  const getSnapshot = useCallback(() => {\n    const status = yjsProvider?.getStatus();\n    return status === \"synchronizing\" || status === \"synchronized\";\n  }, [yjsProvider]);\n\n  const subscribe = useCallback(\n    (callback: () => void) => {\n      if (yjsProvider === undefined) return () => {};\n      yjsProvider.on(\"status\", callback);\n      return () => {\n        yjsProvider.off(\"status\", callback);\n      };\n    },\n    [yjsProvider]\n  );\n\n  return useSyncExternalStore(subscribe, getSnapshot, getSnapshot);\n}\n\nconst YChangeMark = Mark.create({\n  name: \"ychange\",\n  inclusive: false,\n  parseHTML() {\n    return [{ tag: \"ychange\" }];\n  },\n  addAttributes() {\n    return {\n      user: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_user\") ?? null,\n        renderHTML: (attributes: { user: string | null }) => {\n          if (!attributes.user) {\n            return {};\n          }\n          return { \"data-ychange-user\": attributes.user };\n        },\n      },\n      type: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_type\") ?? null,\n        renderHTML: (attributes: { type: string | null }) => {\n          if (!attributes.type) {\n            return {};\n          }\n          return {\n            \"data-ychange-type\": attributes.type,\n            class: `lb-changed-${attributes.type}`,\n          };\n        },\n      },\n      color: {\n        default: null,\n        parseHTML: (element) => {\n          return element.getAttribute(\"ychange_color\") ?? null;\n        },\n        renderHTML: () => {\n          // attributes: { color: { light: string; dark: string } | null }\n          return {}; // we don't need this color attribute for now\n        },\n      },\n    };\n  },\n  renderHTML({ HTMLAttributes }) {\n    return [\"ychange\", HTMLAttributes, 0];\n  },\n});\n\nexport const useLiveblocksExtension = (\n  opts?: LiveblocksExtensionOptions\n): Extension => {\n  const options = {\n    ...DEFAULT_OPTIONS,\n    ...opts,\n  };\n  const [editor, setEditor] = useState<Editor | null>(null);\n  const room = useRoom();\n\n  // TODO: we don't need these things if comments isn't turned on...\n  // TODO: we don't have a reference to the editor here, need to figure this out\n  useCommentsErrorListener((error) => {\n    // If thread creation fails, we remove the thread id from the associated nodes and unwrap the nodes if they are no longer associated with any threads\n    if (error instanceof CreateThreadError) {\n      // handleThreadDelete(error.context.threadId);\n    }\n  });\n  const isEditorReady = useIsEditorReady();\n  const client = useClient();\n  const store = getUmbrellaStoreForClient(client);\n  const roomId = room.id;\n  const yjsProvider = useYjsProvider();\n\n  // If the user provided initialContent, wait for ready and then set it\n  useEffect(() => {\n    if (!isEditorReady || !yjsProvider || !options.initialContent || !editor)\n      return;\n\n    // As noted in the tiptap documentation, you may not set initial content with collaboration.\n    // The docs provide the following workaround:\n    const ydoc = (yjsProvider as LiveblocksYjsProvider).getYDoc();\n    const hasContentSet = ydoc.getMap(\"liveblocks_config\").get(\"hasContentSet\");\n    if (!hasContentSet) {\n      ydoc.getMap(\"liveblocks_config\").set(\"hasContentSet\", true);\n      editor.commands.setContent(options.initialContent);\n    }\n  }, [isEditorReady, yjsProvider, options.initialContent, editor]);\n\n  useReportTextEditor(\n    TextEditorType.TipTap,\n    options.field ?? DEFAULT_OPTIONS.field\n  );\n\n  const createTextMention = useCreateTextMention();\n  const deleteTextMention = useDeleteTextMention();\n\n  return Extension.create<never, LiveblocksExtensionStorage>({\n    name: \"liveblocksExtension\",\n\n    onCreate() {\n      setEditor(this.editor);\n      if (this.editor.options.content) {\n        console.warn(\n          \"[Liveblocks] Initial content must be set in the useLiveblocksExtension hook option. Remove content from your editor options.\"\n        );\n      }\n      if (\n        options.mentions &&\n        this.editor.extensionManager.extensions.find(\n          (e) => e.name.toLowerCase() === \"mention\"\n        )\n      ) {\n        console.warn(\n          \"[Liveblocks] Liveblocks own mention plugin is enabled, using another mention plugin may cause a conflict.\"\n        );\n      }\n      const self = room.getSelf();\n      const updateUser = ({\n        info,\n        id: userId,\n      }: User<JsonObject, BaseUserMeta>) => {\n        if (!info) {\n          return;\n        }\n        const { user: storedUser } =\n          this.storage.provider.awareness.getLocalState() as {\n            user: IUserInfo;\n          };\n        this.storage.permanentUserData?.setUserMapping(\n          this.storage.doc,\n          this.storage.doc.clientID,\n          userId ?? \"Unknown\" // TODO: change this to the user's ID so we can map it to the user's name\n        );\n        if (\n          info.name !== storedUser?.name ||\n          info.color !== storedUser?.color\n        ) {\n          this.editor.commands.updateUser({\n            name: info.name,\n            color: info.color,\n          });\n        }\n      };\n      // if we already have user info, we update the user\n      if (self?.info) {\n        updateUser(self);\n      }\n      // we also listen in case the user info changes\n      this.storage.unsubs.push(room.events.self.subscribe(updateUser));\n      if (options.comments) {\n        const commentMarkType = getMarkType(\n          LIVEBLOCKS_COMMENT_MARK_TYPE,\n          this.editor.schema\n        );\n        this.storage.unsubs.push(\n          // Subscribe to threads so we can update comment marks if they become resolved/deleted\n          store.outputs.threads.subscribe(() => {\n            const threadMap = new Map(\n              store.outputs.threads\n                .get()\n                .findMany(roomId, { resolved: false }, \"asc\")\n                .map((thread) => [thread.id, true])\n            );\n            function isComment(mark: PMMark): mark is PMMark & {\n              attrs: { orphan: boolean; threadId: string };\n            } {\n              return mark.type.name === LIVEBLOCKS_COMMENT_MARK_TYPE;\n            }\n            // when threads change, find marks and update them if needed\n            this.editor.state.doc.descendants((node, pos) => {\n              node.marks.forEach((mark) => {\n                if (isComment(mark)) {\n                  const markThreadId = mark.attrs.threadId;\n                  const isOrphan = !threadMap.has(markThreadId);\n                  if (isOrphan !== mark.attrs.orphan) {\n                    const { tr } = this.editor.state;\n                    const trimmedFrom = Math.max(pos, 0);\n                    const trimmedTo = Math.min(\n                      pos + node.nodeSize,\n                      this.editor.state.doc.content.size - 1\n                    );\n                    tr.removeMark(trimmedFrom, trimmedTo, commentMarkType);\n                    tr.addMark(\n                      trimmedFrom,\n                      trimmedTo,\n                      commentMarkType.create({\n                        ...mark.attrs,\n                        orphan: isOrphan,\n                      })\n                    );\n                    this.editor.view.dispatch(tr);\n                  }\n                }\n              });\n            });\n          })\n        );\n      }\n    },\n    onDestroy() {\n      this.storage.unsubs.forEach((unsub) => unsub());\n    },\n    addGlobalAttributes() {\n      return [\n        {\n          types: [\"paragraph\", \"heading\"],\n          attributes: {\n            ychange: { default: null },\n          },\n        },\n      ];\n    },\n    addStorage() {\n      if (!providersMap.has(room.id)) {\n        const doc = new Doc();\n        docMap.set(room.id, doc);\n        pudMap.set(room.id, new PermanentUserData(doc));\n        providersMap.set(\n          room.id,\n          new LiveblocksYjsProvider(room, doc, {\n            offlineSupport_experimental: options.offlineSupport_experimental,\n          })\n        );\n      }\n      return {\n        doc: docMap.get(room.id)!,\n        provider: providersMap.get(room.id)!,\n        permanentUserData: pudMap.get(room.id)!,\n        unsubs: [],\n      };\n    },\n    addExtensions() {\n      const extensions: AnyExtension[] = [\n        YChangeMark,\n        LiveblocksCollab.configure({\n          ySyncOptions: {\n            permanentUserData: this.storage.permanentUserData,\n          },\n          document: this.storage.doc,\n          field: options.field,\n        }),\n        CollaborationCursor.configure({\n          provider: this.storage.provider,\n        }),\n      ];\n\n      if (options.comments) {\n        extensions.push(CommentsExtension);\n      }\n      if (options.mentions) {\n        extensions.push(\n          MentionExtension.configure({\n            onCreateMention: createTextMention,\n            onDeleteMention: deleteTextMention,\n          })\n        );\n      }\n      if (options.ai) {\n        const resolveAiPrompt = async ({\n          prompt,\n          selectionText,\n          context,\n          signal,\n          retryCount = 0,\n        }: ResolveAiPromptArgs): Promise<AiResponse> => {\n          const result = await room[kInternal].executeContextualPrompt({\n            prompt,\n            selectionText,\n            context,\n            signal,\n          });\n\n          if (retryCount > 3) {\n            throw new Error(\"Failed to resolve AI prompt\");\n          }\n\n          const retry = () => {\n            return resolveAiPrompt({\n              prompt,\n              selectionText,\n              context,\n              signal,\n              retryCount: retryCount + 1,\n            });\n          };\n          // TODO: proper backoff\n          try {\n            const parsedResponse = JSON.parse(result) as JsonObject;\n            const type =\n              typeof parsedResponse.type === \"string\"\n                ? parsedResponse.type\n                : \"\";\n            if (\n              typeof parsedResponse.content === \"string\" &&\n              [\"insert\", \"modification\", \"other\"].includes(type)\n            ) {\n              return parsedResponse as AiResponse;\n            }\n            return retry();\n          } catch (e) {\n            return retry();\n          }\n        };\n        extensions.push(\n          AiExtension.configure({\n            resolveAiPrompt,\n            ...(typeof options.ai === \"boolean\" ? {} : options.ai),\n            doc: this.storage.doc,\n            pud: this.storage.permanentUserData,\n          })\n        );\n      }\n\n      return extensions;\n    },\n  });\n};\n"],"names":["useYjsProvider","useCallback","useSyncExternalStore","Mark","useState","useRoom","useCommentsErrorListener","CreateThreadError","useClient","getUmbrellaStoreForClient","useEffect","useReportTextEditor","TextEditorType","useCreateTextMention","useDeleteTextMention","Extension","getMarkType","LIVEBLOCKS_COMMENT_MARK_TYPE","Doc","PermanentUserData","LiveblocksYjsProvider","CommentsExtension","MentionExtension","kInternal","AiExtension"],"mappings":";;;;;;;;;;;;;;;;AAwCA,MAAM,YAAA,uBAAmB,GAGvB,EAAA,CAAA;AAEF,MAAM,MAAA,uBAAa,GAAiB,EAAA,CAAA;AAIpC,MAAM,MAAA,uBAAa,GAA+B,EAAA,CAAA;AAElD,MAAM,eAAqE,GAAA;AAAA,EACzE,KAAO,EAAA,SAAA;AAAA,EACP,QAAU,EAAA,IAAA;AAAA,EACV,QAAU,EAAA,IAAA;AAAA,EACV,2BAA6B,EAAA,KAAA;AAC/B,CAAA,CAAA;AAEA,MAAM,gBAAA,GAAmB,cAAc,MAAO,CAAA;AAAA,EAE5C,QAAW,GAAA;AACT,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,KAAK,CACrE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,wIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IAAA,IACE,CAAC,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA;AAAA,MACvC,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,WAAA;AAAA,KAEpB,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,yIAAA;AAAA,OACF,CAAA;AAAA,KACF;AAEA,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,MAAM,CACtE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,oIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IACE,IAAA,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA,CAAK,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,SAAS,CACxE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,kKAAA;AAAA,OACF,CAAA;AAAA,KACF;AAAA,GACF;AACF,CAAC,CAAA,CAAA;AAOM,SAAS,gBAA4B,GAAA;AAC1C,EAAA,MAAM,cAAcA,uBAAe,EAAA,CAAA;AAEnC,EAAM,MAAA,WAAA,GAAcC,kBAAY,MAAM;AACpC,IAAM,MAAA,MAAA,GAAS,aAAa,SAAU,EAAA,CAAA;AACtC,IAAO,OAAA,MAAA,KAAW,mBAAmB,MAAW,KAAA,cAAA,CAAA;AAAA,GAClD,EAAG,CAAC,WAAW,CAAC,CAAA,CAAA;AAEhB,EAAA,MAAM,SAAY,GAAAA,iBAAA;AAAA,IAChB,CAAC,QAAyB,KAAA;AACxB,MAAA,IAAI,WAAgB,KAAA,KAAA,CAAA;AAAW,QAAA,OAAO,MAAM;AAAA,SAAC,CAAA;AAC7C,MAAY,WAAA,CAAA,EAAA,CAAG,UAAU,QAAQ,CAAA,CAAA;AACjC,MAAA,OAAO,MAAM;AACX,QAAY,WAAA,CAAA,GAAA,CAAI,UAAU,QAAQ,CAAA,CAAA;AAAA,OACpC,CAAA;AAAA,KACF;AAAA,IACA,CAAC,WAAW,CAAA;AAAA,GACd,CAAA;AAEA,EAAO,OAAAC,0BAAA,CAAqB,SAAW,EAAA,WAAA,EAAa,WAAW,CAAA,CAAA;AACjE,CAAA;AAEA,MAAM,WAAA,GAAcC,UAAK,MAAO,CAAA;AAAA,EAC9B,IAAM,EAAA,SAAA;AAAA,EACN,SAAW,EAAA,KAAA;AAAA,EACX,SAAY,GAAA;AACV,IAAA,OAAO,CAAC,EAAE,GAAK,EAAA,SAAA,EAAW,CAAA,CAAA;AAAA,GAC5B;AAAA,EACA,aAAgB,GAAA;AACd,IAAO,OAAA;AAAA,MACL,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA,EAAE,mBAAqB,EAAA,UAAA,CAAW,IAAK,EAAA,CAAA;AAAA,SAChD;AAAA,OACF;AAAA,MACA,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA;AAAA,YACL,qBAAqB,UAAW,CAAA,IAAA;AAAA,YAChC,KAAA,EAAO,cAAc,UAAW,CAAA,IAAA,CAAA,CAAA;AAAA,WAClC,CAAA;AAAA,SACF;AAAA,OACF;AAAA,MACA,KAAO,EAAA;AAAA,QACL,OAAS,EAAA,IAAA;AAAA,QACT,SAAA,EAAW,CAAC,OAAY,KAAA;AACtB,UAAO,OAAA,OAAA,CAAQ,YAAa,CAAA,eAAe,CAAK,IAAA,IAAA,CAAA;AAAA,SAClD;AAAA,QACA,YAAY,MAAM;AAEhB,UAAA,OAAO,EAAC,CAAA;AAAA,SACV;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AAAA,EACA,UAAA,CAAW,EAAE,cAAA,EAAkB,EAAA;AAC7B,IAAO,OAAA,CAAC,SAAW,EAAA,cAAA,EAAgB,CAAC,CAAA,CAAA;AAAA,GACtC;AACF,CAAC,CAAA,CAAA;AAEY,MAAA,sBAAA,GAAyB,CACpC,IACc,KAAA;AACd,EAAA,MAAM,OAAU,GAAA;AAAA,IACd,GAAG,eAAA;AAAA,IACH,GAAG,IAAA;AAAA,GACL,CAAA;AACA,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAIC,eAAwB,IAAI,CAAA,CAAA;AACxD,EAAA,MAAM,OAAOC,eAAQ,EAAA,CAAA;AAIrB,EAAAC,gCAAA,CAAyB,CAAC,KAAU,KAAA;AAElC,IAAA,IAAI,iBAAiBC,0BAAmB,EAAA;AAAA,KAExC;AAAA,GACD,CAAA,CAAA;AACD,EAAA,MAAM,gBAAgB,gBAAiB,EAAA,CAAA;AACvC,EAAA,MAAM,SAASC,iBAAU,EAAA,CAAA;AACzB,EAAM,MAAA,KAAA,GAAQC,mCAA0B,MAAM,CAAA,CAAA;AAC9C,EAAA,MAAM,SAAS,IAAK,CAAA,EAAA,CAAA;AACpB,EAAA,MAAM,cAAcT,uBAAe,EAAA,CAAA;AAGnC,EAAAU,eAAA,CAAU,MAAM;AACd,IAAA,IAAI,CAAC,aAAiB,IAAA,CAAC,eAAe,CAAC,OAAA,CAAQ,kBAAkB,CAAC,MAAA;AAChE,MAAA,OAAA;AAIF,IAAM,MAAA,IAAA,GAAQ,YAAsC,OAAQ,EAAA,CAAA;AAC5D,IAAA,MAAM,gBAAgB,IAAK,CAAA,MAAA,CAAO,mBAAmB,CAAA,CAAE,IAAI,eAAe,CAAA,CAAA;AAC1E,IAAA,IAAI,CAAC,aAAe,EAAA;AAClB,MAAA,IAAA,CAAK,MAAO,CAAA,mBAAmB,CAAE,CAAA,GAAA,CAAI,iBAAiB,IAAI,CAAA,CAAA;AAC1D,MAAO,MAAA,CAAA,QAAA,CAAS,UAAW,CAAA,OAAA,CAAQ,cAAc,CAAA,CAAA;AAAA,KACnD;AAAA,KACC,CAAC,aAAA,EAAe,aAAa,OAAQ,CAAA,cAAA,EAAgB,MAAM,CAAC,CAAA,CAAA;AAE/D,EAAAC,4BAAA;AAAA,IACEC,qBAAe,CAAA,MAAA;AAAA,IACf,OAAA,CAAQ,SAAS,eAAgB,CAAA,KAAA;AAAA,GACnC,CAAA;AAEA,EAAA,MAAM,oBAAoBC,6BAAqB,EAAA,CAAA;AAC/C,EAAA,MAAM,oBAAoBC,6BAAqB,EAAA,CAAA;AAE/C,EAAA,OAAOC,eAAU,MAA0C,CAAA;AAAA,IACzD,IAAM,EAAA,qBAAA;AAAA,IAEN,QAAW,GAAA;AACT,MAAA,SAAA,CAAU,KAAK,MAAM,CAAA,CAAA;AACrB,MAAI,IAAA,IAAA,CAAK,MAAO,CAAA,OAAA,CAAQ,OAAS,EAAA;AAC/B,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,8HAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAA,IACE,OAAQ,CAAA,QAAA,IACR,IAAK,CAAA,MAAA,CAAO,iBAAiB,UAAW,CAAA,IAAA;AAAA,QACtC,CAAC,CAAA,KAAM,CAAE,CAAA,IAAA,CAAK,aAAkB,KAAA,SAAA;AAAA,OAElC,EAAA;AACA,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,2GAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAM,MAAA,IAAA,GAAO,KAAK,OAAQ,EAAA,CAAA;AAC1B,MAAA,MAAM,aAAa,CAAC;AAAA,QAClB,IAAA;AAAA,QACA,EAAI,EAAA,MAAA;AAAA,OACgC,KAAA;AACpC,QAAA,IAAI,CAAC,IAAM,EAAA;AACT,UAAA,OAAA;AAAA,SACF;AACA,QAAM,MAAA,EAAE,MAAM,UAAW,EAAA,GACvB,KAAK,OAAQ,CAAA,QAAA,CAAS,UAAU,aAAc,EAAA,CAAA;AAGhD,QAAA,IAAA,CAAK,QAAQ,iBAAmB,EAAA,cAAA;AAAA,UAC9B,KAAK,OAAQ,CAAA,GAAA;AAAA,UACb,IAAA,CAAK,QAAQ,GAAI,CAAA,QAAA;AAAA,UACjB,MAAU,IAAA,SAAA;AAAA,SACZ,CAAA;AACA,QAAA,IACE,KAAK,IAAS,KAAA,UAAA,EAAY,QAC1B,IAAK,CAAA,KAAA,KAAU,YAAY,KAC3B,EAAA;AACA,UAAK,IAAA,CAAA,MAAA,CAAO,SAAS,UAAW,CAAA;AAAA,YAC9B,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,OAAO,IAAK,CAAA,KAAA;AAAA,WACb,CAAA,CAAA;AAAA,SACH;AAAA,OACF,CAAA;AAEA,MAAA,IAAI,MAAM,IAAM,EAAA;AACd,QAAA,UAAA,CAAW,IAAI,CAAA,CAAA;AAAA,OACjB;AAEA,MAAK,IAAA,CAAA,OAAA,CAAQ,OAAO,IAAK,CAAA,IAAA,CAAK,OAAO,IAAK,CAAA,SAAA,CAAU,UAAU,CAAC,CAAA,CAAA;AAC/D,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,MAAM,eAAkB,GAAAC,gBAAA;AAAA,UACtBC,kCAAA;AAAA,UACA,KAAK,MAAO,CAAA,MAAA;AAAA,SACd,CAAA;AACA,QAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,IAAA;AAAA,UAElB,KAAM,CAAA,OAAA,CAAQ,OAAQ,CAAA,SAAA,CAAU,MAAM;AACpC,YAAA,MAAM,YAAY,IAAI,GAAA;AAAA,cACpB,KAAA,CAAM,QAAQ,OACX,CAAA,GAAA,GACA,QAAS,CAAA,MAAA,EAAQ,EAAE,QAAU,EAAA,KAAA,IAAS,KAAK,CAAA,CAC3C,IAAI,CAAC,MAAA,KAAW,CAAC,MAAO,CAAA,EAAA,EAAI,IAAI,CAAC,CAAA;AAAA,aACtC,CAAA;AACA,YAAA,SAAS,UAAU,IAEjB,EAAA;AACA,cAAO,OAAA,IAAA,CAAK,KAAK,IAAS,KAAAA,kCAAA,CAAA;AAAA,aAC5B;AAEA,YAAA,IAAA,CAAK,OAAO,KAAM,CAAA,GAAA,CAAI,WAAY,CAAA,CAAC,MAAM,GAAQ,KAAA;AAC/C,cAAK,IAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,CAAC,IAAS,KAAA;AAC3B,gBAAI,IAAA,SAAA,CAAU,IAAI,CAAG,EAAA;AACnB,kBAAM,MAAA,YAAA,GAAe,KAAK,KAAM,CAAA,QAAA,CAAA;AAChC,kBAAA,MAAM,QAAW,GAAA,CAAC,SAAU,CAAA,GAAA,CAAI,YAAY,CAAA,CAAA;AAC5C,kBAAI,IAAA,QAAA,KAAa,IAAK,CAAA,KAAA,CAAM,MAAQ,EAAA;AAClC,oBAAA,MAAM,EAAE,EAAA,EAAO,GAAA,IAAA,CAAK,MAAO,CAAA,KAAA,CAAA;AAC3B,oBAAA,MAAM,WAAc,GAAA,IAAA,CAAK,GAAI,CAAA,GAAA,EAAK,CAAC,CAAA,CAAA;AACnC,oBAAA,MAAM,YAAY,IAAK,CAAA,GAAA;AAAA,sBACrB,MAAM,IAAK,CAAA,QAAA;AAAA,sBACX,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,GAAA,CAAI,QAAQ,IAAO,GAAA,CAAA;AAAA,qBACvC,CAAA;AACA,oBAAG,EAAA,CAAA,UAAA,CAAW,WAAa,EAAA,SAAA,EAAW,eAAe,CAAA,CAAA;AACrD,oBAAG,EAAA,CAAA,OAAA;AAAA,sBACD,WAAA;AAAA,sBACA,SAAA;AAAA,sBACA,gBAAgB,MAAO,CAAA;AAAA,wBACrB,GAAG,IAAK,CAAA,KAAA;AAAA,wBACR,MAAQ,EAAA,QAAA;AAAA,uBACT,CAAA;AAAA,qBACH,CAAA;AACA,oBAAK,IAAA,CAAA,MAAA,CAAO,IAAK,CAAA,QAAA,CAAS,EAAE,CAAA,CAAA;AAAA,mBAC9B;AAAA,iBACF;AAAA,eACD,CAAA,CAAA;AAAA,aACF,CAAA,CAAA;AAAA,WACF,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA,SAAY,GAAA;AACV,MAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,OAAA,CAAQ,CAAC,KAAA,KAAU,OAAO,CAAA,CAAA;AAAA,KAChD;AAAA,IACA,mBAAsB,GAAA;AACpB,MAAO,OAAA;AAAA,QACL;AAAA,UACE,KAAA,EAAO,CAAC,WAAA,EAAa,SAAS,CAAA;AAAA,UAC9B,UAAY,EAAA;AAAA,YACV,OAAA,EAAS,EAAE,OAAA,EAAS,IAAK,EAAA;AAAA,WAC3B;AAAA,SACF;AAAA,OACF,CAAA;AAAA,KACF;AAAA,IACA,UAAa,GAAA;AACX,MAAA,IAAI,CAAC,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAG,EAAA;AAC9B,QAAM,MAAA,GAAA,GAAM,IAAIC,OAAI,EAAA,CAAA;AACpB,QAAO,MAAA,CAAA,GAAA,CAAI,IAAK,CAAA,EAAA,EAAI,GAAG,CAAA,CAAA;AACvB,QAAA,MAAA,CAAO,IAAI,IAAK,CAAA,EAAA,EAAI,IAAIC,qBAAA,CAAkB,GAAG,CAAC,CAAA,CAAA;AAC9C,QAAa,YAAA,CAAA,GAAA;AAAA,UACX,IAAK,CAAA,EAAA;AAAA,UACL,IAAIC,2BAAsB,CAAA,IAAA,EAAM,GAAK,EAAA;AAAA,YACnC,6BAA6B,OAAQ,CAAA,2BAAA;AAAA,WACtC,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAO,OAAA;AAAA,QACL,GAAK,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACvB,QAAU,EAAA,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QAClC,iBAAmB,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACrC,QAAQ,EAAC;AAAA,OACX,CAAA;AAAA,KACF;AAAA,IACA,aAAgB,GAAA;AACd,MAAA,MAAM,UAA6B,GAAA;AAAA,QACjC,WAAA;AAAA,QACA,iBAAiB,SAAU,CAAA;AAAA,UACzB,YAAc,EAAA;AAAA,YACZ,iBAAA,EAAmB,KAAK,OAAQ,CAAA,iBAAA;AAAA,WAClC;AAAA,UACA,QAAA,EAAU,KAAK,OAAQ,CAAA,GAAA;AAAA,UACvB,OAAO,OAAQ,CAAA,KAAA;AAAA,SAChB,CAAA;AAAA,QACD,oBAAoB,SAAU,CAAA;AAAA,UAC5B,QAAA,EAAU,KAAK,OAAQ,CAAA,QAAA;AAAA,SACxB,CAAA;AAAA,OACH,CAAA;AAEA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,UAAA,CAAW,KAAKC,mCAAiB,CAAA,CAAA;AAAA,OACnC;AACA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAW,UAAA,CAAA,IAAA;AAAA,UACTC,kCAAiB,SAAU,CAAA;AAAA,YACzB,eAAiB,EAAA,iBAAA;AAAA,YACjB,eAAiB,EAAA,iBAAA;AAAA,WAClB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAA,IAAI,QAAQ,EAAI,EAAA;AACd,QAAA,MAAM,kBAAkB,OAAO;AAAA,UAC7B,MAAA;AAAA,UACA,aAAA;AAAA,UACA,OAAA;AAAA,UACA,MAAA;AAAA,UACA,UAAa,GAAA,CAAA;AAAA,SACiC,KAAA;AAC9C,UAAA,MAAM,MAAS,GAAA,MAAM,IAAK,CAAAC,gBAAA,CAAA,CAAW,uBAAwB,CAAA;AAAA,YAC3D,MAAA;AAAA,YACA,aAAA;AAAA,YACA,OAAA;AAAA,YACA,MAAA;AAAA,WACD,CAAA,CAAA;AAED,UAAA,IAAI,aAAa,CAAG,EAAA;AAClB,YAAM,MAAA,IAAI,MAAM,6BAA6B,CAAA,CAAA;AAAA,WAC/C;AAEA,UAAA,MAAM,QAAQ,MAAM;AAClB,YAAA,OAAO,eAAgB,CAAA;AAAA,cACrB,MAAA;AAAA,cACA,aAAA;AAAA,cACA,OAAA;AAAA,cACA,MAAA;AAAA,cACA,YAAY,UAAa,GAAA,CAAA;AAAA,aAC1B,CAAA,CAAA;AAAA,WACH,CAAA;AAEA,UAAI,IAAA;AACF,YAAM,MAAA,cAAA,GAAiB,IAAK,CAAA,KAAA,CAAM,MAAM,CAAA,CAAA;AACxC,YAAA,MAAM,OACJ,OAAO,cAAA,CAAe,IAAS,KAAA,QAAA,GAC3B,eAAe,IACf,GAAA,EAAA,CAAA;AACN,YACE,IAAA,OAAO,cAAe,CAAA,OAAA,KAAY,QAClC,IAAA,CAAC,QAAU,EAAA,cAAA,EAAgB,OAAO,CAAA,CAAE,QAAS,CAAA,IAAI,CACjD,EAAA;AACA,cAAO,OAAA,cAAA,CAAA;AAAA,aACT;AACA,YAAA,OAAO,KAAM,EAAA,CAAA;AAAA,mBACN,CAAP,EAAA;AACA,YAAA,OAAO,KAAM,EAAA,CAAA;AAAA,WACf;AAAA,SACF,CAAA;AACA,QAAW,UAAA,CAAA,IAAA;AAAA,UACTC,wBAAY,SAAU,CAAA;AAAA,YACpB,eAAA;AAAA,YACA,GAAI,OAAO,OAAA,CAAQ,OAAO,SAAY,GAAA,KAAK,OAAQ,CAAA,EAAA;AAAA,YACnD,GAAA,EAAK,KAAK,OAAQ,CAAA,GAAA;AAAA,YAClB,GAAA,EAAK,KAAK,OAAQ,CAAA,iBAAA;AAAA,WACnB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAEA,MAAO,OAAA,UAAA,CAAA;AAAA,KACT;AAAA,GACD,CAAA,CAAA;AACH;;;;;"}
+{"version":3,"file":"LiveblocksExtension.js","sources":["../src/LiveblocksExtension.ts"],"sourcesContent":["import type {\n  BaseUserMeta,\n  IUserInfo,\n  JsonObject,\n  User,\n} from \"@liveblocks/core\";\nimport { kInternal, TextEditorType } from \"@liveblocks/core\";\nimport { useClient, useRoom } from \"@liveblocks/react\";\nimport {\n  getUmbrellaStoreForClient,\n  useCreateTextMention,\n  useDeleteTextMention,\n  useReportTextEditor,\n  useYjsProvider,\n} from \"@liveblocks/react/_private\";\nimport { LiveblocksYjsProvider } from \"@liveblocks/yjs\";\nimport type { AnyExtension, Editor } from \"@tiptap/core\";\nimport { Extension, getMarkType, Mark } from \"@tiptap/core\";\nimport Collaboration from \"@tiptap/extension-collaboration\";\nimport CollaborationCursor from \"@tiptap/extension-collaboration-cursor\";\nimport type { Mark as PMMark } from \"@tiptap/pm/model\";\nimport { useCallback, useEffect, useState, useSyncExternalStore } from \"react\";\nimport { Doc, PermanentUserData } from \"yjs\";\n\nimport { AiExtension } from \"./ai/AiExtension\";\nimport { CommentsExtension } from \"./comments/CommentsExtension\";\nimport { MentionExtension } from \"./mentions/MentionExtension\";\nimport type {\n  AiResponse,\n  LiveblocksExtensionOptions,\n  LiveblocksExtensionStorage,\n  ResolveAiPromptArgs,\n} from \"./types\";\nimport { LIVEBLOCKS_COMMENT_MARK_TYPE } from \"./types\";\n\nconst providersMap = new Map<\n  string,\n  LiveblocksYjsProvider<any, any, any, any, any>\n>();\n\nconst docMap = new Map<string, Doc>();\n\ntype WithRequired<T, K extends keyof T> = T & { [P in K]-?: T[P] };\n\nconst pudMap = new Map<string, PermanentUserData>();\n\nconst DEFAULT_OPTIONS: WithRequired<LiveblocksExtensionOptions, \"field\"> = {\n  field: \"default\",\n  comments: true,\n  mentions: true,\n  offlineSupport_experimental: false,\n};\n\nconst LiveblocksCollab = Collaboration.extend({\n  // Override the onCreate method to warn users about potential misconfigurations\n  onCreate() {\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"doc\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap document extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      !this.editor.extensionManager.extensions.find(\n        (e) => e.name === \"paragraph\"\n      )\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap paragraph extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"text\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap text extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      this.editor.extensionManager.extensions.find((e) => e.name === \"history\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The history extension is enabled, Liveblocks extension provides its own. Please remove or disable the History plugin to prevent unwanted conflicts.\"\n      );\n    }\n  },\n});\n\n/**\n * Returns whether the editor has loaded the initial text contents from the\n * server and is ready to be used.\n *\n */\nexport function useIsEditorReady(): boolean {\n  const yjsProvider = useYjsProvider();\n\n  const getSnapshot = useCallback(() => {\n    const status = yjsProvider?.getStatus();\n    return status === \"synchronizing\" || status === \"synchronized\";\n  }, [yjsProvider]);\n\n  const subscribe = useCallback(\n    (callback: () => void) => {\n      if (yjsProvider === undefined) return () => {};\n      yjsProvider.on(\"status\", callback);\n      return () => {\n        yjsProvider.off(\"status\", callback);\n      };\n    },\n    [yjsProvider]\n  );\n\n  return useSyncExternalStore(subscribe, getSnapshot, getSnapshot);\n}\n\nconst YChangeMark = Mark.create({\n  name: \"ychange\",\n  inclusive: false,\n  parseHTML() {\n    return [{ tag: \"ychange\" }];\n  },\n  addAttributes() {\n    return {\n      user: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_user\") ?? null,\n        renderHTML: (attributes: { user: string | null }) => {\n          if (!attributes.user) {\n            return {};\n          }\n          return { \"data-ychange-user\": attributes.user };\n        },\n      },\n      type: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_type\") ?? null,\n        renderHTML: (attributes: { type: string | null }) => {\n          if (!attributes.type) {\n            return {};\n          }\n          return {\n            \"data-ychange-type\": attributes.type,\n            class: `lb-root lb-tiptap-change-${attributes.type}`,\n          };\n        },\n      },\n      color: {\n        default: null,\n        parseHTML: (element) => {\n          return element.getAttribute(\"ychange_color\") ?? null;\n        },\n        renderHTML: () => {\n          // attributes: { color: { light: string; dark: string } | null }\n          return {}; // we don't need this color attribute for now\n        },\n      },\n    };\n  },\n  renderHTML({ HTMLAttributes }) {\n    return [\"ychange\", HTMLAttributes, 0];\n  },\n});\n\nexport const useLiveblocksExtension = (\n  opts?: LiveblocksExtensionOptions\n): Extension => {\n  const options = {\n    ...DEFAULT_OPTIONS,\n    ...opts,\n  };\n  const [editor, setEditor] = useState<Editor | null>(null);\n  const room = useRoom();\n\n  // TODO: we don't need these things if comments isn't turned on...\n  // TODO: we don't have a reference to the editor here, need to figure this out\n  // useErrorListener((error) => {\n  //   // If thread creation fails, we remove the thread id from the associated nodes and unwrap the nodes if they are no longer associated with any threads\n  //   if (\n  //     error.context.type === \"CREATE_THREAD_ERROR\" &&\n  //     error.context.roomId === room.id\n  //   ) {\n  //     handleThreadDelete(error.context.threadId);\n  //   }\n  // });\n\n  const isEditorReady = useIsEditorReady();\n  const client = useClient();\n  const store = getUmbrellaStoreForClient(client);\n  const roomId = room.id;\n  const yjsProvider = useYjsProvider();\n\n  // If the user provided initialContent, wait for ready and then set it\n  useEffect(() => {\n    if (!isEditorReady || !yjsProvider || !options.initialContent || !editor)\n      return;\n\n    // As noted in the tiptap documentation, you may not set initial content with collaboration.\n    // The docs provide the following workaround:\n    const ydoc = (yjsProvider as LiveblocksYjsProvider).getYDoc();\n    const hasContentSet = ydoc.getMap(\"liveblocks_config\").get(\"hasContentSet\");\n    if (!hasContentSet) {\n      ydoc.getMap(\"liveblocks_config\").set(\"hasContentSet\", true);\n      editor.commands.setContent(options.initialContent);\n    }\n  }, [isEditorReady, yjsProvider, options.initialContent, editor]);\n\n  useReportTextEditor(\n    TextEditorType.TipTap,\n    options.field ?? DEFAULT_OPTIONS.field\n  );\n\n  const createTextMention = useCreateTextMention();\n  const deleteTextMention = useDeleteTextMention();\n\n  return Extension.create<never, LiveblocksExtensionStorage>({\n    name: \"liveblocksExtension\",\n\n    onCreate() {\n      setEditor(this.editor);\n      if (this.editor.options.content) {\n        console.warn(\n          \"[Liveblocks] Initial content must be set in the useLiveblocksExtension hook option. Remove content from your editor options.\"\n        );\n      }\n      if (\n        options.mentions &&\n        this.editor.extensionManager.extensions.find(\n          (e) => e.name.toLowerCase() === \"mention\"\n        )\n      ) {\n        console.warn(\n          \"[Liveblocks] Liveblocks own mention plugin is enabled, using another mention plugin may cause a conflict.\"\n        );\n      }\n      const self = room.getSelf();\n      const updateUser = ({\n        info,\n        id: userId,\n      }: User<JsonObject, BaseUserMeta>) => {\n        if (!info) {\n          return;\n        }\n        const { user: storedUser } =\n          this.storage.provider.awareness.getLocalState() as {\n            user: IUserInfo;\n          };\n        this.storage.permanentUserData?.setUserMapping(\n          this.storage.doc,\n          this.storage.doc.clientID,\n          userId ?? \"Unknown\" // TODO: change this to the user's ID so we can map it to the user's name\n        );\n        if (\n          info.name !== storedUser?.name ||\n          info.color !== storedUser?.color\n        ) {\n          this.editor.commands.updateUser({\n            name: info.name,\n            color: info.color,\n          });\n        }\n      };\n      // if we already have user info, we update the user\n      if (self?.info) {\n        updateUser(self);\n      }\n      // we also listen in case the user info changes\n      this.storage.unsubs.push(room.events.self.subscribe(updateUser));\n      if (options.comments) {\n        const commentMarkType = getMarkType(\n          LIVEBLOCKS_COMMENT_MARK_TYPE,\n          this.editor.schema\n        );\n        this.storage.unsubs.push(\n          // Subscribe to threads so we can update comment marks if they become resolved/deleted\n          store.outputs.threads.subscribe(() => {\n            const threadMap = new Map(\n              store.outputs.threads\n                .get()\n                .findMany(roomId, { resolved: false }, \"asc\")\n                .map((thread) => [thread.id, true])\n            );\n            function isComment(mark: PMMark): mark is PMMark & {\n              attrs: { orphan: boolean; threadId: string };\n            } {\n              return mark.type.name === LIVEBLOCKS_COMMENT_MARK_TYPE;\n            }\n            // when threads change, find marks and update them if needed\n            this.editor.state.doc.descendants((node, pos) => {\n              node.marks.forEach((mark) => {\n                if (isComment(mark)) {\n                  const markThreadId = mark.attrs.threadId;\n                  const isOrphan = !threadMap.has(markThreadId);\n                  if (isOrphan !== mark.attrs.orphan) {\n                    const { tr } = this.editor.state;\n                    const trimmedFrom = Math.max(pos, 0);\n                    const trimmedTo = Math.min(\n                      pos + node.nodeSize,\n                      this.editor.state.doc.content.size - 1\n                    );\n                    tr.removeMark(trimmedFrom, trimmedTo, commentMarkType);\n                    tr.addMark(\n                      trimmedFrom,\n                      trimmedTo,\n                      commentMarkType.create({\n                        ...mark.attrs,\n                        orphan: isOrphan,\n                      })\n                    );\n                    this.editor.view.dispatch(tr);\n                  }\n                }\n              });\n            });\n          })\n        );\n      }\n    },\n    onDestroy() {\n      this.storage.unsubs.forEach((unsub) => unsub());\n    },\n    addGlobalAttributes() {\n      return [\n        {\n          types: [\"paragraph\", \"heading\"],\n          attributes: {\n            ychange: { default: null },\n          },\n        },\n      ];\n    },\n    addStorage() {\n      if (!providersMap.has(room.id)) {\n        const doc = new Doc();\n        docMap.set(room.id, doc);\n        pudMap.set(room.id, new PermanentUserData(doc));\n        providersMap.set(\n          room.id,\n          new LiveblocksYjsProvider(room, doc, {\n            offlineSupport_experimental: options.offlineSupport_experimental,\n          })\n        );\n      }\n      return {\n        doc: docMap.get(room.id)!,\n        provider: providersMap.get(room.id)!,\n        permanentUserData: pudMap.get(room.id)!,\n        unsubs: [],\n      };\n    },\n    addExtensions() {\n      const extensions: AnyExtension[] = [\n        YChangeMark,\n        LiveblocksCollab.configure({\n          ySyncOptions: {\n            permanentUserData: this.storage.permanentUserData,\n          },\n          document: this.storage.doc,\n          field: options.field,\n        }),\n        CollaborationCursor.configure({\n          provider: this.storage.provider,\n        }),\n      ];\n\n      if (options.comments) {\n        extensions.push(CommentsExtension);\n      }\n      if (options.mentions) {\n        extensions.push(\n          MentionExtension.configure({\n            onCreateMention: createTextMention,\n            onDeleteMention: deleteTextMention,\n          })\n        );\n      }\n      if (options.ai) {\n        const resolveAiPrompt = async ({\n          prompt,\n          selectionText,\n          context,\n          signal,\n        }: ResolveAiPromptArgs): Promise<AiResponse> => {\n          const result = await room[kInternal].executeContextualPrompt({\n            prompt,\n            selectionText,\n            context,\n            signal,\n          });\n\n          const parsedResponse = JSON.parse(result) as JsonObject;\n          const type =\n            typeof parsedResponse.type === \"string\" ? parsedResponse.type : \"\";\n          if (\n            typeof parsedResponse.content === \"string\" &&\n            [\"insert\", \"modification\", \"other\"].includes(type)\n          ) {\n            return parsedResponse as AiResponse;\n          }\n\n          throw new Error(\"Failed to resolve AI prompt\");\n        };\n        extensions.push(\n          AiExtension.configure({\n            resolveAiPrompt,\n            ...(typeof options.ai === \"boolean\" ? {} : options.ai),\n            doc: this.storage.doc,\n            pud: this.storage.permanentUserData,\n          })\n        );\n      }\n\n      return extensions;\n    },\n  });\n};\n"],"names":["useYjsProvider","useCallback","useSyncExternalStore","Mark","useState","useRoom","useClient","getUmbrellaStoreForClient","useEffect","useReportTextEditor","TextEditorType","useCreateTextMention","useDeleteTextMention","Extension","getMarkType","LIVEBLOCKS_COMMENT_MARK_TYPE","Doc","PermanentUserData","LiveblocksYjsProvider","CommentsExtension","MentionExtension","kInternal","AiExtension"],"mappings":";;;;;;;;;;;;;;;;AAmCA,MAAM,YAAA,uBAAmB,GAGvB,EAAA,CAAA;AAEF,MAAM,MAAA,uBAAa,GAAiB,EAAA,CAAA;AAIpC,MAAM,MAAA,uBAAa,GAA+B,EAAA,CAAA;AAElD,MAAM,eAAqE,GAAA;AAAA,EACzE,KAAO,EAAA,SAAA;AAAA,EACP,QAAU,EAAA,IAAA;AAAA,EACV,QAAU,EAAA,IAAA;AAAA,EACV,2BAA6B,EAAA,KAAA;AAC/B,CAAA,CAAA;AAEA,MAAM,gBAAA,GAAmB,cAAc,MAAO,CAAA;AAAA,EAE5C,QAAW,GAAA;AACT,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,KAAK,CACrE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,wIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IAAA,IACE,CAAC,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA;AAAA,MACvC,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,WAAA;AAAA,KAEpB,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,yIAAA;AAAA,OACF,CAAA;AAAA,KACF;AAEA,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,MAAM,CACtE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,oIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IACE,IAAA,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA,CAAK,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,SAAS,CACxE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,kKAAA;AAAA,OACF,CAAA;AAAA,KACF;AAAA,GACF;AACF,CAAC,CAAA,CAAA;AAOM,SAAS,gBAA4B,GAAA;AAC1C,EAAA,MAAM,cAAcA,uBAAe,EAAA,CAAA;AAEnC,EAAM,MAAA,WAAA,GAAcC,kBAAY,MAAM;AACpC,IAAM,MAAA,MAAA,GAAS,aAAa,SAAU,EAAA,CAAA;AACtC,IAAO,OAAA,MAAA,KAAW,mBAAmB,MAAW,KAAA,cAAA,CAAA;AAAA,GAClD,EAAG,CAAC,WAAW,CAAC,CAAA,CAAA;AAEhB,EAAA,MAAM,SAAY,GAAAA,iBAAA;AAAA,IAChB,CAAC,QAAyB,KAAA;AACxB,MAAA,IAAI,WAAgB,KAAA,KAAA,CAAA;AAAW,QAAA,OAAO,MAAM;AAAA,SAAC,CAAA;AAC7C,MAAY,WAAA,CAAA,EAAA,CAAG,UAAU,QAAQ,CAAA,CAAA;AACjC,MAAA,OAAO,MAAM;AACX,QAAY,WAAA,CAAA,GAAA,CAAI,UAAU,QAAQ,CAAA,CAAA;AAAA,OACpC,CAAA;AAAA,KACF;AAAA,IACA,CAAC,WAAW,CAAA;AAAA,GACd,CAAA;AAEA,EAAO,OAAAC,0BAAA,CAAqB,SAAW,EAAA,WAAA,EAAa,WAAW,CAAA,CAAA;AACjE,CAAA;AAEA,MAAM,WAAA,GAAcC,UAAK,MAAO,CAAA;AAAA,EAC9B,IAAM,EAAA,SAAA;AAAA,EACN,SAAW,EAAA,KAAA;AAAA,EACX,SAAY,GAAA;AACV,IAAA,OAAO,CAAC,EAAE,GAAK,EAAA,SAAA,EAAW,CAAA,CAAA;AAAA,GAC5B;AAAA,EACA,aAAgB,GAAA;AACd,IAAO,OAAA;AAAA,MACL,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA,EAAE,mBAAqB,EAAA,UAAA,CAAW,IAAK,EAAA,CAAA;AAAA,SAChD;AAAA,OACF;AAAA,MACA,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA;AAAA,YACL,qBAAqB,UAAW,CAAA,IAAA;AAAA,YAChC,KAAA,EAAO,4BAA4B,UAAW,CAAA,IAAA,CAAA,CAAA;AAAA,WAChD,CAAA;AAAA,SACF;AAAA,OACF;AAAA,MACA,KAAO,EAAA;AAAA,QACL,OAAS,EAAA,IAAA;AAAA,QACT,SAAA,EAAW,CAAC,OAAY,KAAA;AACtB,UAAO,OAAA,OAAA,CAAQ,YAAa,CAAA,eAAe,CAAK,IAAA,IAAA,CAAA;AAAA,SAClD;AAAA,QACA,YAAY,MAAM;AAEhB,UAAA,OAAO,EAAC,CAAA;AAAA,SACV;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AAAA,EACA,UAAA,CAAW,EAAE,cAAA,EAAkB,EAAA;AAC7B,IAAO,OAAA,CAAC,SAAW,EAAA,cAAA,EAAgB,CAAC,CAAA,CAAA;AAAA,GACtC;AACF,CAAC,CAAA,CAAA;AAEY,MAAA,sBAAA,GAAyB,CACpC,IACc,KAAA;AACd,EAAA,MAAM,OAAU,GAAA;AAAA,IACd,GAAG,eAAA;AAAA,IACH,GAAG,IAAA;AAAA,GACL,CAAA;AACA,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAIC,eAAwB,IAAI,CAAA,CAAA;AACxD,EAAA,MAAM,OAAOC,eAAQ,EAAA,CAAA;AAcrB,EAAA,MAAM,gBAAgB,gBAAiB,EAAA,CAAA;AACvC,EAAA,MAAM,SAASC,iBAAU,EAAA,CAAA;AACzB,EAAM,MAAA,KAAA,GAAQC,mCAA0B,MAAM,CAAA,CAAA;AAC9C,EAAA,MAAM,SAAS,IAAK,CAAA,EAAA,CAAA;AACpB,EAAA,MAAM,cAAcP,uBAAe,EAAA,CAAA;AAGnC,EAAAQ,eAAA,CAAU,MAAM;AACd,IAAA,IAAI,CAAC,aAAiB,IAAA,CAAC,eAAe,CAAC,OAAA,CAAQ,kBAAkB,CAAC,MAAA;AAChE,MAAA,OAAA;AAIF,IAAM,MAAA,IAAA,GAAQ,YAAsC,OAAQ,EAAA,CAAA;AAC5D,IAAA,MAAM,gBAAgB,IAAK,CAAA,MAAA,CAAO,mBAAmB,CAAA,CAAE,IAAI,eAAe,CAAA,CAAA;AAC1E,IAAA,IAAI,CAAC,aAAe,EAAA;AAClB,MAAA,IAAA,CAAK,MAAO,CAAA,mBAAmB,CAAE,CAAA,GAAA,CAAI,iBAAiB,IAAI,CAAA,CAAA;AAC1D,MAAO,MAAA,CAAA,QAAA,CAAS,UAAW,CAAA,OAAA,CAAQ,cAAc,CAAA,CAAA;AAAA,KACnD;AAAA,KACC,CAAC,aAAA,EAAe,aAAa,OAAQ,CAAA,cAAA,EAAgB,MAAM,CAAC,CAAA,CAAA;AAE/D,EAAAC,4BAAA;AAAA,IACEC,qBAAe,CAAA,MAAA;AAAA,IACf,OAAA,CAAQ,SAAS,eAAgB,CAAA,KAAA;AAAA,GACnC,CAAA;AAEA,EAAA,MAAM,oBAAoBC,6BAAqB,EAAA,CAAA;AAC/C,EAAA,MAAM,oBAAoBC,6BAAqB,EAAA,CAAA;AAE/C,EAAA,OAAOC,eAAU,MAA0C,CAAA;AAAA,IACzD,IAAM,EAAA,qBAAA;AAAA,IAEN,QAAW,GAAA;AACT,MAAA,SAAA,CAAU,KAAK,MAAM,CAAA,CAAA;AACrB,MAAI,IAAA,IAAA,CAAK,MAAO,CAAA,OAAA,CAAQ,OAAS,EAAA;AAC/B,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,8HAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAA,IACE,OAAQ,CAAA,QAAA,IACR,IAAK,CAAA,MAAA,CAAO,iBAAiB,UAAW,CAAA,IAAA;AAAA,QACtC,CAAC,CAAA,KAAM,CAAE,CAAA,IAAA,CAAK,aAAkB,KAAA,SAAA;AAAA,OAElC,EAAA;AACA,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,2GAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAM,MAAA,IAAA,GAAO,KAAK,OAAQ,EAAA,CAAA;AAC1B,MAAA,MAAM,aAAa,CAAC;AAAA,QAClB,IAAA;AAAA,QACA,EAAI,EAAA,MAAA;AAAA,OACgC,KAAA;AACpC,QAAA,IAAI,CAAC,IAAM,EAAA;AACT,UAAA,OAAA;AAAA,SACF;AACA,QAAM,MAAA,EAAE,MAAM,UAAW,EAAA,GACvB,KAAK,OAAQ,CAAA,QAAA,CAAS,UAAU,aAAc,EAAA,CAAA;AAGhD,QAAA,IAAA,CAAK,QAAQ,iBAAmB,EAAA,cAAA;AAAA,UAC9B,KAAK,OAAQ,CAAA,GAAA;AAAA,UACb,IAAA,CAAK,QAAQ,GAAI,CAAA,QAAA;AAAA,UACjB,MAAU,IAAA,SAAA;AAAA,SACZ,CAAA;AACA,QAAA,IACE,KAAK,IAAS,KAAA,UAAA,EAAY,QAC1B,IAAK,CAAA,KAAA,KAAU,YAAY,KAC3B,EAAA;AACA,UAAK,IAAA,CAAA,MAAA,CAAO,SAAS,UAAW,CAAA;AAAA,YAC9B,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,OAAO,IAAK,CAAA,KAAA;AAAA,WACb,CAAA,CAAA;AAAA,SACH;AAAA,OACF,CAAA;AAEA,MAAA,IAAI,MAAM,IAAM,EAAA;AACd,QAAA,UAAA,CAAW,IAAI,CAAA,CAAA;AAAA,OACjB;AAEA,MAAK,IAAA,CAAA,OAAA,CAAQ,OAAO,IAAK,CAAA,IAAA,CAAK,OAAO,IAAK,CAAA,SAAA,CAAU,UAAU,CAAC,CAAA,CAAA;AAC/D,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,MAAM,eAAkB,GAAAC,gBAAA;AAAA,UACtBC,kCAAA;AAAA,UACA,KAAK,MAAO,CAAA,MAAA;AAAA,SACd,CAAA;AACA,QAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,IAAA;AAAA,UAElB,KAAM,CAAA,OAAA,CAAQ,OAAQ,CAAA,SAAA,CAAU,MAAM;AACpC,YAAA,MAAM,YAAY,IAAI,GAAA;AAAA,cACpB,KAAA,CAAM,QAAQ,OACX,CAAA,GAAA,GACA,QAAS,CAAA,MAAA,EAAQ,EAAE,QAAU,EAAA,KAAA,IAAS,KAAK,CAAA,CAC3C,IAAI,CAAC,MAAA,KAAW,CAAC,MAAO,CAAA,EAAA,EAAI,IAAI,CAAC,CAAA;AAAA,aACtC,CAAA;AACA,YAAA,SAAS,UAAU,IAEjB,EAAA;AACA,cAAO,OAAA,IAAA,CAAK,KAAK,IAAS,KAAAA,kCAAA,CAAA;AAAA,aAC5B;AAEA,YAAA,IAAA,CAAK,OAAO,KAAM,CAAA,GAAA,CAAI,WAAY,CAAA,CAAC,MAAM,GAAQ,KAAA;AAC/C,cAAK,IAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,CAAC,IAAS,KAAA;AAC3B,gBAAI,IAAA,SAAA,CAAU,IAAI,CAAG,EAAA;AACnB,kBAAM,MAAA,YAAA,GAAe,KAAK,KAAM,CAAA,QAAA,CAAA;AAChC,kBAAA,MAAM,QAAW,GAAA,CAAC,SAAU,CAAA,GAAA,CAAI,YAAY,CAAA,CAAA;AAC5C,kBAAI,IAAA,QAAA,KAAa,IAAK,CAAA,KAAA,CAAM,MAAQ,EAAA;AAClC,oBAAA,MAAM,EAAE,EAAA,EAAO,GAAA,IAAA,CAAK,MAAO,CAAA,KAAA,CAAA;AAC3B,oBAAA,MAAM,WAAc,GAAA,IAAA,CAAK,GAAI,CAAA,GAAA,EAAK,CAAC,CAAA,CAAA;AACnC,oBAAA,MAAM,YAAY,IAAK,CAAA,GAAA;AAAA,sBACrB,MAAM,IAAK,CAAA,QAAA;AAAA,sBACX,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,GAAA,CAAI,QAAQ,IAAO,GAAA,CAAA;AAAA,qBACvC,CAAA;AACA,oBAAG,EAAA,CAAA,UAAA,CAAW,WAAa,EAAA,SAAA,EAAW,eAAe,CAAA,CAAA;AACrD,oBAAG,EAAA,CAAA,OAAA;AAAA,sBACD,WAAA;AAAA,sBACA,SAAA;AAAA,sBACA,gBAAgB,MAAO,CAAA;AAAA,wBACrB,GAAG,IAAK,CAAA,KAAA;AAAA,wBACR,MAAQ,EAAA,QAAA;AAAA,uBACT,CAAA;AAAA,qBACH,CAAA;AACA,oBAAK,IAAA,CAAA,MAAA,CAAO,IAAK,CAAA,QAAA,CAAS,EAAE,CAAA,CAAA;AAAA,mBAC9B;AAAA,iBACF;AAAA,eACD,CAAA,CAAA;AAAA,aACF,CAAA,CAAA;AAAA,WACF,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA,SAAY,GAAA;AACV,MAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,OAAA,CAAQ,CAAC,KAAA,KAAU,OAAO,CAAA,CAAA;AAAA,KAChD;AAAA,IACA,mBAAsB,GAAA;AACpB,MAAO,OAAA;AAAA,QACL;AAAA,UACE,KAAA,EAAO,CAAC,WAAA,EAAa,SAAS,CAAA;AAAA,UAC9B,UAAY,EAAA;AAAA,YACV,OAAA,EAAS,EAAE,OAAA,EAAS,IAAK,EAAA;AAAA,WAC3B;AAAA,SACF;AAAA,OACF,CAAA;AAAA,KACF;AAAA,IACA,UAAa,GAAA;AACX,MAAA,IAAI,CAAC,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAG,EAAA;AAC9B,QAAM,MAAA,GAAA,GAAM,IAAIC,OAAI,EAAA,CAAA;AACpB,QAAO,MAAA,CAAA,GAAA,CAAI,IAAK,CAAA,EAAA,EAAI,GAAG,CAAA,CAAA;AACvB,QAAA,MAAA,CAAO,IAAI,IAAK,CAAA,EAAA,EAAI,IAAIC,qBAAA,CAAkB,GAAG,CAAC,CAAA,CAAA;AAC9C,QAAa,YAAA,CAAA,GAAA;AAAA,UACX,IAAK,CAAA,EAAA;AAAA,UACL,IAAIC,2BAAsB,CAAA,IAAA,EAAM,GAAK,EAAA;AAAA,YACnC,6BAA6B,OAAQ,CAAA,2BAAA;AAAA,WACtC,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAO,OAAA;AAAA,QACL,GAAK,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACvB,QAAU,EAAA,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QAClC,iBAAmB,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACrC,QAAQ,EAAC;AAAA,OACX,CAAA;AAAA,KACF;AAAA,IACA,aAAgB,GAAA;AACd,MAAA,MAAM,UAA6B,GAAA;AAAA,QACjC,WAAA;AAAA,QACA,iBAAiB,SAAU,CAAA;AAAA,UACzB,YAAc,EAAA;AAAA,YACZ,iBAAA,EAAmB,KAAK,OAAQ,CAAA,iBAAA;AAAA,WAClC;AAAA,UACA,QAAA,EAAU,KAAK,OAAQ,CAAA,GAAA;AAAA,UACvB,OAAO,OAAQ,CAAA,KAAA;AAAA,SAChB,CAAA;AAAA,QACD,oBAAoB,SAAU,CAAA;AAAA,UAC5B,QAAA,EAAU,KAAK,OAAQ,CAAA,QAAA;AAAA,SACxB,CAAA;AAAA,OACH,CAAA;AAEA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,UAAA,CAAW,KAAKC,mCAAiB,CAAA,CAAA;AAAA,OACnC;AACA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAW,UAAA,CAAA,IAAA;AAAA,UACTC,kCAAiB,SAAU,CAAA;AAAA,YACzB,eAAiB,EAAA,iBAAA;AAAA,YACjB,eAAiB,EAAA,iBAAA;AAAA,WAClB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAA,IAAI,QAAQ,EAAI,EAAA;AACd,QAAA,MAAM,kBAAkB,OAAO;AAAA,UAC7B,MAAA;AAAA,UACA,aAAA;AAAA,UACA,OAAA;AAAA,UACA,MAAA;AAAA,SAC8C,KAAA;AAC9C,UAAA,MAAM,MAAS,GAAA,MAAM,IAAK,CAAAC,gBAAA,CAAA,CAAW,uBAAwB,CAAA;AAAA,YAC3D,MAAA;AAAA,YACA,aAAA;AAAA,YACA,OAAA;AAAA,YACA,MAAA;AAAA,WACD,CAAA,CAAA;AAED,UAAM,MAAA,cAAA,GAAiB,IAAK,CAAA,KAAA,CAAM,MAAM,CAAA,CAAA;AACxC,UAAA,MAAM,OACJ,OAAO,cAAA,CAAe,IAAS,KAAA,QAAA,GAAW,eAAe,IAAO,GAAA,EAAA,CAAA;AAClE,UACE,IAAA,OAAO,cAAe,CAAA,OAAA,KAAY,QAClC,IAAA,CAAC,QAAU,EAAA,cAAA,EAAgB,OAAO,CAAA,CAAE,QAAS,CAAA,IAAI,CACjD,EAAA;AACA,YAAO,OAAA,cAAA,CAAA;AAAA,WACT;AAEA,UAAM,MAAA,IAAI,MAAM,6BAA6B,CAAA,CAAA;AAAA,SAC/C,CAAA;AACA,QAAW,UAAA,CAAA,IAAA;AAAA,UACTC,wBAAY,SAAU,CAAA;AAAA,YACpB,eAAA;AAAA,YACA,GAAI,OAAO,OAAA,CAAQ,OAAO,SAAY,GAAA,KAAK,OAAQ,CAAA,EAAA;AAAA,YACnD,GAAA,EAAK,KAAK,OAAQ,CAAA,GAAA;AAAA,YAClB,GAAA,EAAK,KAAK,OAAQ,CAAA,iBAAA;AAAA,WACnB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAEA,MAAO,OAAA,UAAA,CAAA;AAAA,KACT;AAAA,GACD,CAAA,CAAA;AACH;;;;;"}

package/dist/LiveblocksExtension.mjs

@@ -1,6 +1,6 @@
 import { TextEditorType, kInternal } from '@liveblocks/core';
-import { useRoom, useCommentsErrorListener, useClient } from '@liveblocks/react';
-import { useYjsProvider, CreateThreadError, getUmbrellaStoreForClient, useReportTextEditor, useCreateTextMention, useDeleteTextMention } from '@liveblocks/react/_private';
+import { useRoom, useClient } from '@liveblocks/react';
+import { useYjsProvider, getUmbrellaStoreForClient, useReportTextEditor, useCreateTextMention, useDeleteTextMention } from '@liveblocks/react/_private';
 import { LiveblocksYjsProvider } from '@liveblocks/yjs';
 import { Mark, Extension, getMarkType } from '@tiptap/core';
 import Collaboration from '@tiptap/extension-collaboration';
@@ -94,7 +94,7 @@ const YChangeMark = Mark.create({
           }
           return {
             "data-ychange-type": attributes.type,
-            class: `lb-changed-${attributes.type}`
+            class: `lb-root lb-tiptap-change-${attributes.type}`
           };
         }
       },
@@ -120,10 +120,6 @@ const useLiveblocksExtension = (opts) => {
   };
   const [editor, setEditor] = useState(null);
   const room = useRoom();
-  useCommentsErrorListener((error) => {
-    if (error instanceof CreateThreadError) {
-    }
-  });
   const isEditorReady = useIsEditorReady();
   const client = useClient();
   const store = getUmbrellaStoreForClient(client);
@@ -291,8 +287,7 @@ const useLiveblocksExtension = (opts) => {
           prompt,
           selectionText,
           context,
-          signal,
-          retryCount = 0
+          signal
         }) => {
           const result = await room[kInternal].executeContextualPrompt({
             prompt,
@@ -300,28 +295,12 @@ const useLiveblocksExtension = (opts) => {
             context,
             signal
           });
-          if (retryCount > 3) {
-            throw new Error("Failed to resolve AI prompt");
-          }
-          const retry = () => {
-            return resolveAiPrompt({
-              prompt,
-              selectionText,
-              context,
-              signal,
-              retryCount: retryCount + 1
-            });
-          };
-          try {
-            const parsedResponse = JSON.parse(result);
-            const type = typeof parsedResponse.type === "string" ? parsedResponse.type : "";
-            if (typeof parsedResponse.content === "string" && ["insert", "modification", "other"].includes(type)) {
-              return parsedResponse;
-            }
-            return retry();
-          } catch (e) {
-            return retry();
+          const parsedResponse = JSON.parse(result);
+          const type = typeof parsedResponse.type === "string" ? parsedResponse.type : "";
+          if (typeof parsedResponse.content === "string" && ["insert", "modification", "other"].includes(type)) {
+            return parsedResponse;
           }
+          throw new Error("Failed to resolve AI prompt");
         };
         extensions.push(
           AiExtension.configure({

package/dist/LiveblocksExtension.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"LiveblocksExtension.mjs","sources":["../src/LiveblocksExtension.ts"],"sourcesContent":["import type {\n  BaseUserMeta,\n  IUserInfo,\n  JsonObject,\n  User,\n} from \"@liveblocks/core\";\nimport { kInternal, TextEditorType } from \"@liveblocks/core\";\nimport {\n  useClient,\n  useCommentsErrorListener,\n  useRoom,\n} from \"@liveblocks/react\";\nimport {\n  CreateThreadError,\n  getUmbrellaStoreForClient,\n  useCreateTextMention,\n  useDeleteTextMention,\n  useReportTextEditor,\n  useYjsProvider,\n} from \"@liveblocks/react/_private\";\nimport { LiveblocksYjsProvider } from \"@liveblocks/yjs\";\nimport type { AnyExtension, Editor } from \"@tiptap/core\";\nimport { Extension, getMarkType, Mark } from \"@tiptap/core\";\nimport Collaboration from \"@tiptap/extension-collaboration\";\nimport CollaborationCursor from \"@tiptap/extension-collaboration-cursor\";\nimport type { Mark as PMMark } from \"@tiptap/pm/model\";\nimport { useCallback, useEffect, useState, useSyncExternalStore } from \"react\";\nimport { Doc, PermanentUserData } from \"yjs\";\n\nimport { AiExtension } from \"./ai/AiExtension\";\nimport { CommentsExtension } from \"./comments/CommentsExtension\";\nimport { MentionExtension } from \"./mentions/MentionExtension\";\nimport type {\n  AiResponse,\n  LiveblocksExtensionOptions,\n  LiveblocksExtensionStorage,\n  ResolveAiPromptArgs,\n} from \"./types\";\nimport { LIVEBLOCKS_COMMENT_MARK_TYPE } from \"./types\";\n\nconst providersMap = new Map<\n  string,\n  LiveblocksYjsProvider<any, any, any, any, any>\n>();\n\nconst docMap = new Map<string, Doc>();\n\ntype WithRequired<T, K extends keyof T> = T & { [P in K]-?: T[P] };\n\nconst pudMap = new Map<string, PermanentUserData>();\n\nconst DEFAULT_OPTIONS: WithRequired<LiveblocksExtensionOptions, \"field\"> = {\n  field: \"default\",\n  comments: true,\n  mentions: true,\n  offlineSupport_experimental: false,\n};\n\nconst LiveblocksCollab = Collaboration.extend({\n  // Override the onCreate method to warn users about potential misconfigurations\n  onCreate() {\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"doc\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap document extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      !this.editor.extensionManager.extensions.find(\n        (e) => e.name === \"paragraph\"\n      )\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap paragraph extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"text\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap text extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      this.editor.extensionManager.extensions.find((e) => e.name === \"history\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The history extension is enabled, Liveblocks extension provides its own. Please remove or disable the History plugin to prevent unwanted conflicts.\"\n      );\n    }\n  },\n});\n\n/**\n * Returns whether the editor has loaded the initial text contents from the\n * server and is ready to be used.\n *\n */\nexport function useIsEditorReady(): boolean {\n  const yjsProvider = useYjsProvider();\n\n  const getSnapshot = useCallback(() => {\n    const status = yjsProvider?.getStatus();\n    return status === \"synchronizing\" || status === \"synchronized\";\n  }, [yjsProvider]);\n\n  const subscribe = useCallback(\n    (callback: () => void) => {\n      if (yjsProvider === undefined) return () => {};\n      yjsProvider.on(\"status\", callback);\n      return () => {\n        yjsProvider.off(\"status\", callback);\n      };\n    },\n    [yjsProvider]\n  );\n\n  return useSyncExternalStore(subscribe, getSnapshot, getSnapshot);\n}\n\nconst YChangeMark = Mark.create({\n  name: \"ychange\",\n  inclusive: false,\n  parseHTML() {\n    return [{ tag: \"ychange\" }];\n  },\n  addAttributes() {\n    return {\n      user: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_user\") ?? null,\n        renderHTML: (attributes: { user: string | null }) => {\n          if (!attributes.user) {\n            return {};\n          }\n          return { \"data-ychange-user\": attributes.user };\n        },\n      },\n      type: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_type\") ?? null,\n        renderHTML: (attributes: { type: string | null }) => {\n          if (!attributes.type) {\n            return {};\n          }\n          return {\n            \"data-ychange-type\": attributes.type,\n            class: `lb-changed-${attributes.type}`,\n          };\n        },\n      },\n      color: {\n        default: null,\n        parseHTML: (element) => {\n          return element.getAttribute(\"ychange_color\") ?? null;\n        },\n        renderHTML: () => {\n          // attributes: { color: { light: string; dark: string } | null }\n          return {}; // we don't need this color attribute for now\n        },\n      },\n    };\n  },\n  renderHTML({ HTMLAttributes }) {\n    return [\"ychange\", HTMLAttributes, 0];\n  },\n});\n\nexport const useLiveblocksExtension = (\n  opts?: LiveblocksExtensionOptions\n): Extension => {\n  const options = {\n    ...DEFAULT_OPTIONS,\n    ...opts,\n  };\n  const [editor, setEditor] = useState<Editor | null>(null);\n  const room = useRoom();\n\n  // TODO: we don't need these things if comments isn't turned on...\n  // TODO: we don't have a reference to the editor here, need to figure this out\n  useCommentsErrorListener((error) => {\n    // If thread creation fails, we remove the thread id from the associated nodes and unwrap the nodes if they are no longer associated with any threads\n    if (error instanceof CreateThreadError) {\n      // handleThreadDelete(error.context.threadId);\n    }\n  });\n  const isEditorReady = useIsEditorReady();\n  const client = useClient();\n  const store = getUmbrellaStoreForClient(client);\n  const roomId = room.id;\n  const yjsProvider = useYjsProvider();\n\n  // If the user provided initialContent, wait for ready and then set it\n  useEffect(() => {\n    if (!isEditorReady || !yjsProvider || !options.initialContent || !editor)\n      return;\n\n    // As noted in the tiptap documentation, you may not set initial content with collaboration.\n    // The docs provide the following workaround:\n    const ydoc = (yjsProvider as LiveblocksYjsProvider).getYDoc();\n    const hasContentSet = ydoc.getMap(\"liveblocks_config\").get(\"hasContentSet\");\n    if (!hasContentSet) {\n      ydoc.getMap(\"liveblocks_config\").set(\"hasContentSet\", true);\n      editor.commands.setContent(options.initialContent);\n    }\n  }, [isEditorReady, yjsProvider, options.initialContent, editor]);\n\n  useReportTextEditor(\n    TextEditorType.TipTap,\n    options.field ?? DEFAULT_OPTIONS.field\n  );\n\n  const createTextMention = useCreateTextMention();\n  const deleteTextMention = useDeleteTextMention();\n\n  return Extension.create<never, LiveblocksExtensionStorage>({\n    name: \"liveblocksExtension\",\n\n    onCreate() {\n      setEditor(this.editor);\n      if (this.editor.options.content) {\n        console.warn(\n          \"[Liveblocks] Initial content must be set in the useLiveblocksExtension hook option. Remove content from your editor options.\"\n        );\n      }\n      if (\n        options.mentions &&\n        this.editor.extensionManager.extensions.find(\n          (e) => e.name.toLowerCase() === \"mention\"\n        )\n      ) {\n        console.warn(\n          \"[Liveblocks] Liveblocks own mention plugin is enabled, using another mention plugin may cause a conflict.\"\n        );\n      }\n      const self = room.getSelf();\n      const updateUser = ({\n        info,\n        id: userId,\n      }: User<JsonObject, BaseUserMeta>) => {\n        if (!info) {\n          return;\n        }\n        const { user: storedUser } =\n          this.storage.provider.awareness.getLocalState() as {\n            user: IUserInfo;\n          };\n        this.storage.permanentUserData?.setUserMapping(\n          this.storage.doc,\n          this.storage.doc.clientID,\n          userId ?? \"Unknown\" // TODO: change this to the user's ID so we can map it to the user's name\n        );\n        if (\n          info.name !== storedUser?.name ||\n          info.color !== storedUser?.color\n        ) {\n          this.editor.commands.updateUser({\n            name: info.name,\n            color: info.color,\n          });\n        }\n      };\n      // if we already have user info, we update the user\n      if (self?.info) {\n        updateUser(self);\n      }\n      // we also listen in case the user info changes\n      this.storage.unsubs.push(room.events.self.subscribe(updateUser));\n      if (options.comments) {\n        const commentMarkType = getMarkType(\n          LIVEBLOCKS_COMMENT_MARK_TYPE,\n          this.editor.schema\n        );\n        this.storage.unsubs.push(\n          // Subscribe to threads so we can update comment marks if they become resolved/deleted\n          store.outputs.threads.subscribe(() => {\n            const threadMap = new Map(\n              store.outputs.threads\n                .get()\n                .findMany(roomId, { resolved: false }, \"asc\")\n                .map((thread) => [thread.id, true])\n            );\n            function isComment(mark: PMMark): mark is PMMark & {\n              attrs: { orphan: boolean; threadId: string };\n            } {\n              return mark.type.name === LIVEBLOCKS_COMMENT_MARK_TYPE;\n            }\n            // when threads change, find marks and update them if needed\n            this.editor.state.doc.descendants((node, pos) => {\n              node.marks.forEach((mark) => {\n                if (isComment(mark)) {\n                  const markThreadId = mark.attrs.threadId;\n                  const isOrphan = !threadMap.has(markThreadId);\n                  if (isOrphan !== mark.attrs.orphan) {\n                    const { tr } = this.editor.state;\n                    const trimmedFrom = Math.max(pos, 0);\n                    const trimmedTo = Math.min(\n                      pos + node.nodeSize,\n                      this.editor.state.doc.content.size - 1\n                    );\n                    tr.removeMark(trimmedFrom, trimmedTo, commentMarkType);\n                    tr.addMark(\n                      trimmedFrom,\n                      trimmedTo,\n                      commentMarkType.create({\n                        ...mark.attrs,\n                        orphan: isOrphan,\n                      })\n                    );\n                    this.editor.view.dispatch(tr);\n                  }\n                }\n              });\n            });\n          })\n        );\n      }\n    },\n    onDestroy() {\n      this.storage.unsubs.forEach((unsub) => unsub());\n    },\n    addGlobalAttributes() {\n      return [\n        {\n          types: [\"paragraph\", \"heading\"],\n          attributes: {\n            ychange: { default: null },\n          },\n        },\n      ];\n    },\n    addStorage() {\n      if (!providersMap.has(room.id)) {\n        const doc = new Doc();\n        docMap.set(room.id, doc);\n        pudMap.set(room.id, new PermanentUserData(doc));\n        providersMap.set(\n          room.id,\n          new LiveblocksYjsProvider(room, doc, {\n            offlineSupport_experimental: options.offlineSupport_experimental,\n          })\n        );\n      }\n      return {\n        doc: docMap.get(room.id)!,\n        provider: providersMap.get(room.id)!,\n        permanentUserData: pudMap.get(room.id)!,\n        unsubs: [],\n      };\n    },\n    addExtensions() {\n      const extensions: AnyExtension[] = [\n        YChangeMark,\n        LiveblocksCollab.configure({\n          ySyncOptions: {\n            permanentUserData: this.storage.permanentUserData,\n          },\n          document: this.storage.doc,\n          field: options.field,\n        }),\n        CollaborationCursor.configure({\n          provider: this.storage.provider,\n        }),\n      ];\n\n      if (options.comments) {\n        extensions.push(CommentsExtension);\n      }\n      if (options.mentions) {\n        extensions.push(\n          MentionExtension.configure({\n            onCreateMention: createTextMention,\n            onDeleteMention: deleteTextMention,\n          })\n        );\n      }\n      if (options.ai) {\n        const resolveAiPrompt = async ({\n          prompt,\n          selectionText,\n          context,\n          signal,\n          retryCount = 0,\n        }: ResolveAiPromptArgs): Promise<AiResponse> => {\n          const result = await room[kInternal].executeContextualPrompt({\n            prompt,\n            selectionText,\n            context,\n            signal,\n          });\n\n          if (retryCount > 3) {\n            throw new Error(\"Failed to resolve AI prompt\");\n          }\n\n          const retry = () => {\n            return resolveAiPrompt({\n              prompt,\n              selectionText,\n              context,\n              signal,\n              retryCount: retryCount + 1,\n            });\n          };\n          // TODO: proper backoff\n          try {\n            const parsedResponse = JSON.parse(result) as JsonObject;\n            const type =\n              typeof parsedResponse.type === \"string\"\n                ? parsedResponse.type\n                : \"\";\n            if (\n              typeof parsedResponse.content === \"string\" &&\n              [\"insert\", \"modification\", \"other\"].includes(type)\n            ) {\n              return parsedResponse as AiResponse;\n            }\n            return retry();\n          } catch (e) {\n            return retry();\n          }\n        };\n        extensions.push(\n          AiExtension.configure({\n            resolveAiPrompt,\n            ...(typeof options.ai === \"boolean\" ? {} : options.ai),\n            doc: this.storage.doc,\n            pud: this.storage.permanentUserData,\n          })\n        );\n      }\n\n      return extensions;\n    },\n  });\n};\n"],"names":[],"mappings":";;;;;;;;;;;;;;AAwCA,MAAM,YAAA,uBAAmB,GAGvB,EAAA,CAAA;AAEF,MAAM,MAAA,uBAAa,GAAiB,EAAA,CAAA;AAIpC,MAAM,MAAA,uBAAa,GAA+B,EAAA,CAAA;AAElD,MAAM,eAAqE,GAAA;AAAA,EACzE,KAAO,EAAA,SAAA;AAAA,EACP,QAAU,EAAA,IAAA;AAAA,EACV,QAAU,EAAA,IAAA;AAAA,EACV,2BAA6B,EAAA,KAAA;AAC/B,CAAA,CAAA;AAEA,MAAM,gBAAA,GAAmB,cAAc,MAAO,CAAA;AAAA,EAE5C,QAAW,GAAA;AACT,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,KAAK,CACrE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,wIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IAAA,IACE,CAAC,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA;AAAA,MACvC,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,WAAA;AAAA,KAEpB,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,yIAAA;AAAA,OACF,CAAA;AAAA,KACF;AAEA,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,MAAM,CACtE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,oIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IACE,IAAA,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA,CAAK,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,SAAS,CACxE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,kKAAA;AAAA,OACF,CAAA;AAAA,KACF;AAAA,GACF;AACF,CAAC,CAAA,CAAA;AAOM,SAAS,gBAA4B,GAAA;AAC1C,EAAA,MAAM,cAAc,cAAe,EAAA,CAAA;AAEnC,EAAM,MAAA,WAAA,GAAc,YAAY,MAAM;AACpC,IAAM,MAAA,MAAA,GAAS,aAAa,SAAU,EAAA,CAAA;AACtC,IAAO,OAAA,MAAA,KAAW,mBAAmB,MAAW,KAAA,cAAA,CAAA;AAAA,GAClD,EAAG,CAAC,WAAW,CAAC,CAAA,CAAA;AAEhB,EAAA,MAAM,SAAY,GAAA,WAAA;AAAA,IAChB,CAAC,QAAyB,KAAA;AACxB,MAAA,IAAI,WAAgB,KAAA,KAAA,CAAA;AAAW,QAAA,OAAO,MAAM;AAAA,SAAC,CAAA;AAC7C,MAAY,WAAA,CAAA,EAAA,CAAG,UAAU,QAAQ,CAAA,CAAA;AACjC,MAAA,OAAO,MAAM;AACX,QAAY,WAAA,CAAA,GAAA,CAAI,UAAU,QAAQ,CAAA,CAAA;AAAA,OACpC,CAAA;AAAA,KACF;AAAA,IACA,CAAC,WAAW,CAAA;AAAA,GACd,CAAA;AAEA,EAAO,OAAA,oBAAA,CAAqB,SAAW,EAAA,WAAA,EAAa,WAAW,CAAA,CAAA;AACjE,CAAA;AAEA,MAAM,WAAA,GAAc,KAAK,MAAO,CAAA;AAAA,EAC9B,IAAM,EAAA,SAAA;AAAA,EACN,SAAW,EAAA,KAAA;AAAA,EACX,SAAY,GAAA;AACV,IAAA,OAAO,CAAC,EAAE,GAAK,EAAA,SAAA,EAAW,CAAA,CAAA;AAAA,GAC5B;AAAA,EACA,aAAgB,GAAA;AACd,IAAO,OAAA;AAAA,MACL,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA,EAAE,mBAAqB,EAAA,UAAA,CAAW,IAAK,EAAA,CAAA;AAAA,SAChD;AAAA,OACF;AAAA,MACA,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA;AAAA,YACL,qBAAqB,UAAW,CAAA,IAAA;AAAA,YAChC,KAAA,EAAO,cAAc,UAAW,CAAA,IAAA,CAAA,CAAA;AAAA,WAClC,CAAA;AAAA,SACF;AAAA,OACF;AAAA,MACA,KAAO,EAAA;AAAA,QACL,OAAS,EAAA,IAAA;AAAA,QACT,SAAA,EAAW,CAAC,OAAY,KAAA;AACtB,UAAO,OAAA,OAAA,CAAQ,YAAa,CAAA,eAAe,CAAK,IAAA,IAAA,CAAA;AAAA,SAClD;AAAA,QACA,YAAY,MAAM;AAEhB,UAAA,OAAO,EAAC,CAAA;AAAA,SACV;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AAAA,EACA,UAAA,CAAW,EAAE,cAAA,EAAkB,EAAA;AAC7B,IAAO,OAAA,CAAC,SAAW,EAAA,cAAA,EAAgB,CAAC,CAAA,CAAA;AAAA,GACtC;AACF,CAAC,CAAA,CAAA;AAEY,MAAA,sBAAA,GAAyB,CACpC,IACc,KAAA;AACd,EAAA,MAAM,OAAU,GAAA;AAAA,IACd,GAAG,eAAA;AAAA,IACH,GAAG,IAAA;AAAA,GACL,CAAA;AACA,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAI,SAAwB,IAAI,CAAA,CAAA;AACxD,EAAA,MAAM,OAAO,OAAQ,EAAA,CAAA;AAIrB,EAAA,wBAAA,CAAyB,CAAC,KAAU,KAAA;AAElC,IAAA,IAAI,iBAAiB,iBAAmB,EAAA;AAAA,KAExC;AAAA,GACD,CAAA,CAAA;AACD,EAAA,MAAM,gBAAgB,gBAAiB,EAAA,CAAA;AACvC,EAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AACzB,EAAM,MAAA,KAAA,GAAQ,0BAA0B,MAAM,CAAA,CAAA;AAC9C,EAAA,MAAM,SAAS,IAAK,CAAA,EAAA,CAAA;AACpB,EAAA,MAAM,cAAc,cAAe,EAAA,CAAA;AAGnC,EAAA,SAAA,CAAU,MAAM;AACd,IAAA,IAAI,CAAC,aAAiB,IAAA,CAAC,eAAe,CAAC,OAAA,CAAQ,kBAAkB,CAAC,MAAA;AAChE,MAAA,OAAA;AAIF,IAAM,MAAA,IAAA,GAAQ,YAAsC,OAAQ,EAAA,CAAA;AAC5D,IAAA,MAAM,gBAAgB,IAAK,CAAA,MAAA,CAAO,mBAAmB,CAAA,CAAE,IAAI,eAAe,CAAA,CAAA;AAC1E,IAAA,IAAI,CAAC,aAAe,EAAA;AAClB,MAAA,IAAA,CAAK,MAAO,CAAA,mBAAmB,CAAE,CAAA,GAAA,CAAI,iBAAiB,IAAI,CAAA,CAAA;AAC1D,MAAO,MAAA,CAAA,QAAA,CAAS,UAAW,CAAA,OAAA,CAAQ,cAAc,CAAA,CAAA;AAAA,KACnD;AAAA,KACC,CAAC,aAAA,EAAe,aAAa,OAAQ,CAAA,cAAA,EAAgB,MAAM,CAAC,CAAA,CAAA;AAE/D,EAAA,mBAAA;AAAA,IACE,cAAe,CAAA,MAAA;AAAA,IACf,OAAA,CAAQ,SAAS,eAAgB,CAAA,KAAA;AAAA,GACnC,CAAA;AAEA,EAAA,MAAM,oBAAoB,oBAAqB,EAAA,CAAA;AAC/C,EAAA,MAAM,oBAAoB,oBAAqB,EAAA,CAAA;AAE/C,EAAA,OAAO,UAAU,MAA0C,CAAA;AAAA,IACzD,IAAM,EAAA,qBAAA;AAAA,IAEN,QAAW,GAAA;AACT,MAAA,SAAA,CAAU,KAAK,MAAM,CAAA,CAAA;AACrB,MAAI,IAAA,IAAA,CAAK,MAAO,CAAA,OAAA,CAAQ,OAAS,EAAA;AAC/B,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,8HAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAA,IACE,OAAQ,CAAA,QAAA,IACR,IAAK,CAAA,MAAA,CAAO,iBAAiB,UAAW,CAAA,IAAA;AAAA,QACtC,CAAC,CAAA,KAAM,CAAE,CAAA,IAAA,CAAK,aAAkB,KAAA,SAAA;AAAA,OAElC,EAAA;AACA,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,2GAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAM,MAAA,IAAA,GAAO,KAAK,OAAQ,EAAA,CAAA;AAC1B,MAAA,MAAM,aAAa,CAAC;AAAA,QAClB,IAAA;AAAA,QACA,EAAI,EAAA,MAAA;AAAA,OACgC,KAAA;AACpC,QAAA,IAAI,CAAC,IAAM,EAAA;AACT,UAAA,OAAA;AAAA,SACF;AACA,QAAM,MAAA,EAAE,MAAM,UAAW,EAAA,GACvB,KAAK,OAAQ,CAAA,QAAA,CAAS,UAAU,aAAc,EAAA,CAAA;AAGhD,QAAA,IAAA,CAAK,QAAQ,iBAAmB,EAAA,cAAA;AAAA,UAC9B,KAAK,OAAQ,CAAA,GAAA;AAAA,UACb,IAAA,CAAK,QAAQ,GAAI,CAAA,QAAA;AAAA,UACjB,MAAU,IAAA,SAAA;AAAA,SACZ,CAAA;AACA,QAAA,IACE,KAAK,IAAS,KAAA,UAAA,EAAY,QAC1B,IAAK,CAAA,KAAA,KAAU,YAAY,KAC3B,EAAA;AACA,UAAK,IAAA,CAAA,MAAA,CAAO,SAAS,UAAW,CAAA;AAAA,YAC9B,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,OAAO,IAAK,CAAA,KAAA;AAAA,WACb,CAAA,CAAA;AAAA,SACH;AAAA,OACF,CAAA;AAEA,MAAA,IAAI,MAAM,IAAM,EAAA;AACd,QAAA,UAAA,CAAW,IAAI,CAAA,CAAA;AAAA,OACjB;AAEA,MAAK,IAAA,CAAA,OAAA,CAAQ,OAAO,IAAK,CAAA,IAAA,CAAK,OAAO,IAAK,CAAA,SAAA,CAAU,UAAU,CAAC,CAAA,CAAA;AAC/D,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,MAAM,eAAkB,GAAA,WAAA;AAAA,UACtB,4BAAA;AAAA,UACA,KAAK,MAAO,CAAA,MAAA;AAAA,SACd,CAAA;AACA,QAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,IAAA;AAAA,UAElB,KAAM,CAAA,OAAA,CAAQ,OAAQ,CAAA,SAAA,CAAU,MAAM;AACpC,YAAA,MAAM,YAAY,IAAI,GAAA;AAAA,cACpB,KAAA,CAAM,QAAQ,OACX,CAAA,GAAA,GACA,QAAS,CAAA,MAAA,EAAQ,EAAE,QAAU,EAAA,KAAA,IAAS,KAAK,CAAA,CAC3C,IAAI,CAAC,MAAA,KAAW,CAAC,MAAO,CAAA,EAAA,EAAI,IAAI,CAAC,CAAA;AAAA,aACtC,CAAA;AACA,YAAA,SAAS,UAAU,IAEjB,EAAA;AACA,cAAO,OAAA,IAAA,CAAK,KAAK,IAAS,KAAA,4BAAA,CAAA;AAAA,aAC5B;AAEA,YAAA,IAAA,CAAK,OAAO,KAAM,CAAA,GAAA,CAAI,WAAY,CAAA,CAAC,MAAM,GAAQ,KAAA;AAC/C,cAAK,IAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,CAAC,IAAS,KAAA;AAC3B,gBAAI,IAAA,SAAA,CAAU,IAAI,CAAG,EAAA;AACnB,kBAAM,MAAA,YAAA,GAAe,KAAK,KAAM,CAAA,QAAA,CAAA;AAChC,kBAAA,MAAM,QAAW,GAAA,CAAC,SAAU,CAAA,GAAA,CAAI,YAAY,CAAA,CAAA;AAC5C,kBAAI,IAAA,QAAA,KAAa,IAAK,CAAA,KAAA,CAAM,MAAQ,EAAA;AAClC,oBAAA,MAAM,EAAE,EAAA,EAAO,GAAA,IAAA,CAAK,MAAO,CAAA,KAAA,CAAA;AAC3B,oBAAA,MAAM,WAAc,GAAA,IAAA,CAAK,GAAI,CAAA,GAAA,EAAK,CAAC,CAAA,CAAA;AACnC,oBAAA,MAAM,YAAY,IAAK,CAAA,GAAA;AAAA,sBACrB,MAAM,IAAK,CAAA,QAAA;AAAA,sBACX,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,GAAA,CAAI,QAAQ,IAAO,GAAA,CAAA;AAAA,qBACvC,CAAA;AACA,oBAAG,EAAA,CAAA,UAAA,CAAW,WAAa,EAAA,SAAA,EAAW,eAAe,CAAA,CAAA;AACrD,oBAAG,EAAA,CAAA,OAAA;AAAA,sBACD,WAAA;AAAA,sBACA,SAAA;AAAA,sBACA,gBAAgB,MAAO,CAAA;AAAA,wBACrB,GAAG,IAAK,CAAA,KAAA;AAAA,wBACR,MAAQ,EAAA,QAAA;AAAA,uBACT,CAAA;AAAA,qBACH,CAAA;AACA,oBAAK,IAAA,CAAA,MAAA,CAAO,IAAK,CAAA,QAAA,CAAS,EAAE,CAAA,CAAA;AAAA,mBAC9B;AAAA,iBACF;AAAA,eACD,CAAA,CAAA;AAAA,aACF,CAAA,CAAA;AAAA,WACF,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA,SAAY,GAAA;AACV,MAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,OAAA,CAAQ,CAAC,KAAA,KAAU,OAAO,CAAA,CAAA;AAAA,KAChD;AAAA,IACA,mBAAsB,GAAA;AACpB,MAAO,OAAA;AAAA,QACL;AAAA,UACE,KAAA,EAAO,CAAC,WAAA,EAAa,SAAS,CAAA;AAAA,UAC9B,UAAY,EAAA;AAAA,YACV,OAAA,EAAS,EAAE,OAAA,EAAS,IAAK,EAAA;AAAA,WAC3B;AAAA,SACF;AAAA,OACF,CAAA;AAAA,KACF;AAAA,IACA,UAAa,GAAA;AACX,MAAA,IAAI,CAAC,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAG,EAAA;AAC9B,QAAM,MAAA,GAAA,GAAM,IAAI,GAAI,EAAA,CAAA;AACpB,QAAO,MAAA,CAAA,GAAA,CAAI,IAAK,CAAA,EAAA,EAAI,GAAG,CAAA,CAAA;AACvB,QAAA,MAAA,CAAO,IAAI,IAAK,CAAA,EAAA,EAAI,IAAI,iBAAA,CAAkB,GAAG,CAAC,CAAA,CAAA;AAC9C,QAAa,YAAA,CAAA,GAAA;AAAA,UACX,IAAK,CAAA,EAAA;AAAA,UACL,IAAI,qBAAsB,CAAA,IAAA,EAAM,GAAK,EAAA;AAAA,YACnC,6BAA6B,OAAQ,CAAA,2BAAA;AAAA,WACtC,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAO,OAAA;AAAA,QACL,GAAK,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACvB,QAAU,EAAA,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QAClC,iBAAmB,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACrC,QAAQ,EAAC;AAAA,OACX,CAAA;AAAA,KACF;AAAA,IACA,aAAgB,GAAA;AACd,MAAA,MAAM,UAA6B,GAAA;AAAA,QACjC,WAAA;AAAA,QACA,iBAAiB,SAAU,CAAA;AAAA,UACzB,YAAc,EAAA;AAAA,YACZ,iBAAA,EAAmB,KAAK,OAAQ,CAAA,iBAAA;AAAA,WAClC;AAAA,UACA,QAAA,EAAU,KAAK,OAAQ,CAAA,GAAA;AAAA,UACvB,OAAO,OAAQ,CAAA,KAAA;AAAA,SAChB,CAAA;AAAA,QACD,oBAAoB,SAAU,CAAA;AAAA,UAC5B,QAAA,EAAU,KAAK,OAAQ,CAAA,QAAA;AAAA,SACxB,CAAA;AAAA,OACH,CAAA;AAEA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,UAAA,CAAW,KAAK,iBAAiB,CAAA,CAAA;AAAA,OACnC;AACA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAW,UAAA,CAAA,IAAA;AAAA,UACT,iBAAiB,SAAU,CAAA;AAAA,YACzB,eAAiB,EAAA,iBAAA;AAAA,YACjB,eAAiB,EAAA,iBAAA;AAAA,WAClB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAA,IAAI,QAAQ,EAAI,EAAA;AACd,QAAA,MAAM,kBAAkB,OAAO;AAAA,UAC7B,MAAA;AAAA,UACA,aAAA;AAAA,UACA,OAAA;AAAA,UACA,MAAA;AAAA,UACA,UAAa,GAAA,CAAA;AAAA,SACiC,KAAA;AAC9C,UAAA,MAAM,MAAS,GAAA,MAAM,IAAK,CAAA,SAAA,CAAA,CAAW,uBAAwB,CAAA;AAAA,YAC3D,MAAA;AAAA,YACA,aAAA;AAAA,YACA,OAAA;AAAA,YACA,MAAA;AAAA,WACD,CAAA,CAAA;AAED,UAAA,IAAI,aAAa,CAAG,EAAA;AAClB,YAAM,MAAA,IAAI,MAAM,6BAA6B,CAAA,CAAA;AAAA,WAC/C;AAEA,UAAA,MAAM,QAAQ,MAAM;AAClB,YAAA,OAAO,eAAgB,CAAA;AAAA,cACrB,MAAA;AAAA,cACA,aAAA;AAAA,cACA,OAAA;AAAA,cACA,MAAA;AAAA,cACA,YAAY,UAAa,GAAA,CAAA;AAAA,aAC1B,CAAA,CAAA;AAAA,WACH,CAAA;AAEA,UAAI,IAAA;AACF,YAAM,MAAA,cAAA,GAAiB,IAAK,CAAA,KAAA,CAAM,MAAM,CAAA,CAAA;AACxC,YAAA,MAAM,OACJ,OAAO,cAAA,CAAe,IAAS,KAAA,QAAA,GAC3B,eAAe,IACf,GAAA,EAAA,CAAA;AACN,YACE,IAAA,OAAO,cAAe,CAAA,OAAA,KAAY,QAClC,IAAA,CAAC,QAAU,EAAA,cAAA,EAAgB,OAAO,CAAA,CAAE,QAAS,CAAA,IAAI,CACjD,EAAA;AACA,cAAO,OAAA,cAAA,CAAA;AAAA,aACT;AACA,YAAA,OAAO,KAAM,EAAA,CAAA;AAAA,mBACN,CAAP,EAAA;AACA,YAAA,OAAO,KAAM,EAAA,CAAA;AAAA,WACf;AAAA,SACF,CAAA;AACA,QAAW,UAAA,CAAA,IAAA;AAAA,UACT,YAAY,SAAU,CAAA;AAAA,YACpB,eAAA;AAAA,YACA,GAAI,OAAO,OAAA,CAAQ,OAAO,SAAY,GAAA,KAAK,OAAQ,CAAA,EAAA;AAAA,YACnD,GAAA,EAAK,KAAK,OAAQ,CAAA,GAAA;AAAA,YAClB,GAAA,EAAK,KAAK,OAAQ,CAAA,iBAAA;AAAA,WACnB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAEA,MAAO,OAAA,UAAA,CAAA;AAAA,KACT;AAAA,GACD,CAAA,CAAA;AACH;;;;"}
+{"version":3,"file":"LiveblocksExtension.mjs","sources":["../src/LiveblocksExtension.ts"],"sourcesContent":["import type {\n  BaseUserMeta,\n  IUserInfo,\n  JsonObject,\n  User,\n} from \"@liveblocks/core\";\nimport { kInternal, TextEditorType } from \"@liveblocks/core\";\nimport { useClient, useRoom } from \"@liveblocks/react\";\nimport {\n  getUmbrellaStoreForClient,\n  useCreateTextMention,\n  useDeleteTextMention,\n  useReportTextEditor,\n  useYjsProvider,\n} from \"@liveblocks/react/_private\";\nimport { LiveblocksYjsProvider } from \"@liveblocks/yjs\";\nimport type { AnyExtension, Editor } from \"@tiptap/core\";\nimport { Extension, getMarkType, Mark } from \"@tiptap/core\";\nimport Collaboration from \"@tiptap/extension-collaboration\";\nimport CollaborationCursor from \"@tiptap/extension-collaboration-cursor\";\nimport type { Mark as PMMark } from \"@tiptap/pm/model\";\nimport { useCallback, useEffect, useState, useSyncExternalStore } from \"react\";\nimport { Doc, PermanentUserData } from \"yjs\";\n\nimport { AiExtension } from \"./ai/AiExtension\";\nimport { CommentsExtension } from \"./comments/CommentsExtension\";\nimport { MentionExtension } from \"./mentions/MentionExtension\";\nimport type {\n  AiResponse,\n  LiveblocksExtensionOptions,\n  LiveblocksExtensionStorage,\n  ResolveAiPromptArgs,\n} from \"./types\";\nimport { LIVEBLOCKS_COMMENT_MARK_TYPE } from \"./types\";\n\nconst providersMap = new Map<\n  string,\n  LiveblocksYjsProvider<any, any, any, any, any>\n>();\n\nconst docMap = new Map<string, Doc>();\n\ntype WithRequired<T, K extends keyof T> = T & { [P in K]-?: T[P] };\n\nconst pudMap = new Map<string, PermanentUserData>();\n\nconst DEFAULT_OPTIONS: WithRequired<LiveblocksExtensionOptions, \"field\"> = {\n  field: \"default\",\n  comments: true,\n  mentions: true,\n  offlineSupport_experimental: false,\n};\n\nconst LiveblocksCollab = Collaboration.extend({\n  // Override the onCreate method to warn users about potential misconfigurations\n  onCreate() {\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"doc\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap document extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      !this.editor.extensionManager.extensions.find(\n        (e) => e.name === \"paragraph\"\n      )\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap paragraph extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n\n    if (\n      !this.editor.extensionManager.extensions.find((e) => e.name === \"text\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The tiptap text extension is required for Liveblocks collaboration. Please add it or use Tiptap StarterKit extension.\"\n      );\n    }\n    if (\n      this.editor.extensionManager.extensions.find((e) => e.name === \"history\")\n    ) {\n      console.warn(\n        \"[Liveblocks] The history extension is enabled, Liveblocks extension provides its own. Please remove or disable the History plugin to prevent unwanted conflicts.\"\n      );\n    }\n  },\n});\n\n/**\n * Returns whether the editor has loaded the initial text contents from the\n * server and is ready to be used.\n *\n */\nexport function useIsEditorReady(): boolean {\n  const yjsProvider = useYjsProvider();\n\n  const getSnapshot = useCallback(() => {\n    const status = yjsProvider?.getStatus();\n    return status === \"synchronizing\" || status === \"synchronized\";\n  }, [yjsProvider]);\n\n  const subscribe = useCallback(\n    (callback: () => void) => {\n      if (yjsProvider === undefined) return () => {};\n      yjsProvider.on(\"status\", callback);\n      return () => {\n        yjsProvider.off(\"status\", callback);\n      };\n    },\n    [yjsProvider]\n  );\n\n  return useSyncExternalStore(subscribe, getSnapshot, getSnapshot);\n}\n\nconst YChangeMark = Mark.create({\n  name: \"ychange\",\n  inclusive: false,\n  parseHTML() {\n    return [{ tag: \"ychange\" }];\n  },\n  addAttributes() {\n    return {\n      user: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_user\") ?? null,\n        renderHTML: (attributes: { user: string | null }) => {\n          if (!attributes.user) {\n            return {};\n          }\n          return { \"data-ychange-user\": attributes.user };\n        },\n      },\n      type: {\n        default: null,\n        parseHTML: (element) => element.getAttribute(\"ychange_type\") ?? null,\n        renderHTML: (attributes: { type: string | null }) => {\n          if (!attributes.type) {\n            return {};\n          }\n          return {\n            \"data-ychange-type\": attributes.type,\n            class: `lb-root lb-tiptap-change-${attributes.type}`,\n          };\n        },\n      },\n      color: {\n        default: null,\n        parseHTML: (element) => {\n          return element.getAttribute(\"ychange_color\") ?? null;\n        },\n        renderHTML: () => {\n          // attributes: { color: { light: string; dark: string } | null }\n          return {}; // we don't need this color attribute for now\n        },\n      },\n    };\n  },\n  renderHTML({ HTMLAttributes }) {\n    return [\"ychange\", HTMLAttributes, 0];\n  },\n});\n\nexport const useLiveblocksExtension = (\n  opts?: LiveblocksExtensionOptions\n): Extension => {\n  const options = {\n    ...DEFAULT_OPTIONS,\n    ...opts,\n  };\n  const [editor, setEditor] = useState<Editor | null>(null);\n  const room = useRoom();\n\n  // TODO: we don't need these things if comments isn't turned on...\n  // TODO: we don't have a reference to the editor here, need to figure this out\n  // useErrorListener((error) => {\n  //   // If thread creation fails, we remove the thread id from the associated nodes and unwrap the nodes if they are no longer associated with any threads\n  //   if (\n  //     error.context.type === \"CREATE_THREAD_ERROR\" &&\n  //     error.context.roomId === room.id\n  //   ) {\n  //     handleThreadDelete(error.context.threadId);\n  //   }\n  // });\n\n  const isEditorReady = useIsEditorReady();\n  const client = useClient();\n  const store = getUmbrellaStoreForClient(client);\n  const roomId = room.id;\n  const yjsProvider = useYjsProvider();\n\n  // If the user provided initialContent, wait for ready and then set it\n  useEffect(() => {\n    if (!isEditorReady || !yjsProvider || !options.initialContent || !editor)\n      return;\n\n    // As noted in the tiptap documentation, you may not set initial content with collaboration.\n    // The docs provide the following workaround:\n    const ydoc = (yjsProvider as LiveblocksYjsProvider).getYDoc();\n    const hasContentSet = ydoc.getMap(\"liveblocks_config\").get(\"hasContentSet\");\n    if (!hasContentSet) {\n      ydoc.getMap(\"liveblocks_config\").set(\"hasContentSet\", true);\n      editor.commands.setContent(options.initialContent);\n    }\n  }, [isEditorReady, yjsProvider, options.initialContent, editor]);\n\n  useReportTextEditor(\n    TextEditorType.TipTap,\n    options.field ?? DEFAULT_OPTIONS.field\n  );\n\n  const createTextMention = useCreateTextMention();\n  const deleteTextMention = useDeleteTextMention();\n\n  return Extension.create<never, LiveblocksExtensionStorage>({\n    name: \"liveblocksExtension\",\n\n    onCreate() {\n      setEditor(this.editor);\n      if (this.editor.options.content) {\n        console.warn(\n          \"[Liveblocks] Initial content must be set in the useLiveblocksExtension hook option. Remove content from your editor options.\"\n        );\n      }\n      if (\n        options.mentions &&\n        this.editor.extensionManager.extensions.find(\n          (e) => e.name.toLowerCase() === \"mention\"\n        )\n      ) {\n        console.warn(\n          \"[Liveblocks] Liveblocks own mention plugin is enabled, using another mention plugin may cause a conflict.\"\n        );\n      }\n      const self = room.getSelf();\n      const updateUser = ({\n        info,\n        id: userId,\n      }: User<JsonObject, BaseUserMeta>) => {\n        if (!info) {\n          return;\n        }\n        const { user: storedUser } =\n          this.storage.provider.awareness.getLocalState() as {\n            user: IUserInfo;\n          };\n        this.storage.permanentUserData?.setUserMapping(\n          this.storage.doc,\n          this.storage.doc.clientID,\n          userId ?? \"Unknown\" // TODO: change this to the user's ID so we can map it to the user's name\n        );\n        if (\n          info.name !== storedUser?.name ||\n          info.color !== storedUser?.color\n        ) {\n          this.editor.commands.updateUser({\n            name: info.name,\n            color: info.color,\n          });\n        }\n      };\n      // if we already have user info, we update the user\n      if (self?.info) {\n        updateUser(self);\n      }\n      // we also listen in case the user info changes\n      this.storage.unsubs.push(room.events.self.subscribe(updateUser));\n      if (options.comments) {\n        const commentMarkType = getMarkType(\n          LIVEBLOCKS_COMMENT_MARK_TYPE,\n          this.editor.schema\n        );\n        this.storage.unsubs.push(\n          // Subscribe to threads so we can update comment marks if they become resolved/deleted\n          store.outputs.threads.subscribe(() => {\n            const threadMap = new Map(\n              store.outputs.threads\n                .get()\n                .findMany(roomId, { resolved: false }, \"asc\")\n                .map((thread) => [thread.id, true])\n            );\n            function isComment(mark: PMMark): mark is PMMark & {\n              attrs: { orphan: boolean; threadId: string };\n            } {\n              return mark.type.name === LIVEBLOCKS_COMMENT_MARK_TYPE;\n            }\n            // when threads change, find marks and update them if needed\n            this.editor.state.doc.descendants((node, pos) => {\n              node.marks.forEach((mark) => {\n                if (isComment(mark)) {\n                  const markThreadId = mark.attrs.threadId;\n                  const isOrphan = !threadMap.has(markThreadId);\n                  if (isOrphan !== mark.attrs.orphan) {\n                    const { tr } = this.editor.state;\n                    const trimmedFrom = Math.max(pos, 0);\n                    const trimmedTo = Math.min(\n                      pos + node.nodeSize,\n                      this.editor.state.doc.content.size - 1\n                    );\n                    tr.removeMark(trimmedFrom, trimmedTo, commentMarkType);\n                    tr.addMark(\n                      trimmedFrom,\n                      trimmedTo,\n                      commentMarkType.create({\n                        ...mark.attrs,\n                        orphan: isOrphan,\n                      })\n                    );\n                    this.editor.view.dispatch(tr);\n                  }\n                }\n              });\n            });\n          })\n        );\n      }\n    },\n    onDestroy() {\n      this.storage.unsubs.forEach((unsub) => unsub());\n    },\n    addGlobalAttributes() {\n      return [\n        {\n          types: [\"paragraph\", \"heading\"],\n          attributes: {\n            ychange: { default: null },\n          },\n        },\n      ];\n    },\n    addStorage() {\n      if (!providersMap.has(room.id)) {\n        const doc = new Doc();\n        docMap.set(room.id, doc);\n        pudMap.set(room.id, new PermanentUserData(doc));\n        providersMap.set(\n          room.id,\n          new LiveblocksYjsProvider(room, doc, {\n            offlineSupport_experimental: options.offlineSupport_experimental,\n          })\n        );\n      }\n      return {\n        doc: docMap.get(room.id)!,\n        provider: providersMap.get(room.id)!,\n        permanentUserData: pudMap.get(room.id)!,\n        unsubs: [],\n      };\n    },\n    addExtensions() {\n      const extensions: AnyExtension[] = [\n        YChangeMark,\n        LiveblocksCollab.configure({\n          ySyncOptions: {\n            permanentUserData: this.storage.permanentUserData,\n          },\n          document: this.storage.doc,\n          field: options.field,\n        }),\n        CollaborationCursor.configure({\n          provider: this.storage.provider,\n        }),\n      ];\n\n      if (options.comments) {\n        extensions.push(CommentsExtension);\n      }\n      if (options.mentions) {\n        extensions.push(\n          MentionExtension.configure({\n            onCreateMention: createTextMention,\n            onDeleteMention: deleteTextMention,\n          })\n        );\n      }\n      if (options.ai) {\n        const resolveAiPrompt = async ({\n          prompt,\n          selectionText,\n          context,\n          signal,\n        }: ResolveAiPromptArgs): Promise<AiResponse> => {\n          const result = await room[kInternal].executeContextualPrompt({\n            prompt,\n            selectionText,\n            context,\n            signal,\n          });\n\n          const parsedResponse = JSON.parse(result) as JsonObject;\n          const type =\n            typeof parsedResponse.type === \"string\" ? parsedResponse.type : \"\";\n          if (\n            typeof parsedResponse.content === \"string\" &&\n            [\"insert\", \"modification\", \"other\"].includes(type)\n          ) {\n            return parsedResponse as AiResponse;\n          }\n\n          throw new Error(\"Failed to resolve AI prompt\");\n        };\n        extensions.push(\n          AiExtension.configure({\n            resolveAiPrompt,\n            ...(typeof options.ai === \"boolean\" ? {} : options.ai),\n            doc: this.storage.doc,\n            pud: this.storage.permanentUserData,\n          })\n        );\n      }\n\n      return extensions;\n    },\n  });\n};\n"],"names":[],"mappings":";;;;;;;;;;;;;;AAmCA,MAAM,YAAA,uBAAmB,GAGvB,EAAA,CAAA;AAEF,MAAM,MAAA,uBAAa,GAAiB,EAAA,CAAA;AAIpC,MAAM,MAAA,uBAAa,GAA+B,EAAA,CAAA;AAElD,MAAM,eAAqE,GAAA;AAAA,EACzE,KAAO,EAAA,SAAA;AAAA,EACP,QAAU,EAAA,IAAA;AAAA,EACV,QAAU,EAAA,IAAA;AAAA,EACV,2BAA6B,EAAA,KAAA;AAC/B,CAAA,CAAA;AAEA,MAAM,gBAAA,GAAmB,cAAc,MAAO,CAAA;AAAA,EAE5C,QAAW,GAAA;AACT,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,KAAK,CACrE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,wIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IAAA,IACE,CAAC,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA;AAAA,MACvC,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,WAAA;AAAA,KAEpB,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,yIAAA;AAAA,OACF,CAAA;AAAA,KACF;AAEA,IACE,IAAA,CAAC,IAAK,CAAA,MAAA,CAAO,gBAAiB,CAAA,UAAA,CAAW,IAAK,CAAA,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,MAAM,CACtE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,oIAAA;AAAA,OACF,CAAA;AAAA,KACF;AACA,IACE,IAAA,IAAA,CAAK,MAAO,CAAA,gBAAA,CAAiB,UAAW,CAAA,IAAA,CAAK,CAAC,CAAM,KAAA,CAAA,CAAE,IAAS,KAAA,SAAS,CACxE,EAAA;AACA,MAAQ,OAAA,CAAA,IAAA;AAAA,QACN,kKAAA;AAAA,OACF,CAAA;AAAA,KACF;AAAA,GACF;AACF,CAAC,CAAA,CAAA;AAOM,SAAS,gBAA4B,GAAA;AAC1C,EAAA,MAAM,cAAc,cAAe,EAAA,CAAA;AAEnC,EAAM,MAAA,WAAA,GAAc,YAAY,MAAM;AACpC,IAAM,MAAA,MAAA,GAAS,aAAa,SAAU,EAAA,CAAA;AACtC,IAAO,OAAA,MAAA,KAAW,mBAAmB,MAAW,KAAA,cAAA,CAAA;AAAA,GAClD,EAAG,CAAC,WAAW,CAAC,CAAA,CAAA;AAEhB,EAAA,MAAM,SAAY,GAAA,WAAA;AAAA,IAChB,CAAC,QAAyB,KAAA;AACxB,MAAA,IAAI,WAAgB,KAAA,KAAA,CAAA;AAAW,QAAA,OAAO,MAAM;AAAA,SAAC,CAAA;AAC7C,MAAY,WAAA,CAAA,EAAA,CAAG,UAAU,QAAQ,CAAA,CAAA;AACjC,MAAA,OAAO,MAAM;AACX,QAAY,WAAA,CAAA,GAAA,CAAI,UAAU,QAAQ,CAAA,CAAA;AAAA,OACpC,CAAA;AAAA,KACF;AAAA,IACA,CAAC,WAAW,CAAA;AAAA,GACd,CAAA;AAEA,EAAO,OAAA,oBAAA,CAAqB,SAAW,EAAA,WAAA,EAAa,WAAW,CAAA,CAAA;AACjE,CAAA;AAEA,MAAM,WAAA,GAAc,KAAK,MAAO,CAAA;AAAA,EAC9B,IAAM,EAAA,SAAA;AAAA,EACN,SAAW,EAAA,KAAA;AAAA,EACX,SAAY,GAAA;AACV,IAAA,OAAO,CAAC,EAAE,GAAK,EAAA,SAAA,EAAW,CAAA,CAAA;AAAA,GAC5B;AAAA,EACA,aAAgB,GAAA;AACd,IAAO,OAAA;AAAA,MACL,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA,EAAE,mBAAqB,EAAA,UAAA,CAAW,IAAK,EAAA,CAAA;AAAA,SAChD;AAAA,OACF;AAAA,MACA,IAAM,EAAA;AAAA,QACJ,OAAS,EAAA,IAAA;AAAA,QACT,WAAW,CAAC,OAAA,KAAY,OAAQ,CAAA,YAAA,CAAa,cAAc,CAAK,IAAA,IAAA;AAAA,QAChE,UAAA,EAAY,CAAC,UAAwC,KAAA;AACnD,UAAI,IAAA,CAAC,WAAW,IAAM,EAAA;AACpB,YAAA,OAAO,EAAC,CAAA;AAAA,WACV;AACA,UAAO,OAAA;AAAA,YACL,qBAAqB,UAAW,CAAA,IAAA;AAAA,YAChC,KAAA,EAAO,4BAA4B,UAAW,CAAA,IAAA,CAAA,CAAA;AAAA,WAChD,CAAA;AAAA,SACF;AAAA,OACF;AAAA,MACA,KAAO,EAAA;AAAA,QACL,OAAS,EAAA,IAAA;AAAA,QACT,SAAA,EAAW,CAAC,OAAY,KAAA;AACtB,UAAO,OAAA,OAAA,CAAQ,YAAa,CAAA,eAAe,CAAK,IAAA,IAAA,CAAA;AAAA,SAClD;AAAA,QACA,YAAY,MAAM;AAEhB,UAAA,OAAO,EAAC,CAAA;AAAA,SACV;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AAAA,EACA,UAAA,CAAW,EAAE,cAAA,EAAkB,EAAA;AAC7B,IAAO,OAAA,CAAC,SAAW,EAAA,cAAA,EAAgB,CAAC,CAAA,CAAA;AAAA,GACtC;AACF,CAAC,CAAA,CAAA;AAEY,MAAA,sBAAA,GAAyB,CACpC,IACc,KAAA;AACd,EAAA,MAAM,OAAU,GAAA;AAAA,IACd,GAAG,eAAA;AAAA,IACH,GAAG,IAAA;AAAA,GACL,CAAA;AACA,EAAA,MAAM,CAAC,MAAA,EAAQ,SAAS,CAAA,GAAI,SAAwB,IAAI,CAAA,CAAA;AACxD,EAAA,MAAM,OAAO,OAAQ,EAAA,CAAA;AAcrB,EAAA,MAAM,gBAAgB,gBAAiB,EAAA,CAAA;AACvC,EAAA,MAAM,SAAS,SAAU,EAAA,CAAA;AACzB,EAAM,MAAA,KAAA,GAAQ,0BAA0B,MAAM,CAAA,CAAA;AAC9C,EAAA,MAAM,SAAS,IAAK,CAAA,EAAA,CAAA;AACpB,EAAA,MAAM,cAAc,cAAe,EAAA,CAAA;AAGnC,EAAA,SAAA,CAAU,MAAM;AACd,IAAA,IAAI,CAAC,aAAiB,IAAA,CAAC,eAAe,CAAC,OAAA,CAAQ,kBAAkB,CAAC,MAAA;AAChE,MAAA,OAAA;AAIF,IAAM,MAAA,IAAA,GAAQ,YAAsC,OAAQ,EAAA,CAAA;AAC5D,IAAA,MAAM,gBAAgB,IAAK,CAAA,MAAA,CAAO,mBAAmB,CAAA,CAAE,IAAI,eAAe,CAAA,CAAA;AAC1E,IAAA,IAAI,CAAC,aAAe,EAAA;AAClB,MAAA,IAAA,CAAK,MAAO,CAAA,mBAAmB,CAAE,CAAA,GAAA,CAAI,iBAAiB,IAAI,CAAA,CAAA;AAC1D,MAAO,MAAA,CAAA,QAAA,CAAS,UAAW,CAAA,OAAA,CAAQ,cAAc,CAAA,CAAA;AAAA,KACnD;AAAA,KACC,CAAC,aAAA,EAAe,aAAa,OAAQ,CAAA,cAAA,EAAgB,MAAM,CAAC,CAAA,CAAA;AAE/D,EAAA,mBAAA;AAAA,IACE,cAAe,CAAA,MAAA;AAAA,IACf,OAAA,CAAQ,SAAS,eAAgB,CAAA,KAAA;AAAA,GACnC,CAAA;AAEA,EAAA,MAAM,oBAAoB,oBAAqB,EAAA,CAAA;AAC/C,EAAA,MAAM,oBAAoB,oBAAqB,EAAA,CAAA;AAE/C,EAAA,OAAO,UAAU,MAA0C,CAAA;AAAA,IACzD,IAAM,EAAA,qBAAA;AAAA,IAEN,QAAW,GAAA;AACT,MAAA,SAAA,CAAU,KAAK,MAAM,CAAA,CAAA;AACrB,MAAI,IAAA,IAAA,CAAK,MAAO,CAAA,OAAA,CAAQ,OAAS,EAAA;AAC/B,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,8HAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAA,IACE,OAAQ,CAAA,QAAA,IACR,IAAK,CAAA,MAAA,CAAO,iBAAiB,UAAW,CAAA,IAAA;AAAA,QACtC,CAAC,CAAA,KAAM,CAAE,CAAA,IAAA,CAAK,aAAkB,KAAA,SAAA;AAAA,OAElC,EAAA;AACA,QAAQ,OAAA,CAAA,IAAA;AAAA,UACN,2GAAA;AAAA,SACF,CAAA;AAAA,OACF;AACA,MAAM,MAAA,IAAA,GAAO,KAAK,OAAQ,EAAA,CAAA;AAC1B,MAAA,MAAM,aAAa,CAAC;AAAA,QAClB,IAAA;AAAA,QACA,EAAI,EAAA,MAAA;AAAA,OACgC,KAAA;AACpC,QAAA,IAAI,CAAC,IAAM,EAAA;AACT,UAAA,OAAA;AAAA,SACF;AACA,QAAM,MAAA,EAAE,MAAM,UAAW,EAAA,GACvB,KAAK,OAAQ,CAAA,QAAA,CAAS,UAAU,aAAc,EAAA,CAAA;AAGhD,QAAA,IAAA,CAAK,QAAQ,iBAAmB,EAAA,cAAA;AAAA,UAC9B,KAAK,OAAQ,CAAA,GAAA;AAAA,UACb,IAAA,CAAK,QAAQ,GAAI,CAAA,QAAA;AAAA,UACjB,MAAU,IAAA,SAAA;AAAA,SACZ,CAAA;AACA,QAAA,IACE,KAAK,IAAS,KAAA,UAAA,EAAY,QAC1B,IAAK,CAAA,KAAA,KAAU,YAAY,KAC3B,EAAA;AACA,UAAK,IAAA,CAAA,MAAA,CAAO,SAAS,UAAW,CAAA;AAAA,YAC9B,MAAM,IAAK,CAAA,IAAA;AAAA,YACX,OAAO,IAAK,CAAA,KAAA;AAAA,WACb,CAAA,CAAA;AAAA,SACH;AAAA,OACF,CAAA;AAEA,MAAA,IAAI,MAAM,IAAM,EAAA;AACd,QAAA,UAAA,CAAW,IAAI,CAAA,CAAA;AAAA,OACjB;AAEA,MAAK,IAAA,CAAA,OAAA,CAAQ,OAAO,IAAK,CAAA,IAAA,CAAK,OAAO,IAAK,CAAA,SAAA,CAAU,UAAU,CAAC,CAAA,CAAA;AAC/D,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,MAAM,eAAkB,GAAA,WAAA;AAAA,UACtB,4BAAA;AAAA,UACA,KAAK,MAAO,CAAA,MAAA;AAAA,SACd,CAAA;AACA,QAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,IAAA;AAAA,UAElB,KAAM,CAAA,OAAA,CAAQ,OAAQ,CAAA,SAAA,CAAU,MAAM;AACpC,YAAA,MAAM,YAAY,IAAI,GAAA;AAAA,cACpB,KAAA,CAAM,QAAQ,OACX,CAAA,GAAA,GACA,QAAS,CAAA,MAAA,EAAQ,EAAE,QAAU,EAAA,KAAA,IAAS,KAAK,CAAA,CAC3C,IAAI,CAAC,MAAA,KAAW,CAAC,MAAO,CAAA,EAAA,EAAI,IAAI,CAAC,CAAA;AAAA,aACtC,CAAA;AACA,YAAA,SAAS,UAAU,IAEjB,EAAA;AACA,cAAO,OAAA,IAAA,CAAK,KAAK,IAAS,KAAA,4BAAA,CAAA;AAAA,aAC5B;AAEA,YAAA,IAAA,CAAK,OAAO,KAAM,CAAA,GAAA,CAAI,WAAY,CAAA,CAAC,MAAM,GAAQ,KAAA;AAC/C,cAAK,IAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,CAAC,IAAS,KAAA;AAC3B,gBAAI,IAAA,SAAA,CAAU,IAAI,CAAG,EAAA;AACnB,kBAAM,MAAA,YAAA,GAAe,KAAK,KAAM,CAAA,QAAA,CAAA;AAChC,kBAAA,MAAM,QAAW,GAAA,CAAC,SAAU,CAAA,GAAA,CAAI,YAAY,CAAA,CAAA;AAC5C,kBAAI,IAAA,QAAA,KAAa,IAAK,CAAA,KAAA,CAAM,MAAQ,EAAA;AAClC,oBAAA,MAAM,EAAE,EAAA,EAAO,GAAA,IAAA,CAAK,MAAO,CAAA,KAAA,CAAA;AAC3B,oBAAA,MAAM,WAAc,GAAA,IAAA,CAAK,GAAI,CAAA,GAAA,EAAK,CAAC,CAAA,CAAA;AACnC,oBAAA,MAAM,YAAY,IAAK,CAAA,GAAA;AAAA,sBACrB,MAAM,IAAK,CAAA,QAAA;AAAA,sBACX,IAAK,CAAA,MAAA,CAAO,KAAM,CAAA,GAAA,CAAI,QAAQ,IAAO,GAAA,CAAA;AAAA,qBACvC,CAAA;AACA,oBAAG,EAAA,CAAA,UAAA,CAAW,WAAa,EAAA,SAAA,EAAW,eAAe,CAAA,CAAA;AACrD,oBAAG,EAAA,CAAA,OAAA;AAAA,sBACD,WAAA;AAAA,sBACA,SAAA;AAAA,sBACA,gBAAgB,MAAO,CAAA;AAAA,wBACrB,GAAG,IAAK,CAAA,KAAA;AAAA,wBACR,MAAQ,EAAA,QAAA;AAAA,uBACT,CAAA;AAAA,qBACH,CAAA;AACA,oBAAK,IAAA,CAAA,MAAA,CAAO,IAAK,CAAA,QAAA,CAAS,EAAE,CAAA,CAAA;AAAA,mBAC9B;AAAA,iBACF;AAAA,eACD,CAAA,CAAA;AAAA,aACF,CAAA,CAAA;AAAA,WACF,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAAA,KACF;AAAA,IACA,SAAY,GAAA;AACV,MAAA,IAAA,CAAK,QAAQ,MAAO,CAAA,OAAA,CAAQ,CAAC,KAAA,KAAU,OAAO,CAAA,CAAA;AAAA,KAChD;AAAA,IACA,mBAAsB,GAAA;AACpB,MAAO,OAAA;AAAA,QACL;AAAA,UACE,KAAA,EAAO,CAAC,WAAA,EAAa,SAAS,CAAA;AAAA,UAC9B,UAAY,EAAA;AAAA,YACV,OAAA,EAAS,EAAE,OAAA,EAAS,IAAK,EAAA;AAAA,WAC3B;AAAA,SACF;AAAA,OACF,CAAA;AAAA,KACF;AAAA,IACA,UAAa,GAAA;AACX,MAAA,IAAI,CAAC,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAG,EAAA;AAC9B,QAAM,MAAA,GAAA,GAAM,IAAI,GAAI,EAAA,CAAA;AACpB,QAAO,MAAA,CAAA,GAAA,CAAI,IAAK,CAAA,EAAA,EAAI,GAAG,CAAA,CAAA;AACvB,QAAA,MAAA,CAAO,IAAI,IAAK,CAAA,EAAA,EAAI,IAAI,iBAAA,CAAkB,GAAG,CAAC,CAAA,CAAA;AAC9C,QAAa,YAAA,CAAA,GAAA;AAAA,UACX,IAAK,CAAA,EAAA;AAAA,UACL,IAAI,qBAAsB,CAAA,IAAA,EAAM,GAAK,EAAA;AAAA,YACnC,6BAA6B,OAAQ,CAAA,2BAAA;AAAA,WACtC,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAO,OAAA;AAAA,QACL,GAAK,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACvB,QAAU,EAAA,YAAA,CAAa,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QAClC,iBAAmB,EAAA,MAAA,CAAO,GAAI,CAAA,IAAA,CAAK,EAAE,CAAA;AAAA,QACrC,QAAQ,EAAC;AAAA,OACX,CAAA;AAAA,KACF;AAAA,IACA,aAAgB,GAAA;AACd,MAAA,MAAM,UAA6B,GAAA;AAAA,QACjC,WAAA;AAAA,QACA,iBAAiB,SAAU,CAAA;AAAA,UACzB,YAAc,EAAA;AAAA,YACZ,iBAAA,EAAmB,KAAK,OAAQ,CAAA,iBAAA;AAAA,WAClC;AAAA,UACA,QAAA,EAAU,KAAK,OAAQ,CAAA,GAAA;AAAA,UACvB,OAAO,OAAQ,CAAA,KAAA;AAAA,SAChB,CAAA;AAAA,QACD,oBAAoB,SAAU,CAAA;AAAA,UAC5B,QAAA,EAAU,KAAK,OAAQ,CAAA,QAAA;AAAA,SACxB,CAAA;AAAA,OACH,CAAA;AAEA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAA,UAAA,CAAW,KAAK,iBAAiB,CAAA,CAAA;AAAA,OACnC;AACA,MAAA,IAAI,QAAQ,QAAU,EAAA;AACpB,QAAW,UAAA,CAAA,IAAA;AAAA,UACT,iBAAiB,SAAU,CAAA;AAAA,YACzB,eAAiB,EAAA,iBAAA;AAAA,YACjB,eAAiB,EAAA,iBAAA;AAAA,WAClB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AACA,MAAA,IAAI,QAAQ,EAAI,EAAA;AACd,QAAA,MAAM,kBAAkB,OAAO;AAAA,UAC7B,MAAA;AAAA,UACA,aAAA;AAAA,UACA,OAAA;AAAA,UACA,MAAA;AAAA,SAC8C,KAAA;AAC9C,UAAA,MAAM,MAAS,GAAA,MAAM,IAAK,CAAA,SAAA,CAAA,CAAW,uBAAwB,CAAA;AAAA,YAC3D,MAAA;AAAA,YACA,aAAA;AAAA,YACA,OAAA;AAAA,YACA,MAAA;AAAA,WACD,CAAA,CAAA;AAED,UAAM,MAAA,cAAA,GAAiB,IAAK,CAAA,KAAA,CAAM,MAAM,CAAA,CAAA;AACxC,UAAA,MAAM,OACJ,OAAO,cAAA,CAAe,IAAS,KAAA,QAAA,GAAW,eAAe,IAAO,GAAA,EAAA,CAAA;AAClE,UACE,IAAA,OAAO,cAAe,CAAA,OAAA,KAAY,QAClC,IAAA,CAAC,QAAU,EAAA,cAAA,EAAgB,OAAO,CAAA,CAAE,QAAS,CAAA,IAAI,CACjD,EAAA;AACA,YAAO,OAAA,cAAA,CAAA;AAAA,WACT;AAEA,UAAM,MAAA,IAAI,MAAM,6BAA6B,CAAA,CAAA;AAAA,SAC/C,CAAA;AACA,QAAW,UAAA,CAAA,IAAA;AAAA,UACT,YAAY,SAAU,CAAA;AAAA,YACpB,eAAA;AAAA,YACA,GAAI,OAAO,OAAA,CAAQ,OAAO,SAAY,GAAA,KAAK,OAAQ,CAAA,EAAA;AAAA,YACnD,GAAA,EAAK,KAAK,OAAQ,CAAA,GAAA;AAAA,YAClB,GAAA,EAAK,KAAK,OAAQ,CAAA,iBAAA;AAAA,WACnB,CAAA;AAAA,SACH,CAAA;AAAA,OACF;AAEA,MAAO,OAAA,UAAA,CAAA;AAAA,KACT;AAAA,GACD,CAAA,CAAA;AACH;;;;"}

package/dist/ai/AiExtension.js

@@ -1,5 +1,6 @@
 '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');
@@ -10,6 +11,8 @@ 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;
 }
@@ -32,8 +35,6 @@ const AiExtension = core.Extension.create({
       name: this.options.name
     };
   },
-  onCreate() {
-  },
   addCommands() {
     return {
       askAi: (prompt) => () => {
@@ -148,20 +149,26 @@ const AiExtension = core.Extension.create({
           abortController
         };
         this.editor.setEditable(false);
-        const executeAiRequest = 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
-          });
-        };
-        executeAiRequest().then((output) => {
+        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;
           }
@@ -195,31 +202,35 @@ const AiExtension = core.Extension.create({
         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);
-        } else {
-          return true;
+          selection = {
+            from,
+            to: targetTo + output.text.length
+          };
         }
-        const { from, to } = this.editor.state.selection;
-        const contentTarget = this.editor.state.selection.empty || output.type === "insert" ? this.editor.state.selection.to : {
-          from,
-          to
-        };
-        const targetTo = output.type === "insert" ? to : from;
         this.storage.state = {
           phase: "reviewing",
           customPrompt: "",
           prompt: currentState.prompt,
           output,
-          contentTarget: { from, to: targetTo + output.text.length }
+          selection
         };
-        return this.editor.commands.insertContentAt(contentTarget, output.text);
+        return true;
       },
       _handleAiToolbarThinkingError: (error) => () => {
         const currentState = this.storage.state;
@@ -266,10 +277,10 @@ const AiExtension = core.Extension.create({
         key: types.AI_TOOLBAR_SELECTION_PLUGIN,
         props: {
           decorations: ({ doc, selection }) => {
-            if (this.storage.state.phase === "closed") {
+            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 } = selection;
+            const { from, to } = this.storage.state.selection ?? selection;
             const decorations = [
               view.Decoration.inline(from, to, {
                 class: "lb-root lb-selection lb-tiptap-active-selection"