114 lines
2.9 KiB
Ruby
114 lines
2.9 KiB
Ruby
require 'zlib'
|
|
require 'json'
|
|
|
|
module Gitlab
|
|
module Ci
|
|
module Build
|
|
module Artifacts
|
|
class Metadata
|
|
VERSION_PATTERN = '[\w\s]+(\d+\.\d+\.\d+)'
|
|
attr_reader :file, :path, :full_version
|
|
|
|
def initialize(file, path)
|
|
@file, @path = file, path
|
|
@full_version = read_version
|
|
end
|
|
|
|
def version
|
|
@full_version.match(/#{VERSION_PATTERN}/).captures.first
|
|
end
|
|
|
|
def errors
|
|
gzip do |gz|
|
|
read_string(gz) # version
|
|
errors = read_string(gz)
|
|
raise StandardError, 'Errors field not found!' unless errors
|
|
JSON.parse(errors)
|
|
end
|
|
end
|
|
|
|
def match!
|
|
gzip do |gz|
|
|
2.times { read_string(gz) } # version and errors fields
|
|
match_entries(gz)
|
|
end
|
|
end
|
|
|
|
def to_entry
|
|
entires, metadata = match!
|
|
Entry.new(@path, entires, metadata)
|
|
end
|
|
|
|
private
|
|
|
|
def match_entries(gz)
|
|
paths, metadata = [], []
|
|
match_pattern = %r{^#{Regexp.escape(@path)}[^/]*/?$}
|
|
invalid_pattern = %r{(^\.?\.?/)|(/\.?\.?/)}
|
|
|
|
until gz.eof? do
|
|
begin
|
|
path = read_string(gz).force_encoding('UTF-8')
|
|
meta = read_string(gz).force_encoding('UTF-8')
|
|
|
|
next unless path.valid_encoding? && meta.valid_encoding?
|
|
next unless path =~ match_pattern
|
|
next if path =~ invalid_pattern
|
|
|
|
paths.push(path)
|
|
metadata.push(JSON.parse(meta, symbolize_names: true))
|
|
rescue JSON::ParserError, Encoding::CompatibilityError
|
|
next
|
|
end
|
|
end
|
|
|
|
[paths, metadata]
|
|
end
|
|
|
|
def read_version
|
|
gzip do |gz|
|
|
version_string = read_string(gz)
|
|
|
|
unless version_string
|
|
raise StandardError, 'Artifacts metadata file empty!'
|
|
end
|
|
|
|
unless version_string =~ /^#{VERSION_PATTERN}/
|
|
raise StandardError, 'Invalid version!'
|
|
end
|
|
|
|
version_string.chomp
|
|
end
|
|
end
|
|
|
|
def read_uint32(gz)
|
|
binary = gz.read(4)
|
|
binary.unpack('L>')[0] if binary
|
|
end
|
|
|
|
def read_string(gz)
|
|
string_size = read_uint32(gz)
|
|
return nil unless string_size
|
|
gz.read(string_size)
|
|
end
|
|
|
|
def gzip
|
|
open do |file|
|
|
gzip = Zlib::GzipReader.new(file)
|
|
begin
|
|
yield gzip
|
|
ensure
|
|
gzip.close
|
|
end
|
|
end
|
|
end
|
|
|
|
def open
|
|
File.open(@file) { |file| yield file }
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|