fix: include proposedCidrs from akamai siteshield map

schnoddelbotz · Nov 14, 2017 · 2fff299 · 2fff299
1 parent 6e4406a
commit 2fff299
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/akamai.go b/akamai.go
@@ -16,9 +16,10 @@ type siteShieldMaps struct {
 }
 
 type siteShieldMap struct {
-	Acknowledged bool     `json:"acknowledged"`
-	ID           int      `json:"id"`
-	CurrentCidrs []string `json:"currentCidrs"`
+	Acknowledged  bool     `json:"acknowledged"`
+	ID            int      `json:"id"`
+	CurrentCidrs  []string `json:"currentCidrs"`
+	ProposedCidrs []string `json:"proposedCidrs"`
 }
 
 func getAkamaiConfig() edgegrid.Config {

diff --git a/main.go b/main.go
@@ -60,6 +60,12 @@ func run() {
 		for _, cidr := range ssMap.CurrentCidrs {
 			ssCidrs[cidr] = struct{}{}
 		}
+		// add proposed (i.e. non-acknowledged) CIDRs
+		for _, cidr := range ssMap.ProposedCidrs {
+			if _, ok := ssCidrs[cidr]; !ok {
+				ssCidrs[cidr] = struct{}{}
+			}
+		}
 	}
 
 	// compare current with desired state