Custom treeview control and cache manager

Author: shana

src/GitHub.Api/Application/ApplicationManagerBase.cs

@@ -12,6 +12,7 @@ abstract class ApplicationManagerBase : IApplicationManager
         protected static ILogging Logger { get; } = Logging.GetLogger<IApplicationManager>();
 
         private RepositoryManager repositoryManager;
+        private IBranchCache branchCache;
 
         public ApplicationManagerBase(SynchronizationContext synchronizationContext)
         {
@@ -21,6 +22,7 @@ public ApplicationManagerBase(SynchronizationContext synchronizationContext)
             UIScheduler = TaskScheduler.FromCurrentSynchronizationContext();
             ThreadingHelper.MainThreadScheduler = UIScheduler;
             TaskManager = new TaskManager(UIScheduler);
+            CacheManager = new CacheManager();
         }
 
         protected void Initialize()
@@ -80,6 +82,11 @@ private async Task SetupGit()
 
         }
 
+        public void SetupCache(IBranchCache bcache)
+        {
+            branchCache = bcache;
+        }
+
         public ITask InitializeRepository()
         {
             Logger.Trace("Running Repository Initialize");
@@ -214,6 +221,7 @@ public void Dispose()
         public ISettings LocalSettings { get; protected set; }
         public ISettings SystemSettings { get; protected set; }
         public ISettings UserSettings { get; protected set; }
+        public CacheManager CacheManager { get; private set; }
         public IUsageTracker UsageTracker { get; protected set; }
 
         protected TaskScheduler UIScheduler { get; private set; }

src/GitHub.Api/Application/IApplicationManager.cs

@@ -15,6 +15,7 @@ public interface IApplicationManager : IDisposable
         ISettings LocalSettings { get; }
         ISettings UserSettings { get; }
         ITaskManager TaskManager { get; }
+        CacheManager CacheManager { get; }
         IGitClient GitClient { get; }
         IUsageTracker UsageTracker { get; }
 

src/GitHub.Api/Cache/CacheManager.cs

@@ -0,0 +1,46 @@
+using System;
+using System.Linq;
+
+namespace GitHub.Unity
+{
+    public class CacheManager
+    {
+        private IBranchCache branchCache;
+        public IBranchCache BranchCache
+        {
+            get { return branchCache; }
+            set
+            {
+                if (branchCache == null)
+                    branchCache = value;
+            }
+        }
+
+        private Action onLocalBranchListChanged;
+
+        public void SetupCache(IBranchCache branchCache, IRepository repository)
+        {
+            if (repository == null)
+                return;
+
+            BranchCache = branchCache;
+            UpdateCache(repository);
+            if (onLocalBranchListChanged != null)
+                repository.OnLocalBranchListChanged -= onLocalBranchListChanged;
+            onLocalBranchListChanged = () =>
+            {
+                if (!ThreadingHelper.InUIThread)
+                    new ActionTask(TaskManager.Instance.Token, () => UpdateCache(repository)) { Affinity = TaskAffinity.UI }.Start();
+                else
+                    UpdateCache(repository);
+            };
+            repository.OnLocalBranchListChanged += onLocalBranchListChanged;
+        }
+
+        private void UpdateCache(IRepository repository)
+        {
+            BranchCache.LocalBranches = repository.LocalBranches.ToList();
+            BranchCache.RemoteBranches = repository.RemoteBranches.ToList();
+        }
+    }
+}

src/GitHub.Api/Cache/IBranchCache.cs

@@ -2,7 +2,7 @@
 
 namespace GitHub.Unity
 {
-    interface IBranchCache
+    public interface IBranchCache
     {
         List<GitBranch> LocalBranches { get; set; }
         List<GitBranch> RemoteBranches { get; set; }

src/GitHub.Api/GitHub.Api.csproj

@@ -110,6 +110,7 @@
     <Compile Include="Authentication\ILoginManager.cs" />
     <Compile Include="Application\ApplicationManagerBase.cs" />
     <Compile Include="Helpers\Constants.cs" />
+    <Compile Include="Cache\CacheManager.cs" />
     <Compile Include="Cache\IBranchCache.cs" />
     <Compile Include="Helpers\Validation.cs" />
     <Compile Include="OutputProcessors\LfsVersionOutputProcessor.cs" />

src/UnityExtension/Assets/Editor/GitHub.Unity/IconsAndLogos/globe.png

@@ -27,13 +27,14 @@ class Styles
                            MinCommitTreePadding = 20f,
                            FoldoutWidth = 11f,
                            FoldoutIndentation = -2f,
+                           TreePadding = 12f,
                            TreeIndentation = 12f,
                            TreeRootIndentation = -5f,
                            TreeVerticalSpacing = 3f,
                            CommitIconSize = 16f,
                            CommitIconHorizontalPadding = -5f,
                            BranchListIndentation = 20f,
-                           BranchListSeperation = 15f,
+                           BranchListSeparation = 15f,
                            RemotesTotalHorizontalMargin = 37f,
                            RemotesNameRatio = .2f,
                            RemotesUserRatio = .2f,
@@ -195,7 +196,7 @@ public static GUIStyle Label
                     label = new GUIStyle(GUI.skin.label);
                     label.name = "CustomLabel";
 
-                    var hierarchyStyle = GUI.skin.FindStyle("PR Label");
+                    GUIStyle hierarchyStyle = GUI.skin.FindStyle("PR Label");
                     label.onNormal.background = hierarchyStyle.onNormal.background;
                     label.onNormal.textColor = hierarchyStyle.onNormal.textColor;
                     label.onFocused.background = hierarchyStyle.onFocused.background;
@@ -798,5 +799,79 @@ public static Texture2D DropdownListIcon
                 return dropdownListIcon;
             }
         }
+
+        private static Texture2D rootFolderIcon;
+        public static Texture2D RootFolderIcon
+        {
+            get
+            {
+                if (rootFolderIcon == null)
+                {
+                    rootFolderIcon = Utility.GetIcon("globe.png", "[email protected]");
+                }
+                return rootFolderIcon;
+            }
+        }
+
+        private static GUIStyle foldout;
+        public static GUIStyle Foldout
+        {
+            get
+            {
+                if (foldout == null)
+                {
+                    foldout = new GUIStyle(EditorStyles.foldout);
+                    foldout.name = "CustomFoldout";
+
+                    foldout.focused.textColor = Color.white;
+                    foldout.onFocused.textColor = Color.white;
+                    foldout.focused.background = foldout.active.background;
+                    foldout.onFocused.background = foldout.onActive.background;
+                }
+
+                return foldout;
+            }
+        }
+
+        private static GUIStyle treeNode;
+        public static GUIStyle TreeNode
+        {
+            get
+            {
+                if (treeNode == null)
+                {
+                    treeNode = new GUIStyle(GUI.skin.label);
+                    treeNode.name = "Custom TreeNode";
+
+                    var color = new Color(62f / 255f, 125f / 255f, 231f / 255f);
+                    var texture = Utility.GetTextureFromColor(color);
+                    treeNode.focused.background = texture;
+                    treeNode.onFocused.background = texture;
+                    treeNode.focused.textColor = Color.white;
+                    treeNode.onFocused.textColor = Color.white;
+                }
+
+                return treeNode;
+            }
+        }
+
+        private static GUIStyle treeNodeActive;
+        public static GUIStyle TreeNodeActive
+        {
+            get
+            {
+                if (treeNodeActive == null)
+                {
+                    treeNodeActive = new GUIStyle(TreeNode);
+                    treeNodeActive.name = "Custom TreeNode Active";
+                    treeNodeActive.fontStyle = FontStyle.Bold;
+                    treeNodeActive.focused.textColor = Color.white;
+                    treeNodeActive.active.textColor = Color.white;
+                }
+
+                return treeNodeActive;
+            }
+        }
+
     }
 }

src/UnityExtension/Assets/Editor/GitHub.Unity/IconsAndLogos/[email protected]

@@ -23,6 +23,18 @@ public static Texture2D GetIcon(string filename, string filename2x = "")
             var iconPath = EntryPoint.Environment.ExtensionInstallPath.Combine("IconsAndLogos", filename).ToString(SlashMode.Forward);
             return AssetDatabase.LoadAssetAtPath<Texture2D>(iconPath);
         }
+
+        public static Texture2D GetTextureFromColor(Color color)
+        {
+            Color[] pix = new Color[1];
+            pix[0] = color;
+
+            Texture2D result = new Texture2D(1, 1);
+            result.SetPixels(pix);
+            result.Apply();
+
+            return result;
+        }
     }
 
     static class StreamExtensions

src/UnityExtension/Assets/Editor/GitHub.Unity/Misc/Styles.cs

@@ -1,5 +1,4 @@
 using System;
-using GitHub.Unity;
 
 namespace GitHub.Unity
 {