1 files changed, 91 insertions, 0 deletions
diff --git a/lib/gitlab/kubernetes/network_policy.rb b/lib/gitlab/kubernetes/network_policy.rb
new file mode 100644
index 00000000000..ea25d81cbd2
--- /dev/null
+++ b/lib/gitlab/kubernetes/network_policy.rb
@@ -0,0 +1,91 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Kubernetes
+    class NetworkPolicy
+      def initialize(name:, namespace:, pod_selector:, ingress:, creation_timestamp: nil, policy_types: ["Ingress"], egress: nil)
+        @name = name
+        @namespace = namespace
+        @creation_timestamp = creation_timestamp
+        @pod_selector = pod_selector
+        @policy_types = policy_types
+        @ingress = ingress
+        @egress = egress
+      end
+
+      def self.from_yaml(manifest)
+        return unless manifest
+
+        policy = YAML.safe_load(manifest, symbolize_names: true)
+        return if !policy[:metadata] || !policy[:spec]
+
+        metadata = policy[:metadata]
+        spec = policy[:spec]
+        self.new(
+          name: metadata[:name],
+          namespace: metadata[:namespace],
+          pod_selector: spec[:podSelector],
+          policy_types: spec[:policyTypes],
+          ingress: spec[:ingress],
+          egress: spec[:egress]
+        )
+      rescue Psych::SyntaxError, Psych::DisallowedClass
+        nil
+      end
+
+      def self.from_resource(resource)
+        return unless resource
+        return if !resource[:metadata] || !resource[:spec]
+
+        metadata = resource[:metadata]
+        spec = resource[:spec].to_h
+        self.new(
+          name: metadata[:name],
+          namespace: metadata[:namespace],
+          creation_timestamp: metadata[:creationTimestamp],
+          pod_selector: spec[:podSelector],
+          policy_types: spec[:policyTypes],
+          ingress: spec[:ingress],
+          egress: spec[:egress]
+        )
+      end
+
+      def generate
+        ::Kubeclient::Resource.new.tap do |resource|
+          resource.metadata = metadata
+          resource.spec = spec
+        end
+      end
+
+      def as_json(opts = nil)
+        {
+          name: name,
+          namespace: namespace,
+          creation_timestamp: creation_timestamp,
+          manifest: manifest
+        }
+      end
+
+      private
+
+      attr_reader :name, :namespace, :creation_timestamp, :pod_selector, :policy_types, :ingress, :egress
+
+      def metadata
+        { name: name, namespace: namespace }
+      end
+
+      def spec
+        {
+          podSelector: pod_selector,
+          policyTypes: policy_types,
+          ingress: ingress,
+          egress: egress
+        }
+      end
+
+      def manifest
+        YAML.dump({ metadata: metadata, spec: spec }.deep_stringify_keys)
+      end
+    end
+  end
+end