Dev

setup/azurecmdfiles.wxi

@@ -1683,6 +1683,9 @@
                         <Component Id="cmpEDB84317615898A496C0FF620B372C87" Guid="*">
                             <File Id="fil56F72CE4FFC9F006CE1B279238995FC8" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\AzureRM.RecoveryServices\Microsoft.ApplicationInsights.dll" />
                         </Component>
+                        <Component Id="cmp66ECF842EEB4EB595925E310358E9750" Guid="*">
+                            <File Id="filF271979440640E53A189C08A7ADBAD7A" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\AzureRM.RecoveryServices\Microsoft.Azure.Commands.RecoveryServices.ARM.dll-help.xml" />
+                        </Component>
                         <Component Id="cmp5CAA968EE7C749E4480233FD90EDFE63" Guid="*">
                             <File Id="filC143DADE40E067E4692D2246F93BA8A4" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\AzureRM.RecoveryServices\Microsoft.Azure.Commands.RecoveryServices.ARM.dll" />
                         </Component>
@@ -5242,6 +5245,7 @@
             <ComponentRef Id="cmpC96DAF4A31F86D19246E39394F496F83" />
             <ComponentRef Id="cmpA77D5875EABE5151973A245C04E3858E" />
             <ComponentRef Id="cmpEDB84317615898A496C0FF620B372C87" />
+            <ComponentRef Id="cmp66ECF842EEB4EB595925E310358E9750" />
             <ComponentRef Id="cmp5CAA968EE7C749E4480233FD90EDFE63" />
             <ComponentRef Id="cmpE7C8C6576C1069564C717CBBBD229F39" />
             <ComponentRef Id="cmpC9F93119DA28BCE2B325C737951EFABA" />

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

@@ -227,9 +227,10 @@
     <Compile Include="Extension\CustomScript\RemoveAzureVMCustomScriptExtensionCommand.cs" />
     <Compile Include="Extension\CustomScript\SetAzureVMCustomScriptExtensionCommand.cs" />
     <Compile Include="Extension\CustomScript\VirtualMachineCustomScriptExtensionContext.cs" />
-    <Compile Include="Extension\Diagnostics\GetAzureVMDiagnosticsExtension.cs" />
-    <Compile Include="Extension\Diagnostics\RemoveAzureVMDiagnosticsExtension.cs" />
-    <Compile Include="Extension\Diagnostics\SetAzureVMDiagnosticsExtension.cs" />
+    <Compile Include="Extension\Diagnostics\DiagnosticsExtensionConstants.cs" />
+    <Compile Include="Extension\Diagnostics\GetAzureRmVMDiagnosticsExtension.cs" />
+    <Compile Include="Extension\Diagnostics\RemoveAzureRmVMDiagnosticsExtension.cs" />
+    <Compile Include="Extension\Diagnostics\SetAzureRmVMDiagnosticsExtension.cs" />
     <Compile Include="Extension\DSC\GetAzureVMDscExtensionStatusCommand.cs" />
     <Compile Include="Extension\DSC\GetDscResourceException.cs" />
     <Compile Include="Extension\DSC\DscExtensionPublishCmdletCommonBase.cs" />

src/ResourceManager/Compute/Commands.Compute/Extension/Diagnostics/VirtualMachineDiagnosticsExtensionCmdletBase.cs → src/ResourceManager/Compute/Commands.Compute/Extension/Diagnostics/DiagnosticsExtensionConstants.cs

@@ -14,19 +14,11 @@
 
 namespace Microsoft.Azure.Commands.Compute
 {
-    public class VirtualMachineDiagnosticsExtensionCmdletBase : VirtualMachineExtensionBaseCmdlet
+    public static class DiagnosticsExtensionConstants
     {
-        protected const string DiagnosticsExtensionNamespace = "Microsoft.Azure.Diagnostics";
-        protected const string DiagnosticsExtensionType = "IaaSDiagnostics";
-
-        protected string StorageAccountName { get; set; }
-        protected string StorageKey { get; set; }
-        protected string Endpoint { get; set; }
-
-        public VirtualMachineDiagnosticsExtensionCmdletBase()
-            : base()
-        {
-
-        }
+        public const string VirtualMachineExtensionResourceType = "Microsoft.Compute/virtualMachines/extensions";
+        public const string ExtensionDefaultName = "Microsoft.Insights.VMDiagnosticsSettings";
+        public const string ExtensionPublisher = "Microsoft.Azure.Diagnostics";
+        public const string ExtensionType = "IaaSDiagnostics";
     }
-}
+}

src/ResourceManager/Compute/Commands.Compute/Extension/Diagnostics/GetAzureVMDiagnosticsExtension.cs → src/ResourceManager/Compute/Commands.Compute/Extension/Diagnostics/GetAzureRmVMDiagnosticsExtension.cs

@@ -12,6 +12,7 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using System;
 using System.Linq;
 using System.Management.Automation;
 using Microsoft.Azure.Commands.Compute.Common;
