Add EnabledState to CustomRule

Author: digavino

src/FrontDoor/FrontDoor/Cmdlets/NewAzureRmFrontDoorWafCustomRuleObject.cs

@@ -53,9 +53,9 @@ public class NewAzureRmFrontDoorWafCustomRuleObject : AzureFrontDoorCmdletBase
         public PSMatchCondition[] MatchCondition { get; set; }
 
         /// <summary>
-        /// Type of Actions. Possible values include: 'Allow', 'Block', 'Log'
+        /// Type of Actions. Possible values include: 'Allow', 'Block', 'Log', 'Redirect'
         /// </summary>
-        [Parameter(Mandatory = true, HelpMessage = "Type of Actions. Possible values include: 'Allow', 'Block', 'Log'. ")]
+        [Parameter(Mandatory = true, HelpMessage = "Type of Actions. Possible values include: 'Allow', 'Block', 'Log', 'Redirect'.")]
         [PSArgumentCompleter("Allow","Block","Log","Redirect")]
         public string Action { get; set; }
 
@@ -77,6 +77,13 @@ public class NewAzureRmFrontDoorWafCustomRuleObject : AzureFrontDoorCmdletBase
         [Parameter(Mandatory = false, HelpMessage = "Rate limit threshold")]
         public int? RateLimitThreshold { get; set; }
 
+        /// <summary>
+        /// Enabled State. Possible values include: 'Enabled', 'Disabled'
+        /// </summary>
+        [Parameter(Mandatory = false, HelpMessage = "Enabled State. Possible values include: 'Enabled', 'Disabled'.")]
+        [PSArgumentCompleter("Enabled", "Disabled")]
+        public string EnabledState { get; set; }
+
         public override void ExecuteCmdlet()
         {
             var CustomRule = new PSCustomRule
@@ -87,7 +94,8 @@ public override void ExecuteCmdlet()
                RuleType = RuleType,
                RateLimitDurationInMinutes = !this.IsParameterBound(c => c.RateLimitDurationInMinutes)? 1 : RateLimitDurationInMinutes,
                RateLimitThreshold = RateLimitThreshold,
-               Action = Action
+               Action = Action,
+               EnabledState = !this.IsParameterBound(c => c.EnabledState) ? "Enabled" : EnabledState
             };
             WriteObject(CustomRule);
         }

src/FrontDoor/FrontDoor/Helpers/ModelExtensions.cs

@@ -334,7 +334,8 @@ public static PSCustomRule ToPSCustomRule(this SdkCustomRule sdkRule)
                 Action = sdkRule.Action,
                 RuleType = sdkRule.RuleType,
                 Priority = sdkRule.Priority,
-                MatchConditions = sdkRule.MatchConditions?.Select(x => x.ToPSMatchCondition()).ToList()
+                MatchConditions = sdkRule.MatchConditions?.Select(x => x.ToPSMatchCondition()).ToList(),
+                EnabledState = sdkRule.EnabledState
             };
         }
 
@@ -448,7 +449,8 @@ public static SdkCustomRule ToSdkCustomRule(this PSCustomRule psRule)
                 Action = psRule.Action,
                 MatchConditions = psRule.MatchConditions?.Select(x => x.ToSdkMatchCondition()).ToList(),
                 Priority = psRule.Priority,
-                RuleType = psRule.RuleType
+                RuleType = psRule.RuleType,
+                EnabledState = psRule.EnabledState
             };
         }
 

src/FrontDoor/FrontDoor/Models/PSCustomRule.cs

@@ -31,5 +31,7 @@ public class PSCustomRule
         public int? RateLimitThreshold { get; set; }
 
         public string Name { get; set; }
+
+        public string EnabledState { get; set; }
     }
 }