adding support for listing pending certificates

Author: dragav

src/ResourceManager/KeyVault/Commands.KeyVault/Commands.KeyVault.csproj

@@ -91,6 +91,7 @@
     <Compile Include="Commands\UndoAzureKeyVaultKeyRemoval.cs" />
     <Compile Include="Commands\UndoAzureKeyVaultSecretRemoval.cs" />
     <Compile Include="Commands\UpdateAzureKeyVaultSecret.cs" />
+    <Compile Include="Models\KeyVaultCertificateFilterOptions.cs" />
     <Compile Include="Models\ManagedStorageAccounts\PSDeletedKeyVaultManagedStorageAccount.cs" />
     <Compile Include="Models\ManagedStorageAccounts\PSDeletedKeyVaultManagedStorageAccountIdentityItem.cs" />
     <Compile Include="Models\ManagedStorageAccounts\PSDeletedKeyVaultManagedStorageSasDefinition.cs" />

src/ResourceManager/KeyVault/Commands.KeyVault/Commands/GetAzureKeyVaultCertificate.cs

@@ -186,6 +186,20 @@ public class GetAzureKeyVaultCertificate : KeyVaultCmdletBase
                    ParameterSetName = ResourceIdByVaultNameParameterSet,
                    HelpMessage = "Specifies whether to show the previously deleted certificates in the output.")]
         public SwitchParameter InRemovedState { get; set; }
+
+        /// <summary>
+        /// Switch specifying whether to include the pending certificates in the enumeration.
+        /// </summary>
+        [Parameter(Mandatory = false,
+                   ParameterSetName = ByVaultNameParameterSet,
+                   HelpMessage = "Specifies whether to include the pending certificates in the output.")]
+        [Parameter(Mandatory = false,
+                   ParameterSetName = InputObjectByVaultNameParameterSet,
+                   HelpMessage = "Specifies whether to include the pending certificates in the output.")]
+        [Parameter(Mandatory = false,
+                   ParameterSetName = ResourceIdByVaultNameParameterSet,
+                   HelpMessage = "Specifies whether to include the pending certificates in the output.")]
+        public SwitchParameter IncludePending { get; set; }
         #endregion
 
         public override void ExecuteCmdlet()
@@ -241,10 +255,11 @@ public override void ExecuteCmdlet()
 
         private void GetAndWriteCertificates(string vaultName)
         {
-            KeyVaultObjectFilterOptions options = new KeyVaultObjectFilterOptions
+            var options = new KeyVaultCertificateFilterOptions
             {
                 VaultName = VaultName,
-                NextLink = null
+                NextLink = null,
+                IncludePending = IncludePending
             };
 
             do
@@ -272,10 +287,11 @@ private void GetAndWriteCertificatesVersions(string vaultName, string name, stri
 
         private void GetAndWriteDeletedCertificates( string vaultName )
         {
-            KeyVaultObjectFilterOptions options = new KeyVaultObjectFilterOptions
+            var options = new KeyVaultCertificateFilterOptions
             {
                 VaultName = VaultName,
-                NextLink = null
+                NextLink = null,
+                IncludePending = IncludePending
             };
 
             do

src/ResourceManager/KeyVault/Commands.KeyVault/Models/IKeyVaultDataServiceClient.cs

@@ -85,9 +85,9 @@ public interface IKeyVaultDataServiceClient
 
         PSDeletedKeyVaultCertificate GetDeletedCertificate( string vaultName, string certName );
 
-        IEnumerable<PSKeyVaultCertificateIdentityItem> GetCertificates(KeyVaultObjectFilterOptions options);
+        IEnumerable<PSKeyVaultCertificateIdentityItem> GetCertificates(KeyVaultCertificateFilterOptions options);
 
-        IEnumerable<PSDeletedKeyVaultCertificateIdentityItem> GetDeletedCertificates( KeyVaultObjectFilterOptions options );
+        IEnumerable<PSDeletedKeyVaultCertificateIdentityItem> GetDeletedCertificates( KeyVaultCertificateFilterOptions options );
 
         IEnumerable<PSKeyVaultCertificateIdentityItem> GetCertificateVersions(KeyVaultObjectFilterOptions options);
 

src/ResourceManager/KeyVault/Commands.KeyVault/Models/KeyVaultCertificateFilterOptions.cs

@@ -0,0 +1,21 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+namespace Microsoft.Azure.Commands.KeyVault.Models
+{
+    public sealed class KeyVaultCertificateFilterOptions : KeyVaultObjectFilterOptions
+    {
+        public bool? IncludePending { get; set; }
+    }
+}

src/ResourceManager/KeyVault/Commands.KeyVault/Models/KeyVaultDataServiceClient.cs

@@ -340,7 +340,7 @@ public PSKeyVaultKey GetKey(string vaultName, string keyName, string keyVersion)
             return new PSKeyVaultKey(keyBundle, this.vaultUriHelper);
         }
 
-        public IEnumerable<PSKeyVaultCertificateIdentityItem> GetCertificates(KeyVaultObjectFilterOptions options)
+        public IEnumerable<PSKeyVaultCertificateIdentityItem> GetCertificates(KeyVaultCertificateFilterOptions options)
         {
             if (options == null)
                 throw new ArgumentNullException(nameof(options));
@@ -355,7 +355,7 @@ public IEnumerable<PSKeyVaultCertificateIdentityItem> GetCertificates(KeyVaultOb
                 IPage<CertificateItem> result;
 
                 if (string.IsNullOrEmpty(options.NextLink))
-                    result = this.keyVaultClient.GetCertificatesAsync(vaultAddress).GetAwaiter().GetResult();
+                    result = this.keyVaultClient.GetCertificatesAsync(vaultAddress, maxresults: null, includePending: options.IncludePending).GetAwaiter().GetResult();
                 else
                     result = this.keyVaultClient.GetCertificatesNextAsync(options.NextLink).GetAwaiter().GetResult();
 
@@ -1667,7 +1667,7 @@ public PSDeletedKeyVaultCertificate GetDeletedCertificate( string vaultName, str
             return new PSDeletedKeyVaultCertificate(deletedCertificate);
         }
 
-        public IEnumerable<PSDeletedKeyVaultCertificateIdentityItem> GetDeletedCertificates( KeyVaultObjectFilterOptions options )
+        public IEnumerable<PSDeletedKeyVaultCertificateIdentityItem> GetDeletedCertificates( KeyVaultCertificateFilterOptions options )
         {
             if ( options == null )
                 throw new ArgumentNullException( nameof( options ) );
@@ -1681,7 +1681,7 @@ public IEnumerable<PSDeletedKeyVaultCertificateIdentityItem> GetDeletedCertifica
                 IPage<DeletedCertificateItem> result;
 
                 if ( string.IsNullOrEmpty( options.NextLink ) )
-                    result = this.keyVaultClient.GetDeletedCertificatesAsync( vaultAddress ).GetAwaiter( ).GetResult( );
+                    result = this.keyVaultClient.GetDeletedCertificatesAsync( vaultAddress, maxresults: null, includePending: options.IncludePending ).GetAwaiter( ).GetResult( );
                 else
                     result = this.keyVaultClient.GetDeletedCertificatesNextAsync( options.NextLink ).GetAwaiter( ).GetResult( );