tecnh-chef 0.9.9.vpc

data/lib/chef/provider/breakpoint.rb

@@ -0,0 +1,36 @@
+#
+# 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.
+#
+
+class Chef
+  class Provider
+    class Breakpoint < Chef::Provider
+      
+      def load_current_resource
+      end
+      
+      def action_break
+        if defined?(Shef) && Shef.running?
+          run_context.resource_collection.iterator.pause
+          @new_resource.updated = true
+          run_context.resource_collection.iterator
+        end
+      end
+      
+    end
+  end
+end

data/lib/chef/provider/cookbook_file.rb

@@ -0,0 +1,96 @@
+#
+# Author:: Daniel DeLeo (<dan@opscode.com>)
+# Copyright:: Copyright (c) 2010 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/file_access_control'
+require 'chef/provider/file'
+require 'tempfile'
+
+class Chef
+  class Provider
+    class CookbookFile < Chef::Provider::File
+      
+      def load_current_resource
+        @current_resource = Chef::Resource::CookbookFile.new(@new_resource.name)
+        @new_resource.path.gsub!(/\\/, "/") # for Windows
+        @current_resource.path(@new_resource.path)
+        @current_resource
+      end
+
+
+      def action_create
+         if file_cache_location
+           Chef::Log.debug("content of file #{@new_resource.path} requires update")
+           backup_new_resource
+           Tempfile.open(::File.basename(@new_resource.name)) do |staging_file|
+             Chef::Log.debug("staging #{file_cache_location} to #{staging_file.path}")
+             staging_file.close
+             stage_file_to_tmpdir(staging_file.path)
+             FileUtils.mv(staging_file.path, @new_resource.path)
+           end
+           @new_resource.updated = true
+         else
+           set_all_access_controls(@new_resource.path)
+         end
+         @new_resource.updated
+       end
+
+      def action_create_if_missing
+        if ::File.exists?(@new_resource.path)
+          Chef::Log.debug("File #{@new_resource.path} exists, taking no action.")
+        else
+          action_create
+        end
+      end
+      
+      def file_cache_location
+        @file_cache_location ||= begin
+          cookbook = run_context.cookbook_collection[resource_cookbook]
+          cookbook.preferred_filename_on_disk_location(node, :files, @new_resource.source, @new_resource.path)
+        end
+      end
+      
+      # Determine the cookbook to get the file from. If new resource sets an 
+      # explicit cookbook, use it, otherwise fall back to the implicit cookbook
+      # i.e., the cookbook the resource was declared in.
+      def resource_cookbook
+        @new_resource.cookbook || @new_resource.cookbook_name
+      end
+      
+      # Copy the file from the cookbook cache to a temporary location and then
+      # set its file access control settings.
+      def stage_file_to_tmpdir(staging_file_location)
+        FileUtils.cp(file_cache_location, staging_file_location)
+        set_all_access_controls(staging_file_location)
+      end
+
+      def set_all_access_controls(file)
+        access_controls = Chef::FileAccessControl.new(@new_resource, file)
+        access_controls.set_all
+        @new_resource.updated = access_controls.modified?
+      end
+
+      def backup_new_resource
+        if ::File.exists?(@new_resource.path)
+          Chef::Log.info "Backing up current file at #{@new_resource.path}"
+          backup @new_resource.path
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/provider/cron.rb

