72 lines
1.8 KiB
Ruby
72 lines
1.8 KiB
Ruby
![]() |
# frozen_string_literal: true
|
||
|
|
||
|
module Gitlab
|
||
|
module Kubernetes
|
||
|
module NetworkPolicyCommon
|
||
|
DISABLED_BY_LABEL = :'network-policy.gitlab.com/disabled_by'
|
||
|
|
||
|
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,
|
||
|
is_autodevops: autodevops?,
|
||
|
is_enabled: enabled?
|
||
|
}
|
||
|
end
|
||
|
|
||
|
def autodevops?
|
||
|
return false unless labels
|
||
|
|
||
|
!labels[:chart].nil? && labels[:chart].start_with?('auto-deploy-app-')
|
||
|
end
|
||
|
|
||
|
# selector selects pods that should be targeted by this
|
||
|
# policy. It can represent podSelector, nodeSelector or
|
||
|
# endpointSelector We can narrow selection by requiring
|
||
|
# this policy to match our custom labels. Since DISABLED_BY
|
||
|
# label will not be on any pod a policy will be effectively disabled.
|
||
|
def enabled?
|
||
|
return true unless selector&.key?(:matchLabels)
|
||
|
|
||
|
!selector[:matchLabels]&.key?(DISABLED_BY_LABEL)
|
||
|
end
|
||
|
|
||
|
def enable
|
||
|
return if enabled?
|
||
|
|
||
|
selector[:matchLabels].delete(DISABLED_BY_LABEL)
|
||
|
end
|
||
|
|
||
|
def disable
|
||
|
selector[:matchLabels] ||= {}
|
||
|
selector[:matchLabels].merge!(DISABLED_BY_LABEL => 'gitlab')
|
||
|
end
|
||
|
|
||
|
private
|
||
|
|
||
|
def metadata
|
||
|
meta = { name: name, namespace: namespace }
|
||
|
meta[:labels] = labels if labels
|
||
|
meta
|
||
|
end
|
||
|
|
||
|
def spec
|
||
|
raise NotImplementedError
|
||
|
end
|
||
|
|
||
|
def manifest
|
||
|
YAML.dump({ metadata: metadata, spec: spec }.deep_stringify_keys)
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
end
|