92 lines
2.4 KiB
Ruby
92 lines
2.4 KiB
Ruby
|
# 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
|