chef 0.7.10

data/lib/chef/provider/user/pw.rb

@@ -0,0 +1,113 @@
+#
+# Author:: Stephen Haynes (<sh@nomitor.com>)
+# Copyright:: Copyright (c) 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/provider/user'
+
+class Chef
+  class Provider
+    class User
+      class Pw < Chef::Provider::User
+
+        def load_current_resource
+          super
+          raise Chef::Exceptions::User, "Could not find binary /usr/sbin/pw for #{@new_resource}" unless ::File.exists?("/usr/sbin/pw")
+        end
+
+        def create_user
+          command = "pw useradd"
+          command << set_options
+          run_command(:command => command)
+          modify_password
+        end
+        
+        def manage_user
+          command = "pw usermod"
+          command << set_options
+          run_command(:command => command)
+          modify_password
+        end
+        
+        def remove_user
+          command = "pw userdel #{@new_resource.username}"
+          command << " -r" if @new_resource.supports[:manage_home]
+          run_command(:command => command)
+        end
+        
+        def check_lock
+          case @current_resource.password
+          when /^\*LOCKED\*/
+            @locked = true
+          else
+            @locked = false
+          end
+          @locked
+        end
+        
+        def lock_user
+          run_command(:command => "pw lock #{@new_resource.username}")
+        end
+        
+        def unlock_user
+          run_command(:command => "pw unlock #{@new_resource.username}")
+        end
+        
+        def set_options
+          opts = " #{@new_resource.username}"
+          
+          field_list = {
+            'comment' => "-c",
+            'home' => "-d",
+            'gid' => "-g",
+            'uid' => "-u",
+            'shell' => "-s"
+          }
+          field_list.sort{ |a,b| a[0] <=> b[0] }.each do |field, option|
+            field_symbol = field.to_sym
+            if @current_resource.send(field_symbol) != @new_resource.send(field_symbol)
+              if @new_resource.send(field_symbol)
+                Chef::Log.debug("Setting #{@new_resource} #{field} to #{@new_resource.send(field_symbol)}")
+                opts << " #{option} '#{@new_resource.send(field_symbol)}'"
+              end
+            end
+          end
+          if @new_resource.supports[:manage_home]
+            Chef::Log.debug("Managing the home directory for #{@new_resource}")
+            opts << " -m"
+          end
+          opts
+        end
+      
+        def modify_password
+          if @current_resource.password != @new_resource.password
+            Chef::Log.debug("#{new_resource}: updating password")
+            command = "pw usermod #{@new_resource.username} -H 0"
+            status = popen4(command, :waitlast => true) do |pid, stdin, stdout, stderr|
+              stdin.puts "#{@new_resource.password}"
+            end
+            
+            unless status.exitstatus == 0
+              raise Chef::Exceptions::User, "pw failed - #{status.inspect}!"
+            end
+          else
+            Chef::Log.debug("#{new_resource}: no change needed to password")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/chef/provider/user/useradd.rb

