tylerrick-chef 0.10.8

data/lib/chef/provider/subversion.rb

@@ -0,0 +1,197 @@
+#
+# 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/command'
+require 'fileutils'
+
+class Chef
+  class Provider
+    class Subversion < Chef::Provider
+
+      include Chef::Mixin::Command
+
+      def load_current_resource
+        @current_resource = Chef::Resource::Subversion.new(@new_resource.name)
+
+        unless [:export, :force_export].include?(Array(@new_resource.action).first)
+          if current_revision = find_current_revision
+            @current_resource.revision current_revision
+          end
+        end
+      end
+
+      def action_checkout
+        assert_target_directory_valid!
+        if target_dir_non_existant_or_empty?
+          run_command(run_options(:command => checkout_command))
+          @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 - nothing to do"
+        end
+      end
+
+      def action_export
+        assert_target_directory_valid!
+        if target_dir_non_existant_or_empty?
+          run_command(run_options(:command => export_command))
+          @new_resource.updated_by_last_action(true)
+        else
+          Chef::Log.debug "#{@new_resource} export destination #{@new_resource.destination} already exists or is a non-empty directory - nothing to do"
+        end
+      end
+
+      def action_force_export
+        assert_target_directory_valid!
+        run_command(run_options(:command => export_command))
+        @new_resource.updated_by_last_action(true)
+      end
+
+      def action_sync
+        assert_target_directory_valid!
+        if ::File.exist?(::File.join(@new_resource.destination, ".svn"))
+          current_rev = find_current_revision
+          Chef::Log.debug "#{@new_resource} current revision: #{current_rev} target revision: #{revision_int}"
+          unless current_revision_matches_target_revision?
+            run_command(run_options(:command => sync_command))
+            Chef::Log.info "#{@new_resource} updated to revision: #{revision_int}"
+            @new_resource.updated_by_last_action(true)
+          end
+        else
+          action_checkout
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+
+      def sync_command
+        c = scm :update, @new_resource.svn_arguments, verbose, authentication, "-r#{revision_int}", @new_resource.destination
+        Chef::Log.debug "#{@new_resource} updated working copy #{@new_resource.destination} to revision #{@new_resource.revision}"
+				c
+      end
+
+      def checkout_command
+        c = scm :checkout, @new_resource.svn_arguments, verbose, authentication,
+            "-r#{revision_int}", @new_resource.repository, @new_resource.destination
+        Chef::Log.info "#{@new_resource} checked out #{@new_resource.repository} at revision #{@new_resource.revision} to #{@new_resource.destination}"
+				c
+      end
+
+      def export_command
+        args = ["--force"]
+        args << @new_resource.svn_arguments << verbose << authentication <<
+            "-r#{revision_int}" << @new_resource.repository << @new_resource.destination
+        c = scm :export, *args
+        Chef::Log.info "#{@new_resource} exported #{@new_resource.repository} at revision #{@new_resource.revision} to #{@new_resource.destination}"
+				c
+      end
+
+      # If the specified revision isn't an integer ("HEAD" for example), look
+      # up the revision id by asking the server
+      # If the specified revision is an integer, trust it.
+      def revision_int
+        @revision_int ||= begin
+          if @new_resource.revision =~ /^\d+$/
+            @new_resource.revision
+          else
+            command = scm(:info, @new_resource.repository, @new_resource.svn_info_args, authentication, "-r#{@new_resource.revision}")
+            status, svn_info, error_message = output_of_command(command, run_options)
+            handle_command_failures(status, "STDOUT: #{svn_info}\nSTDERR: #{error_message}")
+            extract_revision_info(svn_info)
+          end
+        end
+      end
+
+      alias :revision_slug :revision_int
+
+      def find_current_revision
+        return nil unless ::File.exist?(::File.join(@new_resource.destination, ".svn"))
+        command = scm(:info)
+        status, svn_info, error_message = output_of_command(command, run_options(:cwd => cwd))
+
+        unless [0,1].include?(status.exitstatus)
+          handle_command_failures(status, "STDOUT: #{svn_info}\nSTDERR: #{error_message}")
+        end
+        extract_revision_info(svn_info)
+      end
+
+      def current_revision_matches_target_revision?
+        (!@current_resource.revision.nil?) && (revision_int.strip.to_i == @current_resource.revision.strip.to_i)
+      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
+      end
+
+      private
+
+      def cwd
+        @new_resource.destination
+      end
+
+      def verbose
+        "-q"
+      end
+
+      def extract_revision_info(svn_info)
+        begin
+          repo_attrs = YAML.load(svn_info)
+        rescue ArgumentError
+          # YAML doesn't appreciate input like "svn: '/tmp/deploydir' is not a working copy\n"
+          return nil
+        end
+        raise "Could not parse `svn info` data: #{svn_info}" unless repo_attrs.kind_of?(Hash)
+        rev = (repo_attrs['Last Changed Rev'] || repo_attrs['Revision']).to_s
+        Chef::Log.debug "#{@new_resource} resolved revision #{@new_resource.revision} to #{rev}"
+        rev
+      end
+
+      # If a username is configured for the SCM, return the command-line
+      # switches for that. Note that we don't need to return the password
+      # switch, since Capistrano will check for that prompt in the output
+      # and will respond appropriately.
+      def authentication
+        return "" unless @new_resource.svn_username
+        result = "--username #{@new_resource.svn_username} "
+        result << "--password #{@new_resource.svn_password} "
+        result
+      end
+
+      def scm(*args)
+        ['svn', *args].compact.join(" ")
+      end
+      
+      # TODO these methods are the same as the git provider...need to REFACTOR
+      # ...the subversion and git providers should extend from the same parent
+      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 target_dir_non_existant_or_empty?
+        !::File.exist?(@new_resource.destination) || Dir.entries(@new_resource.destination).sort == ['.','..']
+      end
+    end
+  end
+end

