From 9157bf460aa8e1afd13c25f46c159090484e6a4b Mon Sep 17 00:00:00 2001
From: Trae Yelovich <trae.yelovich@broadcom.com>
Date: Wed, 27 Sep 2023 10:45:16 -0400
Subject: [PATCH] feat: update USS directory icon once collapsed

Signed-off-by: Trae Yelovich <trae.yelovich@broadcom.com>
---
 packages/zowe-explorer/src/uss/USSTree.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/packages/zowe-explorer/src/uss/USSTree.ts b/packages/zowe-explorer/src/uss/USSTree.ts
index c53a3b3fd3..ed2d116179 100644
--- a/packages/zowe-explorer/src/uss/USSTree.ts
+++ b/packages/zowe-explorer/src/uss/USSTree.ts
@@ -80,6 +80,15 @@ export class USSTree extends ZoweTreeProvider implements IZoweTree<IZoweUSSTreeN
             treeDataProvider: this,
             canSelectMany: true,
         });
+        this.treeView.onDidCollapseElement((e) => {
+            if (contextually.isUssDirectory(e.element)) {
+                const newIcon = getIconByNode(e.element);
+                if (newIcon) {
+                    e.element.iconPath = newIcon;
+                    this.mOnDidChangeTreeData.fire(e.element);
+                }
+            }
+        })
     }
 
     /**