mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 01:15:22 +00:00
715d908613
Notable refactoring: - Use stdlib "context" in place of "golang.org/x/net/context" - Go-kit no longer wraps errors, so we remove the unwrap in transport_error.go - Use MakeHandler when setting up endpoint tests (fixes test bug caught during this refactoring) Closes #1411.
43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-kit/kit/endpoint"
|
|
"github.com/kolide/kolide/server/kolide"
|
|
)
|
|
|
|
type importRequest struct {
|
|
// Config contains a JSON osquery config supplied by the end user
|
|
Config string `json:"config"`
|
|
// ExternalPackConfigs contains a map of external Pack configs keyed by
|
|
// Pack name, this includes external packs referenced by the globbing
|
|
// feature. Not in the case of globbed packs, we expect the user to
|
|
// generate unique pack names since we don't know what they are, these
|
|
// names must be included in the GlobPackNames field so that we can
|
|
// validate that they've been accounted for.
|
|
ExternalPackConfigs map[string]string `json:"external_pack_configs"`
|
|
// GlobPackNames list of user generated names for external packs
|
|
// referenced by the glob feature, the JSON for the globbed packs
|
|
// is stored in ExternalPackConfigs keyed by the GlobPackName
|
|
GlobPackNames []string `json:"glob_pack_names"`
|
|
}
|
|
|
|
type importResponse struct {
|
|
Response *kolide.ImportConfigResponse `json:"response,omitempty"`
|
|
Err error `json:"error,omitempty"`
|
|
}
|
|
|
|
func (ir importResponse) error() error { return ir.Err }
|
|
|
|
func makeImportConfigEndpoint(svc kolide.Service) endpoint.Endpoint {
|
|
return func(ctx context.Context, request interface{}) (interface{}, error) {
|
|
config := request.(kolide.ImportConfig)
|
|
resp, err := svc.ImportConfig(ctx, &config)
|
|
if err != nil {
|
|
return importResponse{Err: err}, nil
|
|
}
|
|
return importResponse{Response: resp}, nil
|
|
}
|
|
}
|