Container support for mercury

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

@@ -55,6 +55,11 @@ public static ContainerBase GetContainerModel(ServiceClientModel.ProtectionConta
                 {
                     containerModel = new AzureFileShareContainer(protectionContainer);
                 }
+                else if (protectionContainer.Properties.GetType() ==
+                    typeof(ServiceClientModel.AzureVMAppContainerProtectionContainer))
+                {
+                    containerModel = new AzureWorkloadContainer(protectionContainer);
+                }
             }
 
             return containerModel;

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Helpers/ServiceClientHelpers.cs

@@ -76,6 +76,9 @@ public static string GetServiceClientProviderType(CmdletModel.WorkloadType workl
                 case CmdletModel.WorkloadType.AzureFiles:
                     providerType = ServiceClientModel.BackupManagementType.AzureStorage.ToString();
                     break;
+                case CmdletModel.WorkloadType.MSSQL:
+                    providerType = ServiceClientModel.BackupManagementType.AzureWorkload.ToString();
+                    break;
                 default:
                     break;
             }
@@ -346,11 +349,14 @@ public static string GetServiceClientWorkloadType(CmdletModel.WorkloadType workl
                 case CmdletModel.WorkloadType.AzureVM:
                     serviceClientWorkloadType = ServiceClientModel.WorkloadType.VM.ToString();
                     break;
+                case CmdletModel.WorkloadType.MSSQL:
+                    serviceClientWorkloadType = ServiceClientModel.WorkloadType.SQLDataBase.ToString();
+                    break;
                 default:
                     break;
             }
 
             return serviceClientWorkloadType;
         }
     }
-}
+}

src/ResourceManager/RecoveryServices/Commands.RecoveryServices.Backup.Models/AzureWorkloadModels/AzureWorkloadContainer.cs

@@ -0,0 +1,32 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.
+// ----------------------------------------------------------------------------------
+
+using Microsoft.Azure.Management.RecoveryServices.Backup.Models;
+
+namespace Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.Models
+{
+    /// <summary>
+    /// Azure Workload specific container class.
+    /// </summary>
+    public class AzureWorkloadContainer : AzureContainer
+    {
+        /// <summary>
+        /// Constructor. Takes the service client object representing the container 
+        /// and converts it in to the PS container model
+        /// </summary>
+        /// <param name="protectionContainerResource">Service client object representing the container</param>
+        public AzureWorkloadContainer(ProtectionContainerResource protectionContainerResource)
+            : base(protectionContainerResource) { }
+    }
+}

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

@@ -37,7 +37,12 @@ public enum ContainerType
         /// <summary>
         /// Represents any Azure Storage containers.
         /// </summary>
-        AzureStorage
+        AzureStorage,
+
+        /// <summary>
+        /// Represents Azure Workload
+        /// </summary>
+        AzureWorkload
     }
 
     /// <summary>
@@ -70,6 +75,11 @@ public enum BackupManagementType
         /// Represents Azure File Storage. https://docs.microsoft.com/en-in/azure/storage/files/storage-files-introduction
         /// </summary>
         AzureStorage,
+
+        /// <summary>
+        /// Represents Azure Workload
+        /// </summary>
+        AzureWorkload
     }
 
     /// <summary>
@@ -103,6 +113,11 @@ public enum WorkloadType
         /// Represents Azure File https://docs.microsoft.com/en-in/azure/storage/files/storage-files-introduction
         /// </summary>
         AzureFiles,
+
+        /// <summary>
+        /// Represents MSSQL in Azure VM.
+        /// </summary>
+        MSSQL,
     }
 
     /// <summary>
@@ -134,6 +149,7 @@ public enum PsBackupProviderTypes
         /// Represents the Azure File provider for powershell cmdlets.
         /// </summary>
         AzureFiles,
+        AzureWorkload,
     }
 
     /// <summary>
@@ -334,4 +350,4 @@ public enum SourceFileType
         File,
         Directory
     }
-}
+}

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

@@ -218,6 +218,8 @@ public static BackupManagementType GetPsBackupManagementType(string backupManage
                     return BackupManagementType.AzureSQL;
                 case ServiceClientModel.BackupManagementType.AzureStorage:
                     return BackupManagementType.AzureStorage;
+                case ServiceClientModel.BackupManagementType.AzureWorkload:
+                    return BackupManagementType.AzureWorkload;
                 default:
                     throw new Exception("Unsupported BackupManagmentType: " + backupManagementType);
             }
@@ -248,6 +250,11 @@ public static ContainerType GetPsContainerType(string containerType)
             {
                 return ContainerType.AzureStorage;
             }
