// Update a WAF Policy returns "OK" response
package main
import (
"context"
"encoding/json"
"fmt"
"os"
"github.com/DataDog/datadog-api-client-go/v2/api/datadog"
"github.com/DataDog/datadog-api-client-go/v2/api/datadogV2"
)
func main() {
body := datadogV2.ApplicationSecurityPolicyUpdateRequest{
Data: datadogV2.ApplicationSecurityPolicyUpdateData{
Attributes: datadogV2.ApplicationSecurityPolicyUpdateAttributes{
Description: "Policy applied to internal web applications.",
IsDefault: false,
Name: "Internal Network Policy",
ProtectionPresets: []string{
"attack-tools",
},
Rules: []datadogV2.ApplicationSecurityPolicyRuleOverride{
{
Blocking: false,
Enabled: true,
ExtendedDataCollection: datadog.PtrBool(false),
Id: "rasp-001-002",
},
},
Rulesets: []datadogV2.ApplicationSecurityPolicyRulesetOverride{
{
Blocking: false,
Enabled: true,
Id: "attack_tool",
},
},
Scope: []datadogV2.ApplicationSecurityPolicyScope{
{
Env: "prod",
Service: "billing-service",
},
},
Version: 0,
},
Type: datadogV2.APPLICATIONSECURITYPOLICYTYPE_POLICY,
},
}
ctx := datadog.NewDefaultContext(context.Background())
ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
configuration := datadog.NewConfiguration()
apiClient := datadog.NewAPIClient(configuration)
api := datadogV2.NewApplicationSecurityApi(apiClient)
resp, r, err := api.UpdateApplicationSecurityWafPolicy(ctx, "policy_id", body)
if err != nil {
fmt.Fprintf(os.Stderr, "Error when calling `ApplicationSecurityApi.UpdateApplicationSecurityWafPolicy`: %v\n", err)
fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
}
responseContent, _ := json.MarshalIndent(resp, "", " ")
fmt.Fprintf(os.Stdout, "Response from `ApplicationSecurityApi.UpdateApplicationSecurityWafPolicy`:\n%s\n", responseContent)
}