tylerrick-chef 0.10.8

data/lib/chef/provider.rb

@@ -0,0 +1,124 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Christopher Walters (<cw@opscode.com>)
+# Copyright:: Copyright (c) 2008, 2009 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/from_file'
+require 'chef/mixin/convert_to_class_name'
+require 'chef/mixin/recipe_definition_dsl_core'
+
+class Chef
+  class Provider
+    
+    include Chef::Mixin::RecipeDefinitionDSLCore
+    
+    attr_accessor :new_resource, :current_resource, :run_context
+    
+    def initialize(new_resource, run_context)
+      @new_resource = new_resource
+      @current_resource = nil
+      @run_context = run_context
+    end
+
+    def node
+      run_context && run_context.node
+    end
+
+    # Used by providers supporting embedded recipes
+    def resource_collection
+      run_context && run_context.resource_collection
+    end
+
+    def cookbook_name
+      new_resource.cookbook_name
+    end
+    
+    def load_current_resource
+      raise Chef::Exceptions::Override, "You must override load_current_resource in #{self.to_s}"
+    end
+    
+    def action_nothing
+      Chef::Log.debug("Doing nothing for #{@new_resource.to_s}")
+      true
+    end
+    
+    protected
+    
+    def recipe_eval(&block)
+      # This block has new resource definitions within it, which
+      # essentially makes it an in-line Chef run. Save our current
+      # run_context and create one anew, so the new Chef run only
+      # executes the embedded resources.
+      #
+      # TODO: timh,cw: 2010-5-14: This means that the resources within
+      # this block cannot interact with resources outside, e.g.,
+      # manipulating notifies.
+      saved_run_context = @run_context
+      @run_context = @run_context.dup
+      @run_context.resource_collection = Chef::ResourceCollection.new
+      instance_eval(&block)
+      Chef::Runner.new(@run_context).converge
+      
+      @run_context = saved_run_context
+    end
+    
+    public
+    
+    class << self
+      include Chef::Mixin::ConvertToClassName
+      
+      def build_from_file(cookbook_name, filename, run_context)
+        pname = filename_to_qualified_string(cookbook_name, filename)
+        
+        # Add log entry if we override an existing light-weight provider.
+        class_name = convert_to_class_name(pname)
+        overriding = Chef::Provider.const_defined?(class_name)
+        Chef::Log.info("#{class_name} light-weight provider already initialized -- overriding!") if overriding
+        
+        new_provider_class = Class.new self do |cls|
+          
+          def load_current_resource
+            # silence Chef::Exceptions::Override exception
+          end
+          
+          class << cls
+            include Chef::Mixin::FromFile
+            
+            # setup DSL's shortcut methods
+            def action(name, &block)
+              define_method("action_#{name.to_s}") do
+                instance_eval(&block)
+              end
+            end
+          end
+          
+          # load provider definition from file
+          cls.class_from_file(filename)
+        end
+        
+        # register new class as a Chef::Provider
+        pname = filename_to_qualified_string(cookbook_name, filename)
+        class_name = convert_to_class_name(pname)
+        Chef::Provider.const_set(class_name, new_provider_class)
+        Chef::Log.debug("Loaded contents of #{filename} into a provider named #{pname} defined in Chef::Provider::#{class_name}")
+        
+        new_provider_class
+      end
+    end
+
+  end
+end

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_by_last_action(true)
+          run_context.resource_collection.iterator
+        end
+      end
+      
+    end
+  end
+end

data/lib/chef/provider/cookbook_file.rb

@@ -0,0 +1,100 @@
+#
+# 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
+        assert_enclosing_directory_exists!
+        if file_cache_location && content_stale?
+          Chef::Log.debug("#{@new_resource} has new contents")
+          backup_new_resource
+          Tempfile.open(::File.basename(@new_resource.name)) do |staging_file|
+            Chef::Log.debug("#{@new_resource} 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
+          Chef::Log.info("#{@new_resource} created file #{@new_resource.path}")
+          @new_resource.updated_by_last_action(true)
+        else
+          set_all_access_controls(@new_resource.path)
+        end
+        @new_resource.updated_by_last_action?
+      end
+
+      def action_create_if_missing
+        if ::File.exists?(@new_resource.path)
+          Chef::Log.debug("#{@new_resource} exists at #{@new_resource.path} 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_by_last_action(access_controls.modified?)
+      end
+
+      def backup_new_resource
+        if ::File.exists?(@new_resource.path)
+          backup @new_resource.path
+        end
+      end
+
+      def content_stale?
+        ( ! ::File.exist?(@new_resource.path)) || ( ! compare_content)
+      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("#{@new_resource} updated crontab entry")
+          @new_resource.updated_by_last_action(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("#{@new_resource} added crontab entry")
+          @new_resource.updated_by_last_action(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.info("#{@new_resource} deleted crontab entry")
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+
+    end
+  end
+end