Adopt to onDidChangeBodyVisibility listener

src/vs/workbench/browser/parts/views/customView.ts

@@ -62,11 +62,7 @@ export class CustomTreeViewPanel extends ViewletPanel {
 		this.treeView = treeView;
 		this.treeView.onDidChangeActions(() => this.updateActions(), this, this.disposables);
 		this.disposables.push(toDisposable(() => this.treeView.setVisibility(false)));
-		this.updateTreeVisibility();
-	}
-
-	setVisible(visible: boolean): void {
-		super.setVisible(visible);
+		this.onDidChangeBodyVisibility(() => this.updateTreeVisibility());
 		this.updateTreeVisibility();
 	}
 
@@ -79,11 +75,6 @@ export class CustomTreeViewPanel extends ViewletPanel {
 		this.treeView.show(container);
 	}
 
-	setExpanded(expanded: boolean): boolean {
-		this.treeView.setVisibility(this.isVisible() && expanded);
-		return super.setExpanded(expanded);
-	}
-
 	layoutBody(size: number): void {
 		this.treeView.layout(size);
 	}