fleet/server/service/client_targets.go
Tomas Touceda a18e09b613
Simplify fleetctl implementation and improve testing (#3830)
* Simplify fleetctl implementation and improve testing

* Add a few more

* Handle not founds better

* Fix tests

* Check that logout ds func is called
2022-01-24 16:40:51 -03:00

40 lines
1,017 B
Go

package service
import (
"fmt"
"github.com/fleetdm/fleet/v4/server/fleet"
)
// SearchTargets searches for the supplied targets in the Fleet instance.
func (c *Client) SearchTargets(query string, hostIDs, labelIDs []uint) (*fleet.TargetSearchResults, error) {
req := searchTargetsRequest{
MatchQuery: query,
Selected: fleet.HostTargets{
LabelIDs: labelIDs,
HostIDs: hostIDs,
// TODO handle TeamIDs
},
}
verb, path := "POST", "/api/v1/fleet/targets"
var responseBody searchTargetsResponse
err := c.authenticatedRequest(req, verb, path, &responseBody)
if err != nil {
return nil, fmt.Errorf("SearchTargets: %s", err)
}
hosts := make([]*fleet.Host, len(responseBody.Targets.Hosts))
for i, h := range responseBody.Targets.Hosts {
hosts[i] = h.HostResponse.Host
}
labels := make([]*fleet.Label, len(responseBody.Targets.Labels))
for i, h := range responseBody.Targets.Labels {
labels[i] = h.Label
}
return &fleet.TargetSearchResults{
Hosts: hosts,
Labels: labels,
}, nil
}