Merge pull request #324 from MabOneSdk/swatim-azuresql

Get Container and Unregister Container for AzureSql

Author: swatimadhukargit

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Helpers/Conversions/ConversionHelpers.cs

@@ -37,10 +37,14 @@ public static AzureRmRecoveryServicesBackupContainerBase GetContainerModel(Prote
                 {
                     containerModel = new AzureRmRecoveryServicesBackupIaasVmContainer(protectionContainer);
                 }
-                if (protectionContainer.Properties.GetType() == typeof(MabProtectionContainer))
+                else if (protectionContainer.Properties.GetType() == typeof(MabProtectionContainer))
                 {
                     containerModel = new AzureRmRecoveryServicesMabContainer(protectionContainer);
                 }
+                else if (protectionContainer.Properties.GetType() == typeof(AzureSqlProtectionContainer))
+                {
+                    containerModel = new AzureRmRecoveryServicesAzureSqlContainer(protectionContainer);
+                }
             }
 
             return containerModel;

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.HydraAdapter/Resources/Microsoft.Azure.Management.RecoveryServices.Backup.dll

@@ -23,9 +23,12 @@ namespace Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.Models
 {
     public class AzureRmRecoveryServicesAzureSqlContainer : AzureRmRecoveryServicesBackupContainerBase
     {
+        public ContainerRegistrationStatus Status { get; set; }
         public AzureRmRecoveryServicesAzureSqlContainer(ProtectionContainerResource protectionContainer)
             : base(protectionContainer)
         {
+            AzureSqlProtectionContainer sqlProtectionContainer = (AzureSqlProtectionContainer)protectionContainer.Properties;
+            Status = EnumUtils.GetEnum<ContainerRegistrationStatus>(sqlProtectionContainer.RegistrationStatus);
         }
     }
 }

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Models/AzureSqlModels/AzureRmRecoveryServicesAzureSqlContainer.cs

@@ -163,6 +163,10 @@ public static ContainerType GetPsContainerType(string containerType)
             {
                 return ContainerType.Windows;
             }
