chef-dk 0.2.0

data/lib/chef-dk/cookbook_metadata.rb

@@ -0,0 +1,36 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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/metadata'
+
+module ChefDK
+
+  # Subclass of Chef's Cookbook::Metadata class that provides the API expected
+  # by CookbookOmnifetch
+  class CookbookMetadata < Chef::Cookbook::Metadata
+
+    def self.from_path(path)
+      metadata_rb_path = File.join(path, "metadata.rb")
+      new.tap { |m| m.from_file(metadata_rb_path) }
+    end
+
+    def cookbook_name
+      name
+    end
+
+  end
+end

data/lib/chef-dk/cookbook_omnifetch.rb

@@ -0,0 +1,29 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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 'cookbook-omnifetch'
+require 'chef-dk/shell_out'
+require 'chef-dk/cookbook_metadata'
+
+# Configure CookbookOmnifetch's dependency injection settings to use our classes and config.
+CookbookOmnifetch.configure do |c|
+  c.cache_path = File.expand_path('~/.chefdk/cache')
+  c.storage_path = Pathname.new(File.expand_path('~/.chefdk/cache/cookbooks'))
+  c.shell_out_class = ChefDK::ShellOut
+  c.cached_cookbook_class = ChefDK::CookbookMetadata
+end
+

data/lib/chef-dk/cookbook_profiler/git.rb

@@ -0,0 +1,95 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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-dk/helpers'
+
+module ChefDK
+  module CookbookProfiler
+    class Git
+
+      include Helpers
+
+      attr_reader :cookbook_path
+
+      def initialize(cookbook_path)
+        @cookbook_path = cookbook_path
+      end
+
+      def profile_data
+        {
+          "scm" => "git",
+          # To get this info, you need to do something like:
+          # figure out branch or assume 'master'
+          # git config --get branch.master.remote
+          # git config --get remote.opscode.url
+          "remote" => remote,
+          "revision" => revision,
+          "working_tree_clean" => clean?,
+          "published" => !unpublished_commits?,
+          "synchronized_remote_branches" => synchronized_remotes
+        }
+      end
+
+      def revision
+        git!("rev-parse HEAD").stdout.strip
+      end
+
+      def clean?
+        git!("diff-files --quiet", returns: [0,1]).exitstatus == 0
+      end
+
+      def unpublished_commits?
+        synchronized_remotes.empty?
+      end
+
+      def synchronized_remotes
+        @synchronized_remotes ||= git!("branch -r --contains #{revision}").stdout.lines.map(&:strip)
+      end
+
+      def remote
+        @remote_url ||=
+          if have_remote?
+            git!("config --get remote.#{remote_name}.url").stdout.strip
+          else
+            nil
+          end
+      end
+
+      def remote_name
+        @remote_name ||= git!("config --get branch.#{current_branch}.remote", returns: [0,1]).stdout.strip
+      end
+
+      def have_remote?
+        !remote_name.empty?
+      end
+
+      def current_branch
+        @current_branch ||= git!('rev-parse --abbrev-ref HEAD').stdout.strip
+      end
+
+      private
+
+      def git!(subcommand, options={})
+        options = { cwd: cookbook_path }.merge(options)
+        cmd = system_command("git #{subcommand}", options)
+        cmd.error!
+        cmd
+      end
+
+    end
+  end
+end

data/lib/chef-dk/cookbook_profiler/identifiers.rb

@@ -0,0 +1,79 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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/cookbook_version_loader'
+require 'digest/sha1'
+require 'chef/digester'
+
+module ChefDK
+  module CookbookProfiler
+    class Identifiers
+
+      attr_reader :cookbook_path
+
+      def initialize(cookbook_path)
+        @cookbook_path = cookbook_path
+      end
+
+      def semver_version
+        cookbook_version.version
+      end
+
+      def content_identifier
+        Digest::SHA1.new.hexdigest(fingerprint_text)
+      end
+
+      def dotted_decimal_identifier
+        hex_id = content_identifier
+        major = hex_id[0...14]
+        minor = hex_id[14...28]
+        patch = hex_id[28..40]
+        decimal_integers =[major, minor, patch].map {|hex| hex.to_i(16) }
+        decimal_integers.join(".")
+      end
+
+      def fingerprint_text
+        files_with_checksums.sort {|a,b| a[0] <=> b[0]}.inject("") do |fingerprint, file_spec|
+          fingerprint << "#{file_spec[0]}:#{file_spec[1]}\n"
+        end
+      end
+
+      def files_with_checksums
+        cookbook_files.inject([]) do |files_with_checksums, (_name, file_info)|
+          files_with_checksums << [file_info["path"], file_info["checksum"]]
+        end
+      end
+
+      def cookbook_files
+        @files ||= cookbook_version.manifest_records_by_path
+      end
+
+      def cookbook_version
+        @cookbook_version ||= cookbook_loader.cookbook_version
+      end
+
+      def cookbook_loader
+        @cookbook_loader ||=
+          begin
+            loader = Chef::Cookbook::CookbookVersionLoader.new(cookbook_path)
+            loader.load_cookbooks
+            loader
+          end
+      end
+    end
+  end
+end

