Added SupportsShouldProcess and ConfirmAction

Author: NimishaS

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

@@ -25,8 +25,7 @@
 namespace Microsoft.Azure.Commands.Compute.Extension.Chef
 {
     [Cmdlet(
-        VerbsCommon.Remove,
-        ProfileNouns.VirtualMachineChefExtension)]
+        VerbsCommon.Remove, ProfileNouns.VirtualMachineChefExtension, SupportsShouldProcess = true)]
     [OutputType(typeof(PSAzureOperationResponse))]
     public class RemoveAzureRmVMChefExtension : VirtualMachineExtensionBaseCmdlet
     {
@@ -97,36 +96,37 @@ public override void ExecuteCmdlet()
                 this.Name = ExtensionDefaultName;
             }
 
-            ExecuteClientAction(() =>
-            {
-                if (string.IsNullOrEmpty(this.Name))
+            ConfirmAction("Remove Chef Extension", this.VMName,
+                () =>
                 {
-                    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)
+                    if (string.IsNullOrEmpty(this.Name))
                     {
-                        WriteWarning(string.Format(CultureInfo.InvariantCulture, Properties.Resources.ChefExtensionNotFound, this.ResourceGroupName, this.VMName));
-                        return;
-                    }
-                    else
-                    {
-                        this.Name = chefExtension.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);
-            });
+                    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/Extension/Chef/SetAzureVMChefExtension.cs

@@ -25,8 +25,7 @@
 namespace Microsoft.Azure.Commands.Compute.Extension.Chef
 {
     [Cmdlet(
-        VerbsCommon.Set,
-        ProfileNouns.VirtualMachineChefExtension)]
+        VerbsCommon.Set, ProfileNouns.VirtualMachineChefExtension, SupportsShouldProcess = true)]
     [OutputType(typeof(PSAzureOperationResponse))]
     public class SetAzureVMChefExtensionCommand : VirtualMachineExtensionBaseCmdlet
     {
@@ -321,28 +320,28 @@ private Hashtable PrivateConfiguration
 
         private void ExecuteCommand()
         {
-            ExecuteClientAction(() =>
-            {
-                var parameters = new VirtualMachineExtension
-                {
-                    Location = this.Location,
-                    Settings = this.PublicConfiguration,
-                    ProtectedSettings = this.PrivateConfiguration,
-                    Publisher = ExtensionDefaultPublisher,
-                    VirtualMachineExtensionType = this.Name,
-                    TypeHandlerVersion = this.TypeHandlerVersion,
-                    AutoUpgradeMinorVersion = this.AutoUpgradeMinorVersion
-                };
-
-                var op = this.VirtualMachineExtensionClient.CreateOrUpdateWithHttpMessagesAsync(
-                    this.ResourceGroupName,
-                    this.VMName,
-                    this.Name,
-                    parameters).GetAwaiter().GetResult();
-
-                var result = Mapper.Map<PSAzureOperationResponse>(op);
-                WriteObject(result);
-            });
+            ConfirmAction("Set Chef Extension", this.VMName,
+                () => {
+                    var parameters = new VirtualMachineExtension
+                    {
+                        Location = this.Location,
+                        Settings = this.PublicConfiguration,
+                        ProtectedSettings = this.PrivateConfiguration,
+                        Publisher = ExtensionDefaultPublisher,
+                        VirtualMachineExtensionType = this.Name,
+                        TypeHandlerVersion = this.TypeHandlerVersion,
+                        AutoUpgradeMinorVersion = this.AutoUpgradeMinorVersion
+                    };
+
+                    var op = this.VirtualMachineExtensionClient.CreateOrUpdateWithHttpMessagesAsync(
+                        this.ResourceGroupName,
+                        this.VMName,
+                        this.Name,
+                        parameters).GetAwaiter().GetResult();
+
+                    var result = Mapper.Map<PSAzureOperationResponse>(op);
+                    WriteObject(result);
+                });
         }
 
         private void SetDefault()