kustomizer 0.1.10 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c546f00a97a3e739f1c2448d97c3e18e0fefbeb84007a09ca0359ced2b3d7874
-  data.tar.gz: 4e6d9f05eea851976414c170a522429d50427582210181c052ad6ac8357ccc4f
+  metadata.gz: 60295db655aa832bb933831f4dbd1854ad7705c3908423cc9e8a4e1d25faef69
+  data.tar.gz: a575e73284ea5bd0877543ff888767ff224b26e3ddde9509c591631492340f60
 SHA512:
-  metadata.gz: 2142716318d3608d8d6598c475d646b2f1c7219a08760fbe7ec9497359a5cc1a8a6996e75475c785374a8b297dd771d1aa488108172925acaed6a53a02939f2b
-  data.tar.gz: d45eed6a3f723f1d60e248bdddcd0a428ab668629c422662c6eddd2f4a9c7f737f17dc79156e7454be29d32dbf1fc69e90b92545d3a95d571e0600476704c9a2
+  metadata.gz: d0c24c3dcfb4ee551be1b85a5c84dd52038015aa6ef4fdd439d80eccf9e4cd1046c04c56076c92a428bbccdb4cacee47dfbc3531077da9906c22c888b4faa939
+  data.tar.gz: 9a5b1b6641c3a9e208dcae6310891462f3cbfc3b609266d0bda66f9e4131fef3a8866b364271ee36170ed8c1cc08c7462e959672fa8284b9e63d17e799c3729e

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    kustomizer (0.1.10)
+    kustomizer (0.1.11)
       accessory (~> 0.1.11)
       base32-multi
 
@@ -33,6 +33,3 @@ DEPENDENCIES
   kustomizer!
   rake (~> 13.0)
   rspec (~> 3.0)
-
-BUNDLED WITH
-   1.17.2

data/README.md

@@ -19,8 +19,8 @@ Enough Library to solve our own problems :wink:)
 Status of `kustomization.yaml` feature support:
 
 * [x] `bases`
