From 185d03c97a71f4a85b2795cc7bf405598a1da22c Mon Sep 17 00:00:00 2001
From: curro-bazan <79470107+curro-bazan@users.noreply.github.com>
Date: Fri, 14 Feb 2025 16:48:39 +0100
Subject: [PATCH] Client update to support edgedeployment in compute (#69)

---
 api.go | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/api.go b/api.go
index c3dd7b3..bcaadc1 100644
--- a/api.go
+++ b/api.go
@@ -2941,9 +2941,24 @@ func (sc *Client) GetEdgeDeployment(corpName, siteName string) (EdgeDeployment,
 }
 
 // CreateOrUpdateEdgeDeployment initializes the Next-Gen WAF deployment in Compute@Edge and configures the site for Edge Deployment.
-func (sc *Client) CreateOrUpdateEdgeDeployment(corpName, siteName string) error {
-	_, err := sc.doRequest("PUT", fmt.Sprintf("/v0/corps/%s/sites/%s/edgeDeployment", corpName, siteName), "")
-
+func (sc *Client) CreateOrUpdateEdgeDeployment(corpName, siteName string, authorizedServices []string) error {
+	payload := make(map[string]interface{})
+	if len(authorizedServices) > 0 {
+		payload["authorizedServices"] = authorizedServices
+	}
+	// Convert payload to JSON if it contains data
+	var b []byte
+	var err error
+	if len(payload) > 0 {
+		b, err = json.Marshal(payload)
+		if err != nil {
+			return err
+		}
+	}
+	_, err = sc.doRequest("PUT", fmt.Sprintf("/v0/corps/%s/sites/%s/edgeDeployment", corpName, siteName), string(b))
+	if err != nil {
+		return err
+	}
 	return err
 }