knife_cookbook_dependencies 0.0.1

data/.gitignore

@@ -0,0 +1,22 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+cookbooks
+*~
+*.tar*
+\#*
+Cookbookfile*

data/.rbenv-version

@@ -0,0 +1 @@
+1.9.3-p125

data/Gemfile

@@ -0,0 +1,3 @@
+source "https://rubygems.org"
+
+gemspec

data/Gemfile.lock

@@ -0,0 +1,90 @@
+PATH
+  remote: .
+  specs:
+    knife_cookbook_dependencies (0.0.1)
+      chef (~> 0.10.0)
+      dep_selector
+      minitar
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    bunny (0.7.9)
+    chef (0.10.8)
+      bunny (>= 0.6.0)
+      erubis
+      highline
+      json (>= 1.4.4, <= 1.6.1)
+      mixlib-authentication (>= 1.1.0)
+      mixlib-cli (>= 1.1.0)
+      mixlib-config (>= 1.1.2)
+      mixlib-log (>= 1.3.0)
+      moneta
+      net-ssh (~> 2.1.3)
+      net-ssh-multi (~> 1.1.0)
+      ohai (>= 0.6.0)
+      rest-client (>= 1.0.4, < 1.7.0)
+      treetop (~> 1.4.9)
+      uuidtools
+    dep_selector (0.0.8)
+    diff-lcs (1.1.3)
+    erubis (2.7.0)
+    highline (1.6.11)
+    ipaddress (0.8.0)
+    json (1.6.1)
+    mime-types (1.18)
+    minitar (0.5.3)
+    mixlib-authentication (1.1.4)
+      mixlib-log
+    mixlib-cli (1.2.2)
+    mixlib-config (1.1.2)
+    mixlib-log (1.3.0)
+    moneta (0.6.0)
+    multi_json (1.3.4)
+    net-ssh (2.1.4)
+    net-ssh-gateway (1.1.0)
+      net-ssh (>= 1.99.1)
+    net-ssh-multi (1.1)
+      net-ssh (>= 2.1.4)
+      net-ssh-gateway (>= 0.99.0)
+    ohai (0.6.12)
+      ipaddress
+      mixlib-cli
+      mixlib-config
+      mixlib-log
+      systemu
+      yajl-ruby
+    polyglot (0.3.3)
+    rake (0.9.2.2)
+    rdoc (3.12)
+      json (~> 1.4)
+    rest-client (1.6.7)
+      mime-types (>= 1.16)
+    rspec (2.9.0)
+      rspec-core (~> 2.9.0)
+      rspec-expectations (~> 2.9.0)
+      rspec-mocks (~> 2.9.0)
+    rspec-core (2.9.0)
+    rspec-expectations (2.9.1)
+      diff-lcs (~> 1.1.3)
+    rspec-mocks (2.9.0)
+    simplecov (0.6.2)
+      multi_json (~> 1.3)
+      simplecov-html (~> 0.5.3)
+    simplecov-html (0.5.3)
+    systemu (2.5.0)
+    treetop (1.4.10)
+      polyglot
+      polyglot (>= 0.3.1)
+    uuidtools (2.1.2)
+    yajl-ruby (1.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  knife_cookbook_dependencies!
+  rake (~> 0.9.0)
+  rdoc (~> 3.0)
+  rspec
+  simplecov

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Josiah Kiehl
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,5 @@
+= Remy, cookbook manager
+
+Gusteau:: What do I always say? Anyone can cook.   
+Remy:: Yeah. Anyone can, that doesn't mean that anyone should.  
+Gusteau:: Well, that is not stopping him. See?

data/Rakefile

@@ -0,0 +1,29 @@
+require "bundler/gem_tasks"
+require 'rdoc/task'
+require 'rspec/core/rake_task'
+
+desc "check documentation coverage"
+task "rdoc:check" do 
+  sh "rdoc -C " + Dir["lib/**/*.rb"].join(" ")
+end
+
+desc "clean up doco/coverage"
+task :clean do
+  sh "rm -fr rdoc coverage"
+end
+
+desc "generate documentation"
+RDoc::Task.new :rdoc do |r|
+  r.main = "README.rdoc"
+  r.rdoc_files.include("README.rdoc", "lib/**/*.rb")
+  r.rdoc_dir = "rdoc"
+end
+
+desc "Run specs"
+RSpec::Core::RakeTask.new do |t|
+  t.pattern = "./spec/**/*_spec.rb" # don't need this, it's default.
+  # Put spec opts in a file named .rspec in root
+end
+
+task :check => [:default, "rdoc:check"]
+task :default => [:clean, :spec]

data/knife_cookbook_dependencies.gemspec

@@ -0,0 +1,30 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/knife_cookbook_dependencies/version', __FILE__)
+
+Gem::Specification.new do |gem|
+
+  # TODO FIXME need to modify all of these
+
+  gem.authors       = ["Josiah Kiehl"]
+  gem.email         = ["josiah@skirmisher.net"]
+  gem.description   = %q{Resolves cookbook dependencies}
+  gem.summary       = gem.description
+  gem.homepage      = "http://github.com/RiotGames/knife_cookbook_dependencies"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "knife_cookbook_dependencies"
+  gem.require_paths = ["lib"]
+  gem.version       = KnifeCookbookDependencies::VERSION
+  
+  # FIXME will want to adjust this later
+  gem.add_runtime_dependency 'dep_selector',    '>= 0' 
+  gem.add_runtime_dependency 'chef',            '~> 0.10.0'
+  gem.add_runtime_dependency 'minitar',         '>= 0'
+
+  gem.add_development_dependency 'rake',        '~> 0.9.0'
+  gem.add_development_dependency 'rspec',       '>= 0'
+  gem.add_development_dependency 'rdoc',        '~> 3.0'
+  gem.add_development_dependency 'simplecov',   '>= 0'
+end

data/lib/chef/knife/cookbook_dependencies_install.rb

@@ -0,0 +1,17 @@
+require 'chef/knife'
+require 'knife_cookbook_dependencies'
+
+module KnifeCookbookDependencies
+  class CookbookDependenciesInstall < Chef::Knife
+    banner "knife cookbook dependencies install"
+
+    def run
+      ui.info 'Reading Cookbookfile'
+      ::KnifeCookbookDependencies.ui = ui
+      ::KnifeCookbookDependencies::Cookbookfile.process_install
+    end
+  end
+  
+  class CookbookDepsInstall < CookbookDependenciesInstall; end
+
+end

data/lib/knife_cookbook_dependencies/cookbook.rb

@@ -0,0 +1,184 @@
+require 'knife_cookbook_dependencies/knife_utils'
+require 'knife_cookbook_dependencies/git'
+require 'chef/knife/cookbook_site_download'
+require 'chef/knife/cookbook_site_show'
+
+module KnifeCookbookDependencies
+  class Cookbook
+    attr_reader :name, :version_constraints
+    attr_accessor :locked_version
+
+    DOWNLOAD_LOCATION = ENV["TMPDIR"] || '/tmp'
+
+    def initialize *args
+      @options = args.last.is_a?(Hash) ? args.pop : {}
+
+      if from_git? and from_path?
+        raise "Invalid: path and git options provided to #{args[0]}. They are mutually exclusive."
+      end
+
+      @options[:path] = File.expand_path(@options[:path]) if from_path?
+      @name, constraint_string = args
+
+      add_version_constraint(if from_path?
+                               "= #{version_from_metadata_file.to_s}"
+                             else
+                               constraint_string
+                             end)
+      @locked_version = DepSelector::Version.new(@options[:locked_version]) if @options[:locked_version]
+    end
+
+    def add_version_constraint constraint_string
+      @version_constraints ||= []
+      @version_constraints << DepSelector::VersionConstraint.new(constraint_string) unless @version_constraints.collect(&:to_s).include? constraint_string
+    end
+
+    def download(show_output = false)
+      return if @downloaded
+      return if !from_git? and downloaded_archive_exists?
+
+      if from_git? 
+        @git ||= KnifeCookbookDependencies::Git.new(@options[:git])
+        @git.clone
+        @git.checkout(@options[:ref]) if @options[:ref]
+        @options[:path] ||= @git.directory
+      elsif from_path?
+        return
+      else
+        csd = Chef::Knife::CookbookSiteDownload.new([name, latest_constrained_version.to_s, "--file", download_filename])
+        output = KnifeCookbookDependencies::KnifeUtils.capture_knife_output(csd)
+
+        if show_output
+          puts output
+        end
+      end
+
+      @downloaded = true
+    end
+
+    def copy_to_cookbooks_directory
+      FileUtils.mkdir_p KnifeCookbookDependencies::COOKBOOKS_DIRECTORY
+
+      target = File.join(KnifeCookbookDependencies::COOKBOOKS_DIRECTORY, @name)
+      FileUtils.rm_rf target
+      FileUtils.cp_r full_path, target
+      FileUtils.rm_rf File.join(target, '.git') if from_git?
+    end
+
+    # TODO: Clean up download repetition functionality here, in #download and the associated test.
+    def unpack(location = unpacked_cookbook_path, do_clean = false, do_download = true)
+      return true if from_path?
+      self.clean(File.join(location, @name)) if do_clean
+      download if do_download
+      fname = download_filename
+      if File.directory? location
+        true # noop
+      elsif downloaded_archive_exists?
+        Archive::Tar::Minitar.unpack(Zlib::GzipReader.new(File.open(fname)), location)
+        true
+      else
+        # TODO: Raise friendly error message class
+        raise "Archive hasn't been downloaded yet"
+      end
+    end
+
+    def dependencies
+      download
+      unpack
+      @dependencies ||= DependencyReader.new(self).read
+    end
+
+    def latest_constrained_version
+      return @locked_version if @locked_version
+      return version_from_metadata_file if from_path? or from_git?
+
+      versions.reverse.each do |v|
+        return v if version_constraints_include? v
+      end
+      KnifeCookbookDependencies.ui.fatal "No version available to fit the following constraints for #{@name}: #{version_constraints.inspect}\nAvailable versions: #{versions.inspect}"
+      exit 1
+    end
+
+    def version_constraints_include? version
+      @version_constraints.inject(true) { |check, constraint| check and constraint.include? version }
+    end
+
+    def versions
+      return [latest_constrained_version] if @locked_version
+      return [version_from_metadata_file] if from_path? or from_git?
+      cookbook_data['versions'].collect { |v| DepSelector::Version.new(v.split(/\//).last.gsub(/_/, '.')) }.sort
+    end
+
+    def version_from_metadata_file
+      # TODO: make a generic metadata file reader to replace
+      # dependencyreader and incorporate pulling the version as
+      # well... knife probably has something like this I can use/steal
+      DepSelector::Version.new(metadata_file.match(/version\s+[\"\']([0-9\.]*)[\"\']/)[1])
+    end
+
+    def cookbook_data
+      css = Chef::Knife::CookbookSiteShow.new([@name])
+      @cookbook_data ||= JSON.parse(KnifeCookbookDependencies::KnifeUtils.capture_knife_output(css))
+    end
+
+    def download_filename
+      return nil if from_path?
+      File.join(DOWNLOAD_LOCATION, "#{@name}-#{latest_constrained_version}.tar.gz")
+    end
+
+    def unpacked_cookbook_path
+      @options[:path] || File.join(File.dirname(download_filename), File.basename(download_filename, '.tar.gz'))
+    end
+
+    def full_path
+      if @git
+        unpacked_cookbook_path
+      else
+        File.join(unpacked_cookbook_path, @name)
+      end
+    end
+
+    def metadata_filename
+      File.join(full_path, "metadata.rb")
+    end
+
+    def metadata_file
+      download
+      unpack
+      File.open(metadata_filename).read
+    end
+
+    def from_path?
+      !!@options[:path]
+    end
+
+    def from_git?
+      !!@options[:git]
+    end
+
+    def git_repo
+      @options[:git]
+    end
+
+    def git_ref
+      (from_git? && @git) ? @git.ref : nil
+    end
+
+    def downloaded_archive_exists?
+      download_filename && File.exists?(download_filename)
+    end
+
+    def clean(location = unpacked_cookbook_path)
+      if @git
+        @git.clean
+      else
+        FileUtils.rm_rf location
+        FileUtils.rm_f download_filename
+      end
+    end
+
+    def == other
+      other.name == @name and other.version_constraints == @version_constraints
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/cookbookfile.rb

@@ -0,0 +1,33 @@
+require 'knife_cookbook_dependencies/dsl'
+
+module KnifeCookbookDependencies
+  class Cookbookfile
+    class << self
+      include DSL
+      def read content
+        # This will populate KnifeCookbookDependencies.shelf. TODO: consider making this
+        # build and return the shelf rather than building the shelf as
+        # a side effect.
+        instance_eval(content)
+      end
+
+      def process_install
+        # TODO: friendly error message when the file doesn't exist
+        
+        filename = KnifeCookbookDependencies::DEFAULT_FILENAME + ".lock"
+        lockfile = false
+
+        if File.exist?(filename)
+          lockfile = true
+        else
+          filename = KnifeCookbookDependencies::DEFAULT_FILENAME unless File.exist?(filename)
+        end
+          
+        read File.open(filename).read
+        KnifeCookbookDependencies.shelf.resolve_dependencies
+        KnifeCookbookDependencies.shelf.populate_cookbooks_directory
+        KnifeCookbookDependencies.shelf.write_lockfile unless lockfile
+      end
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/dependency_reader.rb

@@ -0,0 +1,46 @@
+module KnifeCookbookDependencies
+  class DependencyReader
+    attr_reader :dependency_list, :cookbook
+    
+    def initialize(cookbook)
+      @cookbook = cookbook
+      @dependency_list = []
+    end
+
+    def read
+      Dir.chdir(@cookbook.full_path) do
+        # XXX this filename is required because it sets __FILE__, which is
+        # used for README.md parsing among other things in metadata.rb files
+        instance_eval(@cookbook.metadata_file, @cookbook.metadata_filename)
+      end
+      @dependency_list
+    end
+    
+    def depends(*args)
+      name, constraint = args
+
+      dependency_cookbook = KnifeCookbookDependencies.shelf.get_cookbook(name) || get_dependency(name)
+      if dependency_cookbook
+        dependency_cookbook.add_version_constraint constraint
+      else
+        @dependency_list << Cookbook.new(*args) 
+      end
+    end
+
+    def get_dependency(name)
+      @dependency_list.select { |c| c.name == name }.first
+    end
+
+    def method_missing(method, *args)
+      # Don't blow up when other metadata DSL methods are called, we
+      # only care about #depends.
+    end
+
+    def name(the_name)
+      # Module#name is defined, so method_missing won't catch it
+      # when running instance_eval on a metadata.rb that overrides
+      # the name
+    end
+
+  end
+end

data/lib/knife_cookbook_dependencies/dsl.rb

@@ -0,0 +1,7 @@
+module KnifeCookbookDependencies
+  module DSL
+    def cookbook *args
+      KnifeCookbookDependencies.shelf.shelve_cookbook *args
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/git.rb

@@ -0,0 +1,86 @@
+require 'tempfile'
+
+module KnifeCookbookDependencies
+  class Git
+    class << self
+      def git
+        @git ||= find_git
+      end
+
+      #
+      # This is to defeat aliases/shell functions called 'git' and a number of
+      # other problems.
+      #
+      def find_git
+        git_path = nil
+        ENV["PATH"].split(File::PATH_SEPARATOR).each do |path|
+          potential_path = File.join(path, 'git')
+          if File.executable?(potential_path)
+            git_path = potential_path
+            break
+          end
+        end
+
+        unless git_path
+          raise "Could not find git. Please ensure it is in your path."
+        end
+
+        return git_path
+      end
+    end
+
+    attr_reader :directory
+    attr_reader :repository
+
+    def initialize(repo)
+      @repository = repo
+    end
+
+    def clone
+      # XXX not sure how resilient this is, maybe a fetch/merge strategy would be better.
+      if @directory
+        Dir.chdir @directory do
+          system(self.class.git, "pull")
+        end
+      else
+        @directory = Dir.mktmpdir
+        system(self.class.git, "clone", @repository, @directory)
+      end
+
+      error_check
+
+    end
+
+    def checkout(ref)
+      clone
+
+      Dir.chdir @directory do
+        system(self.class.git, "checkout", "-q", ref)
+      end
+
+      error_check
+    end
+
+    def ref
+      return nil unless @directory
+
+      this_ref = nil
+
+      Dir.chdir @directory do
+        this_ref = `#{self.class.git} rev-parse HEAD`.strip
+      end
+
+      return this_ref
+    end
+
+    def clean
+      FileUtils.rm_rf @directory if @directory
+    end
+
+    def error_check
+      if $?.exitstatus != 0
+        raise "Did not succeed executing git; check the output above."
+      end
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/knife_utils.rb

@@ -0,0 +1,13 @@
+require 'stringio'
+require 'chef/config'
+
+module KnifeCookbookDependencies
+  module KnifeUtils
+    def self.capture_knife_output(knife_obj)
+      knife_obj.ui = Chef::Knife::UI.new(StringIO.new, StringIO.new, StringIO.new, { :format => :json })
+      knife_obj.run
+      knife_obj.ui.stdout.rewind
+      knife_obj.ui.stdout.read
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/lockfile.rb

@@ -0,0 +1,25 @@
+require 'knife_cookbook_dependencies/cookbookfile'
+
+module KnifeCookbookDependencies
+  class Lockfile
+    def initialize(cookbooks)
+      @cookbooks = cookbooks
+    end
+
+    def write(filename = KnifeCookbookDependencies::DEFAULT_FILENAME)
+      content = @cookbooks.map do |cookbook|
+                  get_cookbook_definition(cookbook)
+                end.join("\n")
+      File.write(filename + ".lock", content)
+    end
+
+    def get_cookbook_definition(cookbook)
+      definition = "cookbook '#{cookbook.name}', :locked_version => '#{cookbook.locked_version}'"
+      if cookbook.git_repo
+        definition += ", :git => '#{cookbook.git_repo}', :ref => '#{cookbook.git_ref || 'HEAD'}'"
+      end
+
+      return definition
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/metacookbook.rb

@@ -0,0 +1,18 @@
+module KnifeCookbookDependencies
+  class MetaCookbook
+    attr_reader :name, :dependencies
+
+    def initialize name, dependencies
+      @name = name
+      @dependencies = dependencies
+    end
+
+    def versions
+      @v ||= [DepSelector::Version.new('1.0.0')]
+    end
+
+    def latest_constrained_version
+      versions.first
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/shelf.rb

@@ -0,0 +1,70 @@
+require 'knife_cookbook_dependencies/lockfile'
+
+module KnifeCookbookDependencies
+  class Shelf
+    META_COOKBOOK_NAME = 'cookbook_dependencies_shelf'
+
+    attr_accessor :cookbooks
+
+    def initialize
+      @cookbooks = []
+    end
+    
+    def shelve_cookbook(*args)
+      @cookbooks << (args.first.is_a?(Cookbook) ? args.first : Cookbook.new(*args))
+    end
+
+    def resolve_dependencies
+      graph = DepSelector::DependencyGraph.new
+
+      # all cookbooks in the Cookbookfile are dependencies of the shelf
+      shelf = MetaCookbook.new(META_COOKBOOK_NAME, @cookbooks) 
+
+      self.class.populate_graph graph, shelf
+
+      selector = DepSelector::Selector.new(graph)
+      solution = selector.find_solution([DepSelector::SolutionConstraint.new(graph.package(META_COOKBOOK_NAME))])
+      solution.delete META_COOKBOOK_NAME
+      solution
+    end
+
+    def write_lockfile
+      KnifeCookbookDependencies::Lockfile.new(@cookbooks).write
+    end
+
+    def get_cookbook(name)
+      @cookbooks.select { |c| c.name == name }.first
+    end
+
+    def populate_cookbooks_directory
+      cookbooks_from_path = @cookbooks.select(&:from_path?) | @cookbooks.select(&:from_git?)
+      
+      resolve_dependencies.each_pair do |cookbook_name, version|
+        cookbook = cookbooks_from_path.select { |c| c.name == cookbook_name }.first || Cookbook.new(cookbook_name, version.to_s)
+        @cookbooks << cookbook
+        cookbook.download
+        cookbook.unpack
+        cookbook.copy_to_cookbooks_directory
+        cookbook.locked_version = version
+      end
+      @cookbooks = @cookbooks.uniq.reject { |x| x.locked_version.nil? }
+    end
+
+    class << self
+      def populate_graph(graph, cookbook)
+        package = graph.package cookbook.name
+        cookbook.versions.each { |v| package.add_version(v) }
+        cookbook.dependencies.each do |dependency|
+          graph = populate_graph(graph, dependency)
+          dep = graph.package(dependency.name)
+          version = package.versions.select { |v| v.version == cookbook.latest_constrained_version }.first
+          dependency.version_constraints.each do |constraint|
+            version.dependencies << DepSelector::Dependency.new(dep, constraint)
+          end
+        end
+
+        graph
+      end
+    end
+  end
+end

data/lib/knife_cookbook_dependencies/version.rb

@@ -0,0 +1,3 @@
+module KnifeCookbookDependencies
+  VERSION = "0.0.1"
+end

data/lib/knife_cookbook_dependencies.rb

@@ -0,0 +1,42 @@
+require 'dep_selector'
+require 'zlib'
+require 'archive/tar/minitar'
+require 'chef/config'
+require 'chef/knife/cookbook_site_download'
+
+require 'knife_cookbook_dependencies/version'
+require 'knife_cookbook_dependencies/shelf'
+require 'knife_cookbook_dependencies/cookbook'
+require 'knife_cookbook_dependencies/metacookbook'
+require 'knife_cookbook_dependencies/dependency_reader'
+require 'knife_cookbook_dependencies/dsl'
+require 'knife_cookbook_dependencies/cookbookfile'
+require 'knife_cookbook_dependencies/git'
+
+module KnifeCookbookDependencies
+  DEFAULT_FILENAME = 'Cookbookfile'
+  COOKBOOKS_DIRECTORY = 'cookbooks'
+
+  class << self
+    def shelf
+      @shelf ||= KnifeCookbookDependencies::Shelf.new
+    end
+
+    def clear_shelf!
+      @shelf = nil
+    end
+
+    def ui=(ui)
+      @ui = ui
+    end
+
+    def ui
+      @ui ||= Chef::Knife::UI.new(STDOUT, STDERR, STDIN, {})
+    end
+
+    def clean
+      clear_shelf!
+      FileUtils.rm_rf COOKBOOKS_DIRECTORY
+    end
+  end
+end

data/spec/fixtures/cookbooks/example_cookbook/README.md

@@ -0,0 +1,12 @@
+Description
+===========
+
+Requirements
+============
+
+Attributes
+==========
+
+Usage
+=====
+

data/spec/fixtures/cookbooks/example_cookbook/metadata.rb

@@ -0,0 +1,6 @@
+maintainer       "Josiah Kiehl"
+maintainer_email "josiah@skirmisher.net"
+license          "DO WHAT YOU WANT CAUSE A PIRATE IS FREE"
+description      "Installs/Configures example_cookbook"
+long_description IO.read(File.join(File.dirname(__FILE__), 'README.md'))
+version          "0.5.0"

data/spec/fixtures/cookbooks/example_cookbook/recipes/default.rb

@@ -0,0 +1,8 @@
+#
+# Cookbook Name:: example_cookbook
+# Recipe:: default
+#
+# Copyright 2012, YOUR_COMPANY_NAME
+#
+# All rights reserved - Do Not Redistribute
+#

data/spec/lib/knife_cookbook_dependencies/cookbook_spec.rb

@@ -0,0 +1,109 @@
+require 'spec_helper'
+
+module KnifeCookbookDependencies
+  describe Cookbook do
+    subject { Cookbook.new('ntp') }
+
+    after do
+      subject.clean
+    end
+
+    describe  do
+
+      before do
+        Cookbook.any_instance.stub(:versions).and_return ['0.1.1', '0.9.0', '1.0.0', '1.1.8'].collect {|v| Gem::Version.new(v) }
+      end
+
+      # FIXME: This test is flakey
+      it "should raise an error if the cookbook is unpacked without being downloaded first" do
+        -> { subject.unpack(subject.unpacked_cookbook_path, true, false) }.should raise_error
+      end
+
+      describe '#unpacked_cookbook_path' do
+        it "should give the path to the directory where the archive should get unpacked" do
+          subject.unpacked_cookbook_path.should == File.join(ENV['TMPDIR'], 'ntp-1.1.8')
+        end
+      end
+
+      it 'should treat cookbooks pulled from a path like a cookbook that has already been unpacked with the path as the unpacked location' do
+        example_cookbook_from_path.unpacked_cookbook_path.should == File.expand_path(File.join(File.dirname(__FILE__), "..", "..", "..", "spec", "fixtures", "cookbooks"))
+      end
+
+      it "should not attempt to download a cookbook being pulled from a path" do
+        Chef::Knife::CookbookSiteDownload.any_instance.should_not_receive(:run)
+        example_cookbook_from_path.download
+      end
+
+      describe "#copy_to" do
+        it "should copy from the unpacked cookbook directory to the target" do
+          example_cookbook_from_path.copy_to_cookbooks_directory
+          File.exists?(File.join(KnifeCookbookDependencies::COOKBOOKS_DIRECTORY, example_cookbook_from_path.name)).should be_true
+        end
+      end
+      
+    end
+
+    describe "#versions" do
+      it "should return the version in the metadata file as the available versions for a path sourced cookbook" do
+        example_cookbook_from_path.versions.should == [DepSelector::Version.new('0.5.0')]
+      end
+    end
+
+    describe '#version_from_metadata_file' do
+      it "should be able to handle single quoted strings" do
+        Cookbook.any_instance.stub(:metadata_file).and_return <<M
+version '1.2.3'
+M
+        subject.version_from_metadata_file.should == DepSelector::Version.new('1.2.3')
+      end
+      it "should be able to handle double quoted strings" do
+        Cookbook.any_instance.stub(:metadata_file).and_return <<M
+version "1.2.3"
+M
+        subject.version_from_metadata_file.should == DepSelector::Version.new('1.2.3')
+      end
+    end
+
+    describe '#version_constraints_include?' do
+      it "should cycle through all the version constraints to confirm that all of them are satisfied" do
+        subject.add_version_constraint ">= 1.0.0"
+        subject.add_version_constraint "< 2.0.0"
+        subject.version_constraints_include?(DepSelector::Version.new('1.0.0')).should be_true
+        subject.version_constraints_include?(DepSelector::Version.new('1.5.0')).should be_true
+        subject.version_constraints_include?(DepSelector::Version.new('2.0.0')).should be_false
+      end
+    end
+
+    describe '#add_version_constraint' do
+      it "should not duplicate version constraints" do
+        subject.add_version_constraint ">= 1.0.0"
+        subject.add_version_constraint ">= 1.0.0"
+        subject.add_version_constraint ">= 1.0.0"
+        subject.add_version_constraint ">= 1.0.0"
+        subject.version_constraints.size.should == 2 # 1 for the
+                                                     # default when
+                                                     # the cookbook
+                                                     # was created in
+                                                     # the subject
+                                                     # instantiation
+                                                     # line
+      end
+    end
+
+    describe '#dependencies' do
+      it "should not contain the cookbook itself" do
+        # TODO: Mock
+        Cookbook.new('riot_base', git: 'git@github.riotgames.com:cookbooks/riot_base.git').dependencies.collect(&:name).include?('riot_base').should_not be_true
+      end
+    end
+
+    # TODO figure out how to test this. Stubs on classes don't clear after a test.
+    # describe '#unpack' do
+    #   it "should not unpack if it is already unpacked" do
+    #     Archive::Tar::Minitar.should_receive(:unpack).once
+    #     subject.unpack
+    #     subject.unpack
+    #   end
+    # end
+  end
+end

data/spec/lib/knife_cookbook_dependencies/cookbookfile_spec.rb

@@ -0,0 +1,26 @@
+require 'spec_helper'
+
+module KnifeCookbookDependencies
+  describe Cookbookfile do
+    describe '::read' do
+      after do
+        KnifeCookbookDependencies.shelf.cookbooks.each(&:clean)
+      end
+
+      it "should read the cookbookfile and build a dependency list" do
+        described_class.read <<COOKBOOKFILE
+cookbook 'ntp', '<= 1.0.0'
+cookbook 'mysql'
+cookbook 'nginx', '< 0.101.2'
+cookbook 'ssh_known_hosts2', :git => 'https://github.com/erikh/chef-ssh_known_hosts2.git'
+COOKBOOKFILE
+
+        ['ntp', 'mysql', 'nginx', 'ssh_known_hosts2'].each do |dep|
+          KnifeCookbookDependencies.shelf.cookbooks.collect(&:name).should include dep
+        end
+
+        KnifeCookbookDependencies.shelf.populate_cookbooks_directory
+      end
+    end
+  end
+end

data/spec/lib/knife_cookbook_dependencies/dependency_reader_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+module KnifeCookbookDependencies
+  describe DependencyReader do
+    subject { DependencyReader.new(Cookbook.new('mysql')) }
+
+    before do
+      @cookbook = Cookbook.new('mysql')
+      @cookbook.download # TODO: mock out
+      @cookbook.unpack
+    end
+
+    after do
+      @cookbook.clean
+    end
+
+    it "should parse the metadata file for dependencies" do
+      subject.read.should == [Cookbook.new('openssl')]
+    end
+
+    it "should not blow up when reading a metadata.rb that overrides the name" do
+      Cookbook.any_instance.stub(:metadata_file).and_return <<M
+name 'dontblowupplease'
+version '1.2.3'
+M
+      -> { subject.read }.should_not raise_error
+    end
+
+    it 'should display a warning when no version is defined in the metadata.rb'
+
+    it "should add a constraint to the cookbook on the shelf instead of adding a new dependency" do
+      KnifeCookbookDependencies.shelf.shelve_cookbook example_cookbook_from_path
+      subject.depends('example_cookbook')
+      subject.dependency_list.should be_empty
+    end
+
+    it "should create a new cookbook if the cookbook is not found on the shelf" do
+      subject.depends('example_cookbook')
+      subject.dependency_list.first.name.should == 'example_cookbook'
+    end
+  end
+end

data/spec/lib/knife_cookbook_dependencies/dsl_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+require 'knife_cookbook_dependencies/dsl'
+
+module KnifeCookbookDependencies
+  describe DSL do
+    include DSL
+
+    describe "#cookbook" do
+      after do
+        KnifeCookbookDependencies.shelf.cookbooks.each(&:clean)
+      end
+
+      it 'should add the cookbooks to the shelf' do
+        cookbook "ntp"
+        cookbook "nginx"
+
+        ['ntp', 'nginx'].each do |cookbook|
+          KnifeCookbookDependencies.shelf.cookbooks.collect(&:name).should include cookbook
+        end
+      end
+
+      it 'should take version constraints' do
+        Cookbook.any_instance.stub(:clean)
+        cookbook 'ntp', '= 1.2.3'
+        KnifeCookbookDependencies.shelf.cookbooks.select {|c| c.name == 'ntp'}.first.version_constraints.first.should == DepSelector::VersionConstraint.new('= 1.2.3')
+      end
+    end
+  end
+end

data/spec/lib/knife_cookbook_dependencies/git_spec.rb

@@ -0,0 +1,58 @@
+require 'spec_helper'
+
+module KnifeCookbookDependencies
+  describe Git do
+
+    it "should find git" do
+      Git.find_git.should_not be_nil
+    end
+
+    it "should raise if it can't find git" do
+      begin
+        path = ENV["PATH"]
+        ENV["PATH"] = ""
+
+        lambda { Git.find_git }.should raise_error
+      ensure
+        ENV["PATH"] = path
+      end
+    end
+
+    describe "instances" do
+      subject { Git.new("https://github.com/erikh/chef-ssh_known_hosts2.git") }
+
+      after do
+        subject.clean
+      end
+
+      it "should be set the repository accessor" do
+        subject.repository.should == "https://github.com/erikh/chef-ssh_known_hosts2.git"
+      end
+
+      it "should be able to clone this repository" do
+        subject.clone
+
+        File.exist?(subject.directory).should be_true
+        File.directory?(subject.directory).should be_true
+      end
+
+      it "should clone, then pull if the directory is the same" do
+        subject.clone
+        dir = subject.directory
+        subject.clone
+        dir2 = subject.directory
+
+        dir.should == dir2
+      end
+
+      it "should checkout the v0.1.0 tag" do
+        lambda { subject.checkout "fart" }.should raise_error
+        lambda { subject.checkout "v0.1.0" }.should_not raise_error
+
+        Dir.chdir subject.directory do
+          %x[git rev-parse v0.1.0].should == %x[git rev-parse HEAD]
+        end
+      end
+    end
+  end
+end

data/spec/lib/knife_cookbook_dependencies/shelf_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+module KnifeCookbookDependencies
+  describe Shelf do
+    describe '#get_cookbook' do
+      it "should return nil if a cookbook doesn't exist on the shelf" do
+        Shelf.new.get_cookbook('arbitrary').should be_nil
+      end
+      it "should return the cookbook if the cookbook exists on the shelf" do
+        s = Shelf.new
+        s.shelve_cookbook example_cookbook_from_path
+        s.get_cookbook(example_cookbook_from_path.name).should_not be_nil
+      end
+    end
+
+    describe "#shelve_cookbook" do
+      subject { Shelf.new }
+      it 'should store shelved cookbooks' do
+        subject.shelve_cookbook 'acookbook'
+        subject.cookbooks.collect(&:name).should include 'acookbook'
+      end
+
+      it 'should take version constraints' do
+        subject.shelve_cookbook 'acookbook', '= 1.2.3'
+        subject.cookbooks.last.version_constraints.should == [DepSelector::VersionConstraint.new('= 1.2.3')]
+      end
+
+      it "should resolve the dependency graph of the cookbooks on the shelf" do
+        subject.shelve_cookbook 'mysql', "= 1.2.4"
+        
+        subject.resolve_dependencies.should == ({"mysql" => DepSelector::Version.new("1.2.4"), "openssl" => DepSelector::Version.new("1.0.0")})
+        Cookbook.new('mysql').clean
+        Cookbook.new('openssl').clean
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,21 @@
+$:.push File.join(File.dirname(__FILE__), '..')
+$:.push File.join(File.dirname(__FILE__), '..', 'lib')
+
+require 'simplecov'
+require 'pp'
+
+RSpec.configure do |config|
+  config.after do
+    KnifeCookbookDependencies.clean
+  end
+end
+
+SimpleCov.start do
+  add_filter 'spec/'
+end
+
+require 'knife_cookbook_dependencies'
+
+def example_cookbook_from_path
+  @example_cookbook_from_path ||= KnifeCookbookDependencies::Cookbook.new('example_cookbook', path: File.join(File.dirname(__FILE__), 'fixtures', 'cookbooks'))
+end

data/todo.txt

@@ -0,0 +1,13 @@
+Add git support to Cookbookfile
+ - Allow a cookbook to be pulled from a git repository
+ - Also support 'ref' to give a commit hash to pull
+ - Possibly support 'branch' to pull the HEAD from a given branch
+Add path support to Cookbookfile
+ - Pull cookbook from a path on the local filesystem
+Add source support to Cookbookfile
+ - Allow unofficial cookbook repositories to be used via 'source' in Cookbookfile
+Create local mock API service for specs
+ - Instead of mocking out methods, spin up a local mock cookbook repo for tests to run against. This should be able to serve cookbooks as gzips. This might require some monkey patching of knife to get it to talk to the mock server instead of the opscode community site.
+Optimize cookbook retrieval
+ - Possible: Pester Opscode till they release an API call to retrieve cookbook dependencies
+ - Possible: Keep an index of dependencies so we don't have to download cookbook versions we've seen before

metadata

@@ -0,0 +1,163 @@
+--- !ruby/object:Gem::Specification
+name: knife_cookbook_dependencies
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Josiah Kiehl
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: dep_selector
+  requirement: &70196647928780 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70196647928780
+- !ruby/object:Gem::Dependency
+  name: chef
+  requirement: &70196647928060 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70196647928060
+- !ruby/object:Gem::Dependency
+  name: minitar
+  requirement: &70196647943200 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70196647943200
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &70196647940420 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+  type: :development
+  prerelease: false
+  version_requirements: *70196647940420
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70196647936660 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70196647936660
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: &70196647944920 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: *70196647944920
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: &70196647960000 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70196647960000
+description: Resolves cookbook dependencies
+email:
+- josiah@skirmisher.net
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rbenv-version
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.rdoc
+- Rakefile
+- knife_cookbook_dependencies.gemspec
+- lib/chef/knife/cookbook_dependencies_install.rb
+- lib/knife_cookbook_dependencies.rb
+- lib/knife_cookbook_dependencies/cookbook.rb
+- lib/knife_cookbook_dependencies/cookbookfile.rb
+- lib/knife_cookbook_dependencies/dependency_reader.rb
+- lib/knife_cookbook_dependencies/dsl.rb
+- lib/knife_cookbook_dependencies/git.rb
+- lib/knife_cookbook_dependencies/knife_utils.rb
+- lib/knife_cookbook_dependencies/lockfile.rb
+- lib/knife_cookbook_dependencies/metacookbook.rb
+- lib/knife_cookbook_dependencies/shelf.rb
+- lib/knife_cookbook_dependencies/version.rb
+- spec/fixtures/cookbooks/example_cookbook/README.md
+- spec/fixtures/cookbooks/example_cookbook/metadata.rb
+- spec/fixtures/cookbooks/example_cookbook/recipes/default.rb
+- spec/lib/knife_cookbook_dependencies/cookbook_spec.rb
+- spec/lib/knife_cookbook_dependencies/cookbookfile_spec.rb
+- spec/lib/knife_cookbook_dependencies/dependency_reader_spec.rb
+- spec/lib/knife_cookbook_dependencies/dsl_spec.rb
+- spec/lib/knife_cookbook_dependencies/git_spec.rb
+- spec/lib/knife_cookbook_dependencies/shelf_spec.rb
+- spec/spec_helper.rb
+- todo.txt
+homepage: http://github.com/RiotGames/knife_cookbook_dependencies
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Resolves cookbook dependencies
+test_files:
+- spec/fixtures/cookbooks/example_cookbook/README.md
+- spec/fixtures/cookbooks/example_cookbook/metadata.rb
+- spec/fixtures/cookbooks/example_cookbook/recipes/default.rb
+- spec/lib/knife_cookbook_dependencies/cookbook_spec.rb
+- spec/lib/knife_cookbook_dependencies/cookbookfile_spec.rb
+- spec/lib/knife_cookbook_dependencies/dependency_reader_spec.rb
+- spec/lib/knife_cookbook_dependencies/dsl_spec.rb
+- spec/lib/knife_cookbook_dependencies/git_spec.rb
+- spec/lib/knife_cookbook_dependencies/shelf_spec.rb
+- spec/spec_helper.rb