Add GitHub Enterprise Server support

src/GitHubExtension/Client/GithubClientProvider.cs

@@ -82,9 +82,16 @@ public async Task<GitHubClient> GetClientForLoggedInDeveloper(bool logRateLimit
         }
 
         if (logRateLimit)
-        {
-            var miscRateLimit = await client.RateLimit.GetRateLimits();
-            Log.Logger()?.ReportInfo($"Rate Limit:  Remaining: {miscRateLimit.Resources.Core.Remaining}  Total: {miscRateLimit.Resources.Core.Limit}  Resets: {miscRateLimit.Resources.Core.Reset.ToStringInvariant()}");
+        {
+            try
+            {
+                var miscRateLimit = await client.RateLimit.GetRateLimits();
+                Log.Logger()?.ReportInfo($"Rate Limit:  Remaining: {miscRateLimit.Resources.Core.Remaining}  Total: {miscRateLimit.Resources.Core.Limit}  Resets: {miscRateLimit.Resources.Core.Reset.ToStringInvariant()}");
+            }
+            catch (Exception ex)
+            {
+                Log.Logger()?.ReportInfo($"Rate limiting not enabled for server: {ex.Message}");
+            }
         }
 
         return client;

src/GitHubExtension/Client/Validation.cs

@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation and Contributors
 // Licensed under the MIT license.
 using GitHubExtension.DataModel;
+using Octokit;
 
 namespace GitHubExtension.Client;
 
@@ -217,4 +218,15 @@ private static string AddProtocolToString(string s)
 
         return n;
     }
+
+    public static bool IsReachableGitHubEnterpriseServerURL(Uri server)
+    {
+        if (new EnterpriseProbe(new ProductHeaderValue(Constants.DEV_HOME_APPLICATION_NAME)).Probe(server).Result != EnterpriseProbeResult.Ok)
+        {
+            Log.Logger()?.ReportError($"EnterpriseServer {server.AbsoluteUri} is not reachable");
+            return false;
+        }
+
+        return true;
+    }
 }

src/GitHubExtension/Constants.cs

@@ -6,5 +6,6 @@ internal class Constants
 {
 #pragma warning disable SA1310 // Field names should not contain underscore
     public const string DEV_HOME_APPLICATION_NAME = "DevHome";
+    public const string GITHUB_COM_URL = "https://api.github.com/";
 #pragma warning restore SA1310 // Field names should not contain underscore
 }

src/GitHubExtension/DataManager/GitHubSearchManger.cs

@@ -3,8 +3,8 @@
 
 using GitHubExtension.Client;
 using GitHubExtension.DataManager;
-using GitHubExtension.DataModel;
-
+using GitHubExtension.DataModel;
+
 namespace GitHubExtension;
 
 public delegate void SearchManagerResultsAvailableEventHandler(IEnumerable<Octokit.Issue> results, string resultType);
@@ -31,7 +31,7 @@ public GitHubSearchManager()
             Environment.FailFast(e.Message, e);
             return null;
         }
