Merge pull request #21 from datainfrahq/networkpolicy

add network policy support

Author: AdheipSingh

builder/builder.go

@@ -10,6 +10,7 @@ type Builder struct {
 	DeploymentOrStatefulset []BuilderDeploymentStatefulSet
 	StorageConfig           []BuilderStorageConfig
 	Service                 []BuilderService
+	NetworkPolicy           []BuilderNetworkPolicy
 	Recorder                BuilderRecorder
 	Context                 BuilderContext
 	Store                   InternalStore

builder/builder_network_policy.go

@@ -0,0 +1,55 @@
+package builder
+
+import (
+	networkingv1 "k8s.io/api/networking/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
+)
+
+type BuilderNetworkPolicy struct {
+	NetworkPolicy *networkingv1.NetworkPolicy
+	CommonBuilder
+}
+
+func ToNewBuilderNetworkPolicy(builder []BuilderService) func(*Builder) {
+	return func(s *Builder) {
+		s.Service = builder
+	}
+}
+
+func (s *Builder) ReconcileNetworkPolicy() (controllerutil.OperationResult, error) {
+
+	var err error
+	var result controllerutil.OperationResult
+
+	for _, np := range s.NetworkPolicy {
+
+		if np.NetworkPolicy != nil {
+
+			makeNp := np.makeNetworkPolicy()
+
+			np.DesiredState = makeNp
+			np.CurrentState = &networkingv1.NetworkPolicy{}
+
+			result, err = np.CreateOrUpdate(s.Context.Context, s.Recorder)
+			if err != nil {
+				return controllerutil.OperationResultNone, nil
+			}
+		}
+	}
+	return result, nil
+}
+
+func (b *BuilderNetworkPolicy) makeNetworkPolicy() *networkingv1.NetworkPolicy {
+	networkPolicy := &networkingv1.NetworkPolicy{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: "networking.k8s.io/v1",
+			Kind:       "NetworkPolicy",
+		},
+		ObjectMeta: b.ObjectMeta,
+		Spec:       *&b.NetworkPolicy.Spec,
+	}
+
+	return networkPolicy
+}