+            else if (containerType ==
+                ServiceClientModel.BackupManagementType.AzureWorkload)
+            {
+                return ContainerType.AzureWorkload;
+            }
             else
             {
                 throw new Exception("Unsupported ContainerType: " + containerType);
@@ -265,14 +272,18 @@ public static WorkloadType GetPsWorkloadType(string workloadType)
             {
                 return WorkloadType.AzureVM;
             }
-            if (workloadType == ServiceClientModel.WorkloadType.AzureSqlDb.ToString())
+            else if (workloadType == ServiceClientModel.WorkloadType.AzureSqlDb.ToString())
             {
                 return WorkloadType.AzureSQLDatabase;
             }
-            if (workloadType == ServiceClientModel.WorkloadType.AzureFileShare)
+            else if (workloadType == ServiceClientModel.WorkloadType.AzureFileShare)
             {
                 return WorkloadType.AzureFiles;
             }
+            else if (workloadType == ServiceClientModel.WorkloadType.SQLDataBase)
+            {
+                return WorkloadType.MSSQL;
+            }
             else
             {
                 throw new Exception("Unsupported WorkloadType: " + workloadType);
@@ -290,14 +301,18 @@ public static string GetServiceClientWorkloadType(string workloadType)
             {
                 return ServiceClientModel.WorkloadType.VM;
             }
-            if (workloadType == WorkloadType.AzureSQLDatabase.ToString())
+            else if (workloadType == WorkloadType.AzureSQLDatabase.ToString())
             {
                 return ServiceClientModel.WorkloadType.AzureSqlDb;
             }
-            if (workloadType == WorkloadType.AzureFiles.ToString())
+            else if (workloadType == WorkloadType.AzureFiles.ToString())
             {
                 return ServiceClientModel.WorkloadType.AzureFileShare;
             }
+            else if (workloadType == WorkloadType.MSSQL.ToString())
+            {
+                return ServiceClientModel.WorkloadType.SQLDataBase;
+            }
             else
             {
                 throw new Exception("Unsupported WorkloadType: " + workloadType);
@@ -343,4 +358,4 @@ public static string GetWorkloadTypeFromArmType(string armType)
             throw new Exception("Unsupported ArmType: " + armType);
         }
     }
-}
+}

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

@@ -64,12 +64,12 @@ public void RefreshContainer(string vaultName = null, string vaultResourceGroupN
             }
         }
 
