fix deco positions

wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentBubbleComponent.js

 /* eslint react/prop-types: 0 */
 import React, { useLayoutEffect, useContext } from 'react';
 import { WaxContext } from 'wax-prosemirror-core';
+import {
+  ySyncPluginKey,
+  relativePositionToAbsolutePosition,
+  absolutePositionToRelativePosition,
+} from 'y-prosemirror';
 
 import CommentBubble from './CommentBubble';
 import { CommentDecorationPluginKey } from '../../../plugins/CommentDecorationPlugin';
@@ -31,6 +36,10 @@ const CommentBubbleComponent = ({ setPosition, position, group }) => {
     event.preventDefault();
     const { selection } = state;
 
+    if (context.app.config.get('config.YjsService')) {
+      return createYjsComments(selection);
+    }
+
     dispatch(
       state.tr.setMeta(CommentDecorationPluginKey, {
         type: 'addComment',
@@ -52,6 +61,47 @@ const CommentBubbleComponent = ({ setPosition, position, group }) => {
     dispatch(state.tr);
   };
 
+  const createYjsComments = selection => {
+    const ystate = ySyncPluginKey.getState(state);
+    const { doc, type, binding } = ystate;
+    const from = absolutePositionToRelativePosition(
+      selection.from,
+      type,
+      binding.mapping,
+    );
+    const to = absolutePositionToRelativePosition(
+      selection.to,
+      type,
+      binding.mapping,
+    );
+
+    dispatch(
+      state.tr.setMeta(CommentDecorationPluginKey, {
+        type: 'addComment',
+        from: relativePositionToAbsolutePosition(
+          doc,
+          type,
+          from,
+          binding.mapping,
+        ),
+        to: relativePositionToAbsolutePosition(doc, type, to, binding.mapping),
+        data: {
+          yjsFrom: selection.from,
+          yjsTo: selection.to,
+          pmFrom: selection.from,
+          pmTo: selection.to,
+          type: 'comment',
+          conversation: [],
+          title: '',
+          group,
+          viewId: activeViewId,
+        },
+      }),
+    );
+
+    dispatch(state.tr);
+  };
+
   const isCommentAllowed = () => {
     let allowed = true;
     state.doc.nodesBetween(

wax-prosemirror-services/src/CommentsService/plugins/CommentState.js

@@ -208,15 +208,10 @@ export default class CommentState {
 
     if (ystate?.binding && ystate?.binding.mapping) {
       this.updateCommentPostions(ystate);
+      this.createDecorations(state);
       return this;
       // eslint-disable-next-line no-else-return
     } else {
-      map.forEach((annotation, _) => {
-        if ('from' in annotation && 'to' in annotation) {
-          annotation.from = transaction.mapping.map(annotation.from);
-          annotation.to = transaction.mapping.map(annotation.to);
-        }
-      });
       this.createDecorations(state);
       return this;
     }