From 2d0b2b027c41be757f5a073abc2fd2e6ec131a4d Mon Sep 17 00:00:00 2001
From: chris <kokosias@yahoo.gr>
Date: Wed, 22 Feb 2017 15:43:23 +0200
Subject: [PATCH] track toggle view working

---
 .../elements/track_change/TrackChangeComponent.js        | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js b/app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js
index 9fe65a5..4239d21 100644
--- a/app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js
+++ b/app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js
@@ -73,8 +73,8 @@ class TrackChangeComponent extends AnnotationComponent {
     provider.resolve(annotation, action)
   }
 
-  getController () {
-    return this.context.controller
+  getEditor () {
+    return this.context.editor
   }
 
   getProvider () {
@@ -82,9 +82,8 @@ class TrackChangeComponent extends AnnotationComponent {
   }
 
   getViewMode () {
-    const controller = this.getController()
-    // const { trackChangesView } = controller.state
-    const trackChangesView = true
+    const editor = this.getEditor()
+    const { trackChangesView } = editor.state
     return trackChangesView
   }
 
-- 
GitLab