microwave 0.1004.1

data/lib/chef/monkey_patches/string.rb

@@ -0,0 +1,49 @@
+#
+# 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.
+#
+
+# == String (Patch)
+# On ruby 1.9, Strings are aware of multibyte characters, so +size+ and +length+
+# give the actual number of characters. In Chef::REST, we need the bytesize
+# so we can correctly set the Content-Length headers, but ruby 1.8.6 and lower
+# don't define String#bytesize. Monkey patching time!
+
+begin
+  require 'enumerator'
+rescue LoadError
+end
+
+class String
+  unless method_defined?(:bytesize)
+    alias :bytesize :size
+  end
+
+  unless method_defined?(:lines)
+    def lines
+      enum_for(:each)
+    end
+  end
+end
+
+# <= 1.8.6 needs some ord! 
+class String
+  unless method_defined?(:ord)
+    def ord
+      self.unpack('c').first
+    end
+  end
+end

data/lib/chef/monkey_patches/tempfile.rb

@@ -0,0 +1,64 @@
+#
+# 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.
+#
+
+# == Tempfile (Patch)
+# Tempfile has a horrible bug where it causes an IOError: closed stream in its
+# finalizer, leading to intermittent application crashes with confusing stack
+# traces. Here we monkey patch the fix into place. You can track the bug on
+# ruby's redmine: http://redmine.ruby-lang.org/issues/show/3119
+#
+# The patch is slightly different for Ruby 1.8 and Ruby 1.9, both patches are
+# included here.
+class Tempfile # :nodoc:
+  # Tempfile has changes between 1.8.x and 1.9.x
+  # so we monkey patch separately
+  if RUBY_VERSION =~ /^1\.8/
+    def unlink
+      # keep this order for thread safeness
+      begin
+        File.unlink(@tmpname) if File.exist?(@tmpname)
+        @@cleanlist.delete(@tmpname)
+        @tmpname = nil
+        ObjectSpace.undefine_finalizer(self)
+      rescue Errno::EACCES
+        # may not be able to unlink on Windows; just ignore
+      end
+    end
+    alias delete unlink
+
+
+  # There is a patch for this, to be merged into 1.9 at some point.
+  # When that happens, we'll want to also check the RUBY_PATCHLEVEL
+  elsif RUBY_VERSION =~ /^1\.9/
+    def unlink
+      # keep this order for thread safeness
+      return unless @tmpname
+      begin
+        if File.exist?(@tmpname)
+          File.unlink(@tmpname)
+        end
+        # remove tmpname from remover
+        @data[0] = @data[2] = nil
+        @tmpname = nil
+      rescue Errno::EACCES
+        # may not be able to unlink on Windows; just ignore
+      end
+    end
+    alias delete unlink
+  end
+end

data/lib/chef/nil_argument.rb

@@ -0,0 +1,3 @@
+class Chef
+  NIL_ARGUMENT = Object.new
+end

data/lib/chef/node.rb

