flock_of_chefs 0.0.1

data/flock_of_chefs.gemspec

@@ -0,0 +1,18 @@
+$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__)) + '/lib/'
+require 'flock_of_chefs/version'
+
+Gem::Specification.new do |s|
+  s.name = 'flock_of_chefs'
+  s.version = FlockOfChefs::VERSION.version
+  s.summary = 'Flock of chefs'
+  s.author = 'Chris Roberts'
+  s.email = 'chrisroberts.code@gmail.com'
+  s.homepage = 'http://github.com/chrisroberts/flock_of_chefs'
+  s.description = 'Chefs flocking about'
+  s.require_path = 'lib'
+  s.has_rdoc = true
+  s.extra_rdoc_files = ['README.md']  
+  s.files = Dir['**/*']
+  s.add_dependency 'chef'
+  s.add_dependency 'dcell', '~> 0.10.0'
+end

data/lib/flock_of_chefs.rb

@@ -0,0 +1,2 @@
+require 'flock_of_chefs/version'
+require 'flock_of_chefs/loader'

data/lib/flock_of_chefs/flocked_api.rb

@@ -0,0 +1,13 @@
+module FlockOfChefs
+  class FlockedApi
+    include Celluloid
+    attr_accessor :chef_app
+    attr_accessor :node
+
+    def run_chef!
+      chef_app.run_chef_client
+    end
+  end
+end
+
+FlockOfChefs::FlockedApi.supervise_as :flock_api

data/lib/flock_of_chefs/flocked_chef.rb

@@ -0,0 +1,93 @@
+require 'thread'
+# This is for hacking up chef internals
+# NOTE: Currently this is only mucking around with
+#   Chef::Application::Client. Hopefully CHEF-3478
+#   will get merged and make it much easier to add
+#   the custom functionality we want
+module FlockOfChefs
+  module FlockedChef
+    def mutex
+      unless(@mutex)
+        @mutex = Mutex.new
+      end
+      @mutex
+    end
+    def run_chef_client
+      puts '*' * 400
+      mutex.synchronize do
+        @chef_client = Chef::Client.new(
+          @chef_client_json, 
+          :override_runlist => config[:override_runlist]
+        )
+        @chef_client_json = nil
+        @chef_client.run
+        @chef_client = nil
+      end
+    end
+
+    def run_application
+      unless Chef::Platform.windows?
+        SELF_PIPE.replace IO.pipe
+
+        trap("USR1") do
+          Chef::Log.info("SIGUSR1 received, waking up")
+          SELF_PIPE[1].putc('.') # wakeup master process from select
+        end
+      end
+
+      if Chef::Config[:version]
+        puts "Chef version: #{::Chef::VERSION}"
+      end
+
+      if Chef::Config[:daemonize]
+        Chef::Daemon.daemonize("chef-client")
+      end
+
+      loop do
+        begin
+          if Chef::Config[:splay]
+            splay = rand Chef::Config[:splay]
+            Chef::Log.debug("Splay sleep #{splay} seconds")
+            sleep splay
+          end
+          run_chef_client
+          if Chef::Config[:interval]
+            Chef::Log.debug("Sleeping for #{Chef::Config[:interval]} seconds")
+            unless SELF_PIPE.empty?
+              client_sleep Chef::Config[:interval]
+            else
+              # Windows
+              sleep Chef::Config[:interval]
+            end
+          else
+            Chef::Application.exit! "Exiting", 0
+          end
+        rescue Chef::Application::Wakeup => e
+          Chef::Log.debug("Received Wakeup signal.  Starting run.")
+          next
+        rescue SystemExit => e
+          raise
+        rescue Exception => e
+          if Chef::Config[:interval]
+            Chef::Log.error("#{e.class}: #{e}")
+            Chef::Application.debug_stacktrace(e)
+            Chef::Log.error("Sleeping for #{Chef::Config[:interval]} seconds before trying again")
+            unless SELF_PIPE.empty?
+              client_sleep Chef::Config[:interval]
+            else
+              # Windows
+              sleep Chef::Config[:interval]
+            end
+            retry
+          else
+            Chef::Application.debug_stacktrace(e)
+            Chef::Application.fatal!("#{e.class}: #{e.message}", 1)
+          end
+        end
+      end
+    end
+  end
+end
+
+puts '***** HOOKING IN BITS'
+Chef::Application::Client.send(:include, FlockOfChefs::FlockedChef)

