package metabase import ( "encoding/json" "fmt" "net/http" "time" "github.com/crowdsecurity/go-cs-lib/version" "github.com/dghubble/sling" log "github.com/sirupsen/logrus" ) type MBClient struct { CTX *sling.Sling Client *http.Client } const ( sessionEndpoint = "login" scanEndpoint = "scan" resetPasswordEndpoint = "reset_password" userEndpoint = "user" databaseEndpoint = "database" ) var ( routes = map[string]string{ sessionEndpoint: "api/session", scanEndpoint: "api/database/2/rescan_values", resetPasswordEndpoint: "api/user/1/password", userEndpoint: "api/user/1", databaseEndpoint: "api/database/2", } ) func NewMBClient(url string) (*MBClient, error) { httpClient := &http.Client{Timeout: 20 * time.Second} return &MBClient{ CTX: sling.New().Client(httpClient).Base(url).Set("User-Agent", fmt.Sprintf("crowdsec/%s", version.String())), Client: httpClient, }, nil } func (h *MBClient) Do(method string, route string, body interface{}) (interface{}, interface{}, error) { var Success interface{} var Error interface{} var resp *http.Response var err error var data []byte if body != nil { data, _ = json.Marshal(body) } switch method { case "POST": log.Debugf("POST /%s", route) log.Debugf("%s", string(data)) resp, err = h.CTX.New().Post(route).BodyJSON(body).Receive(&Success, &Error) case "GET": log.Debugf("GET /%s", route) resp, err = h.CTX.New().Get(route).Receive(&Success, &Error) case "PUT": log.Debugf("PUT /%s", route) log.Debugf("%s", string(data)) resp, err = h.CTX.New().Put(route).BodyJSON(body).Receive(&Success, &Error) case "DELETE": default: return nil, nil, fmt.Errorf("unsupported method '%s'", method) } if Error != nil { return Success, Error, fmt.Errorf("http error: %v", Error) } if resp != nil && resp.StatusCode != 200 && resp.StatusCode != 202 { return Success, Error, fmt.Errorf("bad status code '%d': (success: %+v) | (error: %+v)", resp.StatusCode, Success, Error) } return Success, Error, err } // Set set headers as key:value func (h *MBClient) Set(key string, value string) { h.CTX = h.CTX.Set(key, value) }