Added Remove-AzureRmVMChefExtension command in ARM

Author: nimisha

src/ResourceManager/Compute/Commands.Compute/Commands.Compute.csproj

@@ -234,6 +234,7 @@
     <Compile Include="Extension\AzureVMBackup\SetAzureVMBackupExtension.cs" />
     <Compile Include="Extension\BGInfo\SetAzureVMBGInfoExtension.cs" />
     <Compile Include="Extension\BGInfo\VirtualMachineBGInfoExtensionContext.cs" />
+    <Compile Include="Extension\Chef\RemoveAzureRmVMChefExtension.cs" />
     <Compile Include="Extension\Chef\SetAzureVMChefExtension.cs" />
     <Compile Include="Extension\CustomScript\GetAzureVMCustomScriptExtensionCommand.cs" />
     <Compile Include="Extension\CustomScript\CustomScriptExtensionPrivateSettings.cs" />

src/ResourceManager/Compute/Commands.Compute/Extension/Chef/RemoveAzureRmVMChefExtension.cs

@@ -0,0 +1,118 @@
+using System;
+using System.Linq;
+using System.Management.Automation;
+using Microsoft.Azure.Commands.Compute.Common;
+using Microsoft.Azure.Management.Compute;
+using System.Globalization;
+using AutoMapper;
+using Microsoft.Azure.Commands.Compute.Models;
+using Microsoft.Azure.Management.Compute.Models;
+
+namespace Microsoft.Azure.Commands.Compute.Extension.Chef
+{
+    [Cmdlet(
+        VerbsCommon.Remove,
+        ProfileNouns.VirtualMachineChefExtension)]
+    [OutputType(typeof(PSAzureOperationResponse))]
+    public class RemoveAzureRmVMChefExtension : VirtualMachineExtensionBaseCmdlet
+    {
+        private string ExtensionDefaultPublisher = "Chef.Bootstrap.WindowsAzure";
+        private string ExtensionDefaultName = "ChefClient";
+        private string LinuxExtensionName = "LinuxChefClient";
+
+        protected const string LinuxParameterSetName = "Linux";
+        protected const string WindowsParameterSetName = "Windows";
+
+        [Parameter(
+           Mandatory = true,
+           Position = 0,
+           ValueFromPipelineByPropertyName = true,
+           HelpMessage = "The resource group name.")]
+        [ValidateNotNullOrEmpty]
+        public string ResourceGroupName { get; set; }
+
+        [Alias("ResourceName")]
+        [Parameter(
+            Mandatory = true,
+            Position = 1,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "The virtual machine name.")]
+        [ValidateNotNullOrEmpty]
+        public string VMName { get; set; }
+
+        [Alias("ExtensionName")]
+        [Parameter(
+            Mandatory = false,
+            Position = 2,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "Extension Name.")]
+        public string Name
+        {
+            get
+            {
+                return this.ExtensionDefaultName;
+            }
+            set
+            {
+                this.ExtensionDefaultName = value;
+            }
+        }
+
+        [Parameter(
+            Mandatory = true,
+            ParameterSetName = LinuxParameterSetName,
+            HelpMessage = "Set extension for Linux.")]
+        public SwitchParameter Linux { get; set; }
+
+        [Parameter(
+            Mandatory = true,
+            ParameterSetName = WindowsParameterSetName,
+            HelpMessage = "Set extension for Windows.")]
+        public SwitchParameter Windows { get; set; }
+
+        public override void ExecuteCmdlet()
+        {
+            base.ExecuteCmdlet();
+
+            if (this.Linux.IsPresent)
+            {
+                this.Name = LinuxExtensionName;
+            }
+            else if (this.Windows.IsPresent)
+            {
+                this.Name = ExtensionDefaultName;
+            }
+
+            ExecuteClientAction(() =>
+            {
+                if (string.IsNullOrEmpty(this.Name))
+                {
+                    VirtualMachine virtualMachine = ComputeClient.ComputeManagementClient.VirtualMachines.Get(
+                        this.ResourceGroupName, this.VMName);
+                    var chefExtension = virtualMachine.Resources != null
+                            ? virtualMachine.Resources.FirstOrDefault(extension =>
+                                extension.Publisher.Equals(ExtensionDefaultPublisher, StringComparison.InvariantCultureIgnoreCase) &&
+                                extension.VirtualMachineExtensionType.Equals(this.Name, StringComparison.InvariantCultureIgnoreCase))
+                            : null;
+
+                    if (chefExtension == null)
+                    {
+                        WriteWarning(string.Format(CultureInfo.InvariantCulture, Properties.Resources.ChefExtensionNotFound, this.ResourceGroupName, this.VMName));
+                        return;
+                    }
+                    else
+                    {
+                        this.Name = chefExtension.Name;
+                    }
+                }
+
+                var op = this.VirtualMachineExtensionClient.DeleteWithHttpMessagesAsync(
+                    this.ResourceGroupName,
+                    this.VMName,
+                    this.Name).GetAwaiter().GetResult();
+                var result = Mapper.Map<PSAzureOperationResponse>(op);
+                WriteObject(result);
+            });
+        }
+    }
+}

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

@@ -532,4 +532,8 @@ The file needs to be a PowerShell script (.ps1 or .psm1) or a ZIP archive (.zip)
   <data name="RemoveScriptExtensionAction" xml:space="preserve">
     <value>Remove custom script extension</value>
   </data>
+  <data name="ChefExtensionNotFound" xml:space="preserve">
+    <value>No chef extension found under resource group '{0}', virtual machine '{1}'.</value>
+    <comment>{0} = resource group name, {1} = virtual machine name</comment>
+  </data>
 </root>