@@ -0,0 +1,107 @@
+#
+# 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/user'
+
+class Chef
+  class Provider
+    class User 
+      class Useradd < Chef::Provider::User
+        def create_user
+          command = "useradd"
+          command << set_options
+          run_command(:command => command)
+        end
+        
+        def manage_user
+          command = "usermod"
+          command << set_options
+          run_command(:command => command)
+        end
+        
+        def remove_user
+          command = "userdel"
+          command << " -r" if @new_resource.supports[:manage_home]
+          command << " #{@new_resource.username}"
+          run_command(:command => command)
+        end
+        
+        def check_lock
+          status = popen4("passwd -S #{@new_resource.username}") do |pid, stdin, stdout, stderr|
+            status_line = stdout.gets.split(' ')
+            case status_line[1]
+            when /^P/
+              @locked = false
+            when /^N/
+              @locked = false
+            when /^L/
+              @locked = true
+            end
+          end
+
+          unless status.exitstatus == 0
+            raise Chef::Exceptions::User, "Cannot determine if #{@new_resource} is locked!"
+          end
+
+          @locked
+        end
+        
+        def lock_user
+          run_command(:command => "usermod -L #{@new_resource.username}")
+        end
+        
+        def unlock_user
+          run_command(:command => "usermod -U #{@new_resource.username}")
+        end
+        
+        def set_options
+          opts = ''
+          
+          field_list = {
+            'comment' => "-c",
+            'gid' => "-g",
+            'uid' => "-u",
+            'shell' => "-s",
+            'password' => "-p"
+          }
+          field_list.sort{ |a,b| a[0] <=> b[0] }.each do |field, option|
+            field_symbol = field.to_sym
+            if @current_resource.send(field_symbol) != @new_resource.send(field_symbol)
+              if @new_resource.send(field_symbol)
+                Chef::Log.debug("Setting #{@new_resource} #{field} to #{@new_resource.send(field_symbol)}")
+                opts << " #{option} '#{@new_resource.send(field_symbol)}'"
+              end
+            end
+          end
+          if @current_resource.home != @new_resource.home && @new_resource.home
+            if @new_resource.supports[:manage_home]
+              Chef::Log.debug("Managing the home directory for #{@new_resource}")
+              opts << " -d '#{@new_resource.home}' -m"
+            else
+              Chef::Log.debug("Setting #{@new_resource} home to #{@new_resource.home}")
+              opts << " -d '#{@new_resource.home}'"
+            end
+          end
+          opts << " #{@new_resource.username}"
+          opts
+        end
+      
+      end
+    end
+  end
+end

data/lib/chef/queue.rb

@@ -0,0 +1,145 @@
+#
+# 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/mixin/params_validate'
+require 'json'
+require 'stomp'
+
+class Chef
+  class Queue
+
+    @client = nil
+    @queue_retry_delay = Chef::Config[:queue_retry_delay]
+    @queue_retry_count = Chef::Config[:queue_retry_count]
+
+    class << self
+      include Chef::Mixin::ParamsValidate
+
+      def connect
+        queue_user     = Chef::Config[:queue_user]
+        queue_password = Chef::Config[:queue_password]
+        queue_host = Chef::Config[:queue_host]
+        queue_port = Chef::Config[:queue_port]
+        queue_retries = 1 unless queue_retries
+
+        # Connection.open(login = "", passcode = "", host='localhost', port=61613, reliable=FALSE, reconnectDelay=5)
+        @client = Stomp::Connection.open(queue_user, queue_password, queue_host, queue_port, false)
+
+      rescue Errno::ECONNREFUSED
+        Chef::Log.error("Connection refused connecting to stomp queue at #{queue_host}:#{queue_port}, retry #{queue_retries}/#{@queue_retry_count}")
+        sleep(@queue_retry_delay)
+        retry if (queue_retries += 1) < @queue_retry_count
+        raise Errno::ECONNREFUSED, "Connection refused connecting to stomp queue at #{queue_host}:#{queue_port}, giving up"
+      rescue Timeout::Error
+        Chef::Log.error("Timeout connecting to stomp queue at #{queue_host}:#{queue_port}, retry #{queue_retries}/#{@queue_retry_count}")
+        sleep(@queue_retry_delay)
+        retry if (queue_retries += 1) < @queue_retry_count
+        raise Timeout::Error, "Timeout connecting to stomp queue at #{queue_host}:#{queue_port}, giving up"
+      else
+        queue_retries = 1 # reset the number of retries on success
+      end
+
+      def make_url(type, name)
+        validate(
+          {
+            :queue_type => type.to_sym,
+            :queue_name => name.to_sym,
+          },
+          {
+            :queue_type => {
+              :equal_to => [ :topic, :queue ],
+            },
+            :queue_name => {
+              :kind_of => [ String, Symbol ],
+            }
+          }
+        )
+        if Chef::Config[:queue_prefix]
+	        queue_prefix = Chef::Config[:queue_prefix]
+          queue_url = "/#{type}/#{queue_prefix}/chef/#{name}"
+	      else
+	        queue_url = "/#{type}/chef/#{name}"
+	      end
+	      queue_url
+      end
+
+      def subscribe(type, name)
+        queue_url = make_url(type, name)
+        Chef::Log.debug("Subscribing to #{queue_url}")
+        connect if @client == nil
+        @client.subscribe(queue_url)
+      end
+
+      def send_msg(type, name, msg)
+        queue_retries = 1 unless queue_retries
+        validate(
+          {
+            :message => msg,
+          },
+          {
+            :message => {
+              :respond_to => :to_json
+            }
+          }
+        )
+        queue_url = make_url(type, name)
+        json = msg.to_json
+        connect if @client == nil
+        Chef::Log.debug("Sending to #{queue_url}: #{json}")
+        begin
+          @client.send(queue_url, json)
+        rescue Errno::EPIPE
+          Chef::Log.debug("Lost connection to stomp queue, reconnecting")
+          connect
+          retry if (queue_retries += 1) < @queue_retry_count
+          raise Errno::EPIPE, "Lost connection to stomp queue, giving up"
+        else
+          queue_retries = 1 # reset the number of retries on success
+        end
+      end
+
+      def receive_msg
+        connect if @client == nil
+        begin
+          raw_msg = @client.receive()
+          Chef::Log.debug("Received Message from #{raw_msg.headers["destination"]} containing: #{raw_msg.body}")
+        rescue
+          Chef::Log.debug("Received nil message from stomp, retrying")
+          retry
+        end
+        msg = JSON.parse(raw_msg.body)
+        return msg, raw_msg.headers
+      end
+
+      def poll_msg
+        connect if @client == nil
+        raw_msg = @client.poll()
+        if raw_msg
+          msg = JSON.parse(raw_msg.body)
+        else
+          nil
+        end
+      end
+
+      def disconnect
+        raise ArgumentError, "You must call connect before you can disconnect!" unless @client
+        @client.disconnect
+      end
+    end
+  end
+end