@@ -0,0 +1,186 @@
+#
+# Author:: Bryan McLellan (btm@loftninjas.org)
+# Copyright:: Copyright (c) 2009 Bryan McLellan
+# 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/mixin/command'
+require 'chef/provider'
+
+class Chef
+  class Provider
+    class Cron < Chef::Provider
+      include Chef::Mixin::Command
+
+      CRON_PATTERN = /([-0-9*,\/]+)\s([-0-9*,\/]+)\s([-0-9*,\/]+)\s([-0-9*,\/]+)\s([-0-9*,\/]+)\s(.*)/
+
+      def initialize(new_resource, run_context)
+        super(new_resource, run_context)
+        @cron_exists = false
+        @cron_empty = false
+      end
+      attr_accessor :cron_exists, :cron_empty
+
+      def load_current_resource
+        crontab_lines = []
+        @current_resource = Chef::Resource::Cron.new(@new_resource.name)
+        @current_resource.user(@new_resource.user)
+        status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
+          stdout.each_line { |line| crontab_lines << line }
+        end
+        if status.exitstatus > 1
+          raise Chef::Exceptions::Cron, "Error determining state of #{@new_resource.name}, exit: #{status.exitstatus}"
+        elsif status.exitstatus == 0
+          cron_found = false
+          crontab_lines.each do |line|
+            case line.chomp
+            when "# Chef Name: #{@new_resource.name}"
+              Chef::Log.debug("Found cron '#{@new_resource.name}'")
+              cron_found = true
+              @cron_exists = true
+              next
+            when /^MAILTO=(\S*)/
+              @current_resource.mailto($1) if cron_found
+              next
+            when /^PATH=(\S*)/
+              @current_resource.path($1) if cron_found
+              next
+            when /^SHELL=(\S*)/
+              @current_resource.shell($1) if cron_found
+              next
+            when /^HOME=(\S*)/
+              @current_resource.home($1) if cron_found
+              next
+            when CRON_PATTERN
+              if cron_found
+                @current_resource.minute($1)
+                @current_resource.hour($2)
+                @current_resource.day($3)
+                @current_resource.month($4)
+                @current_resource.weekday($5)
+                @current_resource.command($6)
+                cron_found=false
+              end
+              next
+            else
+              next
+            end
+          end
+          Chef::Log.debug("Cron '#{@new_resource.name}' not found") unless @cron_exists
+        elsif status.exitstatus == 1
+          Chef::Log.debug("Cron empty for '#{@new_resource.user}'")
+          @cron_empty = true
+        end
+
+        @current_resource
+      end
+
+      def compare_cron
+        [ :minute, :hour, :day, :month, :weekday, :command, :mailto, :path, :shell, :home ].any? do |cron_var|
+          !@new_resource.send(cron_var).nil? && @new_resource.send(cron_var) != @current_resource.send(cron_var)
+        end
+      end
+
+      def action_create
+        crontab = String.new
+        newcron = String.new
+        cron_found = false
+
+        newcron << "# Chef Name: #{new_resource.name}\n"
+        [ :mailto, :path, :shell, :home ].each do |v|
+          newcron << "#{v.to_s.upcase}=#{@new_resource.send(v)}\n" if @new_resource.send(v)
+        end
+        newcron << "#{@new_resource.minute} #{@new_resource.hour} #{@new_resource.day} #{@new_resource.month} #{@new_resource.weekday} #{@new_resource.command}\n"
+
+        if @cron_exists
+          unless compare_cron
+            Chef::Log.debug("Skipping existing cron entry '#{@new_resource.name}'")
+            return
+          end
+          status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
+            stdout.each_line do |line|
+              case line.chomp
+              when "# Chef Name: #{@new_resource.name}"
+                cron_found = true
+                next
+              when CRON_PATTERN
+                if cron_found
+                  cron_found = false
+                  crontab << newcron
+                  next
+                end
+              else
+                next if cron_found
+              end
+              crontab << line
+            end
+          end
+
+          status = popen4("crontab -u #{@new_resource.user} -", :waitlast => true) do |pid, stdin, stdout, stderr|
+            crontab.each_line { |line| stdin.puts "#{line}" }
+          end
+          Chef::Log.info("Updated cron '#{@new_resource.name}'")
+          @new_resource.updated = true
+        else
+          unless @cron_empty
+            status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
+              stdout.each { |line| crontab << line }
+            end
+          end
+
+          crontab << newcron
+
+          status = popen4("crontab -u #{@new_resource.user} -", :waitlast => true) do |pid, stdin, stdout, stderr|
+            crontab.each_line { |line| stdin.puts "#{line}" }
+          end
+          Chef::Log.info("Added cron '#{@new_resource.name}'")
+          @new_resource.updated = true
+        end
+      end
+
+      def action_delete
+        if @cron_exists
+          crontab = String.new
+          cron_found = false
+          status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
+            stdout.each_line do |line|
+              case line.chomp
+              when "# Chef Name: #{@new_resource.name}"
+                cron_found = true
+                next
+              when CRON_PATTERN
+                if cron_found
+                  cron_found = false
+                  next
+                end
+              else
+                next if cron_found
+              end
+              crontab << line
+            end
+          end
+
+          status = popen4("crontab -u #{@new_resource.user} -", :waitlast => true) do |pid, stdin, stdout, stderr|
+            crontab.each_line { |line| stdin.puts "#{line}" }
+          end
+          Chef::Log.debug("Deleted cron '#{@new_resource.name}'")
+          @new_resource.updated = true
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/provider/deploy.rb

