Updating ChangesView to update tree icons accordingly

Author: StanleyGoldman

src/UnityExtension/Assets/Editor/GitHub.Unity/UI/ChangesView.cs

@@ -34,6 +34,7 @@ class ChangesView : Subview
         public override void OnEnable()
         {
             base.OnEnable();
+            UpdateTreeIcons();
             AttachHandlers(Repository);
 
             if (Repository != null)
@@ -190,18 +191,40 @@ private void BuildTree()
             if (treeChanges == null)
             {
                 treeChanges = new Tree();
-                treeChanges.NodeIcon = Styles.BranchIcon;
-                treeChanges.FolderIcon = Styles.FolderIcon;
                 treeChanges.DisplayRootNode = false;
                 treeChanges.Selectable = true;
                 treeChanges.PathIgnoreRoot = Environment.RepositoryPath + Environment.FileSystem.DirectorySeparatorChar;
                 treeChanges.PathSeparator = Environment.FileSystem.DirectorySeparatorChar.ToString();
+
+                UpdateTreeIcons();
             }
 
             treeChanges.Load(gitStatusEntries.Select(entry => new GitStatusEntryTreeData(entry)).Cast<ITreeData>(), "Changes");
             Redraw();
         }
 
+        private void UpdateTreeIcons()
+        {
+            var loaded = false;
+
+            if (treeChanges.NodeIcon == null)
+            {
+                loaded = true;
+                treeChanges.NodeIcon = Styles.BranchIcon;
+            }
+
+            if (treeChanges.FolderIcon == null)
+            {
+                loaded = true;
+                treeChanges.FolderIcon = Styles.FolderIcon;
+            }
+
+            if (loaded)
+            {
+                treeChanges.LoadNodeIcons();
+            }
+        }
+
         private void OnCommitDetailsAreaGUI()
         {
             GUILayout.BeginHorizontal();