debian-mirror-gitlab/app/services/packages/helm/extract_file_metadata_service.rb
2023-02-26 17:17:37 +05:30

58 lines
1.6 KiB
Ruby

# frozen_string_literal: true
require 'rubygems/package'
module Packages
module Helm
class ExtractFileMetadataService
ExtractionError = Class.new(StandardError)
# Charts must be smaller than 1M because of the storage limitations of Kubernetes objects.
# based on https://helm.sh/docs/chart_template_guide/accessing_files/
MAX_FILE_SIZE = 1.megabytes.freeze
def initialize(package_file)
@package_file = package_file
end
def execute
raise ExtractionError, 'invalid package file' unless valid_package_file?
metadata
end
private
def valid_package_file?
@package_file &&
@package_file.package&.helm? &&
@package_file.file.size > 0 # rubocop:disable Style/ZeroLengthPredicate
end
def metadata
YAML.safe_load(chart_yaml_content)
rescue Psych::Exception => e
raise ExtractionError, "Error while parsing Chart.yaml: #{e.message}"
end
def chart_yaml_content
@package_file.file.use_open_file do |file|
tar_reader = Gem::Package::TarReader.new(Zlib::GzipReader.new(file))
chart_yaml = tar_reader.find do |entry|
next unless entry.file?
entry.full_name.end_with?('/Chart.yaml')
end
raise ExtractionError, 'Chart.yaml not found within a directory' unless chart_yaml
raise ExtractionError, 'Chart.yaml too big' if chart_yaml.size > MAX_FILE_SIZE
chart_yaml.read
ensure
tar_reader.close
end
end
end
end
end