Set RS VaultContext cmdlet

Author: pragrawa

src/ResourceManager/RecoveryServices.Backup/Cmdlets/Vault/SetAzureRmRecoveryServicesVaultContext.cs

@@ -0,0 +1,70 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.Text;
+using System.Threading.Tasks;
+using System.Collections.Generic;
+using System.Management.Automation;
+using Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.Models;
+using Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.ProviderModel;
+using Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.HydraAdapter;
+
+namespace Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets
+{
+    /// <summary>
+    /// To set RecoveryServicesVaultContext
+    /// </summary>
+    [Cmdlet(VerbsCommon.Set, "AzureRmRecoveryServicesVaultContext")]
+    public class SetAzureRmRecoveryServicesVaultContext : RecoveryServicesBackupCmdletBase
+    {
+        [Parameter(Mandatory = true, HelpMessage = "", ValueFromPipeline = true)]
+        [ValidateNotNullOrEmpty]
+        public ARSVault Vault { get; set; }        
+
+        public override void ExecuteCmdlet()
+        {
+            ExecutionBlock(() =>
+            {
+                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.CloudServiceNameNullOrEmpty,
+                        Vault.ResouceGroupName);
+                }
+
+                AzureRmRecoveryServicesVaultCreds vaultCreds = new AzureRmRecoveryServicesVaultCreds(Vault.Name,
+                                                               Vault.ResouceGroupName, Vault.Location);
+
+                ClientProxyBase.UpdateCurrentVaultContext(vaultCreds);
+
+                //Add validation to check vault exist or not (see if we can resuse existing ASR code).
+
+                this.WriteObject(vaultCreds);
+            });
+           
+        }
+    }
+}

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.HydraAdapter/ClientProxyBase.cs

@@ -19,6 +19,7 @@
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
+using Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.Models;
 
 namespace Microsoft.Azure.Commands.RecoveryServices.Backup.Cmdlets.HydraAdapter
 {
@@ -37,6 +38,8 @@ public class ClientProxyBase
         private CancellationTokenSource cancellationTokenSource = new CancellationTokenSource();
         public CancellationToken CmdletCancellationToken;
 
+        public static AzureRmRecoveryServicesVaultCreds recoveryServicesVaultCreds = new AzureRmRecoveryServicesVaultCreds();
+
         public ClientProxyBase(params object[] parameters)
         {
             Parameters = parameters;
@@ -56,5 +59,19 @@ public string GetClientRequestId()
         {
             return ClientRequestId;
         }
+
+        public static void UpdateCurrentVaultContext(AzureRmRecoveryServicesVaultCreds vaultCreds)
+        {
+            object updateVaultContextOneAtATime = new object();
+            lock (updateVaultContextOneAtATime)
+            {
+                recoveryServicesVaultCreds.ResourceName =
+                    vaultCreds.ResourceName;
+                recoveryServicesVaultCreds.ResourceGroupName =
+                    vaultCreds.ResourceGroupName;
+                recoveryServicesVaultCreds.Location =
+                    vaultCreds.Location;                
+            }
+        }
     }
 }

src/ResourceManager/RecoveryServices.Backup/Commands.RecoveryServices.Backup.Models/BaseObjects.cs

@@ -49,6 +49,10 @@ public class AzureRmRecoveryServicesVaultCreds : AzureRmRecoveryServicesObjectBa
         /// </summary>
         public string Location { get; set; }
 
+        public AzureRmRecoveryServicesVaultCreds()
+        {           
+        }
+
         public AzureRmRecoveryServicesVaultCreds(string resourceName, string resourceGroupName, string location)
         {
             ResourceName = resourceName;