@@ -26,7 +27,7 @@ namespace Microsoft.Azure.Commands.Compute
         ProfileNouns.VirtualMachineDiagnosticsExtension),
     OutputType(
         typeof(PSVirtualMachineExtension))]
-    public class GetAzureVMDiagnosticsExtensionCommand : VirtualMachineExtensionBaseCmdlet
+    public class GetAzureRmVMDiagnosticsExtensionCommand : VirtualMachineExtensionBaseCmdlet
     {
         [Parameter(
            Mandatory = true,
@@ -47,11 +48,10 @@ public class GetAzureVMDiagnosticsExtensionCommand : VirtualMachineExtensionBase
 
         [Alias("ExtensionName")]
         [Parameter(
-            Mandatory = true,
+            Mandatory = false,
             Position = 2,
             ValueFromPipelineByPropertyName = true,
             HelpMessage = "Extension Name.")]
-        [ValidateNotNullOrEmpty]
         public string Name { get; set; }
 
         [Parameter(
@@ -67,6 +67,25 @@ public override void ExecuteCmdlet()
 
             ExecuteClientAction(() =>
             {
+                if (string.IsNullOrEmpty(this.Name))
+                {
+                    var virtualMachine = ComputeClient.ComputeManagementClient.VirtualMachines.Get(this.ResourceGroupName, this.VMName).VirtualMachine;
+                    var diagnosticsExtension = virtualMachine.Extensions != null
+                            ? virtualMachine.Extensions.FirstOrDefault(extension =>
+                                extension.Publisher.Equals(DiagnosticsExtensionConstants.ExtensionPublisher, StringComparison.InvariantCultureIgnoreCase) &&
+                                extension.ExtensionType.Equals(DiagnosticsExtensionConstants.ExtensionType, StringComparison.InvariantCultureIgnoreCase))
+                            : null;
+
+                    if (diagnosticsExtension == null)
+                    {
+                        WriteObject(null);
+                        return;
+                    }
+                    else {
+                        this.Name = diagnosticsExtension.Name;
+                    }
+                }
+
                 VirtualMachineExtensionGetResponse virtualMachineExtensionGetResponse = null;
                 if (Status.IsPresent)
                 {

src/ResourceManager/Compute/Commands.Compute/Extension/Diagnostics/RemoveAzureVMDiagnosticsExtension.cs → src/ResourceManager/Compute/Commands.Compute/Extension/Diagnostics/RemoveAzureRmVMDiagnosticsExtension.cs

@@ -13,16 +13,18 @@
 // ----------------------------------------------------------------------------------
 
 using System;
+using System.Linq;
 using System.Management.Automation;
 using Microsoft.Azure.Commands.Compute.Common;
 using Microsoft.Azure.Management.Compute;
+using System.Globalization;
 
 namespace Microsoft.Azure.Commands.Compute
 {
     [Cmdlet(
         VerbsCommon.Remove,
         ProfileNouns.VirtualMachineDiagnosticsExtension)]
-    public class RemoveAzureVMDiagnosticsExtensionCommand : VirtualMachineExtensionBaseCmdlet
+    public class RemoveAzureRmVMDiagnosticsExtensionCommand : VirtualMachineExtensionBaseCmdlet
     {
         [Parameter(
            Mandatory = true,
@@ -43,11 +45,10 @@ public class RemoveAzureVMDiagnosticsExtensionCommand : VirtualMachineExtensionB
 
         [Alias("ExtensionName")]
         [Parameter(
-            Mandatory = true,
+            Mandatory = false,
             Position = 2,
             ValueFromPipelineByPropertyName = true,
             HelpMessage = "Extension Name.")]
-        [ValidateNotNullOrEmpty]
         public string Name { get; set; }
 
         public override void ExecuteCmdlet()
@@ -56,6 +57,26 @@ public override void ExecuteCmdlet()
 
             ExecuteClientAction(() =>
             {
+                if (string.IsNullOrEmpty(this.Name))
+                {
+                    var virtualMachine = ComputeClient.ComputeManagementClient.VirtualMachines.Get(this.ResourceGroupName, this.VMName).VirtualMachine;
+                    var diagnosticsExtension = virtualMachine.Extensions != null
+                            ? virtualMachine.Extensions.FirstOrDefault(extension =>
+                                extension.Publisher.Equals(DiagnosticsExtensionConstants.ExtensionPublisher, StringComparison.InvariantCultureIgnoreCase) &&
+                                extension.ExtensionType.Equals(DiagnosticsExtensionConstants.ExtensionType, StringComparison.InvariantCultureIgnoreCase))
+                            : null;
+
+                    if (diagnosticsExtension == null)
+                    {
+                        WriteWarning(string.Format(CultureInfo.InvariantCulture, Properties.Resources.DiagnosticsExtensionNotFound, this.ResourceGroupName, this.VMName));
+                        return;
+                    }
+                    else
+                    {
+                        this.Name = diagnosticsExtension.Name;
+                    }
+                }
+
                 var op = this.VirtualMachineExtensionClient.Delete(this.ResourceGroupName, this.VMName,
                     this.Name);
                 WriteObject(op);