update comments coposition when writing or toggling track changes view mode

app/components/SimpleEditor/ContainerEditor.js

@@ -46,25 +46,24 @@ class ContainerEditor extends SubstanceContainerEditor {
   }
 
   didMount () {
-    // TODO -- replace with super.didMount()
-    Surface.prototype.didMount.apply(this, arguments)
-
-    this.container.on('nodes:changed', this.onContainerChange, this)
-
-    // this.context.documentSession.on('didUpdate', (change, info) => {
-    //   console.log('\n did update')
-    //   console.log(this.getCommandStates())
-    // }, this)
+    super.didMount(this, arguments)
 
     if (this.isEmpty()) this.createText()
     this.focus()
 
+    const documentSession = this.getDocumentSession()
+
     if (this.isReadOnlyMode()) {
-      const documentSession = this.getDocumentSession()
       documentSession.on('didUpdate', this.disableToolbar, this)
 
       this.addTargetToLinks()
     }
+
+    documentSession.on('didUpdate', this.setCommentstops, this)
+  }
+
+  setCommentstops () {
+    this.context.commentsProvider.update()
   }
 
   onTextInput (event) {

app/components/SimpleEditor/Editor.js

@@ -42,6 +42,10 @@ class Editor extends ProseEditor {
   updateTrackChangeView () {
     this.extendState({ trackChangesView: !this.state.trackChangesView })
     // this.getChildContext().commentsProvider.update()
+    const surface = this.surfaceManager.getSurface(this.props.containerId)
+    setTimeout(function () {
+      surface.setCommentstops()
+    }, 250)
   }
 
   willUpdateState () {}