data/lib/chef/provider/template.rb

@@ -0,0 +1,104 @@
+#--
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Daniel DeLeo (<dan@opscode.com>)
+# Copyright:: Copyright (c) 2008, 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/provider/file'
+require 'chef/mixin/template'
+require 'chef/mixin/checksum'
+require 'chef/file_access_control'
+
+class Chef
+  class Provider
+
+    class Template < Chef::Provider::File
+
+      include Chef::Mixin::Checksum
+      include Chef::Mixin::Template
+
+      def load_current_resource
+        super
+        @current_resource.checksum(checksum(@current_resource.path)) if ::File.exist?(@current_resource.path)
+      end
+
+      def action_create
+        render_with_context(template_location) do |rendered_template|
+          rendered(rendered_template)
+          if ::File.exist?(@new_resource.path) && content_matches?
+            Chef::Log.debug("#{@new_resource} content has not changed.")
+            set_all_access_controls(@new_resource.path)
+          else
+            backup
+            set_all_access_controls(rendered_template.path)
+            FileUtils.mv(rendered_template.path, @new_resource.path)
+            Chef::Log.info("#{@new_resource} updated content")
+            @new_resource.updated_by_last_action(true)
+          end
+        end
+      end
+
+      def action_create_if_missing
+        if ::File.exists?(@new_resource.path)
+          Chef::Log.debug("#{@new_resource} exists - taking no action")
+        else
+          action_create
+        end
+      end
+
+      def template_location
+        @template_file_cache_location ||= begin
+          if @new_resource.local
+            @new_resource.source
+          else
+            cookbook = run_context.cookbook_collection[resource_cookbook]
+            cookbook.preferred_filename_on_disk_location(node, :templates, @new_resource.source)
+          end
+        end
+      end
+      
+      def resource_cookbook
+        @new_resource.cookbook || @new_resource.cookbook_name
+      end
+
+      def rendered(rendered_template)
+        @new_resource.checksum(checksum(rendered_template.path))
+        Chef::Log.debug("Current content's checksum:  #{@current_resource.checksum}")
+        Chef::Log.debug("Rendered content's checksum: #{@new_resource.checksum}")
+      end
+
+      def content_matches?
+        @current_resource.checksum == @new_resource.checksum
+      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
+
+      private
+
+      def render_with_context(template_location, &block)
+        context = {}
+        context.merge!(@new_resource.variables)
+        context[:node] = node
+        render_template(IO.read(template_location), context, &block)
+      end
+
+    end
+  end
+end

data/lib/chef/provider/user.rb

