Vendor the external cookbooks
Knife-Zero doesn't include Berkshelf support, so vendoring everything in the repo is convenient again
This commit is contained in:
156
cookbooks/elasticsearch/libraries/helpers.rb
Normal file
156
cookbooks/elasticsearch/libraries/helpers.rb
Normal file
@@ -0,0 +1,156 @@
|
||||
module ElasticsearchCookbook
|
||||
# Helper methods included by various providers and passed to the template engine
|
||||
module Helpers
|
||||
def find_es_resource(run_context, resource_type, resource)
|
||||
resource_name = resource.name
|
||||
instance_name = resource.instance_name
|
||||
|
||||
# if we are truly given a specific name to find
|
||||
name_match = find_exact_resource(run_context, resource_type, resource_name) rescue nil
|
||||
return name_match if name_match
|
||||
|
||||
# first try by instance name attribute
|
||||
name_instance = find_instance_name_resource(run_context, resource_type, instance_name) rescue nil
|
||||
return name_instance if name_instance
|
||||
|
||||
# otherwise try the defaults
|
||||
name_default = find_exact_resource(run_context, resource_type, 'default') rescue nil
|
||||
name_elasticsearch = find_exact_resource(run_context, resource_type, 'elasticsearch') rescue nil
|
||||
|
||||
# if we found exactly one default name that matched
|
||||
return name_default if name_default && !name_elasticsearch
|
||||
return name_elasticsearch if name_elasticsearch && !name_default
|
||||
|
||||
raise "Could not find exactly one #{resource_type} resource, and no specific resource or instance name was given"
|
||||
end
|
||||
|
||||
# find exactly the resource name and type, but raise if there's multiple matches
|
||||
# see https://github.com/chef/chef/blob/master/lib/chef/resource_collection/resource_set.rb#L80
|
||||
def find_exact_resource(run_context, resource_type, resource_name)
|
||||
rc = run_context.resource_collection
|
||||
result = rc.find(resource_type => resource_name)
|
||||
|
||||
if result && result.is_a?(Array)
|
||||
str = ''
|
||||
str << "more than one #{resource_type} was found, "
|
||||
str << 'you must specify a precise resource name'
|
||||
raise str
|
||||
end
|
||||
|
||||
result
|
||||
end
|
||||
|
||||
def find_instance_name_resource(run_context, resource_type, instance_name)
|
||||
results = []
|
||||
rc = run_context.resource_collection
|
||||
|
||||
rc.each do |r|
|
||||
next unless r.resource_name == resource_type && r.instance_name == instance_name
|
||||
results << r
|
||||
end
|
||||
|
||||
if !results.empty? && results.length > 1
|
||||
str = ''
|
||||
str << "more than one #{resource_type} was found, "
|
||||
str << 'you must specify a precise instance name'
|
||||
raise str
|
||||
elsif !results.empty?
|
||||
return results.first
|
||||
end
|
||||
|
||||
nil # falsey
|
||||
end
|
||||
|
||||
def determine_download_url(new_resource, node)
|
||||
platform_family = node['platform_family']
|
||||
|
||||
version_key = 'download_urls'
|
||||
if Gem::Version.new(new_resource.version) >= Gem::Version.new('7.0.0')
|
||||
version_key = 'download_urls_v7'
|
||||
end
|
||||
|
||||
url_string = nil
|
||||
if new_resource.download_url
|
||||
url_string = new_resource.download_url
|
||||
elsif new_resource.type == 'tarball'
|
||||
url_string = node['elasticsearch'][version_key]['tarball']
|
||||
elsif new_resource.type == 'package' && node['elasticsearch']['download_urls'][platform_family]
|
||||
url_string = node['elasticsearch'][version_key][platform_family]
|
||||
end
|
||||
|
||||
if url_string && new_resource.version
|
||||
return format(url_string, new_resource.version)
|
||||
elsif url_string
|
||||
return url_string
|
||||
end
|
||||
end
|
||||
|
||||
def determine_download_checksum(new_resource, node)
|
||||
platform_family = node['platform_family']
|
||||
|
||||
# for the sake of finding correct attribute data, use rhel for amazon too
|
||||
# See https://github.com/elastic/cookbook-elasticsearch/issues/609
|
||||
platform_family = 'rhel' if platform_family == 'amazon'
|
||||
|
||||
install_type = new_resource.type
|
||||
version = new_resource.version
|
||||
|
||||
if new_resource.download_checksum
|
||||
new_resource.download_checksum
|
||||
elsif install_type == 'tarball'
|
||||
node && version &&
|
||||
node['elasticsearch'] &&
|
||||
node['elasticsearch']['checksums'] &&
|
||||
node['elasticsearch']['checksums'][version] &&
|
||||
node['elasticsearch']['checksums'][version]['tarball']
|
||||
elsif install_type == 'package' && node['elasticsearch']['checksums'][version] && node['elasticsearch']['checksums'][version][platform_family]
|
||||
node && version && platform_family &&
|
||||
node['elasticsearch'] &&
|
||||
node['elasticsearch']['checksums'] &&
|
||||
node['elasticsearch']['checksums'][version] &&
|
||||
node['elasticsearch']['checksums'][version][platform_family]
|
||||
end
|
||||
end
|
||||
|
||||
# proxy helper for chef sets JVM 8 proxy options
|
||||
def get_java_proxy_arguments(enabled = true)
|
||||
return '' unless enabled
|
||||
require 'uri'
|
||||
output = ''
|
||||
|
||||
if Chef::Config[:http_proxy] && !Chef::Config[:http_proxy].empty?
|
||||
parsed_uri = URI(Chef::Config[:http_proxy])
|
||||
output += "-Dhttp.proxyHost=#{parsed_uri.host} -Dhttp.proxyPort=#{parsed_uri.port} "
|
||||
end
|
||||
|
||||
if Chef::Config[:https_proxy] && !Chef::Config[:https_proxy].empty?
|
||||
parsed_uri = URI(Chef::Config[:https_proxy])
|
||||
output += "-Dhttps.proxyHost=#{parsed_uri.host} -Dhttps.proxyPort=#{parsed_uri.port} "
|
||||
end
|
||||
|
||||
output
|
||||
rescue
|
||||
''
|
||||
end
|
||||
end
|
||||
|
||||
class HashAndMashBlender
|
||||
attr_accessor :target
|
||||
def initialize(hash_or_mash_or_whatever)
|
||||
self.target = hash_or_mash_or_whatever
|
||||
end
|
||||
|
||||
def to_hash
|
||||
target.each_with_object({}) do |(k, v), hsh|
|
||||
hsh[k] =
|
||||
if v.respond_to?(:to_hash)
|
||||
self.class.new(v).to_hash
|
||||
elsif v.respond_to?(:to_a)
|
||||
v.to_a
|
||||
else
|
||||
v
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user