microwave 0.1004.1

data/lib/chef/provider/file.rb

@@ -0,0 +1,222 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# 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 'chef/config'
+require 'chef/log'
+require 'chef/resource/file'
+require 'chef/mixin/checksum'
+require 'chef/provider'
+require 'etc'
+require 'fileutils'
+
+class Chef
+  class Provider
+    class File < Chef::Provider
+      include Chef::Mixin::Checksum
+
+      def negative_complement(big)
+        if big > 1073741823 # Fixnum max
+          big -= (2**32) # diminished radix wrap to negative
+        end
+        big
+      end
+
+      def octal_mode(mode)
+        ((mode.respond_to?(:oct) ? mode.oct : mode.to_i) & 007777)
+      end
+
+      private :negative_complement, :octal_mode
+
+      def load_current_resource
+        @current_resource = Chef::Resource::File.new(@new_resource.name)
+        @new_resource.path.gsub!(/\\/, "/") # for Windows
+        @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(octal_mode(cstats.mode))
+        end
+        @current_resource
+      end
+
+      # Compare the content of a file.  Returns true if they are the same, false if they are not.
+      def compare_content
+        checksum(@current_resource.path) == new_resource_content_checksum
+      end
+
+      # Set the content of the file, assuming it is not set correctly already.
+      def set_content
+        unless compare_content
+          backup @new_resource.path if ::File.exists?(@new_resource.path)
+          ::File.open(@new_resource.path, "w") {|f| f.write @new_resource.content }
+          Chef::Log.info("#{@new_resource} contents updated")
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+
+      # Compare the ownership of a file.  Returns true if they are the same, false if they are not.
+      def compare_owner
+        return false if @new_resource.owner.nil?
+
+        @set_user_id = case @new_resource.owner
+        when /^\d+$/, Integer
+          @new_resource.owner.to_i
+        else
+          # This raises an ArgumentError if you can't find the user
+          Etc.getpwnam(@new_resource.owner).uid
+        end
+
+        @set_user_id == @current_resource.owner
+      end
+
+      # Set the ownership on the file, assuming it is not set correctly already.
+      def set_owner
+        unless compare_owner
+          @set_user_id = negative_complement(@set_user_id)
+          ::File.chown(@set_user_id, nil, @new_resource.path)
+          Chef::Log.info("#{@new_resource} owner changed to #{@set_user_id}")
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+
+      # Compares the group of a file.  Returns true if they are the same, false if they are not.
+      def compare_group
+        return false if @new_resource.group.nil?
+
+        @set_group_id = case @new_resource.group
+        when /^\d+$/, Integer
+          @new_resource.group.to_i
+        else
+          Etc.getgrnam(@new_resource.group).gid
+        end
+
+        @set_group_id == @current_resource.group
+      end
+
+      def set_group
+        unless compare_group
+          @set_group_id = negative_complement(@set_group_id)
+          ::File.chown(nil, @set_group_id, @new_resource.path)
+          Chef::Log.info("#{@new_resource} group changed to #{@set_group_id}")
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+
+      def compare_mode
+        case @new_resource.mode
+        when /^\d+$/, Integer
+          octal_mode(@new_resource.mode) == octal_mode(@current_resource.mode)
+        else
+          false
+        end
+      end
+
+      def set_mode
+        unless compare_mode && @new_resource.mode != nil
+          # CHEF-174, bad mojo around treating integers as octal.  If a string is passed, we try to do the "right" thing
+          ::File.chmod(octal_mode(@new_resource.mode), @new_resource.path)
+          Chef::Log.info("#{@new_resource} mode changed to #{sprintf("%o" % octal_mode(@new_resource.mode))}")
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+
+      def action_create
+        assert_enclosing_directory_exists!
+        unless ::File.exists?(@new_resource.path)
+          ::File.open(@new_resource.path, "w+") {|f| f.write @new_resource.content }
+          @new_resource.updated_by_last_action(true)
+          Chef::Log.info("#{@new_resource} created file #{@new_resource.path}")
+        else
+          set_content unless @new_resource.content.nil?
+        end
+        set_owner unless @new_resource.owner.nil?
+        set_group unless @new_resource.group.nil?
+        set_mode unless @new_resource.mode.nil?
+      end
+
+      def action_create_if_missing
+        action_create
+      end
+
+      def action_delete
+        if ::File.exists?(@new_resource.path)
+          if ::File.writable?(@new_resource.path)
+            backup unless ::File.symlink?(@new_resource.path)
+            ::File.delete(@new_resource.path)
+            Chef::Log.info("#{@new_resource} deleted file at #{@new_resource.path}")
+            @new_resource.updated_by_last_action(true)
+          else
+            raise "Cannot delete #{@new_resource} at #{@new_resource_path}!"
+          end
+        end
+      end
+
+      def action_touch
+        action_create
+        time = Time.now
+        ::File.utime(time, time, @new_resource.path)
+        Chef::Log.info("#{@new_resource} updated atime and mtime to #{time}")
+        @new_resource.updated_by_last_action(true)
+      end
+
+      def backup(file=nil)
+        file ||= @new_resource.path
+        if @new_resource.backup != false && @new_resource.backup > 0 && ::File.exist?(file)
+          time = Time.now
+          savetime = time.strftime("%Y%m%d%H%M%S")
+          backup_filename = "#{@new_resource.path}.chef-#{savetime}"
+          backup_filename = backup_filename.sub(/^([A-Za-z]:)/, "") #strip drive letter on Windows
+          # if :file_backup_path is nil, we fallback to the old behavior of
+          # keeping the backup in the same directory. We also need to to_s it
+          # so we don't get a type error around implicit to_str conversions.
+          prefix = Chef::Config[:file_backup_path].to_s
+          backup_path = ::File.join(prefix, backup_filename)
+          FileUtils.mkdir_p(::File.dirname(backup_path)) if Chef::Config[:file_backup_path]
+          FileUtils.cp(file, backup_path, :preserve => true)
+          Chef::Log.info("#{@new_resource} backed up to #{backup_path}")
+
+          # Clean up after the number of backups
+          slice_number = @new_resource.backup
+          backup_files = Dir[::File.join(prefix, ".#{@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|
+              FileUtils.rm(backup_to_delete)
+              Chef::Log.info("#{@new_resource} removed backup at #{backup_to_delete}")
+            end
+          end
+        end
+      end
+
+      private
+
+      def assert_enclosing_directory_exists!
+        enclosing_dir = ::File.dirname(@new_resource.path)
+        unless ::File.directory?(enclosing_dir)
+          msg = "Cannot create a file at #{@new_resource.path} because the enclosing directory (#{enclosing_dir}) does not exist"
+          raise Chef::Exceptions::EnclosingDirectoryDoesNotExist, msg
+        end
+      end
+
+      def new_resource_content_checksum
+        @new_resource.content && Digest::SHA2.hexdigest(@new_resource.content)
+      end
+    end
+  end
+end

data/lib/chef/provider/git.rb

@@ -0,0 +1,243 @@
+#
+# Author:: Daniel DeLeo (<dan@kallistec.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# 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 'chef/log'
+require 'chef/provider'
+require 'chef/mixin/shell_out'
+require 'fileutils'
+
+class Chef
+  class Provider
+    class Git < Chef::Provider
+
+      include Chef::Mixin::ShellOut
+
+      def load_current_resource
+        @current_resource = Chef::Resource::Git.new(@new_resource.name)
+        if current_revision = find_current_revision
+          @current_resource.revision current_revision
+        end
+      end
+
+      def action_checkout
+        assert_target_directory_valid!
+
+        if target_dir_non_existent_or_empty?
+          clone
+          checkout
+          enable_submodules
+          add_remotes
+          @new_resource.updated_by_last_action(true)
+        else
+          Chef::Log.debug "#{@new_resource} checkout destination #{@new_resource.destination} already exists or is a non-empty directory"
+        end
+      end
+
+      def action_export
+        action_checkout
+        FileUtils.rm_rf(::File.join(@new_resource.destination,".git"))
+        @new_resource.updated_by_last_action(true)
+      end
+
+      def action_sync
+        assert_target_directory_valid!
+
+        if existing_git_clone?
+          current_rev = find_current_revision
+          Chef::Log.debug "#{@new_resource} current revision: #{current_rev} target revision: #{target_revision}"
+          unless current_revision_matches_target_revision?
+            fetch_updates
+            enable_submodules
+            Chef::Log.info "#{@new_resource} updated to revision #{target_revision}"
+            @new_resource.updated_by_last_action(true)
+          end
+          add_remotes
+        else
+          action_checkout
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+
+      def assert_target_directory_valid!
+        target_parent_directory = ::File.dirname(@new_resource.destination)
+        unless ::File.directory?(target_parent_directory)
+          msg = "Cannot clone #{@new_resource} to #{@new_resource.destination}, the enclosing directory #{target_parent_directory} does not exist"
+          raise Chef::Exceptions::MissingParentDirectory, msg
+        end
+      end
+
+      def existing_git_clone?
+        ::File.exist?(::File.join(@new_resource.destination, ".git"))
+      end
+
+      def target_dir_non_existent_or_empty?
+        !::File.exist?(@new_resource.destination) || Dir.entries(@new_resource.destination).sort == ['.','..']
+      end
+
+      def find_current_revision
+        Chef::Log.debug("#{@new_resource} finding current git revision")
+        if ::File.exist?(::File.join(cwd, ".git"))
+          # 128 is returned when we're not in a git repo. this is fine
+          result = shell_out!('git rev-parse HEAD', :cwd => cwd, :returns => [0,128]).stdout.strip
+        end
+        sha_hash?(result) ? result : nil
+      end
+
+      def add_remotes
+        if (@new_resource.additional_remotes.length > 0)
+          @new_resource.additional_remotes.each_pair do |remote_name, remote_url|
+            Chef::Log.info "#{@new_resource} adding git remote #{remote_name} = #{remote_url}"
+            command = "git remote add #{remote_name} #{remote_url}"
+            if shell_out(command, run_options(:cwd => @new_resource.destination, :command_log_level => :info)).exitstatus != 0
+              @new_resource.updated_by_last_action(true)
+            end
+          end
+        end
+      end
+
+      def clone
+        remote = @new_resource.remote
+
+        args = []
+        args << "-o #{remote}" unless remote == 'origin'
+        args << "--depth #{@new_resource.depth}" if @new_resource.depth
+
+        Chef::Log.info "#{@new_resource} cloning repo #{@new_resource.repository} to #{@new_resource.destination}"
+
+        clone_cmd = "git clone #{args.join(' ')} #{@new_resource.repository} #{@new_resource.destination}"
+        shell_out!(clone_cmd, run_options(:command_log_level => :info))
+      end
+
+      def checkout
+        sha_ref = target_revision
+        # checkout into a local branch rather than a detached HEAD
+        shell_out!("git checkout -b deploy #{sha_ref}", run_options(:cwd => @new_resource.destination))
+        Chef::Log.info "#{@new_resource} checked out branch: #{@new_resource.revision} reference: #{sha_ref}"
+      end
+
+      def enable_submodules
+        if @new_resource.enable_submodules
+          Chef::Log.info "#{@new_resource} enabling git submodules"
+          command = "git submodule init && git submodule update"
+          shell_out!(command, run_options(:cwd => @new_resource.destination, :command_log_level => :info))
+        end
+      end
+
+      def fetch_updates
+        setup_remote_tracking_branches if @new_resource.remote != 'origin'
+
+        # since we're in a local branch already, just reset to specified revision rather than merge
+        fetch_command = "git fetch #{@new_resource.remote} && git fetch #{@new_resource.remote} --tags && git reset --hard #{target_revision}"
+        Chef::Log.debug "Fetching updates from #{new_resource.remote} and resetting to revison #{target_revision}"
+        shell_out!(fetch_command, run_options(:cwd => @new_resource.destination))
+      end
+
+      # Use git-config to setup a remote tracking branches. Could use
+      # git-remote but it complains when a remote of the same name already
+      # exists, git-config will just silenty overwrite the setting every
+      # time. This could cause wierd-ness in the remote cache if the url
+      # changes between calls, but as long as the repositories are all
+      # based from each other it should still work fine.
+      def setup_remote_tracking_branches
+        command = []
+
+        Chef::Log.debug "#{@new_resource} configuring remote tracking branches for repository #{@new_resource.repository} "+
+                        "at remote #{@new_resource.remote}"
+        command << "git config remote.#{@new_resource.remote}.url #{@new_resource.repository}"
+        command << "git config remote.#{@new_resource.remote}.fetch +refs/heads/*:refs/remotes/#{@new_resource.remote}/*"
+        shell_out!(command.join(" && "), run_options(:cwd => @new_resource.destination))
+      end
+
+      def current_revision_matches_target_revision?
+        (!@current_resource.revision.nil?) && (target_revision.strip.to_i(16) == @current_resource.revision.strip.to_i(16))
+      end
+
+      def target_revision
+        @target_revision ||= begin
+          assert_revision_not_remote
+
+          if sha_hash?(@new_resource.revision)
+            @target_revision = @new_resource.revision
+          else
+            resolved_reference = remote_resolve_reference
+            @target_revision = extract_revision(resolved_reference)
+          end
+        end
+      end
+
+      alias :revision_slug :target_revision
+
+      def remote_resolve_reference
+        Chef::Log.debug("#{@new_resource} resolving remote reference")
+        command = git('ls-remote', @new_resource.repository, @new_resource.revision)
+        shell_out!(command, run_options).stdout
+      end
+
+      private
+
+      def run_options(run_opts={})
+        run_opts[:user] = @new_resource.user if @new_resource.user
+        run_opts[:group] = @new_resource.group if @new_resource.group
+        run_opts[:environment] = {"GIT_SSH" => @new_resource.ssh_wrapper} if @new_resource.ssh_wrapper
+        run_opts[:command_log_prepend] = @new_resource.to_s
+        run_opts[:command_log_level] ||= :debug
+        if run_opts[:command_log_level] == :info
+          if STDOUT.tty? && Chef::Log.info?
+            run_opts[:live_stream] = STDOUT
+          end
+        end
+        run_opts
+      end
+
+      def cwd
+        @new_resource.destination
+      end
+
+      def git(*args)
+        ["git", *args].compact.join(" ")
+      end
+
+      def sha_hash?(string)
+        string =~ /^[0-9a-f]{40}$/
+      end
+
+      def assert_revision_not_remote
+        if @new_resource.revision =~ /^origin\//
+          reference = @new_resource.revision
+          error_text =  "Deploying remote branches is not supported. " +
+                        "Specify the remote branch as a local branch for " +
+                        "the git repository you're deploying from " +
+                        "(ie: '#{reference.gsub('origin/', '')}' rather than '#{reference}')."
+          raise RuntimeError, error_text
+        end
+      end
+
+      def extract_revision(resolved_reference)
+        unless resolved_reference =~ /^([0-9a-f]{40})\s+(\S+)/
+          msg = "Unable to parse SHA reference for '#{@new_resource.revision}' in repository '#{@new_resource.repository}'. "
+          msg << "Verify your (case-sensitive) repository URL and revision.\n"
+          msg << "`git ls-remote` output: #{resolved_reference}"
+          raise Chef::Exceptions::UnresolvableGitReference, msg
+        end
+        $1
+      end
+
+    end
+  end
+end