Update VMSS models/NewAZVMss to include APAE, Update Set Configs for VM + VMSS for APAE

Author: ookoka

src/Compute/Compute/Generated/VirtualMachineScaleSet/Config/SetAzVmssProxyAgentSetting.cs

@@ -69,6 +69,12 @@ public class SetAzureVmssProxySetting : Microsoft.Azure.Commands.ResourceManager
             HelpMessage = "Specifies the InVMAccessControlProfileVersion resource id in the IMDS enpoint. Format of /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/inVMAccessControlProfiles/{profile}/versions/{version}")]
         public string ImdsProfile { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "Specifies whether to add installation of the ProxyAgent extension.")]
+        public bool? AddProxyAgentExtension { get; set; }
+
         public override void ExecuteCmdlet()
         {
             if (this.VirtualMachineScaleSet.VirtualMachineProfile == null)
@@ -96,6 +102,11 @@ public override void ExecuteCmdlet()
                 })
             };
 
+            if (this.IsParameterBound(c => c.AddProxyAgentExtension))
+            {
+                this.VirtualMachineScaleSet.VirtualMachineProfile.SecurityProfile.ProxyAgentSettings.AddProxyAgentExtension = this.AddProxyAgentExtension;
+            }
+
             WriteObject(this.VirtualMachineScaleSet);
         }
     }

src/Compute/Compute/Manual/PSVirtualMachineScaleSetVMProfile.cs

@@ -41,5 +41,6 @@ public class PSVirtualMachineScaleSetVMProfile
         public PSApplicationProfile ApplicationProfile { get; set; }
         public SecurityPostureReference SecurityPostureReference { get; set; }
         public System.DateTime? TimeCreated { get; private set; }
+        public bool? AddProxyAgentExtension { get; set; }
     }
 }

src/Compute/Compute/Manual/VirtualMachineScaleSetCreateOrUpdateMethod.cs

@@ -292,6 +292,13 @@ public partial class NewAzureRmVmss : ComputeAutomationBaseCmdlet
             HelpMessage = "Specifies whether Metadata Security Protocol(ProxyAgent) feature should be enabled or not.")]
         public SwitchParameter EnableProxyAgent { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            ParameterSetName = SimpleParameterSet,
+            HelpMessage = "Specifies whether to add installation of the ProxyAgent extension.")]
+        public SwitchParameter AddProxyAgentExtension { get; set; }
+
         private void ConfigureSecuritySettings()
         {
             if (SecurityType?.ToLower() == SecurityTypes.TrustedLaunch ||

src/Compute/Compute/Strategies/ComputeRp/VirtualMachineScaleSetStrategy.cs

@@ -252,7 +252,8 @@ internal static ResourceConfig<VirtualMachineScaleSet> CreateVirtualMachineScale
             string ifNoneMatch = null,
             string securityPostureId = null,
             string[] securityPostureExcludeExtension = null,
-            bool? enableProxyAgent = null
+            bool? enableProxyAgent = null,
+            bool? addProxyAgentExtension = null,
             )
             => Strategy.CreateResourceConfig(
                 resourceGroup: resourceGroup,
@@ -285,7 +286,7 @@ internal static ResourceConfig<VirtualMachineScaleSet> CreateVirtualMachineScale
                                 EncryptionAtHost = encryptionAtHost,
                                 UefiSettings = (enableVtpm != null || enableSecureBoot != null) ? new UefiSettings(enableSecureBoot, enableVtpm) : null,
                                 SecurityType = securityType,
-                                ProxyAgentSettings = enableProxyAgent == true ? new ProxyAgentSettings(enableProxyAgent) : null
+                                ProxyAgentSettings = (enableProxyAgent == true || addProxyAgentExtension == true) ? new ProxyAgentSettings(enableProxyAgent) { AddProxyAgentExtension = addProxyAgentExtension } : null,
                             },
                             OsProfile = new VirtualMachineScaleSetOSProfile
                             {

src/Compute/Compute/VirtualMachine/Config/SetAzVMProxyAgentSetting.cs

@@ -73,6 +73,12 @@ public class SetAzureVMProxySetting : Microsoft.Azure.Commands.ResourceManager.C
             HelpMessage = "Increase the value of this parameter allows users to reset the key used for securing communication channel between guest and host.")]
         public int? KeyIncarnationId { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "Specifies whether to add installation of the ProxyAgent extension.")]
+        public bool? AddProxyAgentExtension { get; set; }
+
         public override void ExecuteCmdlet()
         {
             if (this.VM.SecurityProfile == null)
@@ -95,6 +101,12 @@ public override void ExecuteCmdlet()
                 KeyIncarnationId = this.KeyIncarnationId
             };
 
+            if (this.IsParameterBound(c => c.AddProxyAgentExtension))
+            {
+                this.VM.SecurityProfile.ProxyAgentSettings.AddProxyAgentExtension = this.AddProxyAgentExtension;
+            }
+
+
             WriteObject(this.VM);
         }
     }