chef-etcd 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: c69d2d4c07991326c2d287ca17b05437eef96f29
+  data.tar.gz: ff90b8aacc2d44b871fb44326b08c082e81c8a59
+SHA512:
+  metadata.gz: 83dbf5d4c8382784d1a7ae4255461f14b2684c9710b3c1e85f01d8bef7517e77278638354636732b1ce473a3669442b93a264fe3b93aefe9ca2d68207047e6a6
+  data.tar.gz: 133d27451007a7c48ad99b3aa5fa2db93ede8959aaa27cecc0d552b2e9f84a32f6856f97ef5ac899eb11c09daf298acbe48d54829184e9973422956b85239fff

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Ranjib Dey
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,85 @@
+# Chef::Etcd
+
+Provides chef resource/provider to access etcd key value pairs and report/event handlers to publish chef run data into etcd.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'chef-etcd'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install chef-etcd
+
+## Usage
+
+1. ```require``` the chef-etcd library in your client config (knife.rb/solo.rb/client.rb) and set ```Chef::Config[:etcd]``` to an existing [etcd](https://github.com/coreos/etcd) node.
+
+```ruby
+require 'chef/etcd'
+node_name 'something'
+etcd_host = '192.168.122.1'
+Chef::Config[:etcd]= {host:etcd_host}
+```
+
+2. you can use the resource/provider to get/set etcd keys
+
+```ruby
+etcd "/test/recipe/set" do
+  action :set
+  value Time.now.to_s
+end
+
+etcd "/test/recipe/delete" do
+  action :set
+  value Time.now.to_s
+end
+
+etcd "/test/recipe/delete" do
+  action :delete
+  value Time.now.to_s
+end
+
+etcd "/test/recipe/test_and_set" do
+  action :set
+  value "0"
+end
+
+etcd "/test/recipe/test_and_set" do
+  action :test_and_set
+  value Time.now.to_s
+  prev_value "0"
+end
+
+etcd "/test/recipe/watch" do
+  action :set
+  value Time.now.to_s
+end
+
+Chef::Log.warn("This will halt the recipe, and you have to update the recipe from outside or do some thread foo here")
+etcd "/test/recipe/watch" do
+  action :watch
+  value Time.now.to_s
+end
+
+```
+3. To publish node data using the report handler, or to monitor realtime chef resource convergence(event handler), add following configurations to your client.rb/solo.rb
+
+```ruby
+event_handlers [Chef::EventDispatch::Etcd.new(host: etcd_host)]
+report_handlers << Chef::Handler::EtcdReport.new(host: etcd_host)
+
+```
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

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

data/chef-etcd.gemspec

@@ -0,0 +1,24 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'chef/etcd/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "chef-etcd"
+  spec.version       = Chef::Etcd::VERSION
+  spec.authors       = ["Ranjib Dey"]
+  spec.email         = ["dey.ranjib@gmail.com"]
+  spec.description   = %q{Etcd-Chef bindings}
+  spec.summary       = %q{Provides chef resource, provider, handlers for etcd}
+  spec.homepage      = "https://github.com/ranjib/chef-etcd"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "etcd"
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+end

data/lib/chef/etcd.rb

@@ -0,0 +1,5 @@
+require "chef/etcd/version"
+require 'chef/resource/etcd'
+require 'chef/provider/etcd'
+require 'chef/handler/etcd_report'
+require 'chef/handler/etcd_event'

data/lib/chef/etcd/version.rb

@@ -0,0 +1,5 @@
+module Chef
+  module Etcd
+    VERSION = "0.0.1"
+  end
+end

data/lib/chef/handler/etcd_event.rb

@@ -0,0 +1,253 @@
+require 'chef/event_dispatch/base'
+require 'etcd'
+
+class Chef
+  module EventDispatch
+    class Etcd < Chef::EventDispatch::Base
+
+      attr_reader :etcd
+
+      def initialize(options={})
+        etcd_host = options[:host] || '127.0.0.1'
+        etcd_port = options[:port] || 4001
+        @etcd = ::Etcd.client(host: etcd_host, port: etcd_port)
+        @global_ttl = options[:global_ttl] || 600
+        @ttls = options[:ttls] || {}
+      end
+
+      def get_ttl(m)
+        if @ttls.has_key?(m)
+          @ttls[m]
+        else
+          @global_ttl
+        end
+      end
+
+      def name_space 
+        '/nodes/' + Chef::Config[:node_name] + '/chef'
+      end
+
+      def etcd_set(m, k, v=nil) # m = method
+        key= name_space + k 
+        value = v || m
+        ttl = get_ttl(m)
+        etcd.set(key, value, ttl)
+      end
+
+      def set_sub_status(m)
+       etcd_set(m, '/sub_status')
+      end
+
+      def run_start(version)
+        etcd_set(__method__, '/status', 'running')
+        etcd_set(__method__, '/version', version)
+      end
+
+      def run_completed(node)
+        etcd_set(__method__, '/status', 'completed')
+      end
+
+      def run_failed(exception)
+        etcd_set(__method__, '/status', 'failed')
+      end
+
+      def run_started(run_status)
+        set_sub_status(__method__)
+      end
+
+
+      def ohai_completed(node)
+        set_sub_status(__method__)
+      end
+
+      def skipping_registration(node_name, config)
+        set_sub_status(__method__)
+      end
+
+      def registration_start(node_name, config)
+        set_sub_status(__method__)
+      end
+
+      def registration_completed
+        set_sub_status(__method__)
+      end
+
+      def registration_failed(node_name, exception, config)
+        set_sub_status(__method__)
+      end
+
+      def node_load_start(node_name, config)
+        set_sub_status(__method__)
+      end
+
+      def node_load_failed(node_name, exception, config)
+        set_sub_status(__method__)
+      end
+
+      def run_list_expand_failed(node, exception)
+        set_sub_status(__method__)
+      end
+
+      def node_load_completed(node, expanded_run_list, config)
+        set_sub_status(__method__)
+      end
+
+      def cookbook_resolution_start(expanded_run_list)
+        set_sub_status(__method__)
+      end
+
+      def cookbook_resolution_failed(expanded_run_list, exception)
+        set_sub_status(__method__)
+      end
+
+      def cookbook_resolution_complete(cookbook_collection)
+        set_sub_status(__method__)
+      end
+
+      def cookbook_clean_start
+        set_sub_status(__method__)
+      end
+
+      def cookbook_clean_complete
+        set_sub_status(__method__)
+      end
+
+      def cookbook_sync_start(cookbook_count)
+        set_sub_status(__method__)
+      end
+
+      def cookbook_sync_failed(cookbooks, exception)
+        set_sub_status(__method__)
+      end
+
+      def cookbook_sync_complete
+        set_sub_status(__method__)
+      end
+
+      def library_load_start(file_count)
+        set_sub_status(__method__)
+      end
+
+      def library_file_load_failed(path, exception)
+        set_sub_status(__method__)
+      end
+
+      def library_load_complete
+        set_sub_status(__method__)
+      end
+
+      def lwrp_load_start(lwrp_file_count)
+        set_sub_status(__method__)
+      end
+
+      def lwrp_file_load_failed(path, exception)
+        set_sub_status(__method__)
+      end
+
+      def lwrp_load_complete
+        set_sub_status(__method__)
+      end
+
+      def attribute_load_start(attribute_file_count)
+        set_sub_status(__method__)
+      end
+
+      def attribute_file_load_failed(path, exception)
+        set_sub_status(__method__)
+      end
+
+      def attribute_load_complete
+        set_sub_status(__method__)
+      end
+
+      def definition_load_start(definition_file_count)
+        set_sub_status(__method__)
+      end
+
+      def definition_file_load_failed(path, exception)
+        set_sub_status(__method__)
+      end
+
+      def definition_load_complete
+        set_sub_status(__method__)
+      end
+
+      def recipe_load_start(recipe_count)
+        set_sub_status(__method__)
+      end
+
+      def recipe_file_load_failed(path, exception)
+        set_sub_status(__method__)
+      end
+
+      def recipe_not_found(exception)
+        set_sub_status(__method__)
+      end
+
+      def recipe_load_complete
+        set_sub_status(__method__)
+      end
+
+      def converge_start(run_context)
+        set_sub_status(__method__)
+      end
+
+      def converge_complete
+        set_sub_status(__method__)
+      end
+
+      def resource_action_start(resource, action, notification_type=nil, notifier=nil)
+        etcd_set(__method__, '/resource_state', 'action_start')
+        etcd_set(__method__, '/resource_name', resource.to_s)
+        etcd_set(__method__, '/resource_action', action.to_s)
+      end
+
+      def resource_failed(resource, action, exception)
+        etcd_set(__method__, '/resource_state', 'failed')
+        etcd_set(__method__, '/resource_name', resource.to_s)
+        etcd_set(__method__, '/resource_action', action.to_s)
+      end
+
+      def resource_completed(resource)
+        etcd_set(__method__, '/resource_state', 'completed')
+        etcd_set(__method__, '/resource_name', resource.to_s)
+      end
+
+      def resource_skipped(resource, action, conditional)
+        etcd_set(__method__, '/resource_state', 'skipped')
+        etcd_set(__method__, '/resource_name', resource.to_s)
+        etcd_set(__method__, '/resource_action', action.to_s)
+      end
+
+      def resource_bypassed(resource, action, current_resource)
+        etcd_set(__method__, '/resource_state', 'bypassed')
+        etcd_set(__method__, '/resource_name', resource.to_s)
+        etcd_set(__method__, '/resource_action', action.to_s)
+      end
+
+      def resource_up_to_date(resource, action)
+        etcd_set(__method__, '/resource_state', 'converged')
+        etcd_set(__method__, '/resource_name', resource.to_s)
+        etcd_set(__method__, '/resource_action', action.to_s)
+      end
+
+
+      def resource_updated(resource, action)
+        etcd_set(__method__, '/resource_state', 'updated')
+        etcd_set(__method__, '/resource_name', resource.to_s)
+        etcd_set(__method__, '/resource_action', action.to_s)
+      end
+
+      def handlers_start(handler_count)
+        set_sub_status(__method__)
+      end
+
+      def handlers_completed
+        set_sub_status(__method__)
+      end
+
+      def msg(message)
+      end
+    end
+  end
+end

data/lib/chef/handler/etcd_report.rb

@@ -0,0 +1,23 @@
+require 'chef/handler'
+require 'etcd'
+require 'json'
+
+class Chef
+  class Handler
+    class EtcdReport < Chef::Handler
+
+      attr_reader :etcd
+
+      def initialize(options={})
+        @etcd = ::Etcd.client(options)
+      end
+
+      def report
+        if success?
+          data = {total: all_resources.size, updated: updated_resources.size, time: elapsed_time}
+          @etcd.set('/nodes/'+Chef::Config[:node_name]+'/chef/report',JSON.dump(data)
+        end
+      end
+    end
+  end
+end

data/lib/chef/provider/etcd.rb

@@ -0,0 +1,104 @@
+require 'chef/provider'
+require 'etcd'
+
+
+
+class Chef
+  class  Provider
+    class Etcd < Chef::Provider
+
+      def initialize(new_resource, run_context)
+        super(new_resource, run_context)
+      end
+
+      def load_current_resource
+        @current_resource ||= Chef::Resource::Etcd.new(new_resource.name)
+        if key_exist?
+          @current_resource.value(current_value)
+        end
+        @current_resource
+      end
+
+      def whyrun_supported?
+        true
+      end
+
+      def config
+        @config ||= Chef::Config[:etcd]
+      end
+
+      def etcd
+        @etcd ||= ::Etcd.client(config)
+      end  
+
+
+      def key_exist?
+        exist = true
+        begin
+          etcd.get(new_resource.key)
+        rescue Net::HTTPServerException => e
+          exist = false
+        end
+        exist
+      end
+
+      def current_value
+        if key_exist?
+          etcd.get(new_resource.key).value
+        else
+          nil
+        end
+      end
+
+      def action_set
+        if @current_resource.value == new_resource.value
+          Chef::Log.debug(" etcd #{new_resource.key} is in sync")
+        else
+          converge_by "will set value of key #{new_resource.key}" do
+            etcd.set(new_resource.key, new_resource.value)
+            new_resource.updated_by_last_action(true)
+          end
+        end
+      end
+      def action_get
+        converge_by "will set value of key #{new_resource.key}" do
+          if key_exist?
+            current_value
+          else
+            nil
+          end
+          new_resource.updated_by_last_action(true)
+        end
+      end
+
+      def action_test_and_set
+        begin
+          etcd.test_and_set(new_resource.key, new_resource.value, new_resource.prev_value, new_resource.ttl)
+        rescue Net::HTTPServerException => e
+          converge_by "will not be able test_and_set value of key #{new_resource.key}" do
+            new_resource.updated_by_last_action(true)
+          end
+        end
+      end
+
+      def action_watch
+        converge_bey "will wait for update from etcd key #{new_resource.key}" do
+          etcd.watch(new_resource.key)
+          new_resource.updated_by_last_action(true)
+        end
+      end
+
+      def action_delete
+        if key_exist?
+          converge_by "will delete etcd key #{new_resource.key}" do
+            etcd.delete(new_resource.key)
+            new_resource.updated_by_last_action(true)
+          end
+        else
+          Chef::Log.debug("etcd key #{new_resource.key} does not exist")
+        end
+      end
+    end
+  end
+end
+

data/lib/chef/resource/etcd.rb

@@ -0,0 +1,40 @@
+require 'chef/resource'
+
+class Chef
+  class  Resource
+
+    class Etcd < Chef::Resource
+
+      identity_attr :key
+      state_attrs :value
+      provides :etcd
+
+      def initialize(name, run_context=nil)
+        super
+        @resource_name = :etcd
+        @action = [:set]
+        @allowed_actions.push(:test_and_set, :delete, :get, :wait, :set)
+        @key = name
+        @value = nil
+        @prev_value = nil
+        @ttl = nil
+      end
+
+      def key(arg=nil)
+        set_or_return(:key, arg, :kind_of => String)
+      end
+
+      def value(arg=nil)
+        set_or_return(:value, arg, :kind_of => String)
+      end
+
+      def prev_value(arg=nil)
+        set_or_return(:prev_value, arg, :kind_of => String)
+      end
+
+      def ttl(arg=nil)
+        set_or_return(:ttl, arg, :kind_of => String)
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,98 @@
+--- !ruby/object:Gem::Specification
+name: chef-etcd
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Ranjib Dey
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: etcd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Etcd-Chef bindings
+email:
+- dey.ranjib@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- chef-etcd.gemspec
+- lib/chef/etcd.rb
+- lib/chef/etcd/version.rb
+- lib/chef/handler/etcd_event.rb
+- lib/chef/handler/etcd_report.rb
+- lib/chef/provider/etcd.rb
+- lib/chef/resource/etcd.rb
+homepage: https://github.com/ranjib/chef-etcd
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Provides chef resource, provider, handlers for etcd
+test_files: []