Merge pull request #190 from MabOneSdk/pragrawa-dev1

Set-RS VaultContext changes

Author: pragrawa

src/ResourceManager/RecoveryServices/Commands.RecoveryServices/Commands.RecoveryServices.csproj

@@ -133,6 +133,7 @@
     <Compile Include="Vault\GetAzureRmRecoveryServicesVaultSettingsFile.cs" />
     <Compile Include="Vault\NewAzureRmRecoveryServicesVault.cs" />
     <Compile Include="Vault\RemoveAzureRmRecoveryServicesVault.cs" />
+    <Compile Include="Vault\SetAzureRmRecoveryServicesVaultContext.cs" />
   </ItemGroup>
   <ItemGroup>
     <EmbeddedResource Include="Properties\Resources.resx">

src/ResourceManager/RecoveryServices/Commands.RecoveryServices/Common/PSRecoveryServicesClient.cs

@@ -66,6 +66,14 @@ public RecoveryServicesManagementClient GetRecoveryServicesClient
         Justification = "For Resource Credentials.")]
         public static ASRVaultCreds asrVaultCreds = new ASRVaultCreds();
 
+        /// <summary>
+        /// Resource credentials holds vault, resource group name, location and other details.
+        /// </summary>
+        [SuppressMessage(
+        "Microsoft.StyleCop.CSharp.MaintainabilityRules",
+        "SA1401:FieldsMustBePrivate",
+        Justification = "For Resource Credentials.")]
+        public static ARSVault arsVaultCreds = new ARSVault();
         /// <summary>
         /// Recovery Services client.
         /// </summary>

src/ResourceManager/RecoveryServices/Commands.RecoveryServices/Common/PSRecoveryServicesVaultClient.cs

@@ -32,6 +32,17 @@ public VaultListResponse GetVaultsInResouceGroup(string resouceGroupName)
             return this.GetRecoveryServicesClient.Vaults.List(resouceGroupName, this.GetRequestHeaders());
         }
 
+        /// <summary>
+        /// Method to get Azure Recovery Services Vault
+        /// </summary>
+        /// <param name="resouceGroupName">Name of the resouce group</param>
+        /// <param name="resourceName">Name of the resource</param>
+        /// <returns>vault response object.</returns>
+        public VaultResponse GetVault(string resouceGroupName, string resourceName)
+        {
+            return this.GetRecoveryServicesClient.Vaults.Get(resouceGroupName, resourceName, this.GetRequestHeaders());
+        }
+
         /// <summary>
         /// Method to create Azure Recovery Services Vault
         /// </summary>

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

@@ -277,4 +277,7 @@ Please provide a storage account with the same location as that of the vault.</v
   <data name="SiteNotFound" xml:space="preserve">
     <value>Site {0} is not associated with the Vault {1}</value>
   </data>
+  <data name="ResourceGroupNameNullOrEmpty" xml:space="preserve">
+    <value>Resource Group name mentioned is either null or empty</value>
+  </data>
 </root>

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

@@ -19,6 +19,7 @@
 using System.Security.Cryptography;
 using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
 using System.Collections.Generic;
+using Microsoft.Azure.Management.RecoveryServices.Models;
 
 namespace Microsoft.Azure.Commands.RecoveryServices
 {
@@ -121,6 +122,19 @@ public static void UpdateCurrentVaultContext(ASRVaultCreds asrVaultCreds)
             }
         }
 
+        /// <summary>
+        /// Updates current Vault context.
+        /// </summary>
+        /// <param name="arsVaultCreds">ARS Vault credentials</param>
+        public static void UpdateCurrentVaultContext(ARSVault arsVaultCreds)
+        {
+            object updateVaultContextOneAtATime = new object();
+            lock (updateVaultContextOneAtATime)
+            {
+                PSRecoveryServicesClient.arsVaultCreds = arsVaultCreds;
+            }
+        }
+
         /// <summary>
         /// method to return the Downloads path for the current user.
         /// </summary>

src/ResourceManager/RecoveryServices/Commands.RecoveryServices/Utilities/Utilities.cs

@@ -0,0 +1,72 @@
+// ----------------------------------------------------------------------------------
+//
+// 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 System;
+using System.Linq;
+using System.Management.Automation;
+using System.Security.Cryptography.X509Certificates;
+using Microsoft.Azure.Common.Authentication.Models;
+using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
+
+namespace Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets
+{
+    /// <summary>
+    /// Used to set RecoveryServices Vault Context
+    /// </summary>
+    [Cmdlet(VerbsCommon.Set, "AzureRmRecoveryServicesVaultContext")]
+    public class SetAzureRmRecoveryServicesVaultContext : RecoveryServicesCmdletBase
+    {
+        [Parameter(Mandatory = true, ValueFromPipeline = true)]
+        [ValidateNotNullOrEmpty]
+        public ARSVault Vault { get; set; }        
+
+        public override void ExecuteCmdlet()
+        {
+            try
+            {
+                base.ExecuteCmdlet();
+
+                // Validate required parameters taken from the Vault.
+                if (string.IsNullOrEmpty(Vault.Name))
+                {
+                    throw new ArgumentException(
+                        Properties.Resources.ResourceNameNullOrEmpty,
+                        Vault.Name);
+                }
+
+                if (string.IsNullOrEmpty(Vault.ResouceGroupName))
+                {
+                    throw new ArgumentException(
+                        Properties.Resources.ResourceGroupNameNullOrEmpty,
+                        Vault.ResouceGroupName);
+                }
+
+                var vault = RecoveryServicesClient.GetVault(Vault.ResouceGroupName, Vault.Name);
+                if(vault == null)
+                {
+                    throw new ArgumentException(
+                        string.Format(Properties.Resources.VaultNotFound, Vault.Name),
+                        Vault.ResouceGroupName);
+                }
+
+                Utilities.UpdateCurrentVaultContext(Vault);
+            }
+            catch (Exception exception)
+            {
+                this.HandleException(exception);
+            }           
+           
+        }
+    }
+}