discourse/chef/cookbooks/chef_handler/providers/default.rb

94 lines
2.9 KiB
Ruby

#
# Author:: Seth Chisamore <schisamo@opscode.com>
# Cookbook Name:: chef_handler
# Provider:: default
#
# Copyright:: 2011-2013, Opscode, Inc <legal@opscode.com>
#
# 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.
#
def whyrun_supported?
true
end
action :enable do
# use load instead of require to ensure the handler file
# is reloaded into memory each chef run. fixes COOK-620
handler = nil
converge_by("load #{@new_resource.source}") do
begin
Object.send(:remove_const, klass)
GC.start
rescue
Chef::Log.debug("#{@new_resource.class_name} has not been loaded.")
end
file_name = @new_resource.source
file_name << ".rb" unless file_name =~ /.*\.rb$/
load file_name
handler = klass.send(:new, *collect_args(@new_resource.arguments))
end
@new_resource.supports.each do |type, enable|
if enable
# we have to re-enable the handler every chef run
# to ensure daemonized Chef always has the latest
# handler code. TODO: add a :reload action
converge_by("enable #{@new_resource} as a #{type} handler") do
Chef::Log.info("Enabling #{@new_resource} as a #{type} handler")
Chef::Config.send("#{type}_handlers").delete_if { |v| v.class.to_s.include? @new_resource.class_name.split('::', 3).last }
Chef::Config.send("#{type}_handlers") << handler
end
end
end
end
action :disable do
@new_resource.supports.each_key do |type|
if enabled?(type)
converge_by("disable #{@new_resource} as a #{type} handler") do
Chef::Log.info("Disabling #{@new_resource} as a #{type} handler")
Chef::Config.send("#{type}_handlers").delete_if { |v| v.class.to_s.include? @new_resource.class_name.split('::', 3).last }
end
end
end
end
def load_current_resource
@current_resource = Chef::Resource::ChefHandler.new(@new_resource.name)
@current_resource.class_name(@new_resource.class_name)
@current_resource.source(@new_resource.source)
@current_resource
end
private
def enabled?(type)
Chef::Config.send("#{type}_handlers").select do |handler|
handler.class.to_s.include? @new_resource.class_name
end.size >= 1
end
def collect_args(resource_args = [])
if resource_args.is_a? Array
resource_args
else
[resource_args]
end
end
def klass
@klass ||= begin
@new_resource.class_name.split('::').inject(Kernel) { |scope, const_name| scope.const_get(const_name) }
end
end