class Cloudkeeper::Managers::ImageManager
Constants
- FORMATS
Public Class Methods
check_file!(file)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 31 def check_file!(file) raise Cloudkeeper::Errors::NoSuchFileError, "No such file #{file.inspect}" unless File.exist?(file) raise Cloudkeeper::Errors::PermissionDeniedError, "Cannot read file #{file.inspect}" unless File.readable?(file) end
check_image_checksum!(filename, checksum)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 66 def check_image_checksum!(filename, checksum) computed = Digest::SHA512.file(filename).hexdigest raise Cloudkeeper::Errors::Image::ChecksumError, "Checksum mismatch, expecting #{checksum.inspect} got #{computed.inspect}" \ unless checksum == computed end
connection_options(uri)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 92 def connection_options(uri) use_ssl = uri.scheme == 'https' ca_path = Cloudkeeper::Settings[:'ca-dir'] if Cloudkeeper::Settings[:'ca-dir'] { use_ssl: use_ssl, ca_path: ca_path } end
file_description(file)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 27 def file_description(file) Cloudkeeper::CommandExecutioner.execute('file', '-b', file) end
format(file)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 18 def format(file) check_file!(file) recognize_format(file) rescue Cloudkeeper::Errors::CommandExecutionError, Cloudkeeper::Errors::NoSuchFileError, Cloudkeeper::Errors::PermissionDeniedError, Cloudkeeper::Errors::Image::Format::NoFormatRecognizedError, Cloudkeeper::Errors::Image::Format::Ova::OvaFormatError => ex raise Cloudkeeper::Errors::Image::Format::RecognitionError, ex, "Cannot recognize image format for file #{file.inspect}" end
generate_filename(uri)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 98 def generate_filename(uri) File.join(Cloudkeeper::Settings[:'image-dir'], Cloudkeeper::Utils::Filename.sanitize(File.basename(uri.path))) end
recognize_format(file)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 36 def recognize_format(file) file_format_string = file_description(file) FORMATS.each do |format, regex| next unless regex =~ file_format_string format_test_method = "#{format}?".to_sym additional_test_result = respond_to?(format_test_method) ? send(format_test_method, file) : true return format if additional_test_result end raise Cloudkeeper::Errors::Image::Format::NoFormatRecognizedError, "No image format recognized for file #{file.inspect}" end
retrieve_image(uri, filename)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 72 def retrieve_image(uri, filename) Net::HTTP.start(uri.host, uri.port, connection_options(uri)) do |http| request = Net::HTTP::Get.new(uri) http.request(request) do |response| if response.is_a? Net::HTTPRedirection retrieve_image URI.join(uri, response.header['location']), filename break end response.value File.open(filename, 'w') { |file| response.read_body { |chunk| file.write(chunk) } } end end rescue Timeout::Error, Errno::EINVAL, Errno::ECONNRESET, Errno::ECONNREFUSED, Net::HTTPBadResponse, Net::HTTPHeaderSyntaxError, EOFError, Net::HTTPServerException, Net::HTTPRetriableError, OpenSSL::SSL::SSLError => ex raise Cloudkeeper::Errors::NetworkConnectionError, ex end
secure_download_image(url, checksum)
click to toggle source
# File lib/cloudkeeper/managers/image_manager.rb, line 50 def secure_download_image(url, checksum) logger.debug "Downloading image from #{url.inspect}" Cloudkeeper::Utils::URL.check!(url) uri = URI.parse url filename = generate_filename(uri) retrieve_image(uri, filename) check_image_checksum!(filename, checksum) Cloudkeeper::Entities::ImageFile.new filename, format(filename), Cloudkeeper::Utils::Checksum.compute(filename), File.size(filename) rescue Cloudkeeper::Errors::InvalidURLError, Cloudkeeper::Errors::Image::Format::RecognitionError, Cloudkeeper::Errors::ArgumentError, Cloudkeeper::Errors::NetworkConnectionError, ::IOError => ex raise Cloudkeeper::Errors::Image::DownloadError, "Image #{url.inspect} download error: #{ex.message}" end