(ctx context.Context, session *sources.Session, URL string, driftnetRequest *DriftnetRequest)
| 215 | } |
| 216 | |
| 217 | func (agent *Agent) queryURL(ctx context.Context, session *sources.Session, URL string, driftnetRequest *DriftnetRequest) (*http.Response, error) { |
| 218 | apiURL := fmt.Sprintf(URL, url.QueryEscape(driftnetRequest.From), processQuery(driftnetRequest.Query)) |
| 219 | |
| 220 | if driftnetRequest.Page > 0 { |
| 221 | pageStr := strconv.Itoa(driftnetRequest.Page) |
| 222 | apiURL = apiURL + "&page=" + pageStr |
| 223 | } |
| 224 | |
| 225 | request, err := sources.NewHTTPRequest(ctx, http.MethodGet, apiURL, nil) |
| 226 | if err != nil { |
| 227 | return nil, err |
| 228 | } |
| 229 | request.Header.Set("Accept", "application/json") |
| 230 | request.Header.Set("Authorization", `Bearer `+session.Keys.DriftnetToken) |
| 231 | return session.Do(request, agent.Name()) |
| 232 | } |
| 233 | |
| 234 | func processQuery(input string) string { |
| 235 | if iputil.IsIP(input) || iputil.IsCIDR(input) { |
no test coverage detected