-* [ ] `commonAnnotations`
-* [ ] `commonLabels`
+* [x] `commonAnnotations`
+* [x] `commonLabels`
 * [ ] `configMapGenerator` and `secretGenerator`
 * [ ] `crds`
 * [x] `generators` (see [Extending Kustomize](https://kubectl.docs.kubernetes.io/guides/extending_kustomize/))

data/lib/kustomize.rb

@@ -20,7 +20,7 @@ module Kustomize
         raise ArgumentError, "must be a kustomization document or a path to one, instead got: #{rel_path_or_rc.inspect}"
       end
 
-    Kustomize::Emitter::FinalizerEmitter.new(base_emitter)
+    Kustomize::Emitter::FinalizerEmitter.new(base_emitter, session: session)
   end
 
   def self.load_doc(rc, session: Kustomize::Session.new, source_path:)

data/lib/kustomize/emitter/document_emitter/kustomization_document_emitter.rb

@@ -6,6 +6,8 @@ require 'kustomize/emitter/generator_plugins_emitter'
 require 'kustomize/transform/json_6902_patch_transform'
 require 'kustomize/transform/image_transform'
 require 'kustomize/transform/namespace_transform'
+require 'kustomize/transform/common_annotations_transform'
+require 'kustomize/transform/common_labels_transform'
 require 'kustomize/transform/fingerprint_suffix_transform'
 require 'kustomize/transform/ref_fixup_transform'
 require 'kustomize/transform/purge_internal_annotations_transform'
@@ -101,12 +103,30 @@ class Kustomize::Emitter::DocumentEmitter::KustomizationDocumentEmitter < Kustom
     end
   end
 
+  def common_annotations_transforms
+    if new_annots = @doc['commonAnnotations']
+      [Kustomize::Transform::CommonAnnotationsTransform.create(new_annots)]
+    else
+      []
+    end
+  end
+
+  def common_labels_transforms
+    if new_labels = @doc['commonLabels']
+      [Kustomize::Transform::CommonLabelsTransform.create(new_labels)]
+    else
+      []
+    end
+  end
+
   def transforms
     return @transforms if @transforms
 
     @transforms = [
       self.namespace_transforms,
       self.image_transforms,
+      self.common_annotations_transforms,
+      self.common_labels_transforms,
       self.json_6902_patch_transforms,
       self.transformer_plugin_transforms
     ].flatten

data/lib/kustomize/emitter/finalizer_emitter.rb

@@ -2,11 +2,14 @@ require 'kustomize/emitter'
 
 require 'kustomize/transform/fingerprint_suffix_transform'
 require 'kustomize/transform/ref_fixup_transform'
+require 'kustomize/transform/filter_for_session_specified_component_transform'
+require 'kustomize/transform/drop_filtered_documents_transform'
 require 'kustomize/transform/purge_internal_annotations_transform'
 
 class Kustomize::Emitter::FinalizerEmitter < Kustomize::Emitter
-  def initialize(input_emitter)
+  def initialize(input_emitter, session:)
     @input_emitter = input_emitter
+    @session = session
   end
 
   def input_emitters
@@ -16,11 +19,20 @@ class Kustomize::Emitter::FinalizerEmitter < Kustomize::Emitter
   def transforms
     return @transforms if @transforms
 
+    final_filters =
+      if comp = @session.only_emit_component
+        [Kustomize::Transform::FilterForSessionSpecifiedComponentTransform.create(comp)]
+      else
+        []
+      end
+
     @transforms = [
       Kustomize::Transform::FingerprintSuffixTransform.instance,
       Kustomize::Transform::RefFixupTransform.instance,
+      final_filters,
+      Kustomize::Transform::DropFilteredDocumentsTransform.instance,
       Kustomize::Transform::PurgeInternalAnnotationsTransform.instance
-    ].flatten
+    ].flatten.compact
   end
 
   def emit

data/lib/kustomize/session.rb

@@ -3,12 +3,14 @@ module Kustomize; end
 require 'kustomize/plugin_manager'
 
 class Kustomize::Session
-  def initialize(load_paths: [])
+  def initialize(load_paths: [], only_emit_component: nil)
     @load_paths = load_paths
     @plugin_manager = Kustomize::PluginManager.new(session: self)
+    @only_emit_component = only_emit_component
   end
 
   attr_reader :plugin_manager
+  attr_accessor :only_emit_component
 
   def builtin_load_paths
     [Pathname.new(__FILE__).expand_path.parent / 'builtin_plugins']

data/lib/kustomize/transform/common_annotations_transform.rb

@@ -0,0 +1,36 @@
+require 'accessory'
+
+require 'kustomize/transform'
+
+class Kustomize::Transform::CommonAnnotationsTransform < Kustomize::Transform
+  include Accessory
+
+  def initialize(new_annots)
+    @new_annots = new_annots
+  end
+
+  BASE_LENS = Lens["metadata", "annotations"]
+
+  LENS_PREFIXES = [
+    Lens["spec", "template"],
+    Lens["spec", "jobTemplate", "spec", "template"]
+  ]
+
+  def rewrite(rc_doc)
+    rc_doc = BASE_LENS.update_in(rc_doc) do |annots|
+      [:set, (annots || {}).merge(@new_annots)]
+    end
+
+    LENS_PREFIXES.inject(rc_doc) do |doc, prefix|
+      prefix.update_in(rc_doc) do |node|
+        next(:keep) unless node.kind_of?(Hash)
+
+        new_node = BASE_LENS.update_in(node) do |annots|
+          [:set, (annots || {}).merge(@new_annots)]
+        end
+
+        [:set, new_node]
+      end
+    end
+  end
+end

data/lib/kustomize/transform/common_labels_transform.rb

@@ -0,0 +1,41 @@
+require 'accessory'
+
+require 'kustomize/transform'
+
+class Kustomize::Transform::CommonLabelsTransform < Kustomize::Transform
+  include Accessory
+
+  def initialize(new_labels)
+    @new_labels = new_labels
+  end
+
+  LENSES_FOR_ALL = [
+    Lens["metadata", "labels"]
+  ]
+
+  LENSES_FOR_KIND = {
+    "Service" => [
+      Lens["spec", "selector"]
+    ],
+
+    "Deployment" => [
+      Lens["spec", "selector", "matchLabels"]
+    ],
+  }
+
+  def rewrite(rc_doc)
+    rc_kind = rc_doc['kind']
+    use_lenses = LENSES_FOR_ALL
+
+    if lenses_for_doc_kind = LENSES_FOR_KIND[rc_kind]
+      use_lenses += lenses_for_doc_kind
+    end
+
+    use_lenses.inject(rc_doc) do |doc, lens|
+      lens.update_in(doc) do |annots|
+        [:set, (annots || {}).merge(@new_labels)]
+      end
+    end
+  end
+
+end

data/lib/kustomize/transform/drop_filtered_documents_transform.rb

@@ -0,0 +1,25 @@
+require 'singleton'
+require 'set'
+
+require 'accessory'
+
+require 'kustomize/transform'
+
+class Kustomize::Transform::DropFilteredDocumentsTransform < Kustomize::Transform
+  include Accessory
+  include Singleton
+
+  ANNOTS_LENS = Lens['metadata', 'annotations']
+
+  DROP_ON_ANNOTS = Set[
+    'config.kubernetes.io/local-config',
+    'kustomizer.covalenthq.com/drop'
+  ]
+
+  def rewrite_all(rcs)
+    rcs.filter do |rc|
+      annot_keys = (ANNOTS_LENS.get_in(rc) || {}).keys
+      not(annot_keys.find{ |k| DROP_ON_ANNOTS.member?(k) })
+    end
+  end
+end

data/lib/kustomize/transform/filter_for_session_specified_component_transform.rb

@@ -0,0 +1,29 @@
+require 'accessory'
+
+require 'kustomize/transform'
+
+class Kustomize::Transform::FilterForSessionSpecifiedComponentTransform < Kustomize::Transform
+  include Accessory
+
+  def initialize(component_name)
+    @component_name = component_name
+  end
+
+  ANNOTS_LENS = Lens['metadata', 'annotations']
+
+  COMPONENT_ANNOT_NAME = 'kustomizer.covalenthq.com/component-name'
+  DROP_ANNOT_NAME = 'kustomizer.covalenthq.com/drop'
+
+  def rewrite(rc)
+    ANNOTS_LENS.update_in(rc) do |orig_annots|
+      orig_annots ||= {}
+
+      if orig_annots[COMPONENT_ANNOT_NAME] == @component_name
+        :keep
+      else
+        new_annots = orig_annots.merge({DROP_ANNOT_NAME => 'true'})
+        [:set, new_annots]
+      end
+    end
+  end
+end

data/lib/kustomize/transform/purge_internal_annotations_transform.rb

@@ -8,23 +8,32 @@ class Kustomize::Transform::PurgeInternalAnnotationsTransform < Kustomize::Trans
   include Accessory
   include Singleton
 
-  ANNOTS_LENS = Lens['metadata', 'annotations']
-
-  INTERNAL_ANNOT_PATTERN = /^kustomizer\.covalenthq\.com\//
-
-  def rewrite(rc)
-    ANNOTS_LENS.update_in(rc) do |orig_annots|
-      next(:keep) unless orig_annots and orig_annots.length.nonzero?
-
-      new_annots =
-        orig_annots.reject{ |k, v| INTERNAL_ANNOT_PATTERN.match?(k) }
-
-      if new_annots.length == orig_annots.length
-        :keep
-      elsif new_annots.empty?
-        :pop
-      else
-        [:set, new_annots]
+  LENSES = [
+    Lens['metadata', 'annotations'],
+    Lens['spec', 'template', 'metadata', 'annotations'],
+    Lens['spec', 'jobTemplate', 'spec', 'template', 'metadata', 'annotations']
+  ]
+
+  INTERNAL_ANNOT_PATTERNS = [
+    /^config\.kubernetes\.io\//,
+    /^kustomizer\.covalenthq\.com\//
+  ]
+
+  def rewrite(rc_doc)
+    LENSES.inject(rc_doc) do |doc, lens|
+      lens.update_in(rc_doc) do |orig_annots|
+        next(:keep) unless orig_annots and orig_annots.length.nonzero?
+
+        new_annots =
+          orig_annots.reject{ |k, v| INTERNAL_ANNOT_PATTERNS.any?{ |pat| pat.match?(k) } }
+
+        if new_annots.length == orig_annots.length
+          :keep
+        elsif new_annots.empty?
+          :pop
+        else
+          [:set, new_annots]
+        end
       end
     end
   end

data/lib/kustomize/version.rb

@@ -1,3 +1,3 @@
 module Kustomize
-  VERSION = "0.1.10"
+  VERSION = "0.1.11"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kustomizer
 version: !ruby/object:Gem::Version
-  version: 0.1.10
+  version: 0.1.11
 platform: ruby
 authors:
 - Levi Aul
-autorequire: 
+autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-02-03 00:00:00.000000000 Z
+date: 2021-02-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: accessory
@@ -79,6 +79,10 @@ files:
 - lib/kustomize/session.rb
 - lib/kustomize/target_spec.rb
 - lib/kustomize/transform.rb
+- lib/kustomize/transform/common_annotations_transform.rb
+- lib/kustomize/transform/common_labels_transform.rb
+- lib/kustomize/transform/drop_filtered_documents_transform.rb
+- lib/kustomize/transform/filter_for_session_specified_component_transform.rb
 - lib/kustomize/transform/fingerprint_suffix_transform.rb
 - lib/kustomize/transform/image_transform.rb
 - lib/kustomize/transform/json_6902_patch_transform.rb
@@ -94,7 +98,7 @@ licenses:
 metadata:
   homepage_uri: https://github.com/tsutsu/kustomize
   source_code_uri: https://github.com/tsutsu/kustomize
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -109,8 +113,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.2.3
+signing_key:
 specification_version: 4
 summary: Pure-ruby impl of Kubernetes kustomize
 test_files: []