-    }
+    }
 
     public async Task SearchForGitHubIssuesOrPRs(Octokit.SearchIssuesRequest request, string initiator, SearchCategory category, RequestOptions? options = null)
     {

src/GitHubExtension/DataManager/IGitHubSearchManager.cs

@@ -2,7 +2,7 @@
 // Licensed under the MIT license.
 
 using GitHubExtension.DataManager;
-
+
 namespace GitHubExtension;
 
 public interface IGitHubSearchManager : IDisposable

src/GitHubExtension/DataModel/DataObjects/Issue.cs

@@ -1,7 +1,6 @@
 // Copyright (c) Microsoft Corporation and Contributors
 // Licensed under the MIT license.
 
-using System.Globalization;
 using Dapper;
 using Dapper.Contrib.Extensions;
 using GitHubExtension.Helpers;

src/GitHubExtension/DeveloperId/CredentialVault.cs

@@ -2,28 +2,33 @@
 // Licensed under the MIT license.
 
 using System.ComponentModel;
-using System.Net;
 using System.Runtime.InteropServices;
 using System.Security;
-using Octokit;
 using Windows.Security.Credentials;
 using static GitHubExtension.DeveloperId.CredentialManager;
 
 namespace GitHubExtension.DeveloperId;
-internal static class CredentialVault
+public class CredentialVault : ICredentialVault
 {
+    private readonly string credentialResourceName;
+
     private static class CredentialVaultConfiguration
     {
         public const string CredResourceName = "GitHubDevHomeExtension";
     }
 
-    internal static void SaveAccessTokenToVault(string loginId, SecureString? accessToken)
+    public CredentialVault(string applicationName = "")
+    {
+        credentialResourceName = string.IsNullOrEmpty(applicationName) ? CredentialVaultConfiguration.CredResourceName : applicationName;
+    }
+
+    public void SaveCredentials(string loginId, SecureString? accessToken)
     {
         // Initialize a credential object.
         var credential = new CREDENTIAL
         {
             Type = CRED_TYPE.GENERIC,
-            TargetName = CredentialVaultConfiguration.CredResourceName + ": " + loginId,
+            TargetName = credentialResourceName + ": " + loginId,
             UserName = loginId,
             Persist = (int)CRED_PERSIST.LocalMachine,
             AttributeCount = 0,
@@ -61,29 +66,26 @@ internal static void SaveAccessTokenToVault(string loginId, SecureString? access
         }
     }
 
-    internal static void RemoveAccessTokenFromVault(string loginId)
+    public PasswordCredential? GetCredentials(string loginId)
     {
-        var targetCredentialToDelete = CredentialVaultConfiguration.CredResourceName + ": " + loginId;
-        var isCredentialDeleted = CredDelete(targetCredentialToDelete, CRED_TYPE.GENERIC, 0);
-        if (!isCredentialDeleted)
-        {
-            Log.Logger()?.ReportInfo($"Deleting credentials from Credential Manager has failed");
-            throw new Win32Exception(Marshal.GetLastWin32Error());
-        }
-    }
-
-    internal static PasswordCredential GetCredentialFromLocker(string loginId)
-    {
-        var credentialNameToRetrieve = CredentialVaultConfiguration.CredResourceName + ": " + loginId;
+        var credentialNameToRetrieve = credentialResourceName + ": " + loginId;
         var ptrToCredential = IntPtr.Zero;
 
         try
         {
             var isCredentialRetrieved = CredRead(credentialNameToRetrieve, CRED_TYPE.GENERIC, 0, out ptrToCredential);
             if (!isCredentialRetrieved)
             {
-                Log.Logger()?.ReportInfo($"Retrieving credentials from Credential Manager has failed");
-                throw new Win32Exception(Marshal.GetLastWin32Error());
+                var error = Marshal.GetLastWin32Error();
+                Log.Logger()?.ReportError($"Retrieving credentials from Credential Manager has failed for {loginId} with {error}");
+
+                // NotFound is expected and can be ignored.
+                if (error == 1168)
+                {
+                    return null;
+                }
+
+                throw new Win32Exception(error);
             }
 
             CREDENTIAL credentialObject;
@@ -96,27 +98,30 @@ internal static PasswordCredential GetCredentialFromLocker(string loginId)
             }
             else
             {
-                Log.Logger()?.ReportInfo("No credentials found for this DeveloperId");
-                throw new ArgumentOutOfRangeException(loginId);
+                Log.Logger()?.ReportError($"No credentials found for this DeveloperId : {loginId}");
+                return null;
             }
 
             var accessTokenInChars = new char[credentialObject.CredentialBlobSize / 2];
             Marshal.Copy(credentialObject.CredentialBlob, accessTokenInChars, 0, accessTokenInChars.Length);
 
-            var accessToken = new SecureString();
+            // convert accessTokenInChars to string
+            string accessTokenString = new (accessTokenInChars);
+
             for (var i = 0; i < accessTokenInChars.Length; i++)
             {
-                accessToken.AppendChar(accessTokenInChars[i]);
-
                 // Zero out characters after they are copied over from an unmanaged to managed type.
                 accessTokenInChars[i] = '\0';
             }
 
-            accessToken.MakeReadOnly();
-
-            var credential = new PasswordCredential(CredentialVaultConfiguration.CredResourceName, loginId, new NetworkCredential(string.Empty, accessToken).Password);
+            var credential = new PasswordCredential(credentialResourceName, loginId, accessTokenString);
             return credential;
         }
+        catch (Exception ex)
+        {
+            Log.Logger()?.ReportError($"Retrieving credentials from Credential Manager has failed unexpectedly: {loginId} : {ex.Message}");
+            throw;
+        }
         finally
         {
             if (ptrToCredential != IntPtr.Zero)
@@ -126,7 +131,18 @@ internal static PasswordCredential GetCredentialFromLocker(string loginId)
         }
     }
 
-    public static IEnumerable<string> GetAllSavedLoginIds()
+    public void RemoveCredentials(string loginId)
+    {
+        var targetCredentialToDelete = credentialResourceName + ": " + loginId;
+        var isCredentialDeleted = CredDelete(targetCredentialToDelete, CRED_TYPE.GENERIC, 0);
+        if (!isCredentialDeleted)
+        {
+            Log.Logger()?.ReportError($"Deleting credentials from Credential Manager has failed for {loginId}");
+            throw new Win32Exception(Marshal.GetLastWin32Error());
+        }
+    }
+
+    public IEnumerable<string> GetAllCredentials()
     {
         var ptrToCredential = IntPtr.Zero;
 
@@ -135,7 +151,7 @@ public static IEnumerable<string> GetAllSavedLoginIds()
             IntPtr[] allCredentials;
             uint count;
 
-            if (CredEnumerate(CredentialVaultConfiguration.CredResourceName + "*", 0, out count, out ptrToCredential) != false)
+            if (CredEnumerate(credentialResourceName + "*", 0, out count, out ptrToCredential) != false)
             {
                 allCredentials = new IntPtr[count];
                 Marshal.Copy(ptrToCredential, allCredentials, 0, (int)count);
@@ -179,4 +195,20 @@ public static IEnumerable<string> GetAllSavedLoginIds()
             }
         }
     }
+
+    public void RemoveAllCredentials()
+    {
+        var allCredentials = GetAllCredentials();
+        foreach (var credential in allCredentials)
+        {
+            try
+            {
+                RemoveCredentials(credential);
+            }
+            catch (Exception ex)
+            {
+                Log.Logger()?.ReportError($"Deleting credentials from Credential Manager has failed unexpectedly: {credential} : {ex.Message}");
+            }
+        }
+    }
 }

src/GitHubExtension/DeveloperId/DeveloperId.cs

@@ -55,15 +55,16 @@ public Windows.Security.Credentials.PasswordCredential GetCredential(bool refres
             return RefreshDeveloperId();
         }
 
-        return CredentialVault.GetCredentialFromLocker(LoginId);
+        var credential = DeveloperIdProvider.GetInstance().GetCredentials(this) ?? throw new InvalidOperationException("Invalid credential present for valid DeveloperId");
+        return credential;
     }
 
     public Windows.Security.Credentials.PasswordCredential RefreshDeveloperId()
     {
         // Setting to MaxValue, since GitHub doesn't forcibly expire tokens currently.
         CredentialExpiryTime = DateTime.MaxValue;
         DeveloperIdProvider.GetInstance().RefreshDeveloperId(this);
-        var credential = CredentialVault.GetCredentialFromLocker(LoginId);
+        var credential = DeveloperIdProvider.GetInstance().GetCredentials(this) ?? throw new InvalidOperationException("Invalid credential present for valid DeveloperId");
         GitHubClient.Credentials = new (credential.Password);
         return credential;
     }