data/lib/chef-dk/cookbook_profiler/null_scm.rb

@@ -0,0 +1,32 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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.
+#
+
+module ChefDK
+  module CookbookProfiler
+    class NullSCM
+
+      def initialize(cookbook_path)
+      end
+
+      def profile_data
+        nil
+      end
+
+    end
+  end
+end
+

data/lib/chef-dk/exceptions.rb

@@ -0,0 +1,46 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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.
+#
+
+module ChefDK
+
+  class CachedCookbookNotFound < StandardError
+  end
+
+  class LocalCookbookNotFound < StandardError
+  end
+
+  class MalformedCookbook < StandardError
+  end
+
+  class MissingComponentError < RuntimeError
+    def initialize(component_name)
+      super("Component #{component_name} is missing.")
+    end
+  end
+
+  class OmnibusInstallNotFound < RuntimeError
+    def initialize()
+      super("Can not find omnibus installation directory for Chef.")
+    end
+  end
+
+  class UnsupportedFeature < StandardError
+  end
+
+  class PolicyfileError < StandardError
+  end
+end

data/lib/chef-dk/generator.rb

@@ -0,0 +1,70 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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.
+#
+
+module ChefDK
+
+  module Generator
+
+    class Context
+
+      attr_accessor :have_git
+
+      attr_accessor :app_root
+      attr_accessor :cookbook_root
+      attr_accessor :app_name
+      attr_accessor :cookbook_name
+      attr_accessor :new_file_basename
+      attr_accessor :content_source
+      attr_accessor :author_name
+      attr_accessor :author_email
+      attr_accessor :license_description
+      attr_accessor :license_text
+
+      attr_accessor :skip_git_init
+    end
+
+    def self.reset
+      @context = nil
+    end
+
+    def self.context
+      @context ||= Context.new
+    end
+
+    module TemplateHelper
+
+      def self.delegate_to_app_context(name)
+        define_method(name) do
+          ChefDK::Generator.context.public_send(name)
+        end
+      end
+
+      # delegate all the attributes of app_config
+      delegate_to_app_context :app_root
+      delegate_to_app_context :cookbook_root
+      delegate_to_app_context :cookbook_name
+      delegate_to_app_context :new_file_basename
+      delegate_to_app_context :content_source
+      delegate_to_app_context :author_name
+      delegate_to_app_context :author_email
+      delegate_to_app_context :license_description
+      delegate_to_app_context :license_text
+
+    end
+
+  end
+end

data/lib/chef-dk/helpers.rb

@@ -0,0 +1,95 @@
+#
+# Copyright:: Copyright (c) 2014 Chef Software 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 'mixlib/shellout'
+require 'chef-dk/exceptions'
+
+module ChefDK
+  module Helpers
+
+    #
+    # Runs given commands using mixlib-shellout
+    #
+    def system_command(*command_args)
+      cmd = Mixlib::ShellOut.new(*command_args)
+      cmd.run_command
+      cmd
+    end
+
+    def err(message)
+      stderr.print("#{message}\n")
+    end
+
+    def msg(message)
+      stdout.print("#{message}\n")
+    end
+
+    def stdout
+      $stdout
+    end
+
+    def stderr
+      $stderr
+    end
+
+    #
+    # Locates the omnibus directories
+    #
+
+    def omnibus_root
+      @omnibus_root ||= omnibus_expand_path(Gem.ruby, "..", "..", "..")
+    end
+
+    def omnibus_apps_dir
+      @ominbus_apps_dir ||= omnibus_expand_path(omnibus_root, "embedded", "apps")
+    end
+
+    def omnibus_bin_dir
+      @omnibus_bin_dir ||= omnibus_expand_path(omnibus_root, "bin")
+    end
+
+    def omnibus_embedded_bin_dir
+      @omnibus_embedded_bin_dir ||= omnibus_expand_path(omnibus_root, "embedded", "bin")
+    end
+
+    private
+
+    def omnibus_expand_path(*paths)
+      dir = File.expand_path(File.join(paths))
+      raise OmnibusInstallNotFound.new() unless ( dir and File.directory?(dir) )
+      dir
+    end
+
+    #
+    # environment vars for omnibus
+    #
+
+    def omnibus_env
+      @omnibus_env ||=
+        begin
+          user_bin_dir = File.expand_path(File.join(Gem.user_dir, 'bin'))
+          {
+            'PATH' => "#{omnibus_bin_dir}:#{user_bin_dir}:#{omnibus_embedded_bin_dir}:#{ENV['PATH']}",
+            'GEM_ROOT' => Gem.default_dir.inspect,
+            'GEM_HOME' => Gem.paths.home,
+            'GEM_PATH' => Gem.path.join(':'),
+          }
+        end
+    end
+
+  end
+end