mirror of
https://github.com/inspec/inspec
synced 2024-11-24 13:43:09 +00:00
35 lines
839 B
Ruby
35 lines
839 B
Ruby
# 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)
|
|
return false if uri.nil? or uri.scheme.nil?
|
|
%{ http https }.include? uri.scheme
|
|
end
|
|
|
|
def resolve(target)
|
|
if target.start_with? 'https://github.com' and target.end_with? '.git'
|
|
url = target.sub(/.git$/, '') + '/archive/master.zip'
|
|
return resolve_zip(url)
|
|
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
|
|
content
|
|
end
|
|
end
|
|
|
|
Vulcano::Targets.add_module('url', UrlHelper.new)
|
|
end
|