allow to configure rule group capacity (#7)

Author: blotus

config.go

@@ -39,6 +39,7 @@ type AclConfig struct {
 	AWSProfile       string `yaml:"aws_profile"`
 	IPHeader         string `yaml:"ip_header"`
 	IPHeaderPosition string `yaml:"ip_header_position"`
+	Capacity         int    `yaml:"capacity"`
 }
 
 var validActions = []string{"ban", "captcha"}
@@ -93,6 +94,12 @@ func getConfigFromEnv(config *bouncerConfig) {
 					acl.IPHeader = value
 				case "IP_HEADER_POSITION":
 					acl.IPHeaderPosition = value
+				case "CAPACITY":
+					acl.Capacity, err = strconv.Atoi(value)
+					if err != nil {
+						log.Warnf("Invalid value for %s: %s", key, value)
+						acl.Capacity = 300
+					}
 				}
 			} else {
 				switch key {
@@ -262,6 +269,10 @@ func newConfig(configPath string) (bouncerConfig, error) {
 		} else {
 			ruleGroupNames[c.RuleGroupName] = true
 		}
+
+		if c.Capacity == 0 {
+			c.Capacity = 300
+		}
 	}
 	return config, nil
 }

waf.go

@@ -116,7 +116,7 @@ func (w *WAF) CreateRuleGroup(ruleGroupName string) error {
 				},
 			},
 			Scope:    aws.String(w.config.Scope),
-			Capacity: aws.Int64(300), //FIXME: update it dynamically
+			Capacity: aws.Int64(int64(w.config.Capacity)), //FIXME: Automatically set capacity if not provided by the user
 			VisibilityConfig: &wafv2.VisibilityConfig{
 				SampledRequestsEnabled:   aws.Bool(false),
 				CloudWatchMetricsEnabled: aws.Bool(false),