103 lines
3.8 KiB
Ruby
103 lines
3.8 KiB
Ruby
#
|
|
# Cookbook:: omnibus_updater
|
|
# Library:: omnitrucker
|
|
#
|
|
# Copyright:: 2014-2017, Heavy Water Ops, LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
require 'chef/rest'
|
|
require 'chef/mash'
|
|
require 'net/http'
|
|
|
|
Chef::Mash = Mash unless Chef.constants.include?(:Mash)
|
|
|
|
module OmnibusTrucker
|
|
class << self
|
|
URL_MAP = {
|
|
p: :platform, pv: :platform_version, m: :machine,
|
|
v: :version, prerelease: :prerelease,
|
|
nightlies: :nightlies
|
|
}.freeze unless defined?(URL_MAP)
|
|
|
|
def build_url(*opts)
|
|
args = node = nil
|
|
opts.each do |o|
|
|
if o.is_a?(Hash)
|
|
args = o
|
|
elsif o.is_a?(Chef::Node)
|
|
node = o
|
|
else
|
|
raise ArgumentError.new "Provided argument is not allowed: #{o.class}"
|
|
end
|
|
end
|
|
args ||= {}
|
|
args = collect_attributes(node).merge(args) if node
|
|
url = args[:url] || "http://www.chef.io/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 = Chef::Mash[
|
|
%w(platform_family platform platform_version).map do |k|
|
|
[k, args[k] || node[k]]
|
|
end
|
|
]
|
|
unless @attrs
|
|
if set['platform'] == 'amazon'
|
|
@attrs = { platform: 'el', platform_version: 6 }
|
|
elsif set['platform_family'] == 'fedora'
|
|
@attrs = { platform: 'el', platform_version: 7 }
|
|
elsif set['platform_family'] == 'rhel'
|
|
@attrs = { platform: 'el', platform_version: set['platform_version'].to_i }
|
|
elsif set['platform'] == 'debian'
|
|
@attrs = { platform: set['platform'], platform_version: set['platform_version'].to_i }
|
|
elsif set['platform'] =~ /opensuse/
|
|
@attrs = { platform: 'suse', platform_version: 13 }
|
|
elsif set['platform_family'] == 'suse'
|
|
@attrs = { platform: 'sles', platform_version: 12 }
|
|
elsif set['platform_family'] == 'mac_os_x'
|
|
major, minor, _patch = set['platform_version'].split('.').map { |v| String(v) }
|
|
minor = [minor.to_i, 11].min # this is somewhat of a hack, we need to prevent this recipe to construct links for 10.12 for which there is no download yet...
|
|
@attrs = { platform: set['platform_family'], platform_version: [[major, minor].join('.'), '10.7'].max { |x, y| Gem::Version.new(x) <=> Gem::Version.new(y) } }
|
|
elsif set['platform_family'] == 'windows'
|
|
@attrs = { platform: set['platform'], platform_version: '2008r2' }
|
|
else
|
|
@attrs = { platform: set['platform'], platform_version: set['platform_version'] }
|
|
end
|
|
@attrs[:machine] = args[:machine] || node['kernel']['machine']
|
|
@attrs[:machine] = 'i386' if set['platform_family'] == 'solaris2' && @attrs[:machine] == 'i86pc'
|
|
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
|
|
Chef::Log.info("Using URL '#{url}' for chef-download") unless url.nil?
|
|
request = Chef::REST::RESTRequest.new(:head, URI.parse(url), nil)
|
|
result = request.call
|
|
result['location'] if result.is_a?(Net::HTTPRedirection)
|
|
end
|
|
end
|
|
end
|