fix(VisibilityToolbar): Apply visibility when viewer changes

CASL · May 11, 2018 · 9547d02 · 9547d02
1 parent f235d8f
commit 9547d02
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/widgets/VisibilityToolbar.js b/src/widgets/VisibilityToolbar.js
@@ -11,17 +11,26 @@ export default class VisibilityToolbar extends React.Component {
     };
 
     this.containerEl = null;
+    this.subscribes = [];
 
     this.toggleDropdown = this.toggleDropdown.bind(this);
     this.onDocMouseDown = this.onDocMouseDown.bind(this);
   }
 
   componentDidMount() {
     document.addEventListener('mousedown', this.onDocMouseDown, true);
+    this.subscribeToViewer(this.props.viewer);
+  }
+
+  componentDidUpdate(prevProps) {
+    if (prevProps.viewer !== this.props.viewer) {
+      this.subscribeToViewer(this.props.viewer);
+    }
   }
 
   componentWillUnmount() {
     document.removeEventListener('mousedown', this.onDocMouseDown, true);
+    this.subscribeToViewer(null);
   }
 
   onDocMouseDown(ev) {
@@ -38,6 +47,15 @@ export default class VisibilityToolbar extends React.Component {
     }
   }
 
+  subscribeToViewer(viewer) {
+    while (this.subscribes.length) {
+      this.subscribes.pop().unsubscribe();
+    }
+    if (viewer) {
+      this.subscribes = [viewer.onModified(viewer.applyVisibility)];
+    }
+  }
+
   toggleDropdown(ev) {
     this.setState(({ dropdownVisible }) => ({
       dropdownVisible: !dropdownVisible,