discourse/chef/cookbooks/omnibus_updater/libraries/omnitrucker.rb

65 lines
1.8 KiB
Ruby
Raw Normal View History

2013-07-05 10:56:57 -04:00
module OmnibusTrucker
class << self
URL_MAP = {
:p => :platform, :pv => :platform_version, :m => :machine,
:v => :version, :prerelease => :prerelease,
:nightlies => :nightlies
}
def build_url(*opts)
args = node = nil
opts.each do |o|
if(o.kind_of?(Hash))
args = o
elsif(o.kind_of?(Chef::Node))
node = o
else
raise ArgumentError.new "Provided argument is not allowed: #{o.class}"
end
end
args ||= {}
if(node)
args = collect_attributes(node).merge(args)
end
url = args[:url] || "http://www.opscode.com/chef/download#{'-server' if args[:server]}"
u_args = URL_MAP.map do |u_k, a_k|
"#{u_k}=#{args[a_k]}" unless args[a_k].nil?
end.compact
"#{url}?#{u_args.join('&')}"
end
def collect_attributes(node, args={})
set = Hash[*(
[:platform_family, :platform, :platform_version].map do |k|
[k, args[k] || node[k]]
end.flatten.compact
)]
unless(@attrs)
if(set[:platform_family] == 'rhel')
@attrs = {:platform => 'el', :platform_version => set[:platform_version].to_i}
else
@attrs = {:platform => set[:platform], :platform_version => set[:platform_version]}
end
@attrs[:machine] = args[:machine] || node[:kernel][:machine]
end
@attrs
end
def url(url_or_node, node = nil)
if(url_or_node.is_a?(Chef::Node))
url = build_url(url_or_node)
node = url_or_node
else
url = url_or_node
raise "Node instance is required for Omnitruck.url!" unless node
end
request = Chef::REST::RESTRequest.new(:head, URI.parse(url), nil)
result = request.call
if(result.kind_of?(Net::HTTPRedirection))
result['location']
end
end
end
end