Add new EdgeDeployment arguments (#51)

* Add new EdgeDeployment arguments * make CreateOrUpdateEdgeDeploymentServiceBody.ActivateVersion a pointer to bool * json.Marshal the CreateOrUpdateEdgeDeploymentServiceBody after sc.fastlyKey check --------- Co-authored-by: Vladimir Vuksan <[email protected]>
signalsciences · Mar 22, 2023 · 91cd112 · 91cd112
1 parent 635a5fd
commit 91cd112
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/api.go b/api.go
@@ -2901,14 +2901,24 @@ func (sc *Client) DeleteEdgeDeployment(corpName, siteName string) error {
 	return err
 }
 
+type CreateOrUpdateEdgeDeploymentServiceBody struct {
+	ActivateVersion *bool `json:"activateVersion,omitempty"` //Activate Fastly VCL service after clone
+	PercentEnabled  int   `json:"percentEnabled,omitempty"`  //Percentage of traffic to send to NGWAF@Edge
+}
+
 // CreateOrUpdateEdgeDeploymentService copies the backends from the Fastly service to the
 // Edge Deployment and pre-configures the Fastly service with an edge dictionary and custom VCL.
-func (sc *Client) CreateOrUpdateEdgeDeploymentService(corpName, siteName, fastlySID string) error {
+func (sc *Client) CreateOrUpdateEdgeDeploymentService(corpName, siteName, fastlySID string, body CreateOrUpdateEdgeDeploymentServiceBody) error {
 	if sc.fastlyKey == "" {
 		return errors.New("please set Fastly-Key with the client.SetFastlyKey method")
 	}
 
-	_, err := sc.doRequest("PUT", fmt.Sprintf("/v0/corps/%s/sites/%s/edgeDeployment/%s", corpName, siteName, fastlySID), "")
+	b, err := json.Marshal(body)
+	if err != nil {
+		return err
+	}
+
+	_, err = sc.doRequest("PUT", fmt.Sprintf("/v0/corps/%s/sites/%s/edgeDeployment/%s", corpName, siteName, fastlySID), string(b))
 
 	return err
 }