diff options
Diffstat (limited to 'chef/lib/chef/provider/file.rb')
-rw-r--r-- | chef/lib/chef/provider/file.rb | 169 |
1 files changed, 169 insertions, 0 deletions
diff --git a/chef/lib/chef/provider/file.rb b/chef/lib/chef/provider/file.rb new file mode 100644 index 0000000000..d25e5decca --- /dev/null +++ b/chef/lib/chef/provider/file.rb @@ -0,0 +1,169 @@ +# +# Author:: Adam Jacob (<adam@hjksolutions.com>) +# Copyright:: Copyright (c) 2008 HJK Solutions, LLC +# License:: Apache License, Version 2.0 +# +# 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 'etc' +require 'fileutils' +require File.join(File.dirname(__FILE__), "..", "mixin", "checksum") +require File.join(File.dirname(__FILE__), "..", "mixin", "generate_url") + +class Chef + class Provider + class File < Chef::Provider + include Chef::Mixin::Checksum + include Chef::Mixin::GenerateURL + + def load_current_resource + @current_resource = Chef::Resource::File.new(@new_resource.name) + @current_resource.path(@new_resource.path) + if ::File.exist?(@current_resource.path) && ::File.readable?(@current_resource.path) + cstats = ::File.stat(@current_resource.path) + @current_resource.owner(cstats.uid) + @current_resource.group(cstats.gid) + @current_resource.mode("%o" % (cstats.mode & 007777)) + @current_resource.checksum(checksum(@current_resource.path)) + end + @current_resource + end + + # Compare the ownership of a file. Returns true if they are the same, false if they are not. + def compare_owner + if @new_resource.owner != nil + case @new_resource.owner + when /^\d+$/, Integer + @set_user_id = @new_resource.owner.to_i + @set_user_id == @current_resource.owner + else + # This raises an ArugmentError if you can't find the user + user_info = Etc.getpwnam(@new_resource.owner) + @set_user_id = user_info.uid + @set_user_id == @current_resource.owner + end + end + end + + # Set the ownership on the file, assuming it is not set correctly already. + def set_owner + unless compare_owner + Chef::Log.info("Setting owner to #{@set_user_id} for #{@new_resource}") + ::File.chown(@set_user_id, nil, @new_resource.path) + @new_resource.updated = true + end + end + + # Compares the group of a file. Returns true if they are the same, false if they are not. + def compare_group + if @new_resource.group != nil + case @new_resource.group + when /^\d+$/, Integer + @set_group_id = @new_resource.group.to_i + @set_group_id == @current_resource.group + else + group_info = Etc.getgrnam(@new_resource.group) + @set_group_id = group_info.gid + @set_group_id == @current_resource.group + end + end + end + + def set_group + unless compare_group + Chef::Log.info("Setting group to #{@set_group_id} for #{@new_resource}") + ::File.chown(nil, @set_group_id, @new_resource.path) + @new_resource.updated = true + end + end + + def compare_mode + if @new_resource.mode != nil + case @new_resource.mode + when /^\d+$/, Integer + real_mode = sprintf("%o" % (@new_resource.mode & 007777)) + real_mode.to_i == @current_resource.mode.to_i + end + end + end + + def set_mode + unless compare_mode && @new_resource.mode != nil + Chef::Log.info("Setting mode to #{sprintf("%o" % (@new_resource.mode & 007777)) + } for #{@new_resource}") + ::File.chmod(@new_resource.mode.to_i, @new_resource.path) + @new_resource.updated = true + end + end + + def action_create + unless ::File.exists?(@new_resource.path) + Chef::Log.info("Creating #{@new_resource} at #{@new_resource.path}") + ::File.open(@new_resource.path, "w+") { |f| } + @new_resource.updated = true + end + set_owner if @new_resource.owner != nil + set_group if @new_resource.group != nil + set_mode if @new_resource.mode != nil + end + + def action_delete + if ::File.exists?(@new_resource.path) && ::File.writable?(@new_resource.path) + backup + Chef::Log.info("Deleting #{@new_resource} at #{@new_resource.path}") + ::File.delete(@new_resource.path) + @new_resource.updated = true + else + raise "Cannot delete #{@new_resource} at #{@new_resource_path}!" + end + end + + def action_touch + action_create + time = Time.now + Chef::Log.info("Updating #{@new_resource} with new atime/mtime of #{time}") + ::File.utime(time, time, @new_resource.path) + @new_resource.updated = true + end + + def backup(file=nil) + file ||= @new_resource.path + if @new_resource.backup && ::File.exist?(file) + time = Time.now + savetime = time.strftime("%Y%m%d%H%M%S") + backup_filename = "#{@new_resource.path}.chef-#{savetime}" + Chef::Log.info("Backing up #{@new_resource} to #{backup_filename}") + FileUtils.cp(file, backup_filename) + + # Clean up after the number of backups + slice_number = @new_resource.backup - 1 + backup_files = Dir["#{@new_resource.path}.chef-*"].sort { |a,b| b <=> a } + if backup_files.length >= @new_resource.backup + remainder = backup_files.slice(slice_number..-1) + remainder.each do |backup_to_delete| + Chef::Log.info("Removing backup of #{@new_resource} at #{backup_to_delete}") + FileUtils.rm(backup_to_delete) + end + end + + end + end + + def generate_url(url, type, args=nil) + generate_cookbook_url(url, @new_resource.cookbook_name, type, @node, args) + end + + end + end +end
\ No newline at end of file |