berkshelf 0.1.1

data/lib/berkshelf/dsl.rb

@@ -0,0 +1,39 @@
+module Berkshelf
+  module DSL
+    @@active_group = nil
+
+    def cookbook(*args)
+      source = CookbookSource.new(*args)
+      source.add_group(@@active_group) if @@active_group
+      add_source(source)
+    end
+
+    def group(*args)
+      @@active_group = args
+      yield
+      @@active_group = nil
+    end
+
+    def metadata(options = {})
+      path = options[:path] || File.expand_path('.')
+
+      metadata_file = Berkshelf.find_metadata(path)
+
+      unless metadata_file
+        raise CookbookNotFound, "No 'metadata.rb' found at #{path}"
+      end
+
+      metadata = Chef::Cookbook::Metadata.new
+      metadata.from_file(metadata_file.to_s)
+
+      name = if metadata.name.empty? || metadata.name.nil?
+        File.basename(File.dirname(metadata_file))
+      else
+        metadata.name
+      end
+
+      source = CookbookSource.new(name, :path => File.dirname(metadata_file))
+      add_source(source)
+    end
+  end
+end

data/lib/berkshelf/errors.rb

@@ -0,0 +1,20 @@
+module Berkshelf
+  class BerkshelfError < StandardError
+    class << self
+      def status_code(code)
+        define_method(:status_code) { code }
+        define_singleton_method(:status_code) { code }
+      end
+    end
+  end
+
+  class BerksfileNotFound < BerkshelfError; status_code(100); end
+  class NoVersionForConstraints < BerkshelfError; status_code(101); end
+  class DownloadFailure < BerkshelfError; status_code(102); end
+  class CookbookNotFound < BerkshelfError; status_code(103); end
+  class GitError < BerkshelfError; status_code(104); end
+  class DuplicateSourceDefined < BerkshelfError; status_code(105); end
+  class NoSolution < BerkshelfError; status_code(106); end
+  class CookbookSyntaxError < BerkshelfError; status_code(107); end
+  class UploadFailure < BerkshelfError; status_code(108); end
+end

data/lib/berkshelf/generator_files/Berksfile.erb

@@ -0,0 +1,3 @@
+<% if options[:metadata_entry] -%>
+metadata
+<% end -%>

data/lib/berkshelf/generator_files/chefignore

