[CE] Port review apps file to CEce-improve-review-apps

Signed-off-by: Rémy Coutable <remy@rymai.me>

2 files changed, 79 insertions, 0 deletions

diff --git a/lib/quality/helm_client.rb b/lib/quality/helm_client.rb
new file mode 100644
index 00000000000..49d953da681
--- /dev/null
+++ b/lib/quality/helm_client.rb
@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+require 'time'
+require_relative '../gitlab/popen' unless defined?(Gitlab::Popen)
+
+module Quality
+  class HelmClient
+    attr_reader :namespace
+
+    Release = Struct.new(:name, :revision, :last_update, :status, :chart, :namespace) do
+      def revision
+        @revision ||= self[:revision].to_i
+      end
+
+      def last_update
+        @last_update ||= Time.parse(self[:last_update])
+      end
+    end
+
+    def initialize(namespace: ENV['KUBE_NAMESPACE'])
+      @namespace = namespace
+    end
+
+    def releases(args: [])
+      command = ['list', %(--namespace "#{namespace}"), *args]
+
+      run_command(command)
+        .stdout
+        .lines
+        .select { |line| line.include?(namespace) }
+        .map { |line| Release.new(*line.split(/\t/).map(&:strip)) }
+    end
+
+    def delete(release_name:)
+      run_command(['delete', '--purge', release_name])
+    end
+
+    private
+
+    def run_command(command)
+      final_command = ['helm', *command].join(' ')
+      puts "Running command: `#{final_command}`" # rubocop:disable Rails/Output
+
+      Gitlab::Popen.popen_with_detail([final_command])
+    end
+  end
+end
diff --git a/lib/quality/kubernetes_client.rb b/lib/quality/kubernetes_client.rb
new file mode 100644
index 00000000000..e366a688e3e
--- /dev/null
+++ b/lib/quality/kubernetes_client.rb
@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require_relative '../gitlab/popen' unless defined?(Gitlab::Popen)
+
+module Quality
+  class KubernetesClient
+    attr_reader :namespace
+
+    def initialize(namespace: ENV['KUBE_NAMESPACE'])
+      @namespace = namespace
+    end
+
+    def cleanup(release_name:)
+      command = ['kubectl']
+      command << %(-n "#{namespace}" get ingress,svc,pdb,hpa,deploy,statefulset,job,pod,secret,configmap,pvc,secret,clusterrole,clusterrolebinding,role,rolebinding,sa 2>&1)
+      command << '|' << %(grep "#{release_name}")
+      command << '|' << "awk '{print $1}'"
+      command << '|' << %(xargs kubectl -n "#{namespace}" delete)
+      command << '||' << 'true'
+
+      run_command(command)
+    end
+
+    private
+
+    def run_command(command)
+      puts "Running command: `#{command.join(' ')}`" # rubocop:disable Rails/Output
+
+      Gitlab::Popen.popen_with_detail(command)
+    end
+  end
+end