mirror of
https://github.com/FairwindsOps/polaris.git
synced 2026-05-07 01:36:41 +00:00
4d96993a18
* added fix command implementation * use node api * fix tests * added hostport mutate rule * update mutating server * fix array reference and add back leading slash * added test and refactor findNodes * more tests * added more test and fix issue with arrays * rename findNode function and ensure we capture exceptions * rename findNode function * append array value at the end and for single item remove brackets * append array value at the end and for single item remove brackets * create array if it does not exists * fix tests * handle some exceptions * fix tests * fix string format * guard for PodResult * fix flag name * fix privilegeEscalation check * fix up mutations for local files * fix pod parsing * fix object values * remove logspam * fix import * update some comments for health probes * add an option to not apply any mutations\, and just adjust yaml formatting * add preliminary support for helm * logspam * change up comment strategy * fix object comments * format * fix tests * add comments * fix key updates * fix mutation tests * tidy * refactor test * add test * add test * add test for object comments Co-authored-by: Robert Brennan <accounts@rbren.io> Co-authored-by: Robert Brennan <contact@rbren.io>
76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
// Copyright 2022 FairwindsOps Inc
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package webhook
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/fairwindsops/polaris/pkg/config"
|
|
"github.com/fairwindsops/polaris/pkg/mutation"
|
|
"github.com/sirupsen/logrus"
|
|
"gomodules.xyz/jsonpatch/v2"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/manager"
|
|
"sigs.k8s.io/controller-runtime/pkg/webhook"
|
|
"sigs.k8s.io/controller-runtime/pkg/webhook/admission"
|
|
"sigs.k8s.io/yaml"
|
|
)
|
|
|
|
// Mutator mutate k8s resources.
|
|
type Mutator struct {
|
|
Client client.Client
|
|
Config config.Configuration
|
|
decoder *admission.Decoder
|
|
}
|
|
|
|
var _ admission.Handler = &Mutator{}
|
|
|
|
// NewMutateWebhook creates a mutating admission webhook for the apiType.
|
|
func NewMutateWebhook(mgr manager.Manager, mutator Mutator) {
|
|
path := "/mutate"
|
|
|
|
mgr.GetWebhookServer().Register(path, &webhook.Admission{Handler: &mutator})
|
|
}
|
|
|
|
func (m *Mutator) mutate(req admission.Request) ([]jsonpatch.Operation, error) {
|
|
results, kubeResources, err := GetValidatedResults(req.AdmissionRequest.Kind.Kind, m.decoder, req, m.Config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
patches := mutation.GetMutationsFromResult(results)
|
|
originalYaml, err := yaml.JSONToYAML(kubeResources.OriginalObjectJSON)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
mutatedYamlStr, err := mutation.ApplyAllMutations(string(originalYaml), patches)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return jsonpatch.CreatePatch(originalYaml, []byte(mutatedYamlStr))
|
|
}
|
|
|
|
// Handle for Validator to run validation checks.
|
|
func (m *Mutator) Handle(ctx context.Context, req admission.Request) admission.Response {
|
|
logrus.Info("Starting request")
|
|
patches, err := m.mutate(req)
|
|
if err != nil {
|
|
return admission.Errored(403, err)
|
|
}
|
|
if patches == nil {
|
|
return admission.Allowed("Allowed")
|
|
}
|
|
return admission.Patched("", patches...)
|
|
}
|