114 lines
3 KiB
Ruby
114 lines
3 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'time'
|
||
|
|
require_relative '../gitlab/popen' unless defined?(Gitlab::Popen)
|
||
|
|
|
||
|
|
module Quality
|
||
|
|
class Helm3Client
|
||
|
|
CommandFailedError = Class.new(StandardError)
|
||
|
|
|
||
|
|
attr_reader :namespace
|
||
|
|
|
||
|
|
RELEASE_JSON_ATTRIBUTES = %w[name revision updated status chart app_version namespace].freeze
|
||
|
|
PAGINATION_SIZE = 256 # Default helm list pagination size
|
||
|
|
|
||
|
|
Release = Struct.new(:name, :revision, :last_update, :status, :chart, :app_version, :namespace) do
|
||
|
|
def revision
|
||
|
|
@revision ||= self[:revision].to_i
|
||
|
|
end
|
||
|
|
|
||
|
|
def status
|
||
|
|
@status ||= self[:status].downcase
|
||
|
|
end
|
||
|
|
|
||
|
|
def last_update
|
||
|
|
@last_update ||= Time.parse(self[:last_update])
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
# A single page of data and the corresponding page number.
|
||
|
|
Page = Struct.new(:releases, :number)
|
||
|
|
|
||
|
|
def initialize(namespace:, tiller_namespace: nil)
|
||
|
|
@namespace = namespace
|
||
|
|
end
|
||
|
|
|
||
|
|
def releases(args: [])
|
||
|
|
each_release(args)
|
||
|
|
end
|
||
|
|
|
||
|
|
def delete(release_name:)
|
||
|
|
run_command([
|
||
|
|
'uninstall',
|
||
|
|
%(--namespace "#{namespace}"),
|
||
|
|
release_name
|
||
|
|
])
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def run_command(command)
|
||
|
|
final_command = ['helm', *command].join(' ')
|
||
|
|
puts "Running command: `#{final_command}`" # rubocop:disable Rails/Output
|
||
|
|
|
||
|
|
result = Gitlab::Popen.popen_with_detail([final_command])
|
||
|
|
|
||
|
|
if result.status.success?
|
||
|
|
result.stdout.chomp.freeze
|
||
|
|
else
|
||
|
|
raise CommandFailedError, "The `#{final_command}` command failed (status: #{result.status}) with the following error:\n#{result.stderr}"
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def raw_releases(page, args = [])
|
||
|
|
command = [
|
||
|
|
'list',
|
||
|
|
%(--namespace "#{namespace}"),
|
||
|
|
%(--max #{PAGINATION_SIZE}),
|
||
|
|
%(--offset #{PAGINATION_SIZE * page}),
|
||
|
|
%(--output json),
|
||
|
|
*args
|
||
|
|
]
|
||
|
|
releases = JSON.parse(run_command(command))
|
||
|
|
|
||
|
|
releases.map do |release|
|
||
|
|
Release.new(*release.values_at(*RELEASE_JSON_ATTRIBUTES))
|
||
|
|
end
|
||
|
|
rescue JSON::ParserError => ex
|
||
|
|
puts "Ignoring this JSON parsing error: #{ex}" # rubocop:disable Rails/Output
|
||
|
|
[]
|
||
|
|
end
|
||
|
|
|
||
|
|
# Fetches data from Helm and yields a Page object for every page
|
||
|
|
# of data, without loading all of them into memory.
|
||
|
|
#
|
||
|
|
# method - The Octokit method to use for getting the data.
|
||
|
|
# args - Arguments to pass to the `helm list` command.
|
||
|
|
def each_releases_page(args, &block)
|
||
|
|
return to_enum(__method__, args) unless block_given?
|
||
|
|
|
||
|
|
page = 0
|
||
|
|
final_args = args.dup
|
||
|
|
|
||
|
|
begin
|
||
|
|
collection = raw_releases(page, final_args)
|
||
|
|
|
||
|
|
yield Page.new(collection, page += 1)
|
||
|
|
end while collection.any?
|
||
|
|
end
|
||
|
|
|
||
|
|
# Iterates over all of the releases.
|
||
|
|
#
|
||
|
|
# args - Any arguments to pass to the `helm list` command.
|
||
|
|
def each_release(args, &block)
|
||
|
|
return to_enum(__method__, args) unless block_given?
|
||
|
|
|
||
|
|
each_releases_page(args) do |page|
|
||
|
|
page.releases.each do |release|
|
||
|
|
yield release
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|