@@ -0,0 +1,186 @@
+#
+# 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/provider'
+require 'chef/mixin/command'
+require 'chef/resource/user'
+require 'etc'
+
+class Chef
+  class Provider
+    class User < Chef::Provider
+      
+      include Chef::Mixin::Command
+      
+      attr_accessor :user_exists, :locked
+      
+      def initialize(new_resource, run_context)
+        super
+        @user_exists = true
+        @locked = nil
+      end
+  
+      def convert_group_name
+        if @new_resource.gid.is_a? String
+          @new_resource.gid(Etc.getgrnam(@new_resource.gid).gid)
+        end
+      rescue ArgumentError => e
+        raise Chef::Exceptions::User, "Couldn't lookup integer GID for group name #{@new_resource.gid}"
+      end
+      
+      def load_current_resource
+        @current_resource = Chef::Resource::User.new(@new_resource.name)
+        @current_resource.username(@new_resource.username)
+      
+        begin
+          user_info = Etc.getpwnam(@new_resource.username)
+        rescue ArgumentError => e
+          @user_exists = false
+          Chef::Log.debug("#{@new_resource} user does not exist")
+          user_info = nil
+        end
+        
+        if user_info
+          @current_resource.uid(user_info.uid)
+          @current_resource.gid(user_info.gid)
+          @current_resource.comment(user_info.gecos)
+          @current_resource.home(user_info.dir)
+          @current_resource.shell(user_info.shell)
+          @current_resource.password(user_info.passwd)
+        
+          if @new_resource.password && @current_resource.password == 'x'
+            begin
+              require 'shadow'
+            rescue LoadError
+              raise Chef::Exceptions::MissingLibrary, "You must have ruby-shadow installed for password support!"
+            else
+              shadow_info = Shadow::Passwd.getspnam(@new_resource.username)
+              @current_resource.password(shadow_info.sp_pwdp)
+            end
+          end
+          
+          if @new_resource.gid
+            convert_group_name
+          end
+        end
+        
+        @current_resource
+      end
+
+      # Check to see if the user needs any changes
+      #
+      # === Returns
+      # <true>:: If a change is required
+      # <false>:: If the users are identical
+      def compare_user
+        [ :uid, :gid, :comment, :home, :shell, :password ].any? do |user_attrib|
+          !@new_resource.send(user_attrib).nil? && @new_resource.send(user_attrib) != @current_resource.send(user_attrib)
+        end
+      end
+      
+      def action_create
+        if !@user_exists
+          create_user
+          Chef::Log.info("#{@new_resource} created")
+          @new_resource.updated_by_last_action(true)
+        elsif compare_user
+          manage_user
+          Chef::Log.info("#{@new_resource} altered")
+          @new_resource.updated_by_last_action(true)
+        end
+      end
+      
+      def action_remove
+        if @user_exists
+          remove_user
+          @new_resource.updated_by_last_action(true)
+          Chef::Log.info("#{@new_resource} removed")
+        end
+      end
+
+      def remove_user
+        raise NotImplementedError
+      end
+
+      def action_manage
+        if @user_exists && compare_user
+          manage_user
+          @new_resource.updated_by_last_action(true)
+          Chef::Log.info("#{@new_resource} managed")
+        end
+      end
+
+      def manage_user
+        raise NotImplementedError
+      end
+
+      def action_modify
+        if @user_exists
+          if compare_user
+            manage_user
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} modified")
+          end
+        else
+          raise Chef::Exceptions::User, "Cannot modify user - does not exist!"
+        end
+      end
+
+      def action_lock
+        if @user_exists
+          if check_lock() == false
+            lock_user
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} locked")
+          else
+            Chef::Log.debug("#{@new_resource} already locked - nothing to do")
+          end
+        else
+          raise Chef::Exceptions::User, "Cannot lock user - does not exist!"
+        end
+      end
+
+      def check_lock
+        raise NotImplementedError
+      end
+
+      def lock_user
+        raise NotImplementedError
+      end
+
+      def action_unlock
+        if @user_exists
+          if check_lock() == true
+            unlock_user
+            @new_resource.updated_by_last_action(true)
+            Chef::Log.info("#{@new_resource} unlocked")
+          else
+            Chef::Log.debug("#{@new_resource} already unlocked - nothing to do")
+          end
+        else
+          raise Chef::Exceptions::User, "Cannot unlock user - does not exist!"
+        end
+      end
+      
+      def unlock_user
+        raise NotImplementedError
+      end
+
+    end
+  end
+end