From e96136b8bade393d336a74c607db18d19d139e96 Mon Sep 17 00:00:00 2001
From: Fawad Ali <m.fawaadali98@gmail.com>
Date: Mon, 15 Feb 2021 23:12:48 +0500
Subject: [PATCH] fix(event-display): persist toggle state of phoenix menu node
 children

---
 .../src/ui/phoenix-menu/phoenix-menu-node.ts      | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/packages/phoenix-event-display/src/ui/phoenix-menu/phoenix-menu-node.ts b/packages/phoenix-event-display/src/ui/phoenix-menu/phoenix-menu-node.ts
index 422c22973..d190335df 100644
--- a/packages/phoenix-event-display/src/ui/phoenix-menu/phoenix-menu-node.ts
+++ b/packages/phoenix-event-display/src/ui/phoenix-menu/phoenix-menu-node.ts
@@ -23,6 +23,11 @@ export class PhoenixMenuNode {
   nodeLevel: number = 0;
   /** Parent of the node. */
   private parent: PhoenixMenuNode;
+  /**
+   * Previous toggle state of child nodes. This is so that the
+   * previous state of child can be restored if we toggle the parent back on.
+   * */
+  private childrenToggleState = {};
 
   /** If the node children are active or not. */
   childrenActive: boolean = false;
@@ -124,7 +129,15 @@ export class PhoenixMenuNode {
     this.onToggle?.(value);
     this.toggleState = value;
     for (const child of this.children) {
-      child.toggleSelfAndDescendants(value);
+      if (!value) {
+        // Save previous toggle state of children and toggle them false
+        this.childrenToggleState[child.name] = child.toggleState;
+        child.toggleSelfAndDescendants(value);
+      } else {
+        // Restore previous toggle state of children
+        child.toggleState = this.childrenToggleState[child.name];
+        child.toggleSelfAndDescendants(child.toggleState);
+      }
     }
   }