78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package apiclient
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/models"
|
|
)
|
|
|
|
// type ApiAlerts service
|
|
|
|
type AuthService service
|
|
|
|
// Don't add it to the models, as they are used with LAPI, but the enroll endpoint is specific to CAPI
|
|
type enrollRequest struct {
|
|
EnrollKey string `json:"attachment_key"`
|
|
Name string `json:"name"`
|
|
Tags []string `json:"tags"`
|
|
}
|
|
|
|
func (s *AuthService) UnregisterWatcher(ctx context.Context) (*Response, error) {
|
|
|
|
u := fmt.Sprintf("%s/watchers", s.client.URLPrefix)
|
|
req, err := s.client.NewRequest("DELETE", u, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, nil)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func (s *AuthService) RegisterWatcher(ctx context.Context, registration models.WatcherRegistrationRequest) (*Response, error) {
|
|
|
|
u := fmt.Sprintf("%s/watchers", s.client.URLPrefix)
|
|
|
|
req, err := s.client.NewRequest("POST", u, ®istration)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, nil)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func (s *AuthService) AuthenticateWatcher(ctx context.Context, auth models.WatcherAuthRequest) (*Response, error) {
|
|
u := fmt.Sprintf("%s/watchers/login", s.client.URLPrefix)
|
|
req, err := s.client.NewRequest("POST", u, &auth)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, nil)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func (s *AuthService) EnrollWatcher(ctx context.Context, enrollKey string, name string, tags []string) (*Response, error) {
|
|
u := fmt.Sprintf("%s/watchers/enroll", s.client.URLPrefix)
|
|
req, err := s.client.NewRequest("POST", u, &enrollRequest{EnrollKey: enrollKey, Name: name, Tags: tags})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.client.Do(ctx, req, nil)
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
return resp, nil
|
|
}
|