@@ -0,0 +1,44 @@
+# Put files/directories that should be ignored in this file.
+# Lines that start with '# ' are comments.
+
+## OS
+.DS_Store
+Icon?
+nohup.out
+
+## EDITORS
+\#*
+.#*
+*~
+*.sw[a-z]
+*.bak
+REVISION
+TAGS*
+tmtags
+*_flymake.*
+*_flymake
+*.tmproj
+.project
+.settings
+mkmf.log
+
+## COMPILED
+a.out
+*.o
+*.pyc
+*.so
+
+## OTHER SCM
+*/.bzr/*
+*/.hg/*
+*/.svn/*
+
+## Don't send rspecs up in cookbook
+.watchr
+.rspec
+spec/*
+spec/fixtures/*
+features/*
+
+# Berkshelf
+Berksfile

data/lib/berkshelf/git.rb

@@ -0,0 +1,70 @@
+module Berkshelf
+  class Git
+    class << self
+      def git(*command)
+        out = quietly {
+          %x{ #{git_cmd} #{command.join(' ')} }
+        }
+        error_check
+        
+        out.chomp
+      end
+
+      def clone(uri, destination = Dir.mktmpdir)
+        git("clone", uri, destination.to_s)
+
+        error_check
+
+        destination
+      end
+
+      def checkout(repo_path, ref)
+        Dir.chdir repo_path do
+          git("checkout", "-q", ref)
+        end
+      end
+
+      def rev_parse(repo_path)
+        Dir.chdir repo_path do
+          git("rev-parse", "HEAD")
+        end
+      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
+          potential_path = File.join(path, 'git.exe')
+          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
+
+      private
+
+        def git_cmd
+          @git_cmd ||= find_git
+        end
+
+        def error_check
+          raise Berkshelf::GitError, "Did not succeed executing git; check the output above." unless $?.success?
+        end
+    end
+  end
+end

data/lib/berkshelf/init_generator.rb

@@ -0,0 +1,38 @@
+require 'thor/group'
+
+module Berkshelf
+  # @author Jamie Winsor <jamie@vialstudios.com>
+  class InitGenerator < Thor::Group
+    include Thor::Actions
+
+    class_option :path,
+      :type => :string,
+      :required => true
+
+    class_option :metadata_entry,
+      :type => :boolean,
+      :default => false
+
+    class_option :chefignore,
+      :type => :boolean,
+      :default => false
+
+    def self.source_root
+      File.expand_path(File.join(File.dirname(__FILE__), "generator_files"))
+    end
+
+    def generate
+      template "Berksfile.erb", File.join(target_path, "Berksfile")
+
+      if options[:chefignore]
+        copy_file "chefignore", File.join(target_path, ".chefignore")
+      end
+    end
+
+    private
+
+      def target_path
+        @target_path ||= File.expand_path(options[:path])
+      end
+  end
+end

data/lib/berkshelf/lockfile.rb

@@ -0,0 +1,42 @@
+module Berkshelf
+  class Lockfile
+    class << self
+      def remove!
+        FileUtils.rm_f DEFAULT_FILENAME
+      end
+    end
+
+    DEFAULT_FILENAME = "#{Berkshelf::DEFAULT_FILENAME}.lock".freeze
+
+    attr_reader :sources
+
+    def initialize(sources)
+      @sources = Array(sources)
+    end
+
+    def write(filename = DEFAULT_FILENAME)
+      content = sources.map { |source| get_source_definition(source) }.join("\n")
+      File.open(filename, "wb") { |f| f.write content }
+    end
+
+    def remove!
+      self.class.remove!
+    end
+
+    private
+
+      def get_source_definition(source)
+        definition = "cookbook '#{source.name}'"
+
+        if source.location.is_a?(CookbookSource::GitLocation)
+          definition += ", :git => '#{source.location.uri}', :ref => '#{source.location.branch || 'HEAD'}'"
+        elsif source.location.is_a?(CookbookSource::PathLocation)
+          definition += ", :path => '#{source.location.path}'"
+        else
+          definition += ", :locked_version => '#{source.locked_version}'"
+        end
+
+        return definition
+      end
+  end
+end

data/lib/berkshelf/resolver.rb

@@ -0,0 +1,176 @@
+module Berkshelf
+  # @author Jamie Winsor <jamie@vialstudios.com>
+  class Resolver
+    extend Forwardable
+    include DepSelector
+
+    def_delegators :@graph, :package, :packages
+
+    def initialize(downloader, sources = Array.new)
+      @downloader = downloader
+      @graph = DependencyGraph.new
+      @sources = Hash.new
+
+      # Dependencies need to be added AFTER the sources. If they are
+      # not, then one of the dependencies of a source that is added
+      # may take precedence over an explicitly set source that appears
+      # later in the iterator.
+      Array(sources).each do |source|
+        add_source(source, false)
+      end
+      add_sources_dependencies
+    end
+
+    # @param [Berkshelf::CookbookSource] source
+    #   source to add
+    # @param [Boolean] include_dependencies
+    #   if true, after adding the source the dependencies defined in the
+    #   sources metadata will be added to the graph and downloaded
+    #
+    # @return [DepSelector::PackageVersion]
+    def add_source(source, include_dependencies = true)
+      raise DuplicateSourceDefined if has_source?(source.name)
+
+      set_source(source.name, source)
+
+      install_or_use_source(source)
+
+      package = add_package(source.name)
+      package_version = add_version(package, Version.new(source.metadata.version))
+      
+      add_dependencies(package_version, source.dependencies) if include_dependencies
+
+      package_version
+    end
+
+    # @param [DepSelector::PackageVersion] parent_pkgver
+    #   the PackageVersion you would like to add the given dependencies to. In this case
+    #   the package version is a version of a Cookbook.
+    # @param [Hash] dependencies
+    #   A hash containing Cookbook names for keys and version constraint strings for
+    #   values. This is the same format obtained by sending the 'dependencies' message
+    #   to an instance of Chef::Cookbook::Metadata.
+    #
+    #   Example:
+    #       { 
+    #         "build-essential" => ">= 0.0.0",
+    #         "ohai" => "~> 1.0.2"
+    #       }
+    def add_dependencies(parent_pkgver, dependencies)
+      dependencies.each do |name, constraint|
+        dep_package = add_package(name)
+        parent_pkgver.dependencies << Dependency.new(dep_package, VersionConstraint.new(constraint))
+
+        unless has_source?(name)
+          source = CookbookSource.new(name, constraint)
+
+          install_or_use_source(source)
+
+          dep_pkgver = add_version(dep_package, Version.new(source.metadata.version))
+          add_dependencies(dep_pkgver, source.dependencies)
+          set_source(source.name, source)
+        end
+      end
+    end
+
+    # @return [Array<Berkshelf::CookbookSource>]
+    #   an array of CookbookSources that are currently added to this resolver
+    def sources
+      @sources.collect { |name, source| source }
+    end
+
+    # @return [Hash]
+    #   a hash containing package names - in this case Cookbook names - as keys and
+    #   their locked version as values.
+    #
+    #   Example:
+    #       { 
+    #         "nginx" => 0.101.0,
+    #         "build-essential" => 1.0.2,
+    #         "runit" => 0.15.0,
+    #         "bluepill" => 1.0.4,
+    #         "ohai" => 1.0.2
+    #       }
+    def resolve
+      quietly { selector.find_solution(solution_constraints) }
+    end
+
+    # @param [#to_s] source
+    #   name of the source to return
+    #
+    # @return [Berkshelf::CookbookSource]
+    def [](source)
+      @sources[source.to_s]
+    end
+    alias_method :get_source, :[]
+
+    # @param [#to_s] source
+    #   the source to test if the resolver has added
+    def has_source?(source)
+      !get_source(source).nil?
+    end
+
+    private
+
+      attr_reader :downloader
+      attr_reader :graph
+
+      # @param [#to_s] source
+      #   name of the source to set
+      # @param [CookbookSource] value
+      #   source to set as value
+      def []=(source, value)
+        @sources[source.to_s] = value
+      end
+      alias_method :set_source, :[]=
+
+      def install_or_use_source(source)
+        if downloader.downloaded?(source)
+          msg = "Using #{source.name} (#{source.metadata.version})"
+
+          if source.location.is_a?(CookbookSource::PathLocation)
+            msg << " at #{source.location}"  
+          end
+
+          Berkshelf.ui.info msg
+        else
+          downloader.download!(source)
+          Berkshelf.ui.info "Installing #{source.name} (#{source.local_version}) from #{source.location}"
+        end
+      end
+
+      def selector
+        Selector.new(graph)
+      end
+
+      def solution_constraints
+        constraints = graph.packages.collect do |name, package|
+          SolutionConstraint.new(package)
+        end
+      end
+
+      # Add the dependencies of each source to the graph
+      def add_sources_dependencies
+        sources.each do |source|
+          package_version = package(source.name)[Version.new(source.metadata.version)]
+          add_dependencies(package_version, source.dependencies)
+        end
+      end
+
+      # @param [String] name
+      #   name of the package to add to the graph
+      def add_package(name)
+        graph.package(name)
+      end
+
+      # Add a version to a package
+      #
+      # @param [DepSelector::Package] package
+      #   the package to add a version to
+      # @param [DepSelector::Version] version
+      #   the version to add the the package
+      def add_version(package, version)
+        package.add_version(version)
+      end
+  end
+end

data/lib/berkshelf/tx_result.rb

@@ -0,0 +1,12 @@
+module Berkshelf
+  # @author Jamie Winsor <jamie@vialstudios.com>
+  class TXResult < Struct.new(:status, :message, :source)
+    def failed?
+      status == :error
+    end
+
+    def success?
+      status == :ok
+    end
+  end
+end