runa-chef 0.8.0.1

data/lib/chef/resource_collection/stepable_iterator.rb

@@ -0,0 +1,124 @@
+# Author:: Daniel DeLeo (<dan@kallistec.com>)
+# Copyright:: Copyright (c) 2009 Daniel DeLeo
+# 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 ResourceCollection
+    class StepableIterator
+      
+      def self.for_collection(new_collection)
+        instance = new(new_collection)
+        instance
+      end
+      
+      attr_accessor :collection
+      attr_reader :position
+      
+      def initialize(collection=[])
+        @position = 0
+        @paused = false
+        @collection = collection
+      end
+      
+      def size
+        collection.size
+      end
+      
+      def each(&block)
+        reset_iteration(block)
+        @iterator_type = :element
+        iterate
+      end
+      
+      def each_index(&block)
+        reset_iteration(block)
+        @iterator_type = :index
+        iterate
+      end
+      
+      def each_with_index(&block)
+        reset_iteration(block)
+        @iterator_type = :element_with_index
+        iterate
+      end
+      
+      def paused?
+        @paused
+      end
+      
+      def pause
+        @paused = true
+      end
+      
+      def resume
+        @paused = false
+        iterate
+      end
+      
+      def rewind
+        @position = 0
+      end
+      
+      def skip_back(skips=1)
+        @position -= skips
+      end
+      
+      def skip_forward(skips=1)
+        @position += skips
+      end
+      
+      def step
+        return nil if @position == size
+        call_iterator_block
+        @position += 1
+      end
+      
+      def iterate_on(iteration_type, &block)
+        @iterator_type = iteration_type
+        @iterator_block = block
+      end
+      
+      private
+      
+      def reset_iteration(iterator_block)
+        @iterator_block = iterator_block
+        @position = 0
+        @paused = false
+      end
+      
+      def iterate
+        while @position < size && !paused?
+          step
+        end
+        collection
+      end
+      
+      def call_iterator_block
+        case @iterator_type
+        when :element
+          @iterator_block.call(collection[@position])
+        when :index
+          @iterator_block.call(@position)
+        when :element_with_index
+          @iterator_block.call(collection[@position], @position)
+        else
+          raise "42error: someone forgot to set @iterator_type, wtf?"
+        end
+      end
+      
+    end
+  end
+end

data/lib/chef/resource_definition.rb

@@ -0,0 +1,67 @@
+#
+# 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/from_file'
+require 'chef/mixin/params_validate'
+
+class Chef
+  class ResourceDefinition
+    
+    include Chef::Mixin::FromFile
+    include Chef::Mixin::ParamsValidate
+    
+    attr_accessor :name, :params, :recipe, :node
+    
+    def initialize(node=nil)
+      @name = nil
+      @params = Hash.new
+      @recipe = nil
+      @node = node
+    end
+    
+    def define(resource_name, prototype_params=nil, &block)
+      unless resource_name.kind_of?(Symbol)
+        raise ArgumentError, "You must use a symbol when defining a new resource!"
+      end
+      @name = resource_name
+      if prototype_params
+        unless prototype_params.kind_of?(Hash)
+          raise ArgumentError, "You must pass a hash as the prototype parameters for a definition."
+        end
+        @params = prototype_params
+      end
+      if Kernel.block_given?
+        @recipe = block
+      else
+        raise ArgumentError, "You must pass a block to a definition."
+      end
+      true
+    end
+    
+    # When we do the resource definition, we're really just setting new values for
+    # the paramaters we prototyped at the top.  This method missing is as simple as
+    # it gets.
+    def method_missing(symbol, *args)
+      @params[symbol] = args.length == 1 ? args[0] : args
+    end
+    
+    def to_s
+      "#{name.to_s}"
+    end
+  end
+end

data/lib/chef/rest.rb