@@ -0,0 +1,469 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Christopher Brown (<cb@opscode.com>)
+# Author:: Christopher Walters (<cw@opscode.com>)
+# Author:: Tim Hinderliter (<tim@opscode.com>)
+# Copyright:: Copyright (c) 2008-2011 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 'forwardable'
+require 'chef/config'
+require 'chef/cookbook/cookbook_collection'
+require 'chef/nil_argument'
+require 'chef/mixin/check_helper'
+require 'chef/mixin/params_validate'
+require 'chef/mixin/from_file'
+require 'chef/mixin/language_include_attribute'
+require 'chef/mixin/deep_merge'
+require 'chef/environment'
+require 'chef/run_list'
+require 'chef/node/attribute'
+require 'chef/mash'
+require 'chef/json_compat'
+
+class Chef
+  class Node
+
+    extend Forwardable
+
+    def_delegators :construct_attributes, :keys, :each_key, :each_value, :key?, :has_key?
+
+    attr_accessor :recipe_list, :run_state, :run_list
+    attr_accessor :override_attrs, :default_attrs, :normal_attrs, :automatic_attrs
+
+    # TODO: 5/18/2010 cw/timh. cookbook_collection should be removed
+    # from here and for any place it's needed, it should be accessed
+    # through a Chef::RunContext
+    attr_accessor :cookbook_collection
+
+    include Chef::Mixin::CheckHelper
+    include Chef::Mixin::FromFile
+    include Chef::Mixin::ParamsValidate
+    include Chef::Mixin::LanguageIncludeAttribute
+
+    # Create a new Chef::Node object.
+    def initialize()
+      @name = nil
+
+      @chef_environment = '_default'
+      @normal_attrs = Mash.new
+      @override_attrs = Mash.new
+      @default_attrs = Mash.new
+      @automatic_attrs = Mash.new
+      @run_list = Chef::RunList.new
+
+      @run_state = {
+        :template_cache => Hash.new,
+        :seen_recipes => Hash.new,
+        :seen_attributes => Hash.new
+      }
+      # TODO: 5/20/2010 need this here as long as other objects try to access
+      # the cookbook collection via Node, otherwise get NoMethodError on nil.
+      @cookbook_collection = CookbookCollection.new
+    end
+
+    # Used by DSL
+    def node
+      self
+    end
+
+    # Find a recipe for this Chef::Node by fqdn.  Will search first for
+    # Chef::Config["node_path"]/fqdn.rb, then hostname.rb, then default.rb.
+    #
+    # Returns a new Chef::Node object.
+    #
+    # Raises an ArgumentError if it cannot find the node.
+    def find_file(fqdn)
+      host_parts = fqdn.split(".")
+      hostname = host_parts[0]
+
+      [fqdn, hostname, "default"].each { |fname|
+       node_file = File.join(Chef::Config[:node_path], "#{fname.to_s}.rb")
+       return self.from_file(node_file) if File.exists?(node_file)
+     }
+
+      raise ArgumentError, "Cannot find a node matching #{fqdn}, not even with default.rb!"
+    end
+
+    # Set the name of this Node, or return the current name.
+    def name(arg=nil)
+      if arg != nil
+        validate(
+                 {:name => arg },
+                 {:name => { :kind_of => String,
+                     :cannot_be => :blank,
+                     :regex => /^[\-[:alnum:]_:.]+$/}
+                 })
+        @name = arg
+      else
+        @name
+      end
+    end
+
+    def chef_environment(arg=nil)
+      set_or_return(
+        :chef_environment,
+        arg,
+        { :regex => /^[\-[:alnum:]_]+$/, :kind_of => String }
+      )
+    end
+
+    # Used by the DSL
+    def attribute
+      construct_attributes
+    end
+
+    def construct_attributes
+      Chef::Node::Attribute.new(normal_attrs, default_attrs, override_attrs, automatic_attrs)
+    end
+
+    def attribute=(value)
+      self.normal_attrs = value
+    end
+
+    # Return an attribute of this node.  Returns nil if the attribute is not found.
+    def [](attrib)
+      construct_attributes[attrib]
+    end
+
+    # Set an attribute of this node
+    def []=(attrib, value)
+      construct_attributes[attrib] = value
+    end
+
+    def store(attrib, value)
+      self[attrib] = value
+    end
+
+    # Set a normal attribute of this node, but auto-vivifiy any Mashes that
+    # might be missing
+    def normal
+      attrs = construct_attributes
+      attrs.set_type = :normal
+      attrs.auto_vivifiy_on_read = true
+      attrs
+    end
+
+    alias_method :set, :normal
+
+    # Set a normal attribute of this node, auto-vivifiying any mashes that are
+    # missing, but if the final value already exists, don't set it
+    def normal_unless
+      attrs = construct_attributes
+      attrs.set_type = :normal
+      attrs.auto_vivifiy_on_read = true
+      attrs.set_unless_value_present = true
+      attrs
+    end
+    alias_method :set_unless, :normal_unless
+
+    # Set a default of this node, but auto-vivifiy any Mashes that might
+    # be missing
+    def default
+      attrs = construct_attributes
+      attrs.set_type = :default
+      attrs.auto_vivifiy_on_read = true
+      attrs
+    end
+
+    # Set a default attribute of this node, auto-vivifiying any mashes that are
+    # missing, but if the final value already exists, don't set it
+    def default_unless
+      attrs = construct_attributes
+      attrs.set_type = :default
+      attrs.auto_vivifiy_on_read = true
+      attrs.set_unless_value_present = true
+      attrs
+    end
+
+    # Set an override attribute of this node, but auto-vivifiy any Mashes that
+    # might be missing
+    def override
+      attrs = construct_attributes
+      attrs.set_type = :override
+      attrs.auto_vivifiy_on_read = true
+      attrs
+    end
+
+    # Set an override attribute of this node, auto-vivifiying any mashes that
+    # are missing, but if the final value already exists, don't set it
+    def override_unless
+      attrs = construct_attributes
+      attrs.set_type = :override
+      attrs.auto_vivifiy_on_read = true
+      attrs.set_unless_value_present = true
+      attrs
+    end
+
+    # Return true if this Node has a given attribute, false if not.  Takes either a symbol or
+    # a string.
+    #
+    # Only works on the top level. Preferred way is to use the normal [] style
+    # lookup and call attribute?()
+    def attribute?(attrib)
+      construct_attributes.attribute?(attrib)
+    end
+
+    # Yield each key of the top level to the block.
+    def each(&block)
+      construct_attributes.each(&block)
+    end
+
+    # Iterates over each attribute, passing the attribute and value to the block.
+    def each_attribute(&block)
+      construct_attributes.each_attribute(&block)
+    end
+
+    # Encouraged to only get used for lookups - while you can do sets from here, it's not as explicit
+    # as using the normal/default/override interface.
+    def method_missing(symbol, *args)
+      attrs = construct_attributes
+      attrs.send(symbol, *args)
+    end
+
+    # Returns true if this Node expects a given recipe, false if not.
+    #
+    # First, the run list is consulted to see whether the recipe is
+    # explicitly included. If it's not there, it looks in
+    # run_state[:seen_recipes], which is populated by include_recipe
+    # statements in the DSL (and thus would not be in the run list).
+    #
+    # NOTE: It's used by cookbook authors
+    def recipe?(recipe_name)
+      run_list.include?(recipe_name) || run_state[:seen_recipes].include?(recipe_name)
+    end
+
+    # Returns true if this Node expects a given role, false if not.
+    def role?(role_name)
+      run_list.include?("role[#{role_name}]")
+    end
+
+    # Returns an Array of roles and recipes, in the order they will be applied.
+    # If you call it with arguments, they will become the new list of roles and recipes.
+    def run_list(*args)
+      args.length > 0 ? @run_list.reset!(args) : @run_list
+    end
+
+    # Returns true if this Node expects a given role, false if not.
+    def run_list?(item)
+      run_list.detect { |r| r == item } ? true : false
+    end
+
+    # Consume data from ohai and Attributes provided as JSON on the command line.
+    def consume_external_attrs(ohai_data, json_cli_attrs)
+      Chef::Log.debug("Extracting run list from JSON attributes provided on command line")
+      consume_attributes(json_cli_attrs)
+
+      @automatic_attrs = ohai_data
+
+      platform, version = Chef::Platform.find_platform_and_version(self)
+      Chef::Log.debug("Platform is #{platform} version #{version}")
+      @automatic_attrs[:platform] = platform
+      @automatic_attrs[:platform_version] = version
+    end
+
+    # Consumes the combined run_list and other attributes in +attrs+
+    def consume_attributes(attrs)
+      normal_attrs_to_merge = consume_run_list(attrs)
+      Chef::Log.debug("Applying attributes from json file")
+      @normal_attrs = Chef::Mixin::DeepMerge.merge(@normal_attrs,normal_attrs_to_merge)
+      self.tags # make sure they're defined
+    end
+
+    # Lazy initializer for tags attribute
+    def tags
+      self[:tags] = [] unless attribute?(:tags)
+      self[:tags]
+    end
+
+    # Extracts the run list from +attrs+ and applies it. Returns the remaining attributes
+    def consume_run_list(attrs)
+      attrs = attrs ? attrs.dup : {}
+      if new_run_list = attrs.delete("recipes") || attrs.delete("run_list")
+        if attrs.key?("recipes") || attrs.key?("run_list")
+          raise Chef::Exceptions::AmbiguousRunlistSpecification, "please set the node's run list using the 'run_list' attribute only."
+        end
+        Chef::Log.info("Setting the run_list to #{new_run_list.inspect} from JSON")
+        run_list(new_run_list)
+      end
+      attrs
+    end
+
+    # Clear defaults and overrides, so that any deleted attributes
+    # between runs are still gone.
+    def reset_defaults_and_overrides
+      @default_attrs = Mash.new
+      @override_attrs = Mash.new
+    end
+
+    # Expands the node's run list and sets the default and override
+    # attributes. Also applies stored attributes (from json provided
+    # on the command line)
+    #
+    # Returns the fully-expanded list of recipes, a RunListExpansion.
+    #
+    #--
+    # TODO: timh/cw, 5-14-2010: Should this method exist? Should we
+    # instead modify default_attrs and override_attrs whenever our
+    # run_list is mutated? Or perhaps do something smarter like
+    # on-demand generation of default_attrs and override_attrs,
+    # invalidated only when run_list is mutated?
+    def expand!(data_source = 'disk')
+      expansion = run_list.expand(chef_environment, data_source)
+      raise Chef::Exceptions::MissingRole if expansion.errors?
+
+      self.tags # make sure they're defined
+
+      @automatic_attrs[:recipes] = expansion.recipes
+      @automatic_attrs[:roles] = expansion.roles
+
+      expansion
+    end
+
+    # Apply the default and overrides attributes from the expansion
+    # passed in, which came from roles.
+    def apply_expansion_attributes(expansion)
+      load_chef_environment_object = (chef_environment == "_default" ? nil : Chef::Environment.load(chef_environment))
+      environment_default_attrs = load_chef_environment_object.nil? ? {} : load_chef_environment_object.default_attributes
+      default_before_roles = Chef::Mixin::DeepMerge.merge(default_attrs, environment_default_attrs)
+      @default_attrs = Chef::Mixin::DeepMerge.merge(default_before_roles, expansion.default_attrs)
+      environment_override_attrs = load_chef_environment_object.nil? ? {} : load_chef_environment_object.override_attributes
+      overrides_before_environments = Chef::Mixin::DeepMerge.merge(override_attrs, expansion.override_attrs)
+      @override_attrs = Chef::Mixin::DeepMerge.merge(overrides_before_environments, environment_override_attrs)
+    end
+
+    # Transform the node to a Hash
+    def to_hash
+      index_hash = Hash.new
+      index_hash["chef_type"] = "node"
+      index_hash["name"] = name
+      index_hash["chef_environment"] = chef_environment
+      attribute.each do |key, value|
+        index_hash[key] = value
+      end
+      index_hash["recipe"] = run_list.recipe_names if run_list.recipe_names.length > 0
+      index_hash["role"] = run_list.role_names if run_list.role_names.length > 0
+      index_hash["run_list"] = run_list.run_list if run_list.run_list.length > 0
+      index_hash
+    end
+
+    def display_hash
+      display = {}
+      display["name"]             = name
+      display["chef_environment"] = chef_environment
+      display["automatic"]        = automatic_attrs
+      display["normal"]           = normal_attrs
+      display["default"]          = default_attrs
+      display["override"]         = override_attrs
+      display["run_list"]         = run_list.run_list
+      display
+    end
+
+    # Serialize this object as a hash
+    def to_json(*a)
+      result = {
+        "name" => name,
+        "chef_environment" => chef_environment,
+        'json_class' => self.class.name,
+        "automatic" => automatic_attrs,
+        "normal" => normal_attrs,
+        "chef_type" => "node",
+        "default" => default_attrs,
+        "override" => override_attrs,
+        "run_list" => run_list.run_list
+      }
+      result.to_json(*a)
+    end
+
+    def update_from!(o)
+      run_list.reset!(o.run_list)
+      @automatic_attrs = o.automatic_attrs
+      @normal_attrs = o.normal_attrs
+      @override_attrs = o.override_attrs
+      @default_attrs = o.default_attrs
+      chef_environment(o.chef_environment)
+      self
+    end
+
+    # Create a Chef::Node from JSON
+    def self.json_create(o)
+      node = new
+      node.name(o["name"])
+      node.chef_environment(o["chef_environment"])
+      if o.has_key?("attributes")
+        node.normal_attrs = o["attributes"]
+      end
+      node.automatic_attrs = Mash.new(o["automatic"]) if o.has_key?("automatic")
+      node.normal_attrs = Mash.new(o["normal"]) if o.has_key?("normal")
+      node.default_attrs = Mash.new(o["default"]) if o.has_key?("default")
+      node.override_attrs = Mash.new(o["override"]) if o.has_key?("override")
+
+      if o.has_key?("run_list")
+        node.run_list.reset!(o["run_list"])
+      else
+        o["recipes"].each { |r| node.recipes << r }
+      end
+      node
+    end
+
+    def self.find_or_create(node_name)
+      load(node_name)
+    rescue Net::HTTPServerException => e
+      raise unless e.response.code == '404'
+      node = build(node_name)
+      node.create
+    end
+
+    def self.build(node_name)
+      node = new
+      node.name(node_name)
+      node.chef_environment(Chef::Config[:environment]) unless Chef::Config[:environment].nil? || Chef::Config[:environment].chop.empty?
+      node
+    end
+
+    def to_s
+      "node[#{name}]"
+    end
+
+    # Load all attribute files for all cookbooks associated with this
+    # node.
+    def load_attributes
+      cookbook_collection.values.each do |cookbook|
+        cookbook.segment_filenames(:attributes).each do |segment_filename|
+          Chef::Log.debug("Node #{name} loading cookbook #{cookbook.name}'s attribute file #{segment_filename}")
+          self.from_file(segment_filename)
+        end
+      end
+    end
+
+    # Used by DSL.
+    # Loads the attribute file specified by the short name of the
+    # file, e.g., loads specified cookbook's
+    #   "attributes/mailservers.rb"
+    # if passed
+    #   "mailservers"
+    def load_attribute_by_short_filename(name, src_cookbook_name)
+      src_cookbook = cookbook_collection[src_cookbook_name]
+      raise Chef::Exceptions::CookbookNotFound, "could not find cookbook #{src_cookbook_name} while loading attribute #{name}" unless src_cookbook
+
+      attribute_filename = src_cookbook.attribute_filenames_by_short_filename[name]
+      raise Chef::Exceptions::AttributeNotFound, "could not find filename for attribute #{name} in cookbook #{src_cookbook_name}" unless attribute_filename
+
+      self.from_file(attribute_filename)
+      self
+    end
+  end
+end