inspec/lib/vulcano/targets/url.rb

39 lines
802 B
Ruby
Raw Normal View History

# encoding: utf-8
require 'uri'
require 'tempfile'
require 'open-uri'
require 'vulcano/targets/zip'
module Vulcano::Targets
class UrlHelper
def handles?(target)
uri = URI.parse(target)
%{ http https }.include? uri.scheme
end
def resolve(target)
2015-09-03 18:35:23 +00:00
if target.start_with? 'https://github.com' and target.end_with? '.git'
url = target.sub(/.git$/,'') + '/archive/master.zip'
return resolve_zip(url)
else
end
end
def resolve_zip(url)
zipfile = Tempfile.new('vulcano-dl-')
zipfile.binmode
zipfile.write(open(url).read)
zipfile.rewind
content = ZipHelper.new.resolve(zipfile.path)
zipfile.close
zipfile.unlink
2015-09-03 18:45:37 +00:00
content
end
end
Vulcano::Targets.add_module('url', UrlHelper.new)
end