-
Notifications
You must be signed in to change notification settings - Fork 70
/
custom_deny_action.go
294 lines (241 loc) · 9.25 KB
/
custom_deny_action.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
package botman
import (
"context"
"encoding/json"
"fmt"
"net/http"
"github.com/akamai/AkamaiOPEN-edgegrid-golang/v9/pkg/session"
validation "github.com/go-ozzo/ozzo-validation/v4"
)
type (
// The CustomDenyAction interface supports creating, retrieving, modifying and removing custom deny action for a
// configuration.
CustomDenyAction interface {
// GetCustomDenyActionList https://techdocs.akamai.com/bot-manager/reference/get-custom-deny-actions
GetCustomDenyActionList(ctx context.Context, params GetCustomDenyActionListRequest) (*GetCustomDenyActionListResponse, error)
// GetCustomDenyAction https://techdocs.akamai.com/bot-manager/reference/get-custom-deny-action
GetCustomDenyAction(ctx context.Context, params GetCustomDenyActionRequest) (map[string]interface{}, error)
// CreateCustomDenyAction https://techdocs.akamai.com/bot-manager/reference/post-custom-deny-action
CreateCustomDenyAction(ctx context.Context, params CreateCustomDenyActionRequest) (map[string]interface{}, error)
// UpdateCustomDenyAction https://techdocs.akamai.com/bot-manager/reference/put-custom-deny-action
UpdateCustomDenyAction(ctx context.Context, params UpdateCustomDenyActionRequest) (map[string]interface{}, error)
// RemoveCustomDenyAction https://techdocs.akamai.com/bot-manager/reference/delete-custom-deny-action
RemoveCustomDenyAction(ctx context.Context, params RemoveCustomDenyActionRequest) error
}
// GetCustomDenyActionListRequest is used to retrieve custom deny actions for a configuration.
GetCustomDenyActionListRequest struct {
ConfigID int64
Version int64
ActionID string
}
// GetCustomDenyActionListResponse is used to retrieve custom deny actions for a configuration.
GetCustomDenyActionListResponse struct {
CustomDenyActions []map[string]interface{} `json:"customDenyActions"`
}
// GetCustomDenyActionRequest is used to retrieve a specific custom deny action
GetCustomDenyActionRequest struct {
ConfigID int64
Version int64
ActionID string
}
// CreateCustomDenyActionRequest is used to create a new custom deny action for a specific configuration.
CreateCustomDenyActionRequest struct {
ConfigID int64
Version int64
JsonPayload json.RawMessage
}
// UpdateCustomDenyActionRequest is used to update an existing custom deny action
UpdateCustomDenyActionRequest struct {
ConfigID int64
Version int64
ActionID string
JsonPayload json.RawMessage
}
// RemoveCustomDenyActionRequest is used to remove an existing custom deny action
RemoveCustomDenyActionRequest struct {
ConfigID int64
Version int64
ActionID string
}
)
// Validate validates a GetCustomDenyActionRequest.
func (v GetCustomDenyActionRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
"ActionID": validation.Validate(v.ActionID, validation.Required),
}.Filter()
}
// Validate validates a GetCustomDenyActionListRequest.
func (v GetCustomDenyActionListRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
}.Filter()
}
// Validate validates a CreateCustomDenyActionRequest.
func (v CreateCustomDenyActionRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
"JsonPayload": validation.Validate(v.JsonPayload, validation.Required),
}.Filter()
}
// Validate validates an UpdateCustomDenyActionRequest.
func (v UpdateCustomDenyActionRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
"ActionID": validation.Validate(v.ActionID, validation.Required),
"JsonPayload": validation.Validate(v.JsonPayload, validation.Required),
}.Filter()
}
// Validate validates a RemoveCustomDenyActionRequest.
func (v RemoveCustomDenyActionRequest) Validate() error {
return validation.Errors{
"ConfigID": validation.Validate(v.ConfigID, validation.Required),
"Version": validation.Validate(v.Version, validation.Required),
"ActionID": validation.Validate(v.ActionID, validation.Required),
}.Filter()
}
func (b *botman) GetCustomDenyAction(ctx context.Context, params GetCustomDenyActionRequest) (map[string]interface{}, error) {
logger := b.Log(ctx)
logger.Debug("GetCustomDenyAction")
if err := params.Validate(); err != nil {
return nil, fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
uri := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/response-actions/custom-deny-actions/%s",
params.ConfigID,
params.Version,
params.ActionID)
req, err := http.NewRequestWithContext(ctx, http.MethodGet, uri, nil)
if err != nil {
return nil, fmt.Errorf("failed to create GetCustomDenyAction request: %w", err)
}
var result map[string]interface{}
resp, err := b.Exec(req, &result)
if err != nil {
return nil, fmt.Errorf("GetCustomDenyAction request failed: %w", err)
}
defer session.CloseResponseBody(resp)
if resp.StatusCode != http.StatusOK {
return nil, b.Error(resp)
}
return result, nil
}
func (b *botman) GetCustomDenyActionList(ctx context.Context, params GetCustomDenyActionListRequest) (*GetCustomDenyActionListResponse, error) {
logger := b.Log(ctx)
logger.Debug("GetCustomDenyActionList")
if err := params.Validate(); err != nil {
return nil, fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
uri := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/response-actions/custom-deny-actions",
params.ConfigID,
params.Version,
)
req, err := http.NewRequestWithContext(ctx, http.MethodGet, uri, nil)
if err != nil {
return nil, fmt.Errorf("failed to create GetlustomDenyList request: %w", err)
}
var result GetCustomDenyActionListResponse
resp, err := b.Exec(req, &result)
if err != nil {
return nil, fmt.Errorf("GetCustomDenyActionList request failed: %w", err)
}
defer session.CloseResponseBody(resp)
if resp.StatusCode != http.StatusOK {
return nil, b.Error(resp)
}
var filteredResult GetCustomDenyActionListResponse
if params.ActionID != "" {
for _, val := range result.CustomDenyActions {
if val["actionId"].(string) == params.ActionID {
filteredResult.CustomDenyActions = append(filteredResult.CustomDenyActions, val)
}
}
} else {
filteredResult = result
}
return &filteredResult, nil
}
func (b *botman) UpdateCustomDenyAction(ctx context.Context, params UpdateCustomDenyActionRequest) (map[string]interface{}, error) {
logger := b.Log(ctx)
logger.Debug("UpdateCustomDenyAction")
if err := params.Validate(); err != nil {
return nil, fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
putURL := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/response-actions/custom-deny-actions/%s",
params.ConfigID,
params.Version,
params.ActionID,
)
req, err := http.NewRequestWithContext(ctx, http.MethodPut, putURL, nil)
if err != nil {
return nil, fmt.Errorf("failed to create UpdateCustomDenyAction request: %w", err)
}
var result map[string]interface{}
resp, err := b.Exec(req, &result, params.JsonPayload)
if err != nil {
return nil, fmt.Errorf("UpdateCustomDenyAction request failed: %w", err)
}
defer session.CloseResponseBody(resp)
if resp.StatusCode != http.StatusOK {
return nil, b.Error(resp)
}
return result, nil
}
func (b *botman) CreateCustomDenyAction(ctx context.Context, params CreateCustomDenyActionRequest) (map[string]interface{}, error) {
logger := b.Log(ctx)
logger.Debug("CreateCustomDenyAction")
if err := params.Validate(); err != nil {
return nil, fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
uri := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/response-actions/custom-deny-actions",
params.ConfigID,
params.Version,
)
req, err := http.NewRequestWithContext(ctx, http.MethodPost, uri, nil)
if err != nil {
return nil, fmt.Errorf("failed to create CreateCustomDenyAction request: %w", err)
}
var result map[string]interface{}
resp, err := b.Exec(req, &result, params.JsonPayload)
if err != nil {
return nil, fmt.Errorf("CreateCustomDenyAction request failed: %w", err)
}
defer session.CloseResponseBody(resp)
if resp.StatusCode != http.StatusCreated {
return nil, b.Error(resp)
}
return result, nil
}
func (b *botman) RemoveCustomDenyAction(ctx context.Context, params RemoveCustomDenyActionRequest) error {
logger := b.Log(ctx)
logger.Debug("RemoveCustomDenyAction")
if err := params.Validate(); err != nil {
return fmt.Errorf("%w: %s", ErrStructValidation, err.Error())
}
uri := fmt.Sprintf(
"/appsec/v1/configs/%d/versions/%d/response-actions/custom-deny-actions/%s",
params.ConfigID,
params.Version,
params.ActionID)
req, err := http.NewRequestWithContext(ctx, http.MethodDelete, uri, nil)
if err != nil {
return fmt.Errorf("failed to create RemoveCustomDenyAction request: %w", err)
}
var result map[string]interface{}
resp, err := b.Exec(req, &result)
if err != nil {
return fmt.Errorf("RemoveCustomDenyAction request failed: %w", err)
}
defer session.CloseResponseBody(resp)
if resp.StatusCode != http.StatusNoContent {
return b.Error(resp)
}
return nil
}