Adding the currentUsername check to ProjectWindowInterface

Author: StanleyGoldman

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

@@ -18,14 +18,15 @@ class ProjectWindowInterface : AssetPostprocessor
         private static List<GitLock> locks = new List<GitLock>();
         private static List<string> guids = new List<string>();
         private static List<string> guidsLocks = new List<string>();
-        private static string loggedInUser;
+        private static string currentUsername;
 
         private static IApplicationManager manager;
         private static bool isBusy = false;
         private static ILogging logger;
         private static ILogging Logger { get { return logger = logger ?? LogHelper.GetLogger<ProjectWindowInterface>(); } }
         private static CacheUpdateEvent lastRepositoryStatusChangedEvent;
         private static CacheUpdateEvent lastLocksChangedEvent;
+        private static CacheUpdateEvent lastCurrentRemoteChangedEvent;
         private static IRepository Repository { get { return manager != null ? manager.Environment.Repository : null; } }
         private static IPlatform Platform { get { return manager != null ? manager.Platform : null; } }
         private static bool IsInitialized { get { return Repository != null; } }
@@ -37,13 +38,14 @@ public static void Initialize(IApplicationManager theManager)
 
             manager = theManager;
 
-            Platform.Keychain.ConnectionsChanged += KeychainMayHaveChanged;
-            KeychainMayHaveChanged();
+            Platform.Keychain.ConnectionsChanged += UpdateCurrentUsername;
+            UpdateCurrentUsername();
 
             if (IsInitialized)
             {
                 Repository.StatusEntriesChanged += RepositoryOnStatusEntriesChanged;
                 Repository.LocksChanged += RepositoryOnLocksChanged;
+                Repository.CurrentRemoteChanged += RepositoryOnCurrentRemoteChanged;
                 ValidateCachedData();
             }
         }
@@ -88,9 +90,39 @@ private static void RepositoryOnLocksChanged(CacheUpdateEvent cacheUpdateEvent)
             }
         }
 
-        private static void KeychainMayHaveChanged()
+        private static void RepositoryOnCurrentRemoteChanged(CacheUpdateEvent cacheUpdateEvent)
         {
-            loggedInUser = Platform.Keychain.Connections.Select(x => x.Username).FirstOrDefault();
+            if (!lastCurrentRemoteChangedEvent.Equals(cacheUpdateEvent))
+            {
+                lastCurrentRemoteChangedEvent = cacheUpdateEvent;
+            }
+        }
+
+        private static void UpdateCurrentUsername()
+        {
+            var username = String.Empty;
+            if (Repository != null)
+            {
+                Connection[] connections;
+                if (!string.IsNullOrEmpty(Repository.CloneUrl))
+                {
+                    var host = Repository.CloneUrl.ToRepositoryUri()
+                                         .GetComponents(UriComponents.Host, UriFormat.SafeUnescaped);
+
+                    connections = Platform.Keychain.Connections.OrderByDescending(x => x.Host == host).ToArray();
+                }
+                else
+                {
+                    connections = Platform.Keychain.Connections.OrderByDescending(HostAddress.IsGitHubDotCom).ToArray();
+                }
+
+                if (connections.Any())
+                {
+                    username = connections.First().Username;
+                }
+            }
+
+            currentUsername = username;
         }
 
         [MenuItem(AssetsMenuRequestLock, true, 10000)]
@@ -202,7 +234,7 @@ private static bool IsObjectLocked(Object selected, bool isLockedByCurrentUser)
             NPath assetPath = AssetDatabase.GetAssetPath(selected.GetInstanceID()).ToNPath();
             NPath repositoryPath = manager.Environment.GetRepositoryPath(assetPath);
 
-            return locks.Any(x => repositoryPath == x.Path && (!isLockedByCurrentUser || x.Owner.Name == loggedInUser));
+            return locks.Any(x => repositoryPath == x.Path && (!isLockedByCurrentUser || x.Owner.Name == currentUsername));
         }
 
         private static ITask CreateUnlockObjectTask(Object selected, bool force)