+            else if (containerType == ContainerType.AzureSqlContainer.ToString())
+            {
+                return ContainerType.AzureSqlContainer;
+            }
             else
             {
                 throw new Exception("Unsupported ContainerType: " + containerType);

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Models/CommonModels/Utils.cs

@@ -414,7 +414,7 @@ Please contact Microsoft for further assistant.</value>
     <value>Please provide AzureRmBackupManagementServer of BackupEngineType as DpmEngine or DpmVenusEngine and provide BackupManagementType as SCDPM. Provided AzureRmBackupManagementServer has BackupEngineType {0} and backupManagementType {1} which is not valid.</value>
   </data>
   <data name="UnsupportedContainerException" xml:space="preserve">
-    <value>Please provide Container of containerType as Windows and backupManagementType as MARS. Provided Container has containerType {0} and backupManagementType {1} which is invalid.</value>
+    <value>Please provide Container of containerType as Windows and backupManagementType as MARS or Container of containerType as AzureSQL and backupManagementType as AzureSQL. Provided Container has containerType {0} and backupManagementType {1} which is invalid.</value>
   </data>
   <data name="GetRPErrorInputDatesShouldBeInUTC" xml:space="preserve">
     <value>Please specify startdate and enddate in UTC format</value>
@@ -447,6 +447,6 @@ Please contact Microsoft for further assistant.</value>
     <value>BackupManagementType provider for ContainerType {0} is incorrect.</value>
   </data>
   <data name="AllowedSqlRetentionRange" xml:space="preserve">
-    <value>RetentionCount in Weeks should be 1-520, in Months should be 1-120 and Years should be 1-10</value>
+    <value>Retention count in Weeks should be in between 1-520, in Months should be in between 1-120 and Years should be in between 1-10</value>
   </data>
 </root>

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Models/Properties/Resources.Designer.cs

@@ -188,7 +188,25 @@ public ProtectionPolicyResponse ModifyPolicy()
 
         public List<Models.AzureRmRecoveryServicesBackupContainerBase> ListProtectionContainers()
         {
-            throw new NotImplementedException();
+            string name = (string)this.ProviderData.ProviderParameters[ContainerParams.Name];
+
+            ProtectionContainerListQueryParams queryParams = new ProtectionContainerListQueryParams();
+
+            queryParams.BackupManagementType = Microsoft.Azure.Management.RecoveryServices.Backup.Models.BackupManagementType.AzureSql.ToString();
+
+            var listResponse = HydraAdapter.ListContainers(queryParams);       
+    
+            List<AzureRmRecoveryServicesBackupContainerBase> containerModels = ConversionHelpers.GetContainerModelList(listResponse);
+
+            if (!string.IsNullOrEmpty(name))
+            {
+                if (containerModels != null)
+                {
+                    containerModels = containerModels.Where(x => x.Name == name).ToList();
+                }
+            }
+
+            return containerModels;
         }
 
         public List<AzureRmRecoveryServicesBackupEngineBase> ListBackupManagementServers()

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Models/Properties/Resources.resx

@@ -58,7 +58,7 @@ public IPsBackupProvider GetProviderInstance(ContainerType containerType, Backup
                         throw new ArgumentException(String.Format(Resources.BackupManagementTypeIncorrectForContainerType, containerType));
                     break;
                 case ContainerType.AzureSqlContainer:
-                    if (backupManagementType == BackupManagementType.AzureSql)
+                    if (backupManagementType == BackupManagementType.AzureSql || backupManagementType == null)
                         providerType = PsBackupProviderTypes.AzureSql;
                     else
                         throw new ArgumentException(String.Format(Resources.BackupManagementTypeRequiredForContainerType, containerType));

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

@@ -36,7 +36,7 @@ public class GetAzureRmRecoveryServicesBackupContainer : RecoveryServicesBackupC
 
         [Parameter(Mandatory = false, Position = 2, HelpMessage = ParamHelpMsg.Container.BackupManagementType)]
         [ValidateNotNullOrEmpty]
-        [ValidateSet("AzureVM", "MARS")]
+        [ValidateSet("AzureVM", "MARS", "AzureSql")]
         public string BackupManagementType { get; set; }
 
         [Parameter(Mandatory = false, Position = 3, HelpMessage = ParamHelpMsg.Container.Name)]

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

@@ -41,12 +41,14 @@ public override void ExecuteCmdlet()
             {
                 base.ExecuteCmdlet();
 
-                if (Container.ContainerType != ContainerType.Windows || Container.BackupManagementType != BackupManagementType.MARS)
+                if (!((Container.ContainerType == ContainerType.Windows && Container.BackupManagementType == BackupManagementType.MARS) ||
+                    (Container.ContainerType == ContainerType.AzureSqlContainer && Container.BackupManagementType == BackupManagementType.AzureSql)))
                 {
                     throw new ArgumentException(String.Format(Resources.UnsupportedContainerException, Container.ContainerType, Container.BackupManagementType));
                 }
-                AzureRmRecoveryServicesMabContainer mabContainer = Container as AzureRmRecoveryServicesMabContainer;
-                string containerName = mabContainer.Name;
+                string containerName = Container.Name;
+                if (Container.ContainerType == ContainerType.AzureSqlContainer)
+                    containerName = ContainerConstansts.SqlContainerNamePrefix + containerName;
                 HydraAdapter.UnregisterContainers(containerName);
             });
         }

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup/Cmdlets/Container/GetAzureRmRecoveryServicesBackupContainer.cs

@@ -24,4 +24,9 @@ public class PolicyConstants
         public const int MinPolicyNameLength = 3;
         public const int MaxPolicyNameLength = 150;                
     }
+
+    public class ContainerConstansts
+    {
+        public const string SqlContainerNamePrefix = "AzureSqlContainer;";
+    }
 }