1 files changed, 67 insertions, 0 deletions
diff --git a/lib/gitlab/kubernetes/helm/v2/patch_command.rb b/lib/gitlab/kubernetes/helm/v2/patch_command.rb
new file mode 100644
index 00000000000..2855e6444b1
--- /dev/null
+++ b/lib/gitlab/kubernetes/helm/v2/patch_command.rb
@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+# PatchCommand is for updating values in installed charts without overwriting
+# existing values.
+module Gitlab
+  module Kubernetes
+    module Helm
+      module V2
+        class PatchCommand < BaseCommand
+          include ClientCommand
+
+          attr_reader :chart, :repository
+          attr_accessor :version
+
+          def initialize(chart:, version:, repository: nil, **args)
+            super(**args)
+
+            # version is mandatory to prevent chart mismatches
+            # we do not want our values interpreted in the context of the wrong version
+            raise ArgumentError, 'version is required' if version.blank?
+
+            @chart = chart
+            @version = version
+            @repository = repository
+          end
+
+          def generate_script
+            super + [
+              init_command,
+              repository_command,
+              repository_update_command,
+              upgrade_command
+            ].compact.join("\n")
+          end
+
+          private
+
+          def upgrade_command
+            command = ['helm', 'upgrade', name, chart] +
+                      reuse_values_flag +
+                      version_flag +
+                      namespace_flag +
+                      value_flag
+
+            command.shelljoin
+          end
+
+          def reuse_values_flag
+            ['--reuse-values']
+          end
+
+          def value_flag
+            ['-f', "/data/helm/#{name}/config/values.yaml"]
+          end
+
+          def namespace_flag
+            ['--namespace', Gitlab::Kubernetes::Helm::NAMESPACE]
+          end
+
+          def version_flag
+            ['--version', version]
+          end
+        end
+      end
+    end
+  end
+end