Adding Chef::CookbookLoader

9 files changed, 253 insertions, 0 deletions

diff --git a/lib/chef/cookbook_loader.rb b/lib/chef/cookbook_loader.rb
new file mode 100644
index 0000000000..d585b1e440
--- /dev/null
+++ b/lib/chef/cookbook_loader.rb
@@ -0,0 +1,116 @@
+#
+# Author:: Adam Jacob (<adam@hjksolutions.com>)
+# Copyright:: Copyright (c) 2008 HJK Solutions, LLC
+# License:: GNU General Public License version 2 or later
+# 
+# This program and entire repository is free software; you can
+# redistribute it and/or modify it under the terms of the GNU 
+# General Public License as published by the Free Software 
+# Foundation; either version 2 of the License, or any later version.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+# 
+
+class Chef
+  class CookbookLoader
+    
+    attr_accessor :cookbook, :config
+    
+    include Enumerable
+    
+    def initialize(config)
+      @config = config
+      @cookbook = Hash.new
+      load_cookbooks
+    end
+    
+    def load_cookbooks
+      cookbook_settings = Hash.new
+      @config.cookbook_path.each do |cb_path|
+        Dir[File.join(cb_path, "*")].each do |cookbook|
+          next unless File.directory?(cookbook)          
+          cookbook_name = File.basename(cookbook).to_sym
+          unless cookbook_settings.has_key?(cookbook_name)
+            cookbook_settings[cookbook_name] = { 
+              :ignore_regexes => Array.new,
+              :attribute_files => Array.new,
+              :definition_files => Array.new,
+              :recipe_files => Array.new,
+            }
+          end
+          ignore_regexes = load_ignore_file(File.join(cookbook, "ignore"))
+          cookbook_settings[cookbook_name][:ignore_regexes].concat(ignore_regexes)
+          load_files_unless_basename(
+            File.join(cookbook, "attributes", "*.rb"), 
+            cookbook_settings[cookbook_name][:attribute_files],
+            cookbook_settings[cookbook_name][:ignore_regexes]
+          )
+          load_files_unless_basename(
+            File.join(cookbook, "definitions", "*.rb"), 
+            cookbook_settings[cookbook_name][:definition_files],
+            cookbook_settings[cookbook_name][:ignore_regexes]
+          )
+          load_files_unless_basename(
+            File.join(cookbook, "recipes", "*.rb"), 
+            cookbook_settings[cookbook_name][:recipe_files],
+            cookbook_settings[cookbook_name][:ignore_regexes]
+          )
+        end
+      end
+      cookbook_settings.each_key do |cookbook|
+        @cookbook[cookbook] = Chef::Cookbook.new(cookbook)
+        @cookbook[cookbook].attribute_files = cookbook_settings[cookbook][:attribute_files]
+        @cookbook[cookbook].definition_files = cookbook_settings[cookbook][:definition_files]
+        @cookbook[cookbook].recipe_files = cookbook_settings[cookbook][:recipe_files]
+      end
+    end
+    
+    def [](cookbook)
+      @cookbook[cookbook.to_sym]
+    end
+    
+    def each
+      @cookbook.each_value do |cobject|
+        yield cobject
+      end
+    end
+    
+    private
+    
+      def load_ignore_file(ignore_file)
+        results = Array.new
+        if File.exists?(ignore_file) && File.readable?(ignore_file)
+          IO.foreach(ignore_file) do |line|
+            next if line =~ /^#/
+            next if line =~ /^\w*$/
+            line.chomp!
+            results << Regexp.new(line)
+          end
+        end
+        results
+      end
+    
+      def load_files_unless_basename(file_glob, result_array, ignore_regexes)
+        Dir[file_glob].each do |file|
+          skip = false
+          ignore_regexes.each do |exp|
+            skip = true if exp.match(file)
+          end
+          next if skip
+          file_basename = File.basename(file)
+          # If we've seen a file with this basename before, skip it.
+          unless result_array.detect { |f| File.basename(f) == file_basename }  
+            result_array << file
+          end
+        end
+      end
+      
+  end
+end
\ No newline at end of file
diff --git a/spec/data/kitchen/openldap/attributes/default.rb b/spec/data/kitchen/openldap/attributes/default.rb
new file mode 100644
index 0000000000..d208959475
--- /dev/null
+++ b/spec/data/kitchen/openldap/attributes/default.rb
@@ -0,0 +1,3 @@
+#
+# Nothing to see here, move along
+#
diff --git a/spec/data/kitchen/openldap/attributes/robinson.rb b/spec/data/kitchen/openldap/attributes/robinson.rb
new file mode 100644
index 0000000000..9d6b44d464
--- /dev/null
+++ b/spec/data/kitchen/openldap/attributes/robinson.rb
@@ -0,0 +1,3 @@
+#
+# Smokey lives here
+#
\ No newline at end of file
diff --git a/spec/data/kitchen/openldap/definitions/client.rb b/spec/data/kitchen/openldap/definitions/client.rb
new file mode 100644
index 0000000000..d4c2263b54
--- /dev/null
+++ b/spec/data/kitchen/openldap/definitions/client.rb
@@ -0,0 +1,3 @@
+#
+# A sad client
+#
diff --git a/spec/data/kitchen/openldap/definitions/drewbarrymore.rb b/spec/data/kitchen/openldap/definitions/drewbarrymore.rb
new file mode 100644
index 0000000000..510f0c35da
--- /dev/null
+++ b/spec/data/kitchen/openldap/definitions/drewbarrymore.rb
@@ -0,0 +1,3 @@
+#
+# Was in people magazine this month... 
+#
\ No newline at end of file
diff --git a/spec/data/kitchen/openldap/recipes/gigantor.rb b/spec/data/kitchen/openldap/recipes/gigantor.rb
new file mode 100644
index 0000000000..70a41960eb
--- /dev/null
+++ b/spec/data/kitchen/openldap/recipes/gigantor.rb
@@ -0,0 +1,3 @@
+cat "blanket" do
+  pretty_kitty true
+end
\ No newline at end of file
diff --git a/spec/data/kitchen/openldap/recipes/ignoreme.rb b/spec/data/kitchen/openldap/recipes/ignoreme.rb
new file mode 100644
index 0000000000..15095986c6
--- /dev/null
+++ b/spec/data/kitchen/openldap/recipes/ignoreme.rb
@@ -0,0 +1,3 @@
+#
+# this file will never be seen
+#
\ No newline at end of file
diff --git a/spec/data/kitchen/openldap/recipes/woot.rb b/spec/data/kitchen/openldap/recipes/woot.rb
new file mode 100644
index 0000000000..44893dae36
--- /dev/null
+++ b/spec/data/kitchen/openldap/recipes/woot.rb
@@ -0,0 +1,3 @@
+#
+# Such a funny word..
+#
diff --git a/spec/unit/cookbook_loader_spec.rb b/spec/unit/cookbook_loader_spec.rb
new file mode 100644
index 0000000000..3da3e644ec
--- /dev/null
+++ b/spec/unit/cookbook_loader_spec.rb
@@ -0,0 +1,116 @@
+#
+# Author:: Adam Jacob (<adam@hjksolutions.com>)
+# Copyright:: Copyright (c) 2008 HJK Solutions, LLC
+# License:: GNU General Public License version 2 or later
+# 
+# This program and entire repository is free software; you can
+# redistribute it and/or modify it under the terms of the GNU 
+# General Public License as published by the Free Software 
+# Foundation; either version 2 of the License, or any later version.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#
+
+require File.join(File.dirname(__FILE__), "..", "spec_helper")
+
+describe Chef::CookbookLoader do
+  before(:each) do
+    config = Chef::Config.new
+    config.cookbook_path [ 
+      File.join(File.dirname(__FILE__), "..", "data", "cookbooks"),
+      File.join(File.dirname(__FILE__), "..", "data", "kitchen") 
+    ]
+    @cl = Chef::CookbookLoader.new(config)
+  end
+  
+  it "should be a Chef::CookbookLoader object" do
+    @cl.should be_kind_of(Chef::CookbookLoader)
+  end
+  
+  it "should return cookbook objects with []" do
+    @cl[:openldap].should be_a_kind_of(Chef::Cookbook)
+  end
+  
+  it "should allow you to look up available cookbooks with [] and a symbol" do
+    @cl[:openldap].name.should eql(:openldap)
+  end
+  
+  it "should allow you to look up available cookbooks with [] and a string" do
+    @cl["openldap"].name.should eql(:openldap)
+  end
+  
+  it "should allow you to iterate over cookbooks with each" do
+    seen = Hash.new
+    @cl.each do |cb|
+      seen[cb.name] = true
+    end
+    seen.should have_key(:openldap)
+    seen.should have_key(:apache2)
+  end
+  
+  it "should find all the cookbooks in the cookbook path" do
+    @cl.config.cookbook_path << File.join(File.dirname(__FILE__), "..", "data", "hidden-cookbooks") 
+    @cl.load_cookbooks
+    @cl.detect { |cb| cb.name == :openldap }.should_not eql(nil)
+    @cl.detect { |cb| cb.name == :apache2 }.should_not eql(nil)
+  end
+  
+  it "should allow you to override an attribute file via cookbook_path" do
+    @cl[:openldap].attribute_files.detect { |f| 
+      f =~ /cookbooks\/openldap\/attributes\/default.rb/
+    }.should_not eql(nil)
+    @cl[:openldap].attribute_files.detect { |f| 
+      f =~ /kitchen\/openldap\/attributes\/default.rb/
+    }.should eql(nil)
+  end
+  
+  it "should load different attribute files from deeper paths" do
+    @cl[:openldap].attribute_files.detect { |f| 
+      f =~ /kitchen\/openldap\/attributes\/robinson.rb/
+    }.should_not eql(nil)
+  end
+  
+  it "should allow you to override a definition file via cookbook_path" do
+    @cl[:openldap].definition_files.detect { |f| 
+      f =~ /cookbooks\/openldap\/definitions\/client.rb/
+    }.should_not eql(nil)
+    @cl[:openldap].definition_files.detect { |f| 
+      f =~ /kitchen\/openldap\/definitions\/client.rb/
+    }.should eql(nil)
+  end
+  
+  it "should load definition files from deeper paths" do
+    @cl[:openldap].definition_files.detect { |f| 
+      f =~ /kitchen\/openldap\/definitions\/drewbarrymore.rb/
+    }.should_not eql(nil)
+  end
+  
+  it "should allow you to override a recipe file via cookbook_path" do
+    @cl[:openldap].recipe_files.detect { |f| 
+      f =~ /cookbooks\/openldap\/recipes\/gigantor.rb/
+    }.should_not eql(nil)
+    @cl[:openldap].recipe_files.detect { |f| 
+      f =~ /kitchen\/openldap\/recipes\/gigantor.rb/
+    }.should eql(nil)
+  end
+  
+  it "should load recipe files from deeper paths" do
+    @cl[:openldap].recipe_files.detect { |f| 
+      f =~ /kitchen\/openldap\/recipes\/woot.rb/
+    }.should_not eql(nil)
+  end
+  
+  it "should allow you to have an 'ignore' file, which skips loading files in later cookbooks" do
+    @cl[:openldap].recipe_files.detect { |f| 
+      f =~ /kitchen\/openldap\/recipes\/ignoreme.rb/
+    }.should eql(nil)
+  end
+  
+end
\ No newline at end of file