40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
|
// Copyright 2021 the Pinniped contributors. All Rights Reserved.
|
||
|
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
|
|
||
|
|
package ownerref
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||
|
|
|
||
|
|
"go.pinniped.dev/internal/kubeclient"
|
||
|
|
)
|
||
|
|
|
||
|
|
func New(ref metav1.OwnerReference) kubeclient.Middleware {
|
||
|
|
return ownerRefMiddleware(ref)
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ kubeclient.Middleware = ownerRefMiddleware(metav1.OwnerReference{})
|
||
|
|
|
||
|
|
type ownerRefMiddleware metav1.OwnerReference
|
||
|
|
|
||
|
|
func (o ownerRefMiddleware) Handles(httpMethod string) bool {
|
||
|
|
return httpMethod == http.MethodPost // only handle create requests
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO this func assumes all objects are namespace scoped and are in the same namespace.
|
||
|
|
// i.e. it assumes all objects are safe to set an owner ref on
|
||
|
|
// i.e. the owner could be namespace scoped and thus cannot own a cluster scoped object
|
||
|
|
// this could be fixed by using a rest mapper to confirm the REST scoping
|
||
|
|
// or we could always use an owner ref to a cluster scoped object
|
||
|
|
func (o ownerRefMiddleware) Mutate(obj metav1.Object) (mutated bool) {
|
||
|
|
// we only want to set the owner ref on create and when one is not already present
|
||
|
|
if len(obj.GetOwnerReferences()) != 0 {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
obj.SetOwnerReferences([]metav1.OwnerReference{metav1.OwnerReference(o)})
|
||
|
|
return true
|
||
|
|
}
|