blob: 77efa65f1d18867c59b1ed20b4ecbc52b6c5cf29 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
# 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
|
