VNGCommands

Author: Khushboo Baheti

src/Network/Network/Models/PSVirtualNetworkGateway.cs

@@ -54,6 +54,9 @@ public class PSVirtualNetworkGateway : PSTopLevelResource
 
         public List<PSVirtualNetworkGatewayNatRule> NatRules { get; set; }
 
+        [Ps1Xml(Target = ViewControl.Table)]
+        public bool EnableBgpRouteTranslationForNat { get; set; }
+
         [JsonIgnore]
         public string IpConfigurationsText
         {

src/Network/Network/VirtualNetworkGateway/NewAzureVirtualNetworkGatewayCommand.cs

@@ -199,6 +199,17 @@ public class NewAzureVirtualNetworkGatewayCommand : VirtualNetworkGatewayBaseCmd
         [ValidateNotNullOrEmpty]
         public PSIpConfigurationBgpPeeringAddress[] IpConfigurationBgpPeeringAddresses { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "The NatRules for Virtual network gateway.")]
+        public PSVirtualNetworkGatewayNatRule[] NatRule { get; set; }
+
+        [Parameter(
+            Mandatory = false,
+            HelpMessage = "Flag to enable BgpRouteTranslationForNat on this VirtualNetworkGateway.")]
+        public SwitchParameter EnableBgpRouteTranslationForNatFlag { get; set; }
+
         [Parameter(
             Mandatory = false,
             ValueFromPipelineByPropertyName = true,
@@ -512,6 +523,18 @@ private PSVirtualNetworkGateway CreateVirtualNetworkGateway()
                 vnetGateway.VpnGatewayGeneration = this.VpnGatewayGeneration;
             }
 
+            if (this.NatRule != null && this.NatRule.Any())
+            {
+                vnetGateway.NatRules = this.NatRule?.ToList();
+            }
+            else
+            {
+                vnetGateway.NatRules = null;
+            }
+
+            // Set the EnableBgpRouteTranslationForNat, if it is specified by customer.
+            vnetGateway.EnableBgpRouteTranslationForNat = EnableBgpRouteTranslationForNatFlag.IsPresent;
+
             // Map to the sdk object
             var vnetGatewayModel = NetworkResourceManagerProfile.Mapper.Map<MNM.VirtualNetworkGateway>(vnetGateway);
             vnetGatewayModel.Tags = TagsConversionHelper.CreateTagDictionary(this.Tag, validate: true);

src/Network/Network/VirtualNetworkGateway/UpdateAzureVirtualNetworkGatewayCommand.cs

@@ -201,6 +201,22 @@ public class SetAzureVirtualNetworkGatewayCommand : VirtualNetworkGatewayBaseCmd
                     HelpMessage = "Custom routes AddressPool specified by customer")]
         public string[] CustomRoute { get; set; }
 
+        [Parameter(
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true,
+            HelpMessage = "The NatRules for Virtual network gateway.")]
+        public PSVirtualNetworkGatewayNatRule[] NatRule { get; set; }
+
+        [Parameter(
+            Mandatory = false,
+            HelpMessage = "Flag to enable BgpRouteTranslationForNat on this VirtualNetworkGateway.")]
+        public SwitchParameter EnableBgpRouteTranslationForNatFlag { get; set; }
+
+        [Parameter(
+            Mandatory = false,
+            HelpMessage = "Flag to disable BgpRouteTranslationForNat on this VirtualNetworkGateway.")]
+        public SwitchParameter DisableBgpRouteTranslationForNatFlag { get; set; }
+
         [Parameter(
             Mandatory = true,
             ParameterSetName = VirtualNetworkGatewayParameterSets.UpdateResourceWithTags,
@@ -442,6 +458,22 @@ public override void Execute()
                 this.VirtualNetworkGateway.CustomRoutes = null;
             }
 
+            if (this.NatRule != null)
+            {
+                this.VirtualNetworkGateway.NatRules = this.NatRule?.ToList();
+            }
+
+            if (this.EnableBgpRouteTranslationForNatFlag.IsPresent)
+            {
+                this.VirtualNetworkGateway.EnableBgpRouteTranslationForNat = true;
+            }
+
+            if (this.DisableBgpRouteTranslationForNatFlag.IsPresent)
+            {
+                this.VirtualNetworkGateway.EnableBgpRouteTranslationForNat = false;
+            }
+
+
             // Map to the sdk object
             MNM.VirtualNetworkGateway sdkVirtualNetworkGateway = NetworkResourceManagerProfile.Mapper.Map<MNM.VirtualNetworkGateway>(this.VirtualNetworkGateway);
             sdkVirtualNetworkGateway.Tags =