knife-stats 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in knife-stats.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/knife-stats.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "knife-stats/version"
+
+Gem::Specification.new do |s|
+  s.name        = "knife-stats"
+  s.version     = Knife::Stats::VERSION
+  s.authors     = ["Stephen Nelson-Smith", "Fletcher Nichol"]
+  s.email       = ["support@atalanta-systems.com"]
+  s.homepage    = "https://github.com/Atalanta/knife-stats"
+  s.summary     = %q{Provides usage stats for Opscode Chef roles, cookbooks and environments}
+  s.description = %q{Provides usage stats for Opscode Chef roles, cookbooks and environments}
+
+  s.rubyforge_project = "knife-stats"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_runtime_dependency "chef", ">= 0.10.0"
+end

data/lib/chef/knife/cookbook_stats.rb

@@ -0,0 +1,80 @@
+require 'yaml'
+
+class Chef
+  class Knife
+    class CookbookStats < Chef::Knife
+  
+      banner "knife cookbook stats -E ENVIRONMENT [-r ROLE]"
+  
+      option :environment,
+        :short => "-E ENVIRONMENT",
+        :long => "--environment ENVIRONMENT",
+        :require => true
+    
+      option :role,
+        :short => "-r ROLE",
+        :long => "--role ROLE"
+  
+      deps do
+        require 'chef/shef/ext'
+      end
+  
+      def run
+        Shef::Extensions.extend_context_object(self)
+        validate_options
+  
+        cookbook_hash = build_cookbook_hash
+        cookbook_hash = filter_cookbook_hash(cookbook_hash) if config[:role]
+        
+        puts cookbook_hash.to_yaml
+      end
+  
+      private
+  
+      def validate_options
+        unless api.get("/environments").keys.include?(config[:environment])
+          ui.fatal "Environment #{config[:environment]} not found."
+          exit 1
+        end
+      
+        if config[:role] && !api.get("/roles").keys.include?(config[:role])
+          ui.fatal "Role #{config[:role]} not found."
+          exit 1
+        end
+      end
+  
+      def filter_cookbook_hash(cookbook_hash)
+        cookbooks_from_role = calculate_cookbooks_from_role
+        selected_cookbooks = cookbooks_from_role & cookbook_hash.keys
+        
+        filtered_cookbook_hash = Hash.new	
+        selected_cookbooks.each do |cb|
+          filtered_cookbook_hash[cb] = cookbook_hash[cb]
+        end
+        filtered_cookbook_hash
+      end
+  
+      def build_cookbook_hash
+        node_names = search(:node, "chef_environment:#{config[:environment]}").map { |n| n.name }
+        
+        cookbook = Hash.new(0)
+        node_names.each do |node|
+          node_cookbooks = api.get("/nodes/#{node}/cookbooks").keys
+          node_cookbooks.each { |cb| cookbook[cb] += 1 }
+        end
+        cookbook
+      end
+      
+      def calculate_cookbooks_from_role
+        rl = Chef::RunList.new("role[#{config[:role]}]")
+        
+        recipes = rl.expand(Chef::Environment.load(config[:environment])).recipes
+        
+        cookbooks = recipes.map do |recipe|
+          Chef::Recipe.parse_recipe_name(recipe).first
+        end
+        cookbooks.uniq!.map { |c| c.to_s }
+      end
+    end
+  end
+end

data/lib/chef/knife/environment_stats.rb

@@ -0,0 +1,27 @@
+require 'yaml'
+
+class Chef
+  class Knife
+    class EnvironmentStats < Chef::Knife
+  
+      banner "knife environment stats"
+  
+      deps do
+        require 'chef/shef/ext'
+      end
+  
+      def run
+        Shef::Extensions.extend_context_object(self)
+  
+        environment_names = api.get("/environments").keys
+        
+        environment = Hash.new
+        environment_names.sort.each do |env|
+          environment[env] = search(:node, "chef_environment:#{env}").size
+        end
+        
+        puts environment.to_yaml
+      end
+    end
+  end
+end

data/lib/chef/knife/role_stats.rb

@@ -0,0 +1,50 @@
+require 'yaml'
+
+class Chef
+  class Knife
+    class RoleStats < Chef::Knife
+  
+      banner "knife role stats [-E ENVIRONMENT]"
+  
+      option :environment,
+        :short => "-E ENVIRONMENT",
+        :long => "--environment ENVIRONMENT"
+    
+      deps do
+        require 'chef/shef/ext'
+      end
+  
+      def run
+        Shef::Extensions.extend_context_object(self)
+        validate_options
+  
+        role_names = api.get("/roles").keys
+        
+        role = Hash.new
+        role_names.sort.each do |r|
+          query = "roles:#{r}#{environment_query}"
+          role[r] = search(:node, query).size
+        end
+        
+        puts role.reject! { |k,v| v == 0 }.to_yaml
+      end
+  
+      private
+  
+      def validate_options
+        if config[:environment] && !api.get("/environments").keys.include?(config[:environment])
+          ui.fatal "Environment #{config[:environment]} not found."
+          exit 1
+        end
+      end
+  
+      def environment_query
+        if config[:environment]
+          " AND chef_environment:#{config[:environment]}"
+        else
+          ""
+        end
+      end
+    end
+  end
+end

data/lib/knife-stats.rb

@@ -0,0 +1,5 @@
+require "knife-stats/version"
+
+require "chef/knife/environemt_stats"
+require "chef/knife/role_stats"
+require "chef/knife/coobook_stats"

data/lib/knife-stats/version.rb

@@ -0,0 +1,5 @@
+module Knife
+  module Stats
+    VERSION = "0.0.1"
+  end
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification
+name: knife-stats
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Stephen Nelson-Smith
+- Fletcher Nichol
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-11-12 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: chef
+  requirement: &70268480 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70268480
+description: Provides usage stats for Opscode Chef roles, cookbooks and environments
+email:
+- support@atalanta-systems.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- knife-stats.gemspec
+- lib/chef/knife/cookbook_stats.rb
+- lib/chef/knife/environment_stats.rb
+- lib/chef/knife/role_stats.rb
+- lib/knife-stats.rb
+- lib/knife-stats/version.rb
+homepage: https://github.com/Atalanta/knife-stats
+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: knife-stats
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Provides usage stats for Opscode Chef roles, cookbooks and environments
+test_files: []