mirror of
https://github.com/fleetdm/fleet
synced 2026-04-21 21:47:20 +00:00
e8152e53fc
<!-- Add the related story/sub-task/bug number, like Resolves #123, or remove if NA --> # Checklist for submitter - [x] Changes file added for user-visible changes in `changes/`, `orbit/changes/` or `ee/fleetd-chrome/changes`. See [Changes files](https://github.com/fleetdm/fleet/blob/main/docs/Contributing/guides/committing-changes.md#changes-files) for more information. ## Testing - [ ] QA'd all new/changed functionality manually
104 lines
3.2 KiB
Go
104 lines
3.2 KiB
Go
package webhooks
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"github.com/fleetdm/fleet/v4/server"
|
|
"github.com/fleetdm/fleet/v4/server/contexts/ctxerr"
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/hashicorp/go-multierror"
|
|
)
|
|
|
|
func TriggerHostStatusWebhook(
|
|
ctx context.Context,
|
|
ds fleet.Datastore,
|
|
logger *slog.Logger,
|
|
) error {
|
|
multiErr := &multierror.Error{}
|
|
multiErr = multierror.Append(multiErr, triggerGlobalHostStatusWebhook(ctx, ds, logger))
|
|
multiErr = multierror.Append(multiErr, triggerTeamHostStatusWebhook(ctx, ds, logger))
|
|
return multiErr.ErrorOrNil()
|
|
}
|
|
|
|
func triggerGlobalHostStatusWebhook(ctx context.Context, ds fleet.Datastore, logger *slog.Logger) error {
|
|
appConfig, err := ds.AppConfig(ctx)
|
|
if err != nil {
|
|
return ctxerr.Wrap(ctx, err, "getting app config")
|
|
}
|
|
|
|
if !appConfig.WebhookSettings.HostStatusWebhook.Enable {
|
|
return nil
|
|
}
|
|
|
|
logger.DebugContext(ctx, "host status webhook triggered", "global", true)
|
|
|
|
return processWebhook(ctx, ds, nil, appConfig.WebhookSettings.HostStatusWebhook, logger)
|
|
}
|
|
|
|
func processWebhook(ctx context.Context, ds fleet.Datastore, teamID *uint, settings fleet.HostStatusWebhookSettings, logger *slog.Logger) error {
|
|
total, unseen, err := ds.TotalAndUnseenHostsSince(ctx, teamID, settings.DaysCount)
|
|
if err != nil {
|
|
return ctxerr.Wrap(ctx, err, "getting total and unseen hosts")
|
|
}
|
|
|
|
unseenCount := len(unseen)
|
|
percentUnseen := float64(unseenCount) * 100.0 / float64(total)
|
|
if percentUnseen >= settings.HostPercentage {
|
|
url := settings.DestinationURL
|
|
|
|
message := fmt.Sprintf(
|
|
"More than %.2f%% of your hosts have not checked into Fleet for more than %d days. "+
|
|
"You've been sent this message because the Host status webhook is enabled in your Fleet instance.",
|
|
percentUnseen, settings.DaysCount,
|
|
)
|
|
payload := map[string]interface{}{
|
|
"text": message,
|
|
"data": map[string]interface{}{
|
|
"unseen_hosts": unseenCount,
|
|
"total_hosts": total,
|
|
"days_unseen": settings.DaysCount,
|
|
"host_ids": unseen,
|
|
},
|
|
}
|
|
if teamID != nil {
|
|
payload["data"].(map[string]any)["fleet_id"] = *teamID
|
|
payload["data"].(map[string]any)["team_id"] = *teamID
|
|
}
|
|
|
|
err = server.PostJSONWithTimeout(ctx, url, &payload, logger)
|
|
if err != nil {
|
|
return ctxerr.Wrapf(ctx, err, "posting to %s", url)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func triggerTeamHostStatusWebhook(ctx context.Context, ds fleet.Datastore, logger *slog.Logger) error {
|
|
teams, err := ds.TeamsSummary(ctx)
|
|
if err != nil {
|
|
return ctxerr.Wrap(ctx, err, "getting teams summary")
|
|
}
|
|
// We try to send a webhook for each team. If one team fails, we continue.
|
|
multiErr := &multierror.Error{}
|
|
for _, teamSummary := range teams {
|
|
id := teamSummary.ID
|
|
team, err := ds.TeamLite(ctx, id)
|
|
if err != nil {
|
|
multiErr = multierror.Append(multiErr, ctxerr.Wrap(ctx, err, "getting team"))
|
|
continue
|
|
}
|
|
if team.Config.WebhookSettings.HostStatusWebhook == nil || !team.Config.WebhookSettings.HostStatusWebhook.Enable {
|
|
continue
|
|
}
|
|
logger.DebugContext(ctx, "host status webhook triggered", "fleet_id", id)
|
|
err = processWebhook(ctx, ds, &id, *team.Config.WebhookSettings.HostStatusWebhook, logger)
|
|
if err != nil {
|
|
multiErr = multierror.Append(multiErr, ctxerr.Wrap(ctx, err, "processing webhook"))
|
|
}
|
|
}
|
|
|
|
return multiErr.ErrorOrNil()
|
|
}
|