runa-chef 0.8.0.1

data/lib/chef/mixin/convert_to_class_name.rb

@@ -0,0 +1,57 @@
+#
+# 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.
+#
+
+class Chef
+  module Mixin
+    module ConvertToClassName
+
+      def convert_to_class_name(str)
+        rname = nil
+        regexp = %r{^(.+?)(_(.+))?$}
+        
+        mn = str.match(regexp)
+        if mn
+          rname = mn[1].capitalize
+
+          while mn && mn[3]
+            mn = mn[3].match(regexp)          
+            rname << mn[1].capitalize if mn
+          end
+        end
+
+        rname
+      end
+      
+      def convert_to_snake_case(str, namespace=nil)
+        str = str.dup
+        str.sub!(/^#{namespace}(\:\:)?/, '') if namespace
+        str.gsub!(/[A-Z]/) {|s| "_" + s}
+        str.downcase!
+        str.sub!(/^\_/, "")
+        str
+      end
+      
+      def filename_to_qualified_string(base, filename)
+        file_base = File.basename(filename, ".rb")
+        base.to_s + (file_base == 'default' ? '' : "_#{file_base}")
+      end
+      
+    end
+  end
+end

data/lib/chef/mixin/create_path.rb

@@ -0,0 +1,56 @@
+#
+# 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.
+
+class Chef
+  module Mixin
+    module CreatePath
+      
+      # Creates a given path, including all directories that lead up to it.
+      # Like mkdir_p, but without the leaking.
+      #
+      # === Parameters
+      # file_path<String, Array>:: A string that represents the path to create, 
+      #   or an Array with the path-parts.
+      #
+      # === Returns
+      # The created file_path.
+      def create_path(file_path)
+        unless file_path.kind_of?(String) || file_path.kind_of?(Array)
+          raise ArgumentError, "file_path must be a string or an array!" 
+        end
+        
+        if file_path.kind_of?(String)
+          file_path = File.expand_path(file_path).split(File::SEPARATOR)
+          file_path.shift if file_path[0] = ''
+          unless file_path[0].match("^#{File::SEPARATOR}")
+            file_path[0] = "#{File::SEPARATOR}#{file_path[0]}"
+          end
+        end
+                
+        file_path.each_index do |i|
+          create_path = File.join(file_path[0, i + 1])
+          unless File.directory?(create_path)
+            Chef::Log.debug("Creating directory #{create_path}")
+            Dir.mkdir(create_path)
+          end 
+        end
+        File.expand_path(File.join(file_path))
+      end
+  
+    end
+  end
+end

data/lib/chef/mixin/deep_merge.rb

@@ -0,0 +1,33 @@
+#
+# Author:: Adam Jacob (<adam@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 'deep_merge'
+
+class Chef
+  module Mixin
+    class DeepMerge
+      def self.merge(first, second)
+        first = Mash.new(first).to_hash unless second.kind_of?(Mash)
+        first = first.to_hash
+        second = Mash.new(second).to_hash unless second.kind_of?(Mash)
+        second = second.to_hash
+
+        Mash.new(first.ko_deep_merge!(second, {:knockout_prefix => '!merge:'}))
+      end
+    end
+  end
+end

data/lib/chef/mixin/find_preferred_file.rb

@@ -0,0 +1,92 @@
+#
+# 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/cookbook_loader'
+
+class Chef
+  module Mixin
+    module FindPreferredFile
+
+      def load_cookbook_files(cookbook_id, file_type)
+        unless file_type == :remote_file || file_type == :template
+          raise ArgumentError, "You must supply :remote_file or :template as the file_type"
+        end
+        
+        cl = Chef::CookbookLoader.new
+        cookbook = cl[cookbook_id]
+        raise NotFound unless cookbook
+
+        files = Hash.new
+        
+        cookbook_method = nil
+        
+        case file_type
+        when :remote_file
+          cookbook_method = :remote_files
+        when :template
+          cookbook_method = :template_files
+        end
+                
+        cookbook.send(cookbook_method).each do |rf|
+          full = File.expand_path(rf)
+          name = File.basename(full)
+          case file_type
+          when :remote_file
+            rf =~ /^.+#{cookbook_id}[\\|\/]files[\\|\/](.+?)[\\|\/]#{name}/
+          when :template
+            rf =~ /^.+#{cookbook_id}[\\|\/]templates[\\|\/](.+?)[\\|\/]#{name}/
+          end
+          singlecopy = $1
+          files[full] = {
+            :name => name,
+            :singlecopy => singlecopy,
+            :file => full,
+          }
+        end
+        Chef::Log.debug("Preferred #{file_type} list: #{files.inspect}")
+        
+        files
+      end
+
+      def find_preferred_file(cookbook_id, file_type, file_name, fqdn, platform, version)
+        file_list = load_cookbook_files(cookbook_id, file_type)
+        
+        preferences = [
+          File.join("host-#{fqdn}", "#{file_name}"),
+          File.join("#{platform}-#{version}", "#{file_name}"),
+          File.join("#{platform}", "#{file_name}"),
+          File.join("default", "#{file_name}")
+        ]
+        
+        file_list_str = file_list.keys.join("\n")
+        Chef::Log.debug("Searching for preferred file in\n#{file_list_str}")
+        
+        preferences.each do |pref|
+          Chef::Log.debug("Looking for #{pref}")
+          matcher = /^(.+#{pref})$/
+          if match = matcher.match(file_list_str)
+            return match[1]
+          end
+        end
+        
+        raise Chef::Exceptions::FileNotFound, "Cannot find a preferred file for #{file_name}!"
+      end
+      
+    end
+  end
+end

data/lib/chef/mixin/from_file.rb

@@ -0,0 +1,50 @@
+#
+# 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.
+#
+
+class Chef
+  module Mixin
+    module FromFile
+    
+      # Loads a given ruby file, and runs instance_eval against it in the context of the current 
+      # object.  
+      #
+      # Raises an IOError if the file cannot be found, or is not readable.
+      def from_file(filename)
+        if File.exists?(filename) && File.readable?(filename)
+          self.instance_eval(IO.read(filename), filename, 1)
+        else
+          raise IOError, "Cannot open or read #{filename}!"
+        end
+      end
+
+      # Loads a given ruby file, and runs class_eval against it in the context of the current 
+      # object.
+      #
+      # Raises an IOError if the file cannot be found, or is not readable.
+      def class_from_file(filename)
+        if File.exists?(filename) && File.readable?(filename)
+          self.class_eval(IO.read(filename), filename, 1)
+        else
+          raise IOError, "Cannot open or read #{filename}!"
+        end
+      end
+
+    end
+  end
+end

data/lib/chef/mixin/generate_url.rb

@@ -0,0 +1,58 @@
+#
+# 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/platform'
+
+class Chef
+  module Mixin
+    module GenerateURL
+    
+      def generate_cookbook_url(url, cookbook, type, node, args=nil)
+        Chef::Log.debug("generating cookbook url for url=#{url}, cookbook=#{cookbook.inspect}, type=#{type}, node=#{node}")
+        new_url = nil
+        if url =~ /^(http|https):\/\//
+          new_url = url
+        else
+          new_url = "cookbooks/#{cookbook}/#{type}?"
+          new_url += "id=#{url}"
+          new_url = generate_cookbook_url_from_uri(new_url, node, args)
+        end
+        Chef::Log.debug("generated cookbook url: #{new_url}")
+        return new_url
+      end
+
+      def generate_cookbook_url_from_uri(uri, node, args=nil)
+        platform, version = Chef::Platform.find_platform_and_version(node)
+        uri =~ /cookbooks\/(.+?)\/(.+)\?/
+        cookbook = $1
+        type = $2
+        if type == "files" || type == "templates"
+          uri += "&platform=#{platform}&version=#{version}&fqdn=#{node[:fqdn]}&node_name=#{node.name}"
+        end
+        if args
+          args.each do |key, value|
+            uri += "&#{key}=#{value}"
+          end
+        end
+
+        uri
+      end
+      
+    end
+  end
+end

data/lib/chef/mixin/language.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/search/query'
+require 'chef/data_bag'
+require 'chef/data_bag_item'
+
+class Chef
+  module Mixin
+    module Language
+
+      # Given a hash similar to the one we use for Platforms, select a value from the hash.  Supports
+      # per platform defaults, along with a single base default. Arrays may be passed as hash keys and
+      # will be expanded.
+      #
+      # === Parameters
+      # platform_hash:: A platform-style hash.
+      #
+      # === Returns
+      # value:: Whatever the most specific value of the hash is.
+      def value_for_platform(platform_hash)
+        result = nil
+        
+        platform_hash.each_pair do |key, value|
+          if key.is_a?(Array)
+            key.each { |array_key| platform_hash[array_key] = value }
+            platform_hash.delete(key)
+          end
+        end
+        if platform_hash.has_key?(@node[:platform])
+          if platform_hash[@node[:platform]].has_key?(@node[:platform_version])
+            result = platform_hash[@node[:platform]][@node[:platform_version]]
+          elsif platform_hash[@node[:platform]].has_key?("default")
+            result = platform_hash[@node[:platform]]["default"]
+          end
+        end
+  
+        unless result
+          if platform_hash.has_key?("default")
+            result = platform_hash["default"]
+          end
+        end  
+  
+        result
+      end
+
+      # Given a list of platforms, returns true if the current recipe is being run on a node with
+      # that platform, false otherwise.
+      #
+      # === Parameters
+      # args:: A list of platforms
+      #
+      # === Returns
+      # true:: If the current platform is in the list
+      # false:: If the current platform is not in the list
+      def platform?(*args)
+        has_platform = false
+  
+        args.flatten.each do |platform|
+          has_platform = true if platform == @node[:platform]
+        end
+  
+        has_platform
+      end
+
+      def search(*args, &block)
+        # If you pass a block, or have at least the start argument, do raw result parsing
+        # 
+        # Otherwise, do the iteration for the end user
+        if Kernel.block_given? || args.length >= 4 
+          Chef::Search::Query.new.search(*args, &block)
+        else 
+          results = Array.new
+          Chef::Search::Query.new.search(*args) do |o|
+            results << o 
+          end
+          results
+        end
+      end
+
+      def data_bag(bag)
+        rbag = Chef::DataBag.load(bag)
+        rbag.keys
+      end
+
+      def data_bag_item(bag, item)
+        Chef::DataBagItem.load(bag, item)
+      end
+
+    end
+  end
+end