146 lines
4.3 KiB
Go
146 lines
4.3 KiB
Go
package apiclient
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/models"
|
|
qs "github.com/google/go-querystring/query"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// type ApiAlerts service
|
|
|
|
type AlertsService service
|
|
|
|
type AlertsListOpts struct {
|
|
ScopeEquals *string `url:"scope,omitempty"`
|
|
ValueEquals *string `url:"value,omitempty"`
|
|
ScenarioEquals *string `url:"scenario,omitempty"`
|
|
IPEquals *string `url:"ip,omitempty"`
|
|
RangeEquals *string `url:"range,omitempty"`
|
|
OriginEquals *string `url:"origin,omitempty"`
|
|
Since *string `url:"since,omitempty"`
|
|
TypeEquals *string `url:"decision_type,omitempty"`
|
|
Until *string `url:"until,omitempty"`
|
|
IncludeSimulated *bool `url:"simulated,omitempty"`
|
|
ActiveDecisionEquals *bool `url:"has_active_decision,omitempty"`
|
|
IncludeCAPI *bool `url:"include_capi,omitempty"`
|
|
Limit *int `url:"limit,omitempty"`
|
|
Contains *bool `url:"contains,omitempty"`
|
|
ListOpts
|
|
}
|
|
|
|
type AlertsDeleteOpts struct {
|
|
ScopeEquals *string `url:"scope,omitempty"`
|
|
ValueEquals *string `url:"value,omitempty"`
|
|
ScenarioEquals *string `url:"scenario,omitempty"`
|
|
IPEquals *string `url:"ip,omitempty"`
|
|
RangeEquals *string `url:"range,omitempty"`
|
|
Since *string `url:"since,omitempty"`
|
|
Until *string `url:"until,omitempty"`
|
|
OriginEquals *string `url:"origin,omitempty"`
|
|
ActiveDecisionEquals *bool `url:"has_active_decision,omitempty"`
|
|
SourceEquals *string `url:"alert_source,omitempty"`
|
|
Contains *bool `url:"contains,omitempty"`
|
|
Limit *int `url:"limit,omitempty"`
|
|
ListOpts
|
|
}
|
|
|
|
func (s *AlertsService) Add(ctx context.Context, alerts models.AddAlertsRequest) (*models.AddAlertsResponse, *Response, error) {
|
|
|
|
var added_ids models.AddAlertsResponse
|
|
|
|
u := fmt.Sprintf("%s/alerts", s.client.URLPrefix)
|
|
req, err := s.client.NewRequest(http.MethodPost, u, &alerts)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, &added_ids)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
return &added_ids, resp, nil
|
|
}
|
|
|
|
// to demo query arguments
|
|
func (s *AlertsService) List(ctx context.Context, opts AlertsListOpts) (*models.GetAlertsResponse, *Response, error) {
|
|
var alerts models.GetAlertsResponse
|
|
var URI string
|
|
u := fmt.Sprintf("%s/alerts", s.client.URLPrefix)
|
|
params, err := qs.Values(opts)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "building query")
|
|
}
|
|
if len(params) > 0 {
|
|
URI = fmt.Sprintf("%s?%s", u, params.Encode())
|
|
} else {
|
|
URI = u
|
|
}
|
|
|
|
req, err := s.client.NewRequest(http.MethodGet, URI, nil)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "building request")
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, &alerts)
|
|
if err != nil {
|
|
return nil, resp, errors.Wrap(err, "performing request")
|
|
}
|
|
return &alerts, resp, nil
|
|
}
|
|
|
|
// to demo query arguments
|
|
func (s *AlertsService) Delete(ctx context.Context, opts AlertsDeleteOpts) (*models.DeleteAlertsResponse, *Response, error) {
|
|
var alerts models.DeleteAlertsResponse
|
|
params, err := qs.Values(opts)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
u := fmt.Sprintf("%s/alerts?%s", s.client.URLPrefix, params.Encode())
|
|
|
|
req, err := s.client.NewRequest(http.MethodDelete, u, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, &alerts)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
return &alerts, resp, nil
|
|
}
|
|
|
|
func (s *AlertsService) DeleteOne(ctx context.Context, alert_id string) (*models.DeleteAlertsResponse, *Response, error) {
|
|
var alerts models.DeleteAlertsResponse
|
|
u := fmt.Sprintf("%s/alerts/%s", s.client.URLPrefix, alert_id)
|
|
|
|
req, err := s.client.NewRequest(http.MethodDelete, u, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, &alerts)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
return &alerts, resp, nil
|
|
}
|
|
|
|
func (s *AlertsService) GetByID(ctx context.Context, alertID int) (*models.Alert, *Response, error) {
|
|
var alert models.Alert
|
|
u := fmt.Sprintf("%s/alerts/%d", s.client.URLPrefix, alertID)
|
|
|
|
req, err := s.client.NewRequest(http.MethodGet, u, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, &alert)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return &alert, resp, nil
|
|
}
|