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
|
package client
import (
"encoding/json"
"fmt"
"git.db.org.ai/dborg/internal/models"
"net/url"
)
func (c *Client) SearchStealerLogs(params *models.SLParams) (*models.SLResponse, error) {
queryParams := url.Values{}
queryParams.Add("query", params.Query)
if params.MaxHits > 0 && params.MaxHits != 10 {
queryParams.Add("max_hits", fmt.Sprintf("%d", params.MaxHits))
}
if params.SortBy != "" {
queryParams.Add("sort_by", params.SortBy)
}
if params.IngestStartDate != "" {
queryParams.Add("ingest_start_date", params.IngestStartDate)
}
if params.IngestEndDate != "" {
queryParams.Add("ingest_end_date", params.IngestEndDate)
}
if params.PostedStartDate != "" {
queryParams.Add("posted_start_date", params.PostedStartDate)
}
if params.PostedEndDate != "" {
queryParams.Add("posted_end_date", params.PostedEndDate)
}
if params.Format != "" && params.Format != "json" {
queryParams.Add("format", params.Format)
}
data, err := c.Get("/sl/search", queryParams)
if err != nil {
return nil, err
}
if params.Format != "" && params.Format != "json" {
return &models.SLResponse{
Message: string(data),
}, nil
}
var response models.SLResponse
if err := json.Unmarshal(data, &response); err != nil {
return nil, fmt.Errorf("failed to parse stealer logs response: %w", err)
}
return &response, nil
}
|