-        public void RegisterContainer(string storageAccountName,
+        public void RegisterContainer(string containerName,
             ProtectionContainerResource protectionContainerResource,
             string vaultName, string vaultResourceGroupName)
         {
             var registerResponse = ServiceClientAdapter.RegisterContainer(
-                            storageAccountName,
+                            containerName,
                             protectionContainerResource,
                             vaultName,
                             vaultResourceGroupName);
@@ -79,7 +79,7 @@ public void RegisterContainer(string storageAccountName,
                 operationId =>
                     ServiceClientAdapter.GetRegisterContainerOperationResult(
                         operationId,
-                        storageAccountName,
+                        containerName,
                         vaultName: vaultName,
                         resourceGroupName: vaultResourceGroupName));
 

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

@@ -0,0 +1,93 @@
+<?xml version="1.0" encoding="utf-8"?>
+<Project ToolsVersion="12.0" DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
+  <Import Project="..\..\..\..\tools\Common.Dependencies.targets" />
+  <Import Project="$(MSBuildExtensionsPath)\$(MSBuildToolsVersion)\Microsoft.Common.props" Condition="Exists('$(MSBuildExtensionsPath)\$(MSBuildToolsVersion)\Microsoft.Common.props')" />
+  <PropertyGroup>
+    <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
+    <Platform Condition=" '$(Platform)' == '' ">AnyCPU</Platform>
+    <ProjectGuid>{02234E90-BCDE-4B20-B1F5-01B1005821DB}</ProjectGuid>
+    <OutputType>Library</OutputType>
+    <AppDesignerFolder>Properties</AppDesignerFolder>
+    <RootNamespace>Microsoft.Azure.Commands.RecoveryServices.Backup.Providers</RootNamespace>
+    <AssemblyName>Microsoft.Azure.Commands.RecoveryServices.Backup.Providers</AssemblyName>
+    <TargetFrameworkVersion>v4.5.2</TargetFrameworkVersion>
+    <FileAlignment>512</FileAlignment>
+    <SolutionDir Condition="$(SolutionDir) == '' Or $(SolutionDir) == '*Undefined*'">..\</SolutionDir>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
+    <DebugSymbols>true</DebugSymbols>
+    <DebugType>full</DebugType>
+    <Optimize>false</Optimize>
+    <OutputPath>bin\Debug\</OutputPath>
+    <DefineConstants>DEBUG;TRACE</DefineConstants>
+    <ErrorReport>prompt</ErrorReport>
+    <WarningLevel>4</WarningLevel>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
+    <DebugType>pdbonly</DebugType>
+    <Optimize>true</Optimize>
+    <OutputPath>bin\Release\</OutputPath>
+    <ErrorReport>prompt</ErrorReport>
+    <WarningLevel>4</WarningLevel>
+    <DefineConstants>TRACE;SIGN</DefineConstants>
+    <SignAssembly>true</SignAssembly>
+    <AssemblyOriginatorKeyFile>MSSharedLibKey.snk</AssemblyOriginatorKeyFile>
+    <DelaySign>true</DelaySign>
+    <Prefer32Bit>false</Prefer32Bit>
+  </PropertyGroup>
+  <ItemGroup>
+    <Reference Include="Microsoft.Azure.Management.RecoveryServices.Backup, Version=3.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
+      <SpecificVersion>False</SpecificVersion>
+      <HintPath>..\..\..\packages\Microsoft.Azure.Management.RecoveryServices.Backup.3.0.1-preview\lib\net452\Microsoft.Azure.Management.RecoveryServices.Backup.dll</HintPath>
+    </Reference>
+    <Reference Include="System" />
+    <Reference Include="System.Core" />
+    <Reference Include="System.Net" />
+    <Reference Include="System.Net.Http" />
+    <Reference Include="System.ServiceProcess" />
+    <Reference Include="System.Xml.Linq" />
+    <Reference Include="System.Data.DataSetExtensions" />
+    <Reference Include="Microsoft.CSharp" />
+    <Reference Include="System.Data" />
+    <Reference Include="System.Xml" />
+  </ItemGroup>
+  <ItemGroup>
+    <Compile Include="AzureWorkloadProviderHelper.cs" />
+    <Compile Include="IPsBackupProvider.cs" />
+    <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="Providers\AzureFilesPsBackupProvider.cs" />
+    <Compile Include="Providers\AzureSqlPsBackupProvider.cs" />
+    <Compile Include="Providers\AzureWorkloadPsBackupProvider.cs" />
+    <Compile Include="Providers\DpmPsBackupProvider.cs" />
+    <Compile Include="Providers\MabPsBackupProvider.cs" />
+    <Compile Include="PsBackupProviderManager.cs" />
+    <Compile Include="Providers\IaasVmPsBackupProvider.cs" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\..\Profile\Commands.Profile\Commands.Profile.csproj">
+      <Project>{142d7b0b-388a-4ceb-a228-7f6d423c5c2e}</Project>
+      <Name>Commands.Profile</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>
+    </ProjectReference>
+    <ProjectReference Include="..\Commands.RecoveryServices.Backup.Logger\Commands.RecoveryServices.Backup.Logger.csproj">
+      <Project>{5e675749-6139-464a-904c-59c0ffdfec82}</Project>
+      <Name>Commands.RecoveryServices.Backup.Logger</Name>
+    </ProjectReference>
+    <ProjectReference Include="..\Commands.RecoveryServices.Backup.Models\Commands.RecoveryServices.Backup.Models.csproj">
+      <Project>{30b92759-50b3-494e-b9f0-ec9a2ce9d57b}</Project>
+      <Name>Commands.RecoveryServices.Backup.Models</Name>
+    </ProjectReference>
+    <ProjectReference Include="..\Commands.RecoveryServices.Backup.ServiceClientAdapter\Commands.RecoveryServices.Backup.ServiceClientAdapter.csproj">
+      <Project>{b758fec1-35c1-4f93-a954-66dd33f6e0ec}</Project>
+      <Name>Commands.RecoveryServices.Backup.ServiceClientAdapter</Name>
+    </ProjectReference>
+  </ItemGroup>
+  <ItemGroup>
+    <None Include="MSSharedLibKey.snk" />
+    <None Include="packages.config" />
+  </ItemGroup>
+  <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
+</Project>

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

@@ -60,5 +60,7 @@ public interface IPsBackupProvider
         RPMountScriptDetails ProvisionItemLevelRecoveryAccess();
 
         void RevokeItemLevelRecoveryAccess();
+
+        void RegisterContainer();
     }
 }

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

@@ -17,7 +17,6 @@
 using Microsoft.Azure.Commands.RecoveryServices.Backup.Helpers;
 using Microsoft.Azure.Commands.RecoveryServices.Backup.Properties;
 using Microsoft.Azure.Management.Internal.Resources.Models;
-using Microsoft.Azure.Management.Internal.Resources.Utilities.Models;
 using Microsoft.Azure.Management.RecoveryServices.Backup.Models;
 using Microsoft.Rest.Azure.OData;
 using System;
@@ -700,6 +699,11 @@ public List<ItemBase> ListProtectedItems()
             return itemModels;
         }
 
+        public void RegisterContainer()
+        {
+            throw new NotImplementedException();
+        }
+
         private RestAzureNS.AzureOperationResponse EnableOrModifyProtection(bool disableWithRetentionData = false)
         {
             string vaultName = (string)ProviderData[VaultParams.VaultName];

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

@@ -485,6 +485,11 @@ public List<ItemBase> ListProtectedItems()
             return itemModels;
         }
 
+        public void RegisterContainer()
+        {
+            throw new NotImplementedException();
+        }
+
         #region private
         private void ValidateAzureSqlWorkloadType(CmdletModel.WorkloadType type)
         {