Merge pull request #19 from AsrOneSdk/sriramvu-dev

Update VM properties

Author: sanjkuma

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/PSRecoveryServicesClient/PSRecoveryServicesVMClient.cs

@@ -47,5 +47,24 @@ public VirtualMachineResponse GetAzureSiteRecoveryVirtualMachine(
         {
             return this.GetSiteRecoveryClient().Vm.Get(protectionContainerId, virtualMachineId, this.GetRequestHeaders());
         }
+
+        /// <summary>
+        /// Updates Virtual Machine properties.
+        /// </summary>
+        /// <param name="protectionContainerId">Protection Container ID</param>
+        /// <param name="virtualMachineId">Virtual Machine ID</param>
+        /// <param name="updateVmPropertiesInput">Update VM properties input</param>
+        /// <returns>Job response</returns>
+        public JobResponse UpdateVmProperties(
+            string protectionContainerId,
+            string virtualMachineId,
+            UpdateVmPropertiesInput updateVmPropertiesInput)
+        {
+            return this.GetSiteRecoveryClient().Vm.UpdateVmProperties(
+                protectionContainerId,
+                virtualMachineId,
+                updateVmPropertiesInput,
+                this.GetRequestHeaders());
+        }
     }
 }

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Service/SetAzureSiteRecoveryVirtualMachine.cs

@@ -0,0 +1,101 @@
+// ----------------------------------------------------------------------------------
+//
+// 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.Management.Automation;
+using Microsoft.Azure.Commands.RecoveryServices.SiteRecovery;
+using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
+using Microsoft.WindowsAzure.Management.SiteRecovery.Models;
+
+namespace Microsoft.Azure.Commands.RecoveryServices
+{
+    /// <summary>
+    /// Set Protection Entity protection state.
+    /// </summary>
+    [Cmdlet(VerbsCommon.Set, "AzureSiteRecoveryVirtualMachine")]
+    [OutputType(typeof(ASRJob))]
+    public class SetAzureSiteRecoveryVirtualMachine : RecoveryServicesCmdletBase
+    {
+        #region Parameters
+        /// <summary>
+        /// Job response.
+        /// </summary>
+        private JobResponse jobResponse = null;
+
+        /// <summary>
+        /// Gets or sets ID of the Virtual Machine.
+        /// </summary>
+        [Parameter(Mandatory = true)]
+        [ValidateNotNullOrEmpty]
+        public ASRVirtualMachine VirtualMachine { get; set; }
+
+        /// <summary>
+        /// Gets or sets Recovery Azure VM given name
+        /// </summary>
+        [Parameter]
+        [ValidateNotNullOrEmpty]
+        public string Name { get; set; }
+
+        /// <summary>
+        /// Gets or sets Recovery Azure VM size
+        /// </summary>
+        [Parameter]
+        [ValidateNotNullOrEmpty]
+        public string Size { get; set; }
+
+        /// <summary>
+        /// Gets or sets Selected Primary Network interface card Id
+        /// </summary>
+        [Parameter]
+        [ValidateNotNullOrEmpty]
+        public string PrimaryNic { get; set; }
+
+        /// <summary>
+        /// Gets or sets Recovery Azure Network Id
+        /// </summary>
+        [Parameter]
+        [ValidateNotNullOrEmpty]
+        public string RecoveryNetworkId { get; set; }
+        #endregion Parameters
+
+        /// <summary>
+        /// ProcessRecord of the command.
+        /// </summary>
+        public override void ExecuteCmdlet()
+        {
+            // validate the srouce nic & target network together
+            UpdateVmPropertiesInput updateVmPropertiesInput = new UpdateVmPropertiesInput();
+            updateVmPropertiesInput.RecoveryAzureVmGivenName = this.Name;
+            updateVmPropertiesInput.RecoveryAzureVmSize = this.Size;
+            updateVmPropertiesInput.SelectedPrimaryNicId = this.PrimaryNic;
+            updateVmPropertiesInput.RecoveryAzureNetworkId = this.RecoveryNetworkId;
+
+            this.jobResponse = RecoveryServicesClient.UpdateVmProperties(
+                this.VirtualMachine.ProtectionContainerId,
+                this.VirtualMachine.ID,
+                updateVmPropertiesInput);
+
+            this.WriteJob(this.jobResponse.Job);
+        }
+
+        /// <summary>
+        /// Writes Job.
+        /// </summary>
+        /// <param name="job">JOB object</param>
+        private void WriteJob(Microsoft.WindowsAzure.Management.SiteRecovery.Models.Job job)
+        {
+            this.WriteObject(new ASRJob(job));
+        }
+    }
+}