Sets SiteRecovery and RecoveryServices Vault context without importing file

Author: sriramvu

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Commands.SiteRecovery.csproj

@@ -208,6 +208,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>
   </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
 </Project>

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Models/PSParameterSets.cs

@@ -233,5 +233,15 @@ internal static class ASRParameterSets
         /// For Disable replication group parameter set.
         /// </summary>
         internal const string DisableReplicationGroup = "DisableReplicationGroup";
+
+        /// <summary>
+        /// Handle ASR Vault.
+        /// </summary>
+        internal const string ASRVault = "AzureSiteRecoveryVault";
+
+        /// <summary>
+        /// Handle ARS Vault.
+        /// </summary>
+        internal const string ARSVault = "AzureRecoveryServicesVault";
     }
 }

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Vault/ImportAzureSiteRecoveryVaultSettingsFile.cs

@@ -51,18 +51,20 @@ public override void ExecuteCmdlet()
             this.WriteVerbose("Vault Settings File path: " + this.Path);
 
             ASRVaultCreds asrVaultCreds = null;
+            ARSVaultCreds arsVaultCreds = null;
+
             if (File.Exists(this.Path))
             {
                 try
                 {
-                    var serializer = new DataContractSerializer(typeof(ASRVaultCreds));
+                    var serializer = new DataContractSerializer(typeof(ARSVaultCreds));
                     using (var s = new FileStream(
                         this.Path,
                         FileMode.Open,
                         FileAccess.Read,
                         FileShare.Read))
                     {
-                        asrVaultCreds = (ASRVaultCreds)serializer.ReadObject(s);
+                        arsVaultCreds = (ARSVaultCreds)serializer.ReadObject(s);
                     }
                 }
                 catch (XmlException xmlException)
@@ -72,8 +74,34 @@ public override void ExecuteCmdlet()
                 }
                 catch (SerializationException serializationException)
                 {
-                    throw new SerializationException(
-                        string.Format(Properties.Resources.InvalidXml, serializationException));
+                    try
+                    {
+                        // moved here as ASR Vault is short lived
+                        var serializer = new DataContractSerializer(typeof(ASRVaultCreds));
+                        using (var s = new FileStream(
+                            this.Path,
+                            FileMode.Open,
+                            FileAccess.Read,
+                            FileShare.Read))
+                        {
+                            asrVaultCreds = (ASRVaultCreds)serializer.ReadObject(s);
+                        }
+                    }
+                    catch
+                    {
+                        throw new SerializationException(
+                            string.Format(Properties.Resources.InvalidXml, serializationException));
+                    }
+                }
+                if (null == asrVaultCreds)
+                {
+                    // Copy ars to asr
+                    asrVaultCreds = new ASRVaultCreds();
+                    asrVaultCreds.ResourceName = arsVaultCreds.ResourceName;
+                    asrVaultCreds.ResourceGroupName = arsVaultCreds.ResourceGroupName;
+                    asrVaultCreds.ResourceNamespace = arsVaultCreds.ResourceNamespace;
+                    asrVaultCreds.ARMResourceType = arsVaultCreds.ARMResourceType;
+                    asrVaultCreds.ChannelIntegrityKey = arsVaultCreds.ChannelIntegrityKey;
                 }
             }
             else

src/ResourceManager/SiteRecovery/Commands.SiteRecovery/Vault/SetAzureSiteRecoveryVaultSettings.cs

@@ -18,6 +18,9 @@
 using System.Management.Automation;
 using Microsoft.WindowsAzure.Commands.Utilities.Common;
 using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
