add admissionRequest in ctx

Author: rahulii

Diff for: apis/contexts.go

@@ -22,6 +22,7 @@ import (
 
 	authenticationv1 "k8s.io/api/authentication/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	admissionv1 "k8s.io/api/admission/v1"
 )
 
 // This is attached to contexts passed to webhook interfaces when
@@ -245,6 +246,7 @@ func IsDryRun(ctx context.Context) bool {
 // This is attached to contexts passed to webhook interfaces with
 // additional context from the HTTP request.
 type httpReq struct{}
+type AdmissionReq struct{}
 
 // WithHTTPRequest associated the HTTP request object the webhook
 // received with the context.
@@ -260,3 +262,18 @@ func GetHTTPRequest(ctx context.Context) *http.Request {
 	}
 	return v.(*http.Request)
 }
+
+// WithAdmissionRequest associated the admissionv1.AdmissionRequest object the webhook
+// received with the context.
+func WithAdmissionRequest(ctx context.Context, r *admissionv1.AdmissionRequest) context.Context {
+	return context.WithValue(ctx, AdmissionReq{}, r)
+}
+
+// GetAdmissionRequest fetches the admissionv1.AdmissionRequest received by the webhook.
+func GetAdmissionRequest(ctx context.Context) *admissionv1.AdmissionRequest {
+	v := ctx.Value(AdmissionReq{})
+	if v == nil {
+		return nil
+	}
+	return v.(*admissionv1.AdmissionRequest)
+}

Diff for: apis/contexts_test.go

@@ -24,6 +24,7 @@ import (
 	"github.com/google/go-cmp/cmp"
 	authenticationv1 "k8s.io/api/authentication/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	admissionv1 "k8s.io/api/admission/v1"
 )
 
 func TestContexts(t *testing.T) {
@@ -241,3 +242,21 @@ func TestGetHTTPRequest(t *testing.T) {
 		t.Errorf("GetHTTPRequest() = %v, wanted %v", got, want)
 	}
 }
+
+func TestGetAdmissionRequest(t *testing.T) {
+	ctx := context.Background()
+
+	if got := GetAdmissionRequest(ctx); got != nil {
+		t.Errorf("GetAdmissionRequest() = %v, wanted %v", got, nil)
+	}
+
+	admReq := admissionv1.AdmissionRequest{
+		Name: "foo",
+	}
+	ctx = WithAdmissionRequest(ctx, &admReq)
+
+	if want, got := &admReq, GetAdmissionRequest(ctx); got != want {
+		t.Errorf("GetAdmissionRequest() = %v, wanted %v", got, want)
+	}
+}
+