data/lib/flock_of_chefs/flocked_report.rb

@@ -0,0 +1,52 @@
+require 'chef/handler'
+
+module FlockOfChefs
+  class FlockedReport < Chef::Handler
+    def report
+      if(DCell.me.nil?)
+        setup_dcell!
+      end
+      dnode = DCell.me
+      if(dnode)
+        dnode[:flock_api].node = node.to_hash
+        Chef::Log.info 'Node information successfully stored in flock'
+      else
+        Chef::Log.warn 'Failed to store node information in flock!'
+      end
+    end
+
+    def setup_dcell!
+      raise 'Flock of chefs cookbook not in use!' unless node[:flock_of_chefs]
+      if(node.recipes.include?('flock_of_chefs::keeper'))
+        # NOTE: just gossip for now, zookeeper later
+        start_args = {
+          :node => node.name,
+          :addr => "tcp://#{node[:ipaddress]}:#{node[:flock_of_chefs][:port]}"
+        }
+        case node[:flock_of_chefs][:registry][:type].to_s
+        when 'zk'
+          require 'dcell/registries/zk'
+          start_args.merge(
+            :registry => {
+              :adapter => 'zk',
+              :host => '127.0.0.1', # allow non-local?
+              :port => 2181
+            }
+          )
+        end
+        DCell.start(start_args)
+      else
+        keeper_node = Chef::Search::Query.new.search(:node, 'recipes:flock_of_chefs\:\:keeper').first.first
+        raise 'Failed to find flock keeper!' unless keeper_node
+        DCell.start(
+          :node => node.name,
+          :addr => "tcp://#{node[:ipaddress]}:#{node[:flock_of_chefs][:port]}",
+          :directory => {
+            :id => keeper_node.name,
+            :address => "tcp://#{keeper_node[:ipaddress]}:#{keeper_node[:flock_of_chefs][:port]}"
+          }
+        )
+      end
+    end
+  end
+end

data/lib/flock_of_chefs/loader.rb

@@ -0,0 +1,8 @@
+require 'dcell'
+require 'flock_of_chefs/flocked_api'
+require 'flock_of_chefs/flocked_chef'
+require 'flock_of_chefs/flocked_report'
+
+# Hook in our report handler
+Chef::Config.report_handlers << FlockOfChefs::FlockedReport.new
+

data/lib/flock_of_chefs/version.rb

@@ -0,0 +1,5 @@
+module FlockOfChefs
+  class Version < Gem::Version
+  end
+  VERSION = Version.new('0.0.1')
+end

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: flock_of_chefs
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Chris Roberts
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: chef
+  requirement: &8367760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *8367760
+- !ruby/object:Gem::Dependency
+  name: dcell
+  requirement: &8366980 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.10.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *8366980
+description: Chefs flocking about
+email: chrisroberts.code@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- README.md
+files:
+- CHANGELOG.md
+- README.md
+- flock_of_chefs.gemspec
+- lib/flock_of_chefs.rb
+- lib/flock_of_chefs/loader.rb
+- lib/flock_of_chefs/flocked_report.rb
+- lib/flock_of_chefs/flocked_chef.rb
+- lib/flock_of_chefs/version.rb
+- lib/flock_of_chefs/flocked_api.rb
+- flock_of_chefs-0.0.1.gem
+homepage: http://github.com/chrisroberts/flock_of_chefs
+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.17
+signing_key: 
+specification_version: 3
+summary: Flock of chefs
+test_files: []
+has_rdoc: true