+using Microsoft.Azure.Commands.RecoveryServices;
+using System.Diagnostics;
+using System.Collections.ObjectModel;
 
 namespace Microsoft.Azure.Commands.SiteRecovery
 {
@@ -31,11 +34,18 @@ public class SetAzureSiteRecoveryVaultSettings : SiteRecoveryCmdletBase
         #region Parameters
 
         /// <summary>
-        /// Gets or sets vault Object.
+        /// Gets or sets ASR vault Object.
         /// </summary>
-        [Parameter(Mandatory = true, ValueFromPipeline = true)]
+        [Parameter(ParameterSetName = ASRParameterSets.ASRVault, Mandatory = true)]
         [ValidateNotNullOrEmpty]
-        public ASRVault Vault { get; set; }
+        public ASRVault ASRVault { get; set; }
+
+        /// <summary>
+        /// Gets or sets ARS vault Object.
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.ARSVault, Mandatory = true)]
+        [ValidateNotNullOrEmpty]
+        public ARSVault ARSVault { get; set; }
 
         #endregion Parameters
 
@@ -44,15 +54,63 @@ public class SetAzureSiteRecoveryVaultSettings : SiteRecoveryCmdletBase
         /// </summary>
         public override void ExecuteCmdlet()
         {
-            // Validate the Vault
-            RecoveryServicesClient.ValidateVaultSettings(
-                this.Vault.Name,
-                this.Vault.ResouceGroupName);
+            switch (this.ParameterSetName)
+            {
+                case ASRParameterSets.ASRVault:
+                    this.setASRVaultContext(this.ASRVault);
+                    break;
+                case ASRParameterSets.ARSVault:
+                    this.setARSVaultContext(this.ARSVault);
+                    break;
+            }
+        }
 
+        /// <summary>
+        /// Set Azure Site Recovery Vault context.
+        /// </summary>
+        private void setASRVaultContext(ASRVault asrVault)
+        {
             // Change the vault context
-            ASRVaultCreds vaultCreds = RecoveryServicesClient.ChangeVaultContext(this.Vault);
+            RecoveryServicesClient.ChangeVaultContext(asrVault);
+
+            // Validate the Vault
+            RecoveryServicesClient.ValidateVaultSettings(
+                asrVault.Name,
+                asrVault.ResouceGroupName);
 
             this.WriteObject(new ASRVaultSettings(PSRecoveryServicesClient.asrVaultCreds));
         }
+
+        /// <summary>
+        /// Set Azure Recovery Services Vault context.
+        /// </summary>
+        private void setARSVaultContext(ARSVault arsVault)
+        {
+            try
+            {
+                using (System.Management.Automation.PowerShell powerShell = System.Management.Automation.PowerShell.Create())
+                {
+                    Collection<PSObject> result =
+                        powerShell
+                        .AddCommand("Get-AzureRmRecoveryServicesVaultSettingsFile")
+                        .AddParameter("Vault", arsVault)
+                        .Invoke();
+
+                    string vaultSettingspath = (string)result[0].Members["FilePath"].Value;
+                    powerShell.Commands.Clear();
+
+                    result =
+                        powerShell
+                        .AddCommand("Import-AzureRmSiteRecoveryVaultSettingsFile")
+                        .AddParameter("Path", vaultSettingspath)
+                        .Invoke();
+                    WriteObject(result);
+                    powerShell.Commands.Clear();
+                }
+            }
+            catch (InvalidOperationException e)
+            {
+            }
+        }
     }
 }

src/ResourceManager/SiteRecovery/SiteRecovery.sln

@@ -21,6 +21,8 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = ".nuget", ".nuget", "{5BA788
 		.nuget\packages.config = .nuget\packages.config
 	EndProjectSection
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Commands.RecoveryServices", "..\RecoveryServices\Commands.RecoveryServices\Commands.RecoveryServices.csproj", "{604260DC-B392-4CF4-81EC-34B14591E2D2}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -51,6 +53,10 @@ Global
 		{5EE72C53-1720-4309-B54B-5FB79703195F}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{5EE72C53-1720-4309-B54B-5FB79703195F}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{5EE72C53-1720-4309-B54B-5FB79703195F}.Release|Any CPU.Build.0 = Release|Any CPU
+		{604260DC-B392-4CF4-81EC-34B14591E2D2}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{604260DC-B392-4CF4-81EC-34B14591E2D2}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{604260DC-B392-4CF4-81EC-34B14591E2D2}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{604260DC-B392-4CF4-81EC-34B14591E2D2}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE