health_inspector 0.0.3 → 0.0.4

data/HISTORY.md

@@ -1,11 +1,15 @@
-## 0.0.3 ( 2012-03-27)
+## 0.0.4 ( 2012-04-09 )
+
+* Add checks for data bags, data bag items, environments, and roles
+
+## 0.0.3 ( 2012-03-27 )
 
 * Read cookbook paths from knife config file instead of hardcoding /cookbooks
 
-## 0.0.2 ( 2012-03-27)
+## 0.0.2 ( 2012-03-27 )
 
 * Make sure we iterate over actual cookbooks in cookbooks folder
 
-## 0.0.1 ( 2012-03-27)
+## 0.0.1 ( 2012-03-27 )
 
 * Initial release

data/lib/health_inspector.rb

@@ -6,11 +6,12 @@ require "health_inspector/check"
 require "health_inspector/inspector"
 require "health_inspector/checklists/base"
 require "health_inspector/checklists/cookbooks"
+require "health_inspector/checklists/data_bags"
+require "health_inspector/checklists/data_bag_items"
+require "health_inspector/checklists/environments"
+require "health_inspector/checklists/roles"
 require "health_inspector/cli"
 require "json"
 
-# TODO:
-# * Check if cookbook version is same as on server
-# * Check if remote origin not in sync with local
 module HealthInspector
 end

data/lib/health_inspector/checklists/base.rb

@@ -5,6 +5,66 @@ module HealthInspector
     class Base
       include Color
 
+      class << self
+        attr_reader :checks, :title
+
+        def add_check(name, &block)
+          @checks ||= []
+          @checks << block
+        end
+
+        def title(val=nil)
+          val.nil? ? @title : @title = val
+        end
+      end
+
+      def self.run(context)
+        new(context).run
+      end
+
+      def initialize(context)
+        @context = context
+      end
+
+      def run
+        banner "Inspecting #{self.class.title}"
+
+        items.each do |item|
+          failures = checks.map { |check| run_check(check, item) }.compact
+
+          if failures.empty?
+            print_success(item.name)
+          else
+            print_failures(item.name, failures)
+          end
+        end
+      end
+
+      def checks
+        self.class.checks
+      end
+
+      class CheckContext
+        include Check
+        attr_accessor :item, :context
+
+        def initialize(check, item, context)
+          @item = item
+          @context = context
+          @check = check
+        end
+
+        def call
+          instance_eval(&@check)
+        end
+      end
+
+      def run_check(check, item)
+        check_context = CheckContext.new(check, item, @context)
+        check_context.call
+        return check_context.failure
+      end
+
       def banner(message)
         puts
         puts message

data/lib/health_inspector/checklists/cookbooks.rb

@@ -1,84 +1,52 @@
 module HealthInspector
   module Checklists
 
-    class Cookbook < Struct.new(:name, :path, :server_version, :local_version)
-      def git_repo?
-        self.path && File.exist?("#{self.path}/.git")
-      end
-    end
-
-    class CookbookCheck < Struct.new(:cookbook)
-      include Check
-    end
+    class Cookbooks < Base
 
-    class CheckLocalCopyExists < CookbookCheck
-      def check
-        failure( "exists on chef server but not locally" ) if cookbook.path.nil?
+      add_check "local copy exists" do
+        failure( "exists on chef server but not locally" ) if item.path.nil?
       end
-    end
 
-    class CheckServerCopyExists < CookbookCheck
-      def check
-        failure( "exists locally but not on chef server" ) if cookbook.server_version.nil?
+      add_check "server copy exists" do
+        failure( "exists locally but not on chef server" ) if item.server_version.nil?
       end
-    end
 
-    class CheckVersionsAreTheSame < CookbookCheck
-      def check
-        if cookbook.local_version && cookbook.server_version &&
-           cookbook.local_version != cookbook.server_version
-          failure "chef server has #{cookbook.server_version} but local version is #{cookbook.local_version}"
+      add_check "versions" do
+        if item.local_version && item.server_version &&
+           item.local_version != item.server_version
+          failure "chef server has #{item.server_version} but local version is #{item.local_version}"
         end
       end
-    end
 
-    class CheckUncommittedChanges < CookbookCheck
-      def check
-        if cookbook.git_repo?
-          result = `cd #{cookbook.path} && git status -s`
+      add_check "uncommitted changes" do
+        if item.git_repo?
+          result = `cd #{item.path} && git status -s`
 
           unless result.empty?
             failure "Uncommitted changes:\n#{result.chomp}"
           end
         end
       end
-    end
 
-    class CheckCommitsNotPushedToRemote < CookbookCheck
-      def check
-        if cookbook.git_repo?
-          result = `cd #{cookbook.path} && git status`
+      add_check "commits not pushed to remote" do
+        if item.git_repo?
+          result = `cd #{item.path} && git status`
 
           if result =~ /Your branch is ahead of (.+)/
             failure "ahead of #{$1}"
           end
         end
       end
-    end
-
-    class Cookbooks < Base
-      def self.run(context)
-        new(context).run
-      end
-
-      def initialize(context)
-        @context = context
-      end
 
-      def run
-        banner "Inspecting cookbooks"
-
-        cookbooks.each do |cookbook|
-          failures = cookbook_checks.map { |c| c.new(cookbook).run_check }.compact
-          if failures.empty?
-            print_success(cookbook.name)
-          else
-            print_failures(cookbook.name, failures)
-          end
+      class Cookbook < Struct.new(:name, :path, :server_version, :local_version)
+        def git_repo?
+          self.path && File.exist?("#{self.path}/.git")
         end
       end
 
-      def cookbooks
+      title "cookbooks"
+
+      def items
         server_cookbooks   = cookbooks_on_server
         local_cookbooks    = cookbooks_in_repo
         all_cookbook_names = ( server_cookbooks.keys + local_cookbooks.keys ).uniq.sort
@@ -120,16 +88,6 @@ module HealthInspector
         path = @context.cookbook_path.find { |f| File.exist?("#{f}/#{name}") }
         path ? File.join(path, name) : nil
       end
-
-      def cookbook_checks
-        [
-          CheckLocalCopyExists,
-          CheckServerCopyExists,
-          CheckVersionsAreTheSame,
-          CheckUncommittedChanges,
-          CheckCommitsNotPushedToRemote
-        ]
-      end
     end
   end
 end

data/lib/health_inspector/checklists/data_bag_items.rb

@@ -0,0 +1,71 @@
+require "chef/data_bag"
+
+module HealthInspector
+  module Checklists
+    class DataBagItems < Base
+      title "data bag items"
+
+      add_check "local copy exists" do
+        failure "exists on server but not locally" unless item.local
+      end
+
+      add_check "server copy exists" do
+        failure "exists locally but not on server" unless item.server
+      end
+
+      add_check "items are the same" do
+        if item.server && item.local
+          failure "#{item.server}\n  is not equal to\n  #{item.local}" unless item.server == item.local
+        end
+      end
+
+      DataBagItem = Struct.new(:name, :server, :local)
+
+      def items
+        server_data_bag_items   = data_bag_items_on_server
+        local_data_bag_items    = data_bag_items_in_repo
+        all_data_bag_item_names = ( server_data_bag_items + local_data_bag_items ).uniq.sort
+
+        all_data_bag_item_names.map do |name|
+          DataBagItem.new.tap do |data_bag_item|
+            data_bag_item.name   = name
+            data_bag_item.server = load_item_from_server(name)
+            data_bag_item.local  = load_item_from_local(name)
+          end
+        end
+      end
+
+      def data_bag_items_on_server
+        @data_bags_on_server ||= Chef::DataBag.list.keys.map do |bag_name|
+          [ bag_name, Chef::DataBag.load(bag_name) ]
+        end.inject([]) do |arr, (bag_name, data_bag)|
+          arr += data_bag.keys.map { |item_name| "#{bag_name}/#{item_name}"}
+        end
+      end
+
+      def data_bag_items_in_repo
+        entries = nil
+
+        Dir.chdir("#{@context.repo_path}/data_bags") do
+          entries = Dir["**/*.json"].map { |entry| entry.gsub('.json', '') }
+        end
+
+        return entries
+      end
+
+      def load_item_from_server(name)
+        bag_name, item_name = name.split("/")
+        Chef::DataBagItem.load(bag_name, item_name).raw_data
+      rescue
+        nil
+      end
+
+      def load_item_from_local(name)
+        JSON.parse( File.read("#{@context.repo_path}/data_bags/#{name}.json") )
+      rescue IOError
+        nil
+      end
+    end
+
+  end
+end

data/lib/health_inspector/checklists/data_bags.rb

@@ -0,0 +1,44 @@
+require "chef/data_bag"
+
+module HealthInspector
+  module Checklists
+    class DataBags < Base
+      title "data bags"
+
+      add_check "local copy exists" do
+        failure "exists on server but not locally" unless item.exists_locally
+      end
+
+      add_check "server copy exists" do
+        failure "exists locally but not on server" unless item.exists_on_server
+      end
+
+      DataBag = Struct.new(:name, :exists_on_server, :exists_locally)
+
+      def items
+        server_data_bags   = data_bags_on_server
+        local_data_bags    = data_bags_in_repo
+        all_data_bag_names = ( server_data_bags + local_data_bags ).uniq.sort
+
+        all_data_bag_names.map do |name|
+          DataBag.new.tap do |data_bag|
+            data_bag.name = name
+            data_bag.exists_on_server = data_bags_on_server.include?(name)
+            data_bag.exists_locally = data_bags_in_repo.include?(name)
+          end
+        end
+      end
+
+      def data_bags_on_server
+        @data_bags_on_server ||= Chef::DataBag.list.keys
+      end
+
+      def data_bags_in_repo
+        @data_bags_in_repo ||= Dir["#{@context.repo_path}/data_bags/*"].entries.
+          select { |e| File.directory?(e) }.
+          map { |e| File.basename(e) }
+      end
+    end
+
+  end
+end

data/lib/health_inspector/checklists/environments.rb

@@ -0,0 +1,65 @@
+require "chef/environment"
+
+module HealthInspector
+  module Checklists
+    class Environments < Base
+      title "environments"
+
+      add_check "local copy exists" do
+        failure "exists on server but not locally" unless item.local
+      end
+
+      add_check "server copy exists" do
+        failure "exists locally but not on server" unless item.server
+      end
+
+      add_check "items are the same" do
+        if item.server && item.local
+          failure "#{item.server}\n  is not equal to\n  #{item.local}" unless item.server == item.local
+        end
+      end
+
+      Environment = Struct.new(:name, :server, :local)
+
+      def items
+        server_items   = items_on_server
+        local_items    = items_in_repo
+        all_item_names = ( server_items + local_items ).uniq.sort
+
+        all_item_names.map do |name|
+          Environment.new.tap do |item|
+            item.name   = name
+            item.server = load_item_from_server(name)
+            item.local  = load_item_from_local(name)
+          end
+        end
+      end
+
+      def items_on_server
+        @items_on_server ||= Chef::Environment.list.keys
+      end
+
+      def items_in_repo
+        Dir.chdir("#{@context.repo_path}/environments") do
+          Dir["*.rb"].map { |e| e.gsub('.rb', '') }
+        end
+      end
+
+      def load_item_from_server(name)
+        env = Chef::Environment.load(name)
+        env.to_hash
+      rescue
+        nil
+      end
+
+      def load_item_from_local(name)
+        env = Chef::Environment.new
+        env.from_file( "#{@context.repo_path}/environments/#{name}.rb" )
+        env.to_hash
+      rescue IOError
+        nil
+      end
+    end
+
+  end
+end

data/lib/health_inspector/checklists/roles.rb

@@ -0,0 +1,65 @@
+require "chef/role"
+
+module HealthInspector
+  module Checklists
+    class Roles < Base
+      title "roles"
+
+      add_check "local copy exists" do
+        failure "exists on server but not locally" unless item.local
+      end
+
+      add_check "server copy exists" do
+        failure "exists locally but not on server" unless item.server
+      end
+
+      add_check "items are the same" do
+        if item.server && item.local
+          failure "#{item.server}\n  is not equal to\n  #{item.local}" unless item.server == item.local
+        end
+      end
+
+      Role = Struct.new(:name, :server, :local)
+
+      def items
+        server_items   = items_on_server
+        local_items    = items_in_repo
+        all_item_names = ( server_items + local_items ).uniq.sort
+
+        all_item_names.map do |name|
+          Role.new.tap do |item|
+            item.name   = name
+            item.server = load_item_from_server(name)
+            item.local  = load_item_from_local(name)
+          end
+        end
+      end
+
+      def items_on_server
+        @items_on_server ||= Chef::Role.list.keys
+      end
+
+      def items_in_repo
+        Dir.chdir("#{@context.repo_path}/roles") do
+          Dir["*.rb"].map { |e| e.gsub('.rb', '') }
+        end
+      end
+
+      def load_item_from_server(name)
+        role = Chef::Role.load(name)
+        role.to_hash
+      rescue
+        nil
+      end
+
+      def load_item_from_local(name)
+        role = Chef::Role.new
+        role.from_file( "#{@context.repo_path}/roles/#{name}.rb" )
+        role.to_hash
+      rescue IOError
+        nil
+      end
+    end
+
+  end
+end

data/lib/health_inspector/context.rb

@@ -11,10 +11,12 @@ module HealthInspector
     end
 
     def config
-      @config ||= begin
-        Chef::Config.from_file(config_path)
-        Chef::Config
-      end
+      @config ||= configure
+    end
+
+    def configure
+      Chef::Config.from_file(config_path)
+      Chef::Config
     end
   end
 end

data/lib/health_inspector/inspector.rb

@@ -6,10 +6,15 @@ module HealthInspector
 
     def initialize(repo_path, config_path)
       @context = Context.new( repo_path, File.join(repo_path, config_path) )
+      @context.configure
     end
 
     def inspect
       Checklists::Cookbooks.run(@context)
+      Checklists::DataBags.run(@context)
+      Checklists::DataBagItems.run(@context)
+      Checklists::Environments.run(@context)
+      Checklists::Roles.run(@context)
     end
   end
 end

data/lib/health_inspector/version.rb

@@ -1,3 +1,3 @@
 module HealthInspector
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: health_inspector
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-03-27 00:00:00.000000000Z
+date: 2012-04-09 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
-  requirement: &2164944480 !ruby/object:Gem::Requirement
+  requirement: &2168869000 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2164944480
+  version_requirements: *2168869000
 - !ruby/object:Gem::Dependency
   name: chef
-  requirement: &2164944060 !ruby/object:Gem::Requirement
+  requirement: &2168868580 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2164944060
+  version_requirements: *2168868580
 description: A tool to inspect your chef repo as is compares to what is on your chef
   server
 email:
@@ -54,6 +54,10 @@ files:
 - lib/health_inspector/check.rb
 - lib/health_inspector/checklists/base.rb
 - lib/health_inspector/checklists/cookbooks.rb
+- lib/health_inspector/checklists/data_bag_items.rb
+- lib/health_inspector/checklists/data_bags.rb
+- lib/health_inspector/checklists/environments.rb
+- lib/health_inspector/checklists/roles.rb
 - lib/health_inspector/cli.rb
 - lib/health_inspector/color.rb
 - lib/health_inspector/context.rb