giraffesoft-chef 0.7.15

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,144 @@
+#
+# 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/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/mixin/recipe_definition_dsl_core'
+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
+    include Chef::Mixin::RecipeDefinitionDSLCore
+    
+    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
+      @collection = collection || Chef::ResourceCollection.new
+      @definitions = definitions || Hash.new
+      @cookbook_loader = cookbook_loader || Chef::CookbookLoader.new
+      @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
+    
+  end
+end

data/lib/chef/resource.rb

@@ -0,0 +1,380 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Christopher Walters (<cw@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 'chef/mixin/check_helper'
+require 'chef/mixin/language'
+require 'chef/mixin/convert_to_class_name'
+require 'chef/resource_collection'
+require 'chef/node'
+
+class Chef
+  class Resource
+        
+    include Chef::Mixin::CheckHelper
+    include Chef::Mixin::ParamsValidate
+    include Chef::Mixin::Language
+    include Chef::Mixin::ConvertToClassName
+    
+    attr_accessor :actions, :params, :provider, :updated, :allowed_actions, :collection, :cookbook_name, :recipe_name, :enclosing_provider
+    attr_reader :resource_name, :source_line, :node
+    
+    def initialize(name, collection=nil, node=nil)
+      @name = name
+      if collection
+        @collection = collection
+      else
+        @collection = Chef::ResourceCollection.new()
+      end      
+      @node = node ? node : Chef::Node.new
+      @noop = nil
+      @before = nil
+      @actions = Hash.new
+      @params = Hash.new
+      @provider = nil
+      @allowed_actions = [ :nothing ]
+      @action = :nothing
+      @updated = false
+      @supports = {}
+      @ignore_failure = false
+      @not_if = nil
+      @only_if = nil
+      sline = caller(4).shift
+      if sline
+        @source_line = sline.gsub!(/^(.+):(.+):.+$/, '\1 line \2')
+        @source_line = ::File.expand_path(@source_line) if @source_line
+      end
+    end
+
+    # If an unknown method is invoked, determine whether the enclosing Provider's
+    # lexical scope can fulfill the request. E.g. This happens when the Resource's
+    # block invokes new_resource.
+    def method_missing(method_symbol, *args, &block)
+      if enclosing_provider && enclosing_provider.respond_to?(method_symbol)
+        enclosing_provider.send(method_symbol, *args, &block)
+      else
+        raise NoMethodError, "undefined method `#{method_symbol.to_s}' for #{self.class.to_s}"
+      end
+    end
+    
+    def load_prior_resource
+      begin
+        prior_resource = @collection.lookup(self.to_s)
+        Chef::Log.debug("Setting #{self.to_s} to the state of the prior #{self.to_s}")
+        prior_resource.instance_variables.each do |iv|
+          unless iv == "@source_line" || iv == "@action"
+            self.instance_variable_set(iv, prior_resource.instance_variable_get(iv))
+          end
+        end
+        true
+      rescue ArgumentError => e
+        true
+      end
+    end
+    
+    def supports(args={})
+      if args.any?
+        @supports = args
+      else
+        @supports
+      end
+    end
+    
+    def provider(arg=nil)
+      klass = if arg.kind_of?(String) || arg.kind_of?(Symbol)
+                lookup_provider_constant(arg)
+              else
+                arg
+              end
+      set_or_return(
+        :provider,
+        klass,
+        :kind_of => [ Class ]
+      )
+    end
+    
+    def action(arg=nil)
+      if arg
+        action_list = arg.kind_of?(Array) ? arg : [ arg ]
+        action_list = action_list.collect { |a| a.to_sym }
+        action_list.each do |action|
+          validate(
+            {
+              :action => action,
+            },
+            {
+              :action => { :kind_of => Symbol, :equal_to => @allowed_actions },
+            }
+          )
+        end
+        @action = action_list
+      else
+        @action
+      end
+    end
+    
+    def name(name=nil)
+      set_if_args(@name, name) do
+        raise ArgumentError, "name must be a string!" unless name.kind_of?(String)
+        @name = name
+      end
+    end
+    
+    def noop(tf=nil)
+      set_if_args(@noop, tf) do 
+        raise ArgumentError, "noop must be true or false!" unless tf == true || tf == false
+        @noop = tf
+      end
+    end
+    
+    def ignore_failure(arg=nil)
+      set_or_return(
+        :ignore_failure,
+        arg,
+        :kind_of => [ TrueClass, FalseClass ]
+      )
+    end
+    
+    def epic_fail(arg=nil)
+      ignore_failure(arg)
+    end
+    
+    def notifies(*args)
+      raise ArgumentError, "Wrong number of arguments (should be 1, 2, or 3)" unless ( args.size > 0 && args.size < 4)
+      if args.size > 1
+        notifies_helper(*args)
+      else
+        resources_array = *args
+        resources_array.each do |resource|
+          resource.each do |key, value|
+            notifies_helper(value[0], key, value[1])    
+          end
+        end 
+      end
+    end
+  
+    def resources(*args)
+      @collection.resources(*args)
+    end
+    
+    def subscribes(action, resources, timing=:delayed)
+      timing = check_timing(timing)
+      rarray = resources.kind_of?(Array) ? resources : [ resources ]
+      rarray.each do |resource|
+        action_sym = action.to_sym
+        if resource.actions.has_key?(action_sym)
+          resource.actions[action_sym][timing] << self
+        else       
+          resource.actions[action_sym] = Hash.new
+          resource.actions[action_sym][:delayed] = Array.new
+          resource.actions[action_sym][:immediate] = Array.new   
+          resource.actions[action_sym][timing] << self
+        end
+      end
+      true
+    end
+    
+    def is(*args)
+      return *args
+    end
+    
+    def to_s
+      "#{@resource_name}[#{@name}]"
+    end
+    
+    # Serialize this object as a hash 
+    def to_json(*a)
+      instance_vars = Hash.new
+      self.instance_variables.each do |iv|
+        instance_vars[iv] = self.instance_variable_get(iv) unless iv == "@collection"
+      end
+      results = {
+        'json_class' => self.class.name,
+        'instance_vars' => instance_vars
+      }
+      results.to_json(*a)
+    end
+    
+    def to_hash
+      instance_vars = Hash.new
+      self.instance_variables.each do |iv|
+        instance_vars[iv.sub(/^@/,'').to_sym] = self.instance_variable_get(iv) unless iv == "@collection"
+      end
+      instance_vars
+    end
+    
+    def only_if(arg=nil, &blk)
+      if Kernel.block_given?
+        @only_if = blk
+      else
+        @only_if = arg if arg
+      end
+      @only_if
+    end
+    
+    def not_if(arg=nil, &blk)
+      if Kernel.block_given?
+        @not_if = blk
+      else
+        @not_if = arg if arg
+      end
+      @not_if
+    end
+    
+    def run_action(action)
+      provider = Chef::Platform.provider_for_node(@node, self)
+      provider.load_current_resource
+      provider.send("action_#{action}")
+    end
+    
+    class << self
+      
+      def json_create(o)
+        resource = self.new(o["instance_vars"]["@name"])
+        o["instance_vars"].each do |k,v|
+          resource.instance_variable_set(k.to_sym, v)
+        end
+        resource
+      end
+      
+      include Chef::Mixin::ConvertToClassName
+      
+      def attribute(attr_name, validation_opts={})
+        # This atrocity is the only way to support 1.8 and 1.9 at the same time
+        # When you're ready to drop 1.8 support, do this:
+        # define_method attr_name.to_sym do |arg=nil|
+        # etc.
+        shim_method=<<-SHIM
+        def #{attr_name}(arg=nil)
+          _set_or_return_#{attr_name}(arg)
+        end
+        SHIM
+        class_eval(shim_method)
+        
+        define_method("_set_or_return_#{attr_name.to_s}".to_sym) do |arg|
+          set_or_return(attr_name.to_sym, arg, validation_opts)
+        end
+      end
+      
+      def build_from_file(cookbook_name, filename)
+        rname = filename_to_qualified_string(cookbook_name, filename)
+          
+        new_resource_class = Class.new self do |cls|
+          
+          # default initialize method that ensures that when initialize is finally
+          # wrapped (see below), super is called in the event that the resource
+          # definer does not implement initialize
+          def initialize(name, collection=nil, node=nil)
+            super(name, collection, node)
+          end
+          
+          @actions_to_create = []
+          
+          class << cls
+            include Chef::Mixin::FromFile
+            
+            def actions_to_create
+              @actions_to_create
+            end
+            
+            define_method(:actions) do |*action_names|
+              actions_to_create.push(*action_names)
+            end
+          end
+          
+          # load resource definition from file
+          cls.class_from_file(filename)
+          
+          # create a new constructor that wraps the old one and adds the actions
+          # specified in the DSL
+          old_init = instance_method(:initialize)
+
+          define_method(:initialize) do |name, *optional_args|
+            collection = optional_args.shift
+            node = optional_args.shift
+            @resource_name = rname.to_sym
+            old_init.bind(self).call(name, collection, node)
+            allowed_actions.push(self.class.actions_to_create).flatten!
+          end
+        end
+        
+        # register new class as a Chef::Resource
+        class_name = convert_to_class_name(rname)
+        Chef::Resource.const_set(class_name, new_resource_class)
+        Chef::Log.debug("Loaded contents of #{filename} into a resource named #{rname} defined in Chef::Resource::#{class_name}")
+        
+        new_resource_class
+      end
+      
+      # Resources that want providers namespaced somewhere other than 
+      # Chef::Provider can set the namespace with +provider_base+
+      # Ex:
+      #   class MyResource < Chef::Resource
+      #     provider_base Chef::Provider::Deploy
+      #     # ...other stuff
+      #   end
+      def provider_base(arg=nil)
+        @provider_base ||= arg
+        @provider_base ||= Chef::Provider
+      end
+      
+    end
+    
+    private
+    
+      def lookup_provider_constant(name)
+        begin
+          self.class.provider_base.const_get(convert_to_class_name(name.to_s))
+        rescue NameError => e
+          if e.to_s =~ /#{self.class.provider_base.to_s}/
+            raise ArgumentError, "No provider found to match '#{name}'"
+          else
+            raise e
+          end
+        end
+      end
+      
+      def check_timing(timing)
+        unless timing == :delayed || timing == :immediate || timing == :immediately
+          raise ArgumentError, "Timing must be :delayed or :immediate(ly), you said #{timing}"
+        end
+        if timing == :immediately
+          timing = :immediate
+        end
+        timing
+      end
+      
+      def notifies_helper(action, resources, timing=:delayed)
+        timing = check_timing(timing)
+        rarray = resources.kind_of?(Array) ? resources : [ resources ]
+        rarray.each do |resource|
+          action_sym = action.to_sym
+          if @actions.has_key?(action_sym)
+            @actions[action_sym][timing] << resource
+          else
+            @actions[action_sym] = Hash.new
+            @actions[action_sym][:delayed] = Array.new
+            @actions[action_sym][:immediate] = Array.new   
+            @actions[action_sym][timing] << resource
+          end
+        end
+        true
+      end
+  end
+end