Integrate AzArtifacts Credential Provider

src/code/CredentialProvider.cs

@@ -160,13 +160,8 @@ internal static PSCredential GetCredentialsFromProvider(Uri uri, PSCmdlet cmdlet
                 if (Environment.OSVersion.Platform == PlatformID.Unix)
                 {
                     FileInfo fileInfo = new FileInfo(credProviderPath);
-                    string resolvedFilePath = Utils.GetCaseInsensitiveFilePath(fileInfo.Directory.FullName, _credProviderDll);
-                    if (resolvedFilePath != null)
-                    {
-                        credProviderPath = resolvedFilePath;
-                    }
-                    else
-                    {
+                    if (!Utils.TryGetCaseInsensitiveFilePath(fileInfo.Directory.FullName, _credProviderDll, out credProviderPath))
+                    { 
                         cmdletPassedIn.WriteError(new ErrorRecord(
                             new FileNotFoundException($"Path found '{credProviderPath}' is not a valid Azure Artifact Credential Provider executable. See https://github.com/NuGet/Home/wiki/NuGet-cross-plat-authentication-plugin#plugin-installation-and-discovery to set up the Credential Provider."),
                             "CredentialProviderFileNotFound",

src/code/FindHelper.cs

@@ -204,13 +204,7 @@ public IEnumerable<PSResourceInfo> FindByResourceName(
                 repositoryNamesToSearch.Add(currentRepository.Name);
 
                 // Set network credentials via passed in credentials, AzArtifacts CredentialProvider, or SecretManagement.
-                if (currentRepository.CredentialProvider.Equals(PSRepositoryInfo.CredentialProviderType.AzArtifacts))
-                { 
-                    _networkCredential = Utils.SetCredentialProviderNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
-                else {
-                    _networkCredential = Utils.SetSecretManagementNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
+                _networkCredential = Utils.SetNetworkCredentials(currentRepository, _networkCredential, _cmdletPassedIn);
 
                 ServerApiCall currentServer = ServerFactory.GetServer(currentRepository, _cmdletPassedIn, _networkCredential);
                 if (currentServer == null)
@@ -398,14 +392,7 @@ public IEnumerable<PSCommandResourceInfo> FindByCommandOrDscResource(
                 repositoryNamesToSearch.Add(currentRepository.Name);
 
                 // Set network credentials via passed in credentials, AzArtifacts CredentialProvider, or SecretManagement.
-                if (currentRepository.CredentialProvider.Equals(PSRepositoryInfo.CredentialProviderType.AzArtifacts))
-                {
-                    _networkCredential = Utils.SetCredentialProviderNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
-                else
-                {
-                    _networkCredential = Utils.SetSecretManagementNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
+                _networkCredential = Utils.SetNetworkCredentials(currentRepository, _networkCredential, _cmdletPassedIn);
 
                 ServerApiCall currentServer = ServerFactory.GetServer(currentRepository, _cmdletPassedIn, _networkCredential);
                 if (currentServer == null)
@@ -612,14 +599,7 @@ public IEnumerable<PSResourceInfo> FindByTag(
                 repositoryNamesToSearch.Add(currentRepository.Name);
 
                 // Set network credentials via passed in credentials, AzArtifacts CredentialProvider, or SecretManagement.
-                if (currentRepository.CredentialProvider.Equals(PSRepositoryInfo.CredentialProviderType.AzArtifacts))
-                {
-                    _networkCredential = Utils.SetCredentialProviderNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
-                else
-                {
-                    _networkCredential = Utils.SetSecretManagementNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
+                _networkCredential = Utils.SetNetworkCredentials(currentRepository, _networkCredential, _cmdletPassedIn);
 
                 ServerApiCall currentServer = ServerFactory.GetServer(currentRepository, _cmdletPassedIn, _networkCredential);
                 if (currentServer == null)

src/code/InstallHelper.cs

@@ -287,14 +287,7 @@ private List<PSResourceInfo> ProcessRepositories(
                 sourceTrusted = currentRepository.Trusted || trustRepository;
 
                 // Set network credentials via passed in credentials, AzArtifacts CredentialProvider, or SecretManagement.
-                if (currentRepository.CredentialProvider.Equals(PSRepositoryInfo.CredentialProviderType.AzArtifacts))
-                {
-                    _networkCredential = Utils.SetCredentialProviderNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
-                else
-                {
-                    _networkCredential = Utils.SetSecretManagementNetworkCredential(currentRepository, _networkCredential, _cmdletPassedIn);
-                }
+                _networkCredential = Utils.SetNetworkCredentials(currentRepository, _networkCredential, _cmdletPassedIn);
 
                 ServerApiCall currentServer = ServerFactory.GetServer(currentRepository, _cmdletPassedIn, _networkCredential);
 

src/code/PublishHelper.cs

@@ -381,14 +381,7 @@ internal void PushResource(string Repository, string modulePrefix, bool SkipDepe
                 }
 
                 // Set network credentials via passed in credentials, AzArtifacts CredentialProvider, or SecretManagement.
-                if (repository.CredentialProvider.Equals(PSRepositoryInfo.CredentialProviderType.AzArtifacts))
-                {
-                    _networkCredential = Utils.SetCredentialProviderNetworkCredential(repository, _networkCredential, _cmdletPassedIn);
-                }
-                else
-                {
-                    _networkCredential = Utils.SetSecretManagementNetworkCredential(repository, _networkCredential, _cmdletPassedIn);
-                }
+                _networkCredential = Utils.SetNetworkCredentials(repository, _networkCredential, _cmdletPassedIn);
 
                 // Check if dependencies already exist within the repo if:
                 // 1) the resource to publish has dependencies and

src/code/RegisterPSResourceRepository.cs

@@ -121,13 +121,12 @@ public object GetDynamicParameters()
             // It should also not appear when using the 'Repositories' parameter set.
             if (ParameterSetName.Equals(PSGalleryParameterSet) || 
                 ParameterSetName.Equals(RepositoriesParameterSet) ||
-                Uri.EndsWith(".azurecr.io") || Uri.EndsWith(".azurecr.io/") || Uri.Contains("mcr.microsoft.com"))
+                Utils.IsContainerRegistry(Uri))
             {
                 return null;
             }
 
-            _credentialProvider = new CredentialProviderDynamicParameters();
-            return _credentialProvider;
+            return new CredentialProviderDynamicParameters();
         }
 
         #endregion

src/code/RepositorySettings.cs

@@ -901,7 +901,7 @@ private static APIVersion GetRepoAPIVersion(Uri repoUri)
                 // repositories with Uri Scheme "temp" may have PSPath Uri's like: "Temp:\repo" and we should consider them as local repositories.
                 return APIVersion.Local;
             }
-            else if (repoUri.AbsoluteUri.EndsWith(".azurecr.io") || repoUri.AbsoluteUri.EndsWith(".azurecr.io/") || repoUri.AbsoluteUri.Contains("mcr.microsoft.com"))
+            else if (Utils.IsContainerRegistry(repoUri.AbsoluteUri))
             {
                 return APIVersion.ContainerRegistry;
             }

src/code/SetPSResourceRepository.cs

@@ -114,13 +114,12 @@ public object GetDynamicParameters()
             if (repository is not null && 
                 (repository.Name.Equals("PSGallery", StringComparison.OrdinalIgnoreCase) ||
                 ParameterSetName.Equals(RepositoriesParameterSet) ||
-                repository.Uri.AbsoluteUri.EndsWith(".azurecr.io") || repository.Uri.AbsoluteUri.EndsWith(".azurecr.io/") || repository.Uri.AbsoluteUri.Contains("mcr.microsoft.com")))
+                Utils.IsContainerRegistry(repository.Uri.AbsoluteUri)))
             {
                 return null;
             }
 
-            _credentialProvider = new CredentialProviderDynamicParameters();
-            return _credentialProvider;
+            return new CredentialProviderDynamicParameters();
         }
 
         #endregion

src/code/Utils.cs

@@ -968,6 +968,10 @@ public static NetworkCredential SetSecretManagementNetworkCredential(
             return networkCredential;
         }
 
+        #endregion
+
+        #region Credential methods
+
         public static NetworkCredential SetCredentialProviderNetworkCredential(
             PSRepositoryInfo repository,
             NetworkCredential networkCredential,
@@ -992,6 +996,37 @@ public static NetworkCredential SetCredentialProviderNetworkCredential(
             return networkCredential;
         }
 
+        public static NetworkCredential SetNetworkCredentials(PSRepositoryInfo repository, NetworkCredential networkCredential, PSCmdlet cmdletPassedIn)
+        {
+            NetworkCredential networkCreds = new NetworkCredential();
+            if (repository.CredentialProvider.Equals(PSRepositoryInfo.CredentialProviderType.AzArtifacts))
+            {
+                cmdletPassedIn.WriteVerbose("Setting credential provider network credentials");
+                networkCreds = Utils.SetCredentialProviderNetworkCredential(repository, networkCredential, cmdletPassedIn);
+            }
+            else
+            {
+                cmdletPassedIn.WriteVerbose("Setting Secret Management network credentials");
+                networkCreds = Utils.SetSecretManagementNetworkCredential(repository, networkCredential, cmdletPassedIn);
+            }
+
+            return networkCreds;
+        }     
+
+        #endregion
+
+        #region Container Registry methods
+
+        public static bool IsContainerRegistry(string uri)
+        {
+            if (uri.EndsWith(".azurecr.io") || uri.EndsWith(".azurecr.io/") || uri.Contains("mcr.microsoft.com") || uri.StartsWith("mcr.microsoft"))
+            {
+                return true;
+            }
+
+            return false;
+        }
+
         #endregion
 
         #region Path methods
@@ -1259,19 +1294,21 @@ internal static void GetMetadataFilesFromPath(string dirPath, string packageName
             }
         }
 
-        internal static string GetCaseInsensitiveFilePath(string directory, string fileName)
+        internal static bool TryGetCaseInsensitiveFilePath(string directory, string fileName, out string fileFound)
         {
+            fileFound = String.Empty;
             var files = Directory.GetFiles(directory);
             foreach (var file in files)
             {
                 if (string.Equals(Path.GetFileName(file), fileName, StringComparison.OrdinalIgnoreCase))
                 {
-                    return file;
+                    fileFound = file;
+                    return true;
                 }
             }
 
             // File not found
-            return null;
+            return false;
         }
 
         #endregion