@@ -0,0 +1,298 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Thom May (<thom@clearairturbulence.org>)
+# Author:: Nuo Yan (<nuo@opscode.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/mixin/params_validate'
+require 'chef/openid_registration'
+require 'net/https'
+require 'uri'
+require 'json'
+require 'tempfile'
+require 'singleton'
+require 'mixlib/authentication/signedheaderauth'
+require 'chef/api_client'
+
+include Mixlib::Authentication::SignedHeaderAuth
+
+class Chef
+  class REST
+
+    class CookieJar < Hash
+      include Singleton
+    end
+    
+    attr_accessor :url, :cookies, :client_name, :signing_key, :signing_key_filename, :sign_on_redirect, :sign_request
+    
+    def initialize(url, client_name=Chef::Config[:node_name], signing_key_filename=Chef::Config[:client_key])
+      @url = url
+      @cookies = CookieJar.instance
+      @client_name = client_name
+      if signing_key_filename
+        @signing_key_filename = signing_key_filename
+        @signing_key = load_signing_key(signing_key_filename) 
+        @sign_request = true
+      else
+        @signing_key = nil
+        @sign_request = false
+      end
+      @sign_on_redirect = true
+    end
+
+    def load_signing_key(key)
+      if File.exists?(key) && File.readable?(key)
+        IO.read(key)
+      else
+        raise Chef::Exceptions::PrivateKeyMissing, "I cannot find #{key}, which you told me to use to sign requests!"
+      end
+    end
+    
+    # Register the client 
+    def register(name=Chef::Config[:node_name], destination=Chef::Config[:client_key])
+
+      if File.exists?(destination)
+        raise Chef::Exceptions::CannotWritePrivateKey, "I cannot write your private key to #{destination} - check permissions?" unless File.writable?(destination)
+      end
+
+      nc = Chef::ApiClient.new
+      nc.name(name)
+      response = nc.save(true, true)
+
+      Chef::Log.debug("Registration response: #{response.inspect}")
+
+      raise Chef::Exceptions::CannotWritePrivateKey, "The response from the server did not include a private key!" unless response.has_key?("private_key")
+
+      begin
+        # Write out the private key
+        file = File.open(destination, "w")
+        file.print(response["private_key"])
+        file.close
+      rescue 
+        raise Chef::Exceptions::CannotWritePrivateKey, "I cannot write your private key to #{destination}"
+      end
+
+      true
+    end
+
+    # Send an HTTP GET request to the path
+    #
+    # === Parameters
+    # path:: The path to GET
+    # raw:: Whether you want the raw body returned, or JSON inflated.  Defaults 
+    #   to JSON inflated.
+    def get_rest(path, raw=false, headers={})
+      run_request(:GET, create_url(path), headers, false, 10, raw)    
+    end                               
+                          
+    # Send an HTTP DELETE request to the path
+    def delete_rest(path, headers={}) 
+      run_request(:DELETE, create_url(path), headers)       
+    end                               
+    
+    # Send an HTTP POST request to the path                                  
+    def post_rest(path, json, headers={})
+      run_request(:POST, create_url(path), headers, json)    
+    end                               
+                                      
+    # Send an HTTP PUT request to the path
+    def put_rest(path, json, headers={})
+      run_request(:PUT, create_url(path), headers, json)
+    end
+    
+    def create_url(path)
+      if path =~ /^(http|https):\/\//
+        URI.parse(path)
+      else
+        URI.parse("#{@url}/#{path}")
+      end
+    end
+    
+    def sign_request(http_method, private_key, user_id, body = "", host="localhost")
+      #body = "" if body == false
+      timestamp = Time.now.utc.iso8601
+      sign_obj = Mixlib::Authentication::SignedHeaderAuth.signing_object(
+                                                         :http_method=>http_method,
+                                                         :body=>body,
+                                                         :user_id=>user_id,
+                                                         :timestamp=>timestamp)
+      signed =  sign_obj.sign(private_key).merge({:host => host})
+      signed.inject({}){|memo, kv| memo["#{kv[0].to_s.upcase}"] = kv[1];memo}
+    end
+    
+    # Actually run an HTTP request.  First argument is the HTTP method,
+    # which should be one of :GET, :PUT, :POST or :DELETE.  Next is the
+    # URL, then an object to include in the body (which will be converted with
+    # .to_json) and finally, the limit of HTTP Redirects to follow (10).
+    #
+    # Typically, you won't use this method -- instead, you'll use one of
+    # the helper methods (get_rest, post_rest, etc.)
+    #
+    # Will return the body of the response on success.
+    def run_request(method, url, headers={}, data=false, limit=10, raw=false)
+      
+      http_retry_delay = Chef::Config[:http_retry_delay] 
+      http_retry_count = Chef::Config[:http_retry_count]
+
+      raise ArgumentError, 'HTTP redirect too deep' if limit == 0 
+
+      http = Net::HTTP.new(url.host, url.port)
+      if url.scheme == "https"
+        http.use_ssl = true 
+        if Chef::Config[:ssl_verify_mode] == :verify_none
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        elsif Chef::Config[:ssl_verify_mode] == :verify_peer
+          http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+        end
+        if Chef::Config[:ssl_ca_path] and File.exists?(Chef::Config[:ssl_ca_path])
+          http.ca_path = Chef::Config[:ssl_ca_path]
+        elsif Chef::Config[:ssl_ca_file] and File.exists?(Chef::Config[:ssl_ca_file])
+          http.ca_file = Chef::Config[:ssl_ca_file]
+        end
+        if Chef::Config[:ssl_client_cert] && File.exists?(Chef::Config[:ssl_client_cert])
+          http.cert = OpenSSL::X509::Certificate.new(File.read(Chef::Config[:ssl_client_cert]))
+          http.key = OpenSSL::PKey::RSA.new(File.read(Chef::Config[:ssl_client_key]))
+        end
+      end
+
+      http.read_timeout = Chef::Config[:rest_timeout]
+
+      unless raw
+        headers = headers.merge({ 
+          'Accept' => "application/json",
+        })
+      end
+      
+      if @cookies.has_key?("#{url.host}:#{url.port}")
+        headers['Cookie'] = @cookies["#{url.host}:#{url.port}"]
+      end
+
+      json_body = data ? data.to_json : nil 
+
+      if @sign_request
+        Chef::Log.debug("Signing the request as #{@client_name}")
+        if json_body
+          headers.merge!(sign_request(method, OpenSSL::PKey::RSA.new(@signing_key), @client_name, json_body, "#{url.host}:#{url.port}"))
+        else
+          headers.merge!(sign_request(method, OpenSSL::PKey::RSA.new(@signing_key), @client_name, "", "#{url.host}:#{url.port}"))
+        end
+      end
+     
+      req = nil
+      case method
+      when :GET
+        req_path = "#{url.path}"
+        req_path << "?#{url.query}" if url.query
+        req = Net::HTTP::Get.new(req_path, headers)
+      when :POST
+        headers["Content-Type"] = 'application/json' if data
+        req_path = "#{url.path}"
+        req_path << "?#{url.query}" if url.query
+        req = Net::HTTP::Post.new(req_path, headers)          
+        req.body = json_body if json_body 
+      when :PUT
+        headers["Content-Type"] = 'application/json' if data
+        req_path = "#{url.path}"
+        req_path << "?#{url.query}" if url.query
+        req = Net::HTTP::Put.new(req_path, headers)
+        req.body = json_body if json_body 
+      when :DELETE
+        req_path = "#{url.path}"
+        req_path << "?#{url.query}" if url.query
+        req = Net::HTTP::Delete.new(req_path, headers)
+      else
+        raise ArgumentError, "You must provide :GET, :PUT, :POST or :DELETE as the method"
+      end
+
+      Chef::Log.debug("Sending HTTP Request via #{req.method} to #{url.host}:#{url.port}#{req.path}")
+      
+      # Optionally handle HTTP Basic Authentication
+      req.basic_auth(url.user, url.password) if url.user
+
+      res = nil
+      tf = nil
+      http_retries = 1
+
+      begin
+        res = http.request(req) do |response|
+          if raw
+            tf = Tempfile.new("chef-rest") 
+            # Stolen from http://www.ruby-forum.com/topic/166423
+            # Kudos to _why!
+            size, total = 0, response.header['Content-Length'].to_i
+            response.read_body do |chunk|
+              tf.write(chunk) 
+              size += chunk.size
+              if size == 0
+                Chef::Log.debug("#{req.path} done (0 length file)")
+              elsif total == 0
+                Chef::Log.debug("#{req.path} (zero content length)")
+              else
+                Chef::Log.debug("#{req.path}" + " %d%% done (%d of %d)" % [(size * 100) / total, size, total])
+              end
+            end
+            tf.close 
+            tf
+          else
+            response.read_body
+          end
+          response
+        end
+
+      rescue Errno::ECONNREFUSED => e
+        Chef::Log.error("Connection refused connecting to #{url.host}:#{url.port} for #{req.path} #{http_retries}/#{http_retry_count}")
+        sleep(http_retry_delay)
+        retry if (http_retries += 1) < http_retry_count
+        raise Errno::ECONNREFUSED, "Connection refused connecting to #{url.host}:#{url.port} for #{req.path}, giving up"
+      rescue Timeout::Error
+        Chef::Log.error("Timeout connecting to #{url.host}:#{url.port} for #{req.path}, retry #{http_retries}/#{http_retry_count}")
+        sleep(http_retry_delay)
+        retry if (http_retries += 1) < http_retry_count
+        raise Timeout::Error, "Timeout connecting to #{url.host}:#{url.port} for #{req.path}, giving up"
+      end
+      
+      if res.kind_of?(Net::HTTPSuccess)
+        if res['set-cookie']
+          @cookies["#{url.host}:#{url.port}"] = res['set-cookie']
+        end
+        if res['content-type'] =~ /json/
+          response_body = res.body.chomp
+          JSON.parse(response_body)
+        else
+          if raw
+            tf
+          else
+            res.body
+          end
+        end
+      elsif res.kind_of?(Net::HTTPFound) or res.kind_of?(Net::HTTPMovedPermanently)
+        if res['set-cookie']
+          @cookies["#{url.host}:#{url.port}"] = res['set-cookie']
+        end
+        @sign_request = false if @sign_on_redirect == false
+        run_request(:GET, create_url(res['location']), {}, false, limit - 1, raw)
+      else
+        if res['content-type'] =~ /json/
+          exception = JSON.parse(res.body)
+          Chef::Log.warn("HTTP Request Returned #{res.code} #{res.message}: #{exception["error"].respond_to?(:join) ? exception["error"].join(", ") : exception["error"]}")
+        end
+        res.error!
+      end
+    end
+    
+  end
+end