@@ -0,0 +1,319 @@
+#
+# 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/mixin/command"
+require "chef/mixin/from_file"
+require "chef/provider/git"
+require "chef/provider/subversion"
+
+class Chef
+  class Provider
+    class Deploy < Chef::Provider
+      
+      include Chef::Mixin::FromFile
+      include Chef::Mixin::Command
+      
+      attr_reader :scm_provider, :release_path
+      
+      def initialize(new_resource, run_context)
+        super(new_resource, run_context)
+        
+        @scm_provider = new_resource.scm_provider.new(new_resource, run_context)
+        
+        # @configuration is not used by Deploy, it is only for backwards compat with
+        # chef-deploy or capistrano hooks that might use it to get environment information
+        @configuration = @new_resource.to_hash
+        @configuration[:environment] = @configuration[:environment] && @configuration[:environment]["RAILS_ENV"]
+      end
+      
+      def load_current_resource
+        @release_path = @new_resource.deploy_to + "/releases/#{release_slug}"
+      end
+      
+      def sudo(command,&block)
+        execute(command, &block)
+      end
+      
+      def run(command, &block)
+        exec = execute(command, &block)
+        exec.user(@new_resource.user) if @new_resource.user
+        exec.group(@new_resource.group) if @new_resource.group
+        exec.cwd(release_path) unless exec.cwd
+        exec.environment(@new_resource.environment) unless exec.environment 
+        exec
+      end
+      
+      def action_deploy
+        if all_releases.include?(release_path)
+          if all_releases[-1] == release_path
+            Chef::Log.debug("Already deployed app at #{release_path}, and it is the latest revision.  Use action :force_deploy to re-deploy this revision.")
+          else
+            Chef::Log.info("Already deployed app at #{release_path}.  Rolling back to it - use action :force_deploy to re-checkout this revision.")
+            action_rollback
+          end
+        else
+          deploy
+          @new_resource.updated = true
+        end
+      end
+      
+      def action_force_deploy
+        if all_releases.include?(release_path)
+          Chef::Log.info("Already deployed app at #{release_path}, forcing.")
+          FileUtils.rm_rf(release_path)
+        end
+        deploy
+        @new_resource.updated = true
+      end
+      
+      def action_rollback
+        if release_path
+          rp_index = all_releases.index(release_path)
+          raise RuntimeError, "There is no release to rollback to!" unless rp_index
+          rp_index += 1
+          releases_to_nuke = all_releases[rp_index..-1]
+        else
+          @release_path = all_releases[-2] 
+          raise RuntimeError, "There is no release to rollback to!" unless @release_path
+          releases_to_nuke = [ all_releases.last ]
+        end
+
+        Chef::Log.info "rolling back to previous release: #{release_path}"
+        symlink
+        Chef::Log.info "restarting with previous release"
+        restart
+        releases_to_nuke.each do |i|
+          Chef::Log.info "Removing release: #{i}"
+          FileUtils.rm_rf i 
+          release_deleted(i)
+        end
+        @new_resource.updated = true
+      end
+      
+      def deploy
+        Chef::Log.info "deploying branch: #{@new_resource.branch}"
+        enforce_ownership
+        update_cached_repo
+        copy_cached_repo
+        install_gems
+        enforce_ownership
+        callback(:before_migrate, @new_resource.before_migrate)
+        migrate
+        callback(:before_symlink, @new_resource.before_symlink)
+        symlink
+        callback(:before_restart, @new_resource.before_restart)
+        restart
+        callback(:after_restart, @new_resource.after_restart)
+        cleanup!
+      end
+      
+      def callback(what, callback_code=nil)
+        @collection = Chef::ResourceCollection.new
+        case callback_code
+        when Proc
+          Chef::Log.info "Running callback #{what} code block"
+          recipe_eval(&callback_code)
+        when String
+          callback_file = "#{release_path}/#{callback_code}"
+          unless ::File.exist?(callback_file)
+            raise RuntimeError, "Can't find your callback file #{callback_file}"
+          end
+          run_callback_from_file(callback_file)
+        when nil
+          run_callback_from_file("#{release_path}/deploy/#{what}.rb")
+        else
+          raise RuntimeError, "You gave me a callback I don't know what to do with: #{callback_code.inspect}"
+        end
+      end
+      
+      def migrate
+        run_symlinks_before_migrate
+        
+        if @new_resource.migrate
+          enforce_ownership
+          
+          environment = @new_resource.environment
+          env_info = environment && environment.map do |key_and_val| 
+            "#{key_and_val.first}='#{key_and_val.last}'"
+          end.join(" ")
+          
+          Chef::Log.info  "Migrating: running #{@new_resource.migration_command} as #{@new_resource.user} " +
+                          "with environment #{env_info}"
+          run_command(run_options(:command => @new_resource.migration_command, :cwd=>release_path))
+        end
+      end
+      
+      def symlink
+        Chef::Log.info "Symlinking"
+        purge_tempfiles_from_current_release
+        link_tempfiles_to_current_release
+        link_current_release_to_production
+      end
+      
+      def restart
+        if restart_cmd = @new_resource.restart_command
+          if restart_cmd.kind_of?(Proc)
+            Chef::Log.info("Restarting app with embedded recipe")
+            recipe_eval(&restart_cmd)
+          else
+            Chef::Log.info("Restarting app with #{@new_resource.restart_command} in #{@new_resource.current_path}")
+            run_command(run_options(:command => @new_resource.restart_command, :cwd => @new_resource.current_path))
+          end
+        end
+      end
+      
+      def cleanup!
+        all_releases[0..-6].each do |old_release|
+          Chef::Log.info "Removing old release #{old_release}"
+          FileUtils.rm_rf(old_release)
+          release_deleted(old_release)
+        end
+      end
+      
+      def all_releases
+        Dir.glob(@new_resource.deploy_to + "/releases/*").sort
+      end
+      
+      def update_cached_repo
+        if @new_resource.svn_force_export
+          svn_force_export
+        else
+          run_scm_sync
+        end
+      end
+      
+      def run_scm_sync
+        Chef::Log.info "updating the cached checkout"
+        @scm_provider.action_sync
+      end
+      
+      def svn_force_export
+        Chef::Log.info "exporting source repository to #{@new_resource.destination}"
+        @scm_provider.action_force_export
+      end
+      
+      def copy_cached_repo
+        Chef::Log.info "copying the cached checkout to #{release_path}"
+        FileUtils.mkdir_p(@new_resource.deploy_to + "/releases")
+        run_command(:command => "cp -RPp #{::File.join(@new_resource.destination, ".")} #{release_path}")
+        release_created(release_path)
+      end
+      
+      def enforce_ownership
+        Chef::Log.info "ensuring proper ownership"
+        FileUtils.chown_R(@new_resource.user, @new_resource.group, @new_resource.deploy_to)
+      end
+      
+      def link_current_release_to_production
+        Chef::Log.info "Linking release #{release_path} into production at #{@new_resource.current_path}"
+        FileUtils.rm_f(@new_resource.current_path)
+        FileUtils.ln_sf(release_path, @new_resource.current_path)
+        enforce_ownership
+      end
+      
+      def run_symlinks_before_migrate
+        links_info = @new_resource.symlink_before_migrate.map { |src, dst| "#{src} => #{dst}" }.join(", ")
+        Chef::Log.info "Making pre-migration symlinks: #{links_info}"
+        @new_resource.symlink_before_migrate.each do |src, dest|
+          FileUtils.ln_sf(@new_resource.shared_path + "/#{src}", release_path + "/#{dest}")
+        end
+      end
+      
+      def link_tempfiles_to_current_release
+        dirs_info = @new_resource.create_dirs_before_symlink.join(",")
+        Chef::Log.info("creating directories before symlink: #{dirs_info}")
+        @new_resource.create_dirs_before_symlink.each { |dir| FileUtils.mkdir_p(release_path + "/#{dir}") }
+        
+        links_info = @new_resource.symlinks.map { |src, dst| "#{src} => #{dst}" }.join(", ")
+        Chef::Log.info("Linking shared paths into current release: #{links_info}")
+        @new_resource.symlinks.each do |src, dest|
+          FileUtils.ln_sf(@new_resource.shared_path + "/#{src}",  release_path + "/#{dest}")
+        end
+        run_symlinks_before_migrate
+        enforce_ownership
+      end
+      
+      def create_dirs_before_symlink
+      end
+      
+      def purge_tempfiles_from_current_release
+        log_info = @new_resource.purge_before_symlink.join(", ")
+        Chef::Log.info("Purging directories in checkout #{log_info}")
+        @new_resource.purge_before_symlink.each { |dir| FileUtils.rm_rf(release_path + "/#{dir}") }
+      end
+      
+      protected
+      
+      # Internal callback, called after copy_cached_repo.
+      # Override if you need to keep state externally.
+      def release_created(release_path)
+      end
+      
+      # Internal callback, called during cleanup! for each old release removed.
+      # Override if you need to keep state externally.
+      def release_deleted(release_path)
+      end
+      
+      def release_slug
+        raise Chef::Exceptions::Override, "You must override release_slug in #{self.to_s}"
+      end
+      
+      def install_gems
+        gem_resource_collection_runner.converge
+      end
+
+      def gem_resource_collection_runner
+        gems_collection = Chef::ResourceCollection.new
+        gem_packages.each { |rbgem| gems_collection << rbgem }
+        gems_run_context = run_context.dup
+        gems_run_context.resource_collection = gems_collection
+        Chef::Runner.new(gems_run_context)
+      end
+
+      def gem_packages
+        return [] unless ::File.exist?("#{release_path}/gems.yml")
+        gems = YAML.load(IO.read("#{release_path}/gems.yml"))
+        
+        gems.map do |g|
+          r = Chef::Resource::GemPackage.new(g[:name])
+          r.version g[:version]
+          r.action :install
+          r.source "http://gems.github.com"
+          r
+        end
+      end
+      
+      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] = @new_resource.environment if @new_resource.environment
+        run_opts
+      end
+      
+      def run_callback_from_file(callback_file)
+        if ::File.exist?(callback_file)
+          Dir.chdir(release_path) do
+            Chef::Log.info "running deploy hook: #{callback_file}"
+            recipe_eval { from_file(callback_file) }
+          end
+        end
+      end
+      
+    end
+  end
+end