data/lib/chef/recipe.rb

@@ -0,0 +1,210 @@
+#
+# 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/resource'
+Dir[File.join(File.dirname(__FILE__), 'resource/**/*.rb')].sort.each { |lib| require lib }
+require 'chef/mixin/from_file'
+require 'chef/mixin/language'
+require 'chef/resource_collection'
+require 'chef/cookbook_loader'
+require 'chef/rest'
+require 'chef/search/result'
+
+class Chef
+  class Recipe
+    
+    include Chef::Mixin::FromFile
+    include Chef::Mixin::Language
+        
+    attr_accessor :cookbook_name, :recipe_name, :recipe, :node, :collection, 
+                  :definitions, :params, :cookbook_loader
+    
+    def initialize(cookbook_name, recipe_name, node, collection=nil, definitions=nil, cookbook_loader=nil)
+      @cookbook_name = cookbook_name
+      @recipe_name = recipe_name
+      @node = node
+      
+      if collection
+        @collection = collection
+      else
+        @collection = Chef::ResourceCollection.new()
+      end
+      
+      if definitions
+        @definitions = definitions
+      else
+        @definitions = Hash.new
+      end
+      
+      if cookbook_loader
+        @cookbook_loader = cookbook_loader
+      else
+        @cookbook_loader = Chef::CookbookLoader.new()
+      end
+      
+      @params = Hash.new      
+    end
+    
+    def include_recipe(*args)
+      args.flatten.each do |recipe|
+        if @node.run_state[:seen_recipes].has_key?(recipe)
+          Chef::Log.debug("I am not loading #{recipe}, because I have already seen it.")
+          next
+        end        
+
+        Chef::Log.debug("Loading Recipe #{recipe} via include_recipe")
+        @node.run_state[:seen_recipes][recipe] = true
+        
+        rmatch = recipe.match(/(.+?)::(.+)/)
+        if rmatch
+          cookbook = @cookbook_loader[rmatch[1]]
+          cookbook.load_recipe(rmatch[2], @node, @collection, @definitions, @cookbook_loader)
+        else
+          cookbook = @cookbook_loader[recipe]
+          cookbook.load_recipe("default", @node, @collection, @definitions, @cookbook_loader)
+        end
+      end
+    end
+    
+    def require_recipe(*args)
+      include_recipe(*args)
+    end
+    
+    def resources(*args)
+      @collection.resources(*args)
+    end
+    
+    def search(type, query, attributes=[], &block)
+      Chef::Log.debug("Searching #{type} index with #{query}")
+      r = Chef::REST.new(Chef::Config[:search_url])
+
+      results = r.get_rest("search/#{type}?q=#{query}&a=#{attributes.join(',')}")
+      Chef::Log.debug("Searching #{type} index with #{query} returned #{results.length} entries")
+      if block
+        results.each do |sr|
+          block.call(sr)
+        end
+      else
+        results
+      end
+    end
+    
+    # Sets a tag, or list of tags, for this node.  Syntactic sugar for
+    # @node[:tags].  
+    #
+    # With no arguments, returns the list of tags.
+    #
+    # === Parameters
+    # tags<Array>:: A list of tags to add - can be a single string
+    #
+    # === Returns
+    # tags<Array>:: The contents of @node[:tags]
+    def tag(*args)
+      if args.length > 0
+        args.each do |tag|
+          @node[:tags] << tag unless @node[:tags].include?(tag)
+        end
+        @node[:tags]
+      else
+        @node[:tags]
+      end
+    end
+    
+    # Returns true if the node is tagged with the supplied list of tags.
+    #
+    # === Parameters
+    # tags<Array>:: A list of tags
+    #
+    # === Returns
+    # true<TrueClass>:: If all the parameters are present
+    # false<FalseClass>:: If any of the parameters are missing
+    def tagged?(*args)
+      args.each do |tag|
+        return false unless @node[:tags].include?(tag)
+      end
+      true
+    end
+    
+    # Removes the list of tags from the node.
+    #
+    # === Parameters
+    # tags<Array>:: A list of tags
+    #
+    # === Returns
+    # tags<Array>:: The current list of @node[:tags]
+    def untag(*args)
+      args.each do |tag|
+        @node[:tags].delete(tag)
+      end
+    end
+        
+    def method_missing(method_symbol, *args, &block)
+      resource = nil
+      # If we have a definition that matches, we want to use that instead.  This should
+      # let you do some really crazy over-riding of "native" types, if you really want
+      # to. 
+      if @definitions.has_key?(method_symbol)
+        # This dupes the high level object, but we still need to dup the params
+        new_def = @definitions[method_symbol].dup
+        new_def.params = new_def.params.dup
+        new_def.node = @node
+        # This sets up the parameter overrides
+        new_def.instance_eval(&block) if block
+        new_recipe = Chef::Recipe.new(@cookbook_name, @recipe_name, @node, @collection, @definitions, @cookbook_loader)
+        new_recipe.params = new_def.params
+        new_recipe.params[:name] = args[0]
+        new_recipe.instance_eval(&new_def.recipe)
+      else
+        method_name = method_symbol.to_s
+        # Otherwise, we're rocking the regular resource call route.
+        rname = nil
+        regexp = %r{^(.+?)(_(.+))?$}
+
+        mn = method_name.match(regexp)
+        if mn
+          rname = "Chef::Resource::#{mn[1].capitalize}"
+
+          while mn && mn[3]
+            mn = mn[3].match(regexp)          
+            rname << mn[1].capitalize if mn
+          end
+        end
+
+        begin
+          args << @collection
+          args << @node
+          resource = eval(rname).new(*args)
+          # If we have a resource like this one, we want to steal it's state
+          resource.load_prior_resource
+          resource.cookbook_name = @cookbook_name
+          resource.recipe_name = @recipe_name
+          resource.params = @params
+          resource.instance_eval(&block) if block
+        rescue Exception => e
+          if e.kind_of?(NameError) && e.to_s =~ /Chef::Resource/
+            raise NameError, "Cannot find #{rname} for #{method_name}\nOriginal: #{e.to_s}"
+          else
+            raise e
+          end
+        end
+        @collection << resource
+        resource
+      end
+    end
+  end
+end