Converted Get-Container cmdlet to be Provider Specifc

Author: dragonfly91

src/ResourceManager/RecoveryServices.Backup/Cmdlets/Container/GetAzureRmRecoveryServicesContainer.cs

@@ -50,42 +50,30 @@ public class GetAzureRmRecoveryServicesContainer : RecoveryServicesBackupVaultCm
         public override void ExecuteCmdlet()
         {
             ExecutionBlock(() =>
-                {
-                    base.ExecuteCmdlet();
-
-                    ProtectionContainerListQueryParams queryParams = new ProtectionContainerListQueryParams();
-
-                    // 1. Filter by Name
-                    queryParams.FriendlyName = Name;
-
-                    // 2. Filter by ContainerType
-                    queryParams.ProviderType = HydraHelpers.GetHydraProviderType(ContainerType);
+            {
+                base.ExecuteCmdlet();
 
-                    // 3. Filter by Status
-                    queryParams.RegistrationStatus = Status.ToString();
+                PsBackupProviderManager providerManager = new PsBackupProviderManager(new Dictionary<System.Enum, object>()
+                {  
+                    {ContainerParams.Vault, Vault},
+                    {ContainerParams.ContainerType, ContainerType},
+                    {ContainerParams.Name, Name},
+                    {ContainerParams.ResourceGroupName, ResourceGroupName},
+                    {ContainerParams.Status, Status},
+                }, HydraAdapter);
 
-                    var listResponse = HydraAdapter.ListContainers(Vault.ResouceGroupName, Vault.Name, queryParams);
+                IPsBackupProvider psBackupProvider = providerManager.GetProviderInstance(ContainerType);
+                var containerModels = psBackupProvider.ListProtectionContainers();
 
-                    List<AzureRmRecoveryServicesContainerBase> containerModels = ConversionHelpers.GetContainerModelList(listResponse);
-
-                    // NOTE: Should move this to provider?
-                    // 4. Filter by RG Name
-                    if (ContainerType == Models.ContainerType.AzureVM &&
-                        !string.IsNullOrEmpty(ResourceGroupName))
-                    {
-                        containerModels = containerModels.Where(containerModel =>
-                            (containerModel as AzureRmRecoveryServicesIaasVmContainer).ResourceGroupName == ResourceGroupName).ToList();
-                    }
-
-                    if (containerModels.Count == 1)
-                    {
-                        WriteObject(containerModels.First());
-                    }
-                    else
-                    {
-                        WriteObject(containerModels);
-                    }
-                });
+                if (containerModels.Count == 1)
+                {
+                    WriteObject(containerModels.First());
+                }
+                else
+                {
+                    WriteObject(containerModels);
+                }
+            });
         }
     }
 }

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Providers/Commands.RecoveryServices.Backup.Providers.csproj

@@ -63,6 +63,10 @@
       <Project>{3819d8a7-c62c-4c47-8ddd-0332d9ce1252}</Project>
       <Name>Commands.ResourceManager.Common</Name>
     </ProjectReference>
+    <ProjectReference Include="..\..\RecoveryServices\Commands.RecoveryServices\Commands.RecoveryServices.csproj">
+      <Project>{604260dc-b392-4cf4-81ec-34b14591e2d2}</Project>
+      <Name>Commands.RecoveryServices</Name>
+    </ProjectReference>
     <ProjectReference Include="..\Commands.RecoveryServices.Backup.Helpers\Commands.RecoveryServices.Backup.Helpers.csproj">
       <Project>{0e1d3f36-e6c8-4764-8c7d-6f9ee537490c}</Project>
       <Name>Commands.RecoveryServices.Backup.Helpers</Name>

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Providers/IPsBackupProvider.cs

@@ -12,6 +12,7 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.Models;
 using Microsoft.Azure.Management.RecoveryServices.Backup.Models;
 using System;
 using System.Collections.Generic;
@@ -44,5 +45,7 @@ public interface IPsBackupProvider
         ProtectionPolicyResponse GetPolicy();
 
         void DeletePolicy();
+
+        List<AzureRmRecoveryServicesContainerBase> ListProtectionContainers();
     }
 }

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Providers/Providers/AzureSqlPsBackupProvider.cs

@@ -76,5 +76,10 @@ public void DeletePolicy()
         {
             throw new NotImplementedException();
         }
+
+        public List<Models.AzureRmRecoveryServicesContainerBase> ListProtectionContainers()
+        {
+            throw new NotImplementedException();
+        }
     }
 }

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Providers/Providers/IaasVmPsBackupProvider.cs

@@ -18,14 +18,20 @@
 using System.Text;
 using System.Threading.Tasks;
 using Microsoft.Azure.Management.RecoveryServices.Backup.Models;
+using Microsoft.Azure.Commands.RecoveryServices.Backup.Helpers;
+using Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.Models;
 
 namespace Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.ProviderModel
 {
     public class IaasVmPsBackupProvider : IPsBackupProvider
     {
+        ProviderData ProviderData { get; set; }
+        HydraAdapter.HydraAdapter HydraAdapter { get; set; }
+
         public void Initialize(ProviderData providerData, HydraAdapter.HydraAdapter hydraAdapter)
         {
-            throw new NotImplementedException();
+            this.ProviderData = providerData;
+            this.HydraAdapter = hydraAdapter;
         }
 
         public BaseRecoveryServicesJobResponse EnableProtection()
@@ -77,5 +83,37 @@ public void DeletePolicy()
         {
             throw new NotImplementedException();
         }
+
+        public List<AzureRmRecoveryServicesContainerBase> ListProtectionContainers()
+        {
+            string name = (string)this.ProviderData.ProviderParameters[ContainerParams.Name];
+            ContainerRegistrationStatus status = (ContainerRegistrationStatus)this.ProviderData.ProviderParameters[ContainerParams.Status];
+            ARSVault vault = (ARSVault)this.ProviderData.ProviderParameters[ContainerParams.Vault];
+            string resourceGroupName = (string)this.ProviderData.ProviderParameters[ContainerParams.ResourceGroupName];
+
+            ProtectionContainerListQueryParams queryParams = new ProtectionContainerListQueryParams();
+
+            // 1. Filter by Name
+            queryParams.FriendlyName = name;
+
+            // 2. Filter by ContainerType
+            queryParams.ProviderType = ProviderType.AzureIaasVM.ToString();
+
+            // 3. Filter by Status
+            queryParams.RegistrationStatus = status.ToString();
+
+            var listResponse = HydraAdapter.ListContainers(vault.ResouceGroupName, vault.Name, queryParams);
+
+            List<AzureRmRecoveryServicesContainerBase> containerModels = ConversionHelpers.GetContainerModelList(listResponse);
+
+            // 4. Filter by RG Name
+            if (!string.IsNullOrEmpty(resourceGroupName))
+            {
+                containerModels = containerModels.Where(containerModel =>
+                    (containerModel as AzureRmRecoveryServicesIaasVmContainer).ResourceGroupName == resourceGroupName).ToList();
+            }
+
+            return containerModels;
+        }
     }
 }