mirror of
https://github.com/fleetdm/fleet
synced 2026-05-23 17:08:53 +00:00
ad99cbd499
# Checklist for submitter If some of the following don't apply, delete the relevant line. <!-- Note that API documentation changes are now addressed by the product design team. --> - [ ] 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/Committing-Changes.md#changes-files) for more information. - [ ] Input data is properly validated, `SELECT *` is avoided, SQL injection is prevented (using placeholders for values in statements) - [ ] Added support on fleet's osquery simulator `cmd/osquery-perf` for new osquery data ingestion features. - [ ] Added/updated tests - [ ] If paths of existing endpoints are modified without backwards compatibility, checked the frontend/CLI for any necessary changes - [ ] If database migrations are included, checked table schema to confirm autoupdate - For database migrations: - [ ] Checked schema for all modified table for columns that will auto-update timestamps during migration. - [ ] Confirmed that updating the timestamps is acceptable, and will not cause unwanted side effects. - [ ] Ensured the correct collation is explicitly set for character columns (`COLLATE utf8mb4_unicode_ci`). - [ ] Manual QA for all new/changed functionality - For Orbit and Fleet Desktop changes: - [ ] Orbit runs on macOS, Linux and Windows. Check if the orbit feature/bugfix should only apply to one platform (`runtime.GOOS`). - [ ] Manual QA must be performed in the three main OSs, macOS, Windows and Linux. - [ ] Auto-update manual QA, from released version of component to new version (see [tools/tuf/test](../tools/tuf/test/README.md)).
121 lines
3.9 KiB
Go
121 lines
3.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/fleetdm/fleet/v4/server/mdm/maintainedapps"
|
|
)
|
|
|
|
type addFleetMaintainedAppRequest struct {
|
|
TeamID *uint `json:"team_id"`
|
|
AppID uint `json:"fleet_maintained_app_id"`
|
|
InstallScript string `json:"install_script"`
|
|
PreInstallQuery string `json:"pre_install_query"`
|
|
PostInstallScript string `json:"post_install_script"`
|
|
SelfService bool `json:"self_service"`
|
|
UninstallScript string `json:"uninstall_script"`
|
|
}
|
|
|
|
type addFleetMaintainedAppResponse struct {
|
|
Err error `json:"error,omitempty"`
|
|
}
|
|
|
|
func (r addFleetMaintainedAppResponse) error() error { return r.Err }
|
|
|
|
func addFleetMaintainedAppEndpoint(ctx context.Context, request interface{}, svc fleet.Service) (errorer, error) {
|
|
req := request.(*addFleetMaintainedAppRequest)
|
|
ctx, cancel := context.WithTimeout(ctx, maintainedapps.InstallerTimeout)
|
|
defer cancel()
|
|
err := svc.AddFleetMaintainedApp(
|
|
ctx,
|
|
req.TeamID,
|
|
req.AppID,
|
|
req.InstallScript,
|
|
req.PreInstallQuery,
|
|
req.PostInstallScript,
|
|
req.UninstallScript,
|
|
req.SelfService,
|
|
)
|
|
if err != nil {
|
|
if errors.Is(err, context.DeadlineExceeded) {
|
|
err = fleet.NewGatewayTimeoutError("Couldn't upload. Request timeout. Please make sure your server and load balancer timeout is long enough.", err)
|
|
}
|
|
|
|
return &addFleetMaintainedAppResponse{Err: err}, nil
|
|
}
|
|
return &addFleetMaintainedAppResponse{}, nil
|
|
}
|
|
|
|
func (svc *Service) AddFleetMaintainedApp(ctx context.Context, teamID *uint, appID uint, installScript, preInstallQuery, postInstallScript, uninstallScript string, selfService bool) error {
|
|
// skipauth: No authorization check needed due to implementation returning
|
|
// only license error.
|
|
svc.authz.SkipAuthorization(ctx)
|
|
|
|
return fleet.ErrMissingLicense
|
|
}
|
|
|
|
type listFleetMaintainedAppsRequest struct {
|
|
fleet.ListOptions
|
|
TeamID uint `query:"team_id"`
|
|
}
|
|
|
|
type listFleetMaintainedAppsResponse struct {
|
|
FleetMaintainedApps []fleet.MaintainedApp `json:"fleet_maintained_apps"`
|
|
Meta *fleet.PaginationMetadata `json:"meta"`
|
|
Err error `json:"error,omitempty"`
|
|
}
|
|
|
|
func (r listFleetMaintainedAppsResponse) error() error { return r.Err }
|
|
|
|
func listFleetMaintainedApps(ctx context.Context, request any, svc fleet.Service) (errorer, error) {
|
|
req := request.(*listFleetMaintainedAppsRequest)
|
|
|
|
req.IncludeMetadata = true
|
|
|
|
apps, meta, err := svc.ListFleetMaintainedApps(ctx, req.TeamID, req.ListOptions)
|
|
if err != nil {
|
|
return listFleetMaintainedAppsResponse{Err: err}, nil
|
|
}
|
|
|
|
return listFleetMaintainedAppsResponse{FleetMaintainedApps: apps, Meta: meta}, nil
|
|
}
|
|
|
|
func (svc *Service) ListFleetMaintainedApps(ctx context.Context, teamID uint, opts fleet.ListOptions) ([]fleet.MaintainedApp, *fleet.PaginationMetadata, error) {
|
|
// skipauth: No authorization check needed due to implementation returning
|
|
// only license error.
|
|
svc.authz.SkipAuthorization(ctx)
|
|
|
|
return nil, nil, fleet.ErrMissingLicense
|
|
}
|
|
|
|
type getFleetMaintainedAppRequest struct {
|
|
AppID uint `url:"app_id"`
|
|
}
|
|
|
|
type getFleetMaintainedAppResponse struct {
|
|
FleetMaintainedApp *fleet.MaintainedApp `json:"fleet_maintained_app"`
|
|
Err error `json:"error,omitempty"`
|
|
}
|
|
|
|
func (r getFleetMaintainedAppResponse) error() error { return r.Err }
|
|
|
|
func getFleetMaintainedApp(ctx context.Context, request any, svc fleet.Service) (errorer, error) {
|
|
req := request.(*getFleetMaintainedAppRequest)
|
|
|
|
app, err := svc.GetFleetMaintainedApp(ctx, req.AppID)
|
|
if err != nil {
|
|
return getFleetMaintainedAppResponse{Err: err}, nil
|
|
}
|
|
|
|
return getFleetMaintainedAppResponse{FleetMaintainedApp: app}, nil
|
|
}
|
|
|
|
func (svc *Service) GetFleetMaintainedApp(ctx context.Context, appID uint) (*fleet.MaintainedApp, error) {
|
|
// skipauth: No authorization check needed due to implementation returning
|
|
// only license error.
|
|
svc.authz.SkipAuthorization(ctx)
|
|
|
|
return nil, fleet.ErrMissingLicense
|
|
}
|