mirror of
https://github.com/crowdsecurity/crowdsec.git
synced 2025-05-11 04:15:54 +02:00
135 lines
3.1 KiB
Go
135 lines
3.1 KiB
Go
package apiclient
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
qs "github.com/google/go-querystring/query"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/models"
|
|
)
|
|
|
|
type AllowlistsService service
|
|
|
|
type AllowlistListOpts struct {
|
|
WithContent bool `url:"with_content,omitempty"`
|
|
}
|
|
|
|
func (s *AllowlistsService) List(ctx context.Context, opts AllowlistListOpts) (*models.GetAllowlistsResponse, *Response, error) {
|
|
u := s.client.URLPrefix + "/allowlists"
|
|
|
|
params, err := qs.Values(opts)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("building query: %w", err)
|
|
}
|
|
|
|
u += "?" + params.Encode()
|
|
|
|
req, err := s.client.PrepareRequest(ctx, http.MethodGet, u, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
allowlists := &models.GetAllowlistsResponse{}
|
|
|
|
resp, err := s.client.Do(ctx, req, allowlists)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
|
|
return allowlists, resp, nil
|
|
}
|
|
|
|
type AllowlistGetOpts struct {
|
|
WithContent bool `url:"with_content,omitempty"`
|
|
}
|
|
|
|
func (s *AllowlistsService) Get(ctx context.Context, name string, opts AllowlistGetOpts) (*models.GetAllowlistResponse, *Response, error) {
|
|
u := s.client.URLPrefix + "/allowlists/" + name
|
|
|
|
params, err := qs.Values(opts)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("building query: %w", err)
|
|
}
|
|
|
|
u += "?" + params.Encode()
|
|
|
|
log.Debugf("GET %s", u)
|
|
|
|
req, err := s.client.PrepareRequest(ctx, http.MethodGet, u, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
allowlist := &models.GetAllowlistResponse{}
|
|
|
|
resp, err := s.client.Do(ctx, req, allowlist)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
|
|
return allowlist, resp, nil
|
|
}
|
|
|
|
func (s *AllowlistsService) CheckIfAllowlisted(ctx context.Context, value string) (bool, *Response, error) {
|
|
escapedValue := url.PathEscape(value)
|
|
u := s.client.URLPrefix + "/allowlists/check/" + escapedValue
|
|
|
|
req, err := s.client.PrepareRequest(ctx, http.MethodHead, u, nil)
|
|
if err != nil {
|
|
return false, nil, err
|
|
}
|
|
|
|
var discardBody any
|
|
|
|
resp, err := s.client.Do(ctx, req, discardBody)
|
|
if err != nil {
|
|
return false, resp, err
|
|
}
|
|
|
|
return resp.Response.StatusCode == http.StatusOK, resp, nil
|
|
}
|
|
|
|
func (s *AllowlistsService) CheckIfAllowlistedWithReason(ctx context.Context, value string) (*models.CheckAllowlistResponse, *Response, error) {
|
|
escapedValue := url.PathEscape(value)
|
|
u := s.client.URLPrefix + "/allowlists/check/" + escapedValue
|
|
|
|
req, err := s.client.PrepareRequest(ctx, http.MethodGet, u, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
body := &models.CheckAllowlistResponse{}
|
|
|
|
resp, err := s.client.Do(ctx, req, body)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
|
|
return body, resp, nil
|
|
}
|
|
|
|
func (s *AllowlistsService) CheckIfAllowlistedBulk(ctx context.Context, values []string) (*models.BulkCheckAllowlistResponse, *Response, error) {
|
|
u := s.client.URLPrefix + "/allowlists/check"
|
|
|
|
body := &models.BulkCheckAllowlistRequest{
|
|
Targets: values,
|
|
}
|
|
|
|
req, err := s.client.PrepareRequest(ctx, http.MethodPost, u, body)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
responseBody := &models.BulkCheckAllowlistResponse{}
|
|
|
|
resp, err := s.client.Do(ctx, req, responseBody)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
|
|
return responseBody, resp, nil
|
|
}
|