sensu 0.9.5 → 0.9.6.beta

data/lib/sensu/settings.rb

@@ -0,0 +1,280 @@
+module Sensu
+  class Settings
+    attr_reader :indifferent_access, :loaded_env, :loaded_files
+
+    def initialize
+      @logger = Cabin::Channel.get
+      @settings = Hash.new
+      @indifferent_access = false
+      @loaded_env = false
+      @loaded_files = Array.new
+    end
+
+    def [](key)
+      unless @indifferent_access
+        indifferent_access!
+      end
+      @settings[key.to_sym]
+    end
+
+    def indifferent_access!
+      @settings = indifferent_access(@settings)
+      @indifferent_access = true
+    end
+
+    def load_env
+      if ENV['RABBITMQ_URL']
+        @settings[:rabbitmq] = ENV['RABBITMQ_URL']
+        @logger.warn('using rabbitmq url environment variable', {
+          :rabbitmq_url => ENV['RABBITMQ_URL']
+        })
+      end
+      ENV['REDIS_URL'] ||= ENV['REDISTOGO_URL']
+      if ENV['REDIS_URL']
+        @settings[:redis] = ENV['REDIS_URL']
+        @logger.warn('using redis url environment variable', {
+          :redis_url => ENV['REDIS_URL']
+        })
+      end
+      ENV['API_PORT'] ||= ENV['PORT']
+      if ENV['API_PORT']
+        @settings[:api] ||= Hash.new
+        @settings[:api][:port] = ENV['API_PORT']
+        @logger.warn('using api port environment variable', {
+          :api_port => ENV['API_PORT']
+        })
+      end
+      @indifferent_access = false
+      @loaded_env = true
+    end
+
+    def load_file(file)
+      if File.readable?(file)
+        begin
+          contents = File.open(file, 'r').read
+          config = JSON.parse(contents, :symbolize_names => true)
+          merged = deep_merge(@settings, config)
+          unless @loaded_files.empty?
+            @logger.warn('config file applied changes', {
+              :config_file => file,
+              :changes => deep_diff(@settings, merged)
+            })
+          end
+          @settings = merged
+          @indifferent_access = false
+          @loaded_files.push(file)
+        rescue JSON::ParserError => error
+          @logger.error('config file must be valid json', {
+            :config_file => file,
+            :error => error.to_s
+          })
+          @logger.warn('ignoring config file', {
+            :config_file => file
+          })
+        end
+      else
+        @logger.error('config file does not exist or is not readable', {
+          :config_file => file
+        })
+        @logger.warn('ignoring config file', {
+          :config_file => file
+        })
+      end
+    end
+
+    def checks
+      @settings[:checks].map do |check_name, check_details|
+        check_details.merge(:name => check_name.to_s)
+      end
+    end
+
+    def handlers
+      @settings[:handlers].map do |handler_name, handler_details|
+        handler_details.merge(:name => handler_name.to_s)
+      end
+    end
+
+    def check_exists?(check_name)
+      @settings[:checks].has_key?(check_name.to_sym)
+    end
+
+    def handler_exists?(handler_name)
+      @settings[:handlers].has_key?(handler_name.to_sym)
+    end
+
+    def validate
+      @logger.debug('validating settings')
+      validate_checks
+      case File.basename($0)
+      when 'rake'
+        validate_client
+        validate_api
+        validate_server
+      when 'sensu-client'
+        validate_client
+      when 'sensu-api'
+        validate_api
+      when 'sensu-server'
+        validate_server
+      end
+      @logger.debug('settings are valid')
+    end
+
+    private
+
+    def indifferent_access(hash)
+      hash = indifferent_hash.merge(hash)
+      hash.each do |key, value|
+        if value.is_a?(Hash)
+          hash[key] = indifferent_access(value)
+        end
+      end
+    end
+
+    def indifferent_hash
+      Hash.new do |hash, key|
+        if key.is_a?(String)
+          hash[key.to_sym]
+        end
+      end
+    end
+
+    def deep_merge(hash_one, hash_two)
+      merged = hash_one.dup
+      hash_two.each do |key, value|
+        merged[key] = case
+        when hash_one[key].is_a?(Hash) && value.is_a?(Hash)
+          deep_merge(hash_one[key], value)
+        when hash_one[key].is_a?(Array) && value.is_a?(Array)
+          hash_one[key].concat(value).uniq
+        else
+          value
+        end
+      end
+      merged
+    end
+
+    def deep_diff(hash_one, hash_two)
+      keys = hash_one.keys.concat(hash_two.keys).uniq
+      keys.inject(Hash.new) do |diff, key|
+        unless hash_one[key] == hash_two[key]
+          if hash_one[key].is_a?(Hash) && hash_two[key].is_a?(Hash)
+            diff[key] = deep_diff(hash_one[key], hash_two[key])
+          else
+            diff[key] = [hash_one[key], hash_two[key]]
+          end
+        end
+        diff
+      end
+    end
+
+    def validate_checks
+      unless @settings[:checks].is_a?(Hash)
+        raise('missing check configuration')
+      end
+      checks.each do |check|
+        unless check[:interval].is_a?(Integer) && check[:interval] > 0
+          raise('missing interval for check: ' + check[:name])
+        end
+        unless check[:command].is_a?(String)
+          raise('missing command for check: ' + check[:name])
+        end
+        unless check[:standalone]
+          unless check[:subscribers].is_a?(Array) && check[:subscribers].count > 0
+            raise('missing subscribers for check: ' + check[:name])
+          end
+          check[:subscribers].each do |subscriber|
+            unless subscriber.is_a?(String) && !subscriber.empty?
+              raise('a check subscriber must be a string for check: ' + check[:name])
+            end
+          end
+        end
+        if check.has_key?(:handler)
+          unless check[:handler].is_a?(String)
+            raise('handler must be a string for check: ' + check[:name])
+          end
+        end
+        if check.has_key?(:handlers)
+          unless check[:handlers].is_a?(Array)
+            raise('handlers must be an array for check: ' + check[:name])
+          end
+        end
+      end
+    end
+
+    def validate_client
+      unless @settings[:client].is_a?(Hash)
+        raise('missing client configuration')
+      end
+      unless @settings[:client][:name].is_a?(String)
+        raise('client must have a name')
+      end
+      unless @settings[:client][:address].is_a?(String)
+        raise('client must have an address')
+      end
+      unless @settings[:client][:subscriptions].is_a?(Array) && @settings[:client][:subscriptions].count > 0
+        raise('client must have subscriptions')
+      end
+      @settings[:client][:subscriptions].each do |subscription|
+        unless subscription.is_a?(String) && !subscription.empty?
+          raise('a client subscription must be a string')
+        end
+      end
+    end
+
+    def validate_api
+      unless @settings[:api].is_a?(Hash)
+        raise('missing api configuration')
+      end
+      unless @settings[:api][:port].is_a?(Integer)
+        raise('api port must be an integer')
+      end
+      if @settings[:api].has_key?(:user) || @settings[:api].has_key?(:password)
+        unless @settings[:api][:user].is_a?(String)
+          raise('api user must be a string')
+        end
+        unless @settings[:api][:password].is_a?(String)
+          raise('api password must be a string')
+        end
+      end
+    end
+
+    def validate_server
+      unless @settings[:handlers].is_a?(Hash)
+        raise('missing handler configuration')
+      end
+      unless @settings[:handlers].include?(:default)
+        raise('missing default handler')
+      end
+      handlers.each do |handler|
+        unless handler[:type].is_a?(String)
+          raise('missing type for handler: ' + handler[:name])
+        end
+        case handler[:type]
+        when 'pipe'
+          unless handler[:command].is_a?(String)
+            raise('missing command for pipe handler: ' + handler[:name])
+          end
+        when 'amqp'
+          unless handler[:exchange].is_a?(Hash)
+            raise('missing exchange details for amqp handler: ' + handler[:name])
+          end
+          unless handler[:exchange][:name].is_a?(String)
+            raise('missing exchange name for amqp handler: ' + handler[:name])
+          end
+          if handler[:exchange].has_key?(:type)
+            unless %w[direct fanout topic].include?(handler[:exchange][:type])
+              raise('invalid exchange type for amqp handler: ' + handler[:name])
+            end
+          end
+        when 'set'
+          unless handler[:handlers].is_a?(Array) && handler[:handlers].count > 0
+            raise('missing handler set for handler: ' + handler[:name])
+          end
+        else
+          raise('unknown type for handler: ' + handler[:name])
+        end
+      end
+    end
+  end
+end

data/lib/sensu/socket.rb

@@ -0,0 +1,52 @@
+module Sensu
+  class Socket < EM::Connection
+    attr_accessor :protocol, :logger, :settings, :amq
+
+    def reply(data)
+      if @protocol == :tcp
+        send_data(data)
+      end
+    end
+
+    def receive_data(data)
+      if data == 'ping'
+        @logger.debug('socket received ping')
+        reply('pong')
+      else
+        @logger.debug('socket received data', {
+          :data => data
+        })
+        begin
+          check = JSON.parse(data, :symbolize_names => true)
+          validates = [:name, :output].all? do |key|
+            check[key].is_a?(String)
+          end
+          check[:issued] = Time.now.to_i
+          check[:status] ||= 0
+          if validates && check[:status].is_a?(Integer)
+            payload = {
+              :client => @settings[:client][:name],
+              :check => check
+            }
+            @logger.info('publishing check result', {
+              :payload => payload
+            })
+            @amq.queue('results').publish(payload.to_json)
+            reply('ok')
+          else
+            @logger.warn('invalid check result', {
+              :check => check
+            })
+            reply('invalid')
+          end
+        rescue JSON::ParserError => error
+          @logger.warn('check result must be valid json', {
+            :data => data,
+            :error => error.to_s
+          })
+          reply('invalid')
+        end
+      end
+    end
+  end
+end

data/sensu.gemspec

@@ -1,32 +1,31 @@
-require File.expand_path("../lib/sensu/version", __FILE__)
+require File.join(File.dirname(__FILE__), 'lib', 'sensu', 'constants')
 
 Gem::Specification.new do |s|
-  s.name        = "sensu"
+  s.name        = 'sensu'
   s.version     = Sensu::VERSION
   s.platform    = Gem::Platform::RUBY
-  s.authors     = ["Sean Porter", "Justin Kolberg"]
-  s.email       = ["sean.porter@sonian.net", "justin.kolberg@sonian.net"]
-  s.homepage    = "https://github.com/sonian/sensu"
-  s.summary     = "A monitoring framework"
-  s.description = "A monitoring framework that aims to be simple, malleable, and scalable. Uses the publish/subscribe model."
-  s.license     = "MIT"
+  s.authors     = ['Sean Porter', 'Justin Kolberg']
+  s.email       = ['sean.porter@sonian.net', 'justin.kolberg@sonian.net']
+  s.homepage    = 'https://github.com/sensu/sensu'
+  s.summary     = 'A monitoring framework'
+  s.description = 'A monitoring framework that aims to be simple, malleable, and scalable. Uses the publish/subscribe model.'
+  s.license     = 'MIT'
   s.has_rdoc    = false
 
-  s.add_dependency("bundler")
-  s.add_dependency("eventmachine", "~> 1.0.0.beta.4")
-  s.add_dependency("amqp", "0.7.4")
-  s.add_dependency("json")
-  s.add_dependency("hashie")
-  s.add_dependency("cabin", "0.1.8")
-  s.add_dependency("ruby-redis")
-  s.add_dependency("async_sinatra", "1.0.0")
-  s.add_dependency("thin")
+  s.add_dependency('bundler')
+  s.add_dependency('eventmachine', '~> 1.0.0.beta.4')
+  s.add_dependency('amqp', '0.7.4')
+  s.add_dependency('json')
+  s.add_dependency('cabin', '0.4.4')
+  s.add_dependency('ruby-redis')
+  s.add_dependency('async_sinatra', '1.0.0')
+  s.add_dependency('thin')
 
-  s.add_development_dependency("rake")
-  s.add_development_dependency("em-spec")
-  s.add_development_dependency("em-http-request")
+  s.add_development_dependency('rake')
+  s.add_development_dependency('em-spec')
+  s.add_development_dependency('em-http-request')
 
-  s.files         = Dir.glob("{bin,lib}/**/*") + %w[sensu.gemspec README.org MIT-LICENSE.txt]
-  s.executables   = Dir.glob("bin/**/*").map { |file| File.basename(file) }
-  s.require_paths = ["lib"]
+  s.files         = Dir.glob('{bin,lib}/**/*') + %w[sensu.gemspec README.org MIT-LICENSE.txt]
+  s.executables   = Dir.glob('bin/**/*').map { |file| File.basename(file) }
+  s.require_paths = ['lib']
 end

metadata

@@ -1,263 +1,249 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: sensu
-version: !ruby/object:Gem::Version 
-  hash: 49
-  prerelease: false
-  segments: 
-  - 0
-  - 9
-  - 5
-  version: 0.9.5
+version: !ruby/object:Gem::Version
+  version: 0.9.6.beta
+  prerelease: 6
 platform: ruby
-authors: 
+authors:
 - Sean Porter
 - Justin Kolberg
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2012-03-28 00:00:00 -07:00
-default_executable: 
-dependencies: 
-- !ruby/object:Gem::Dependency 
+date: 2012-05-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
   name: bundler
-  prerelease: false
-  requirement: &id001 !ruby/object:Gem::Requirement 
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id001
-- !ruby/object:Gem::Dependency 
-  name: eventmachine
   prerelease: false
-  requirement: &id002 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
+    requirements:
     - - ~>
-      - !ruby/object:Gem::Version 
-        hash: 62196363
-        segments: 
-        - 1
-        - 0
-        - 0
-        - beta
-        - 4
+      - !ruby/object:Gem::Version
         version: 1.0.0.beta.4
   type: :runtime
-  version_requirements: *id002
-- !ruby/object:Gem::Dependency 
-  name: amqp
   prerelease: false
-  requirement: &id003 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.0.beta.4
+- !ruby/object:Gem::Dependency
+  name: amqp
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - "="
-      - !ruby/object:Gem::Version 
-        hash: 11
-        segments: 
-        - 0
-        - 7
-        - 4
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
         version: 0.7.4
   type: :runtime
-  version_requirements: *id003
-- !ruby/object:Gem::Dependency 
-  name: json
   prerelease: false
-  requirement: &id004 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.7.4
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id004
-- !ruby/object:Gem::Dependency 
-  name: hashie
   prerelease: false
-  requirement: &id005 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
-  type: :runtime
-  version_requirements: *id005
-- !ruby/object:Gem::Dependency 
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
   name: cabin
-  prerelease: false
-  requirement: &id006 !ruby/object:Gem::Requirement 
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - "="
-      - !ruby/object:Gem::Version 
-        hash: 11
-        segments: 
-        - 0
-        - 1
-        - 8
-        version: 0.1.8
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.4.4
   type: :runtime
-  version_requirements: *id006
-- !ruby/object:Gem::Dependency 
-  name: ruby-redis
   prerelease: false
-  requirement: &id007 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.4.4
+- !ruby/object:Gem::Dependency
+  name: ruby-redis
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id007
-- !ruby/object:Gem::Dependency 
-  name: async_sinatra
   prerelease: false
-  requirement: &id008 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: async_sinatra
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - "="
-      - !ruby/object:Gem::Version 
-        hash: 23
-        segments: 
-        - 1
-        - 0
-        - 0
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
         version: 1.0.0
   type: :runtime
-  version_requirements: *id008
-- !ruby/object:Gem::Dependency 
-  name: thin
   prerelease: false
-  requirement: &id009 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: thin
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id009
-- !ruby/object:Gem::Dependency 
-  name: rake
   prerelease: false
-  requirement: &id010 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :development
-  version_requirements: *id010
-- !ruby/object:Gem::Dependency 
-  name: em-spec
   prerelease: false
-  requirement: &id011 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: em-spec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :development
-  version_requirements: *id011
-- !ruby/object:Gem::Dependency 
-  name: em-http-request
   prerelease: false
-  requirement: &id012 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: em-http-request
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :development
-  version_requirements: *id012
-description: A monitoring framework that aims to be simple, malleable, and scalable. Uses the publish/subscribe model.
-email: 
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A monitoring framework that aims to be simple, malleable, and scalable.
+  Uses the publish/subscribe model.
+email:
 - sean.porter@sonian.net
 - justin.kolberg@sonian.net
-executables: 
+executables:
 - sensu-api
 - sensu-client
 - sensu-server
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
+files:
 - bin/sensu-api
 - bin/sensu-client
 - bin/sensu-server
 - lib/sensu/api.rb
+- lib/sensu/base.rb
+- lib/sensu/cli.rb
 - lib/sensu/client.rb
-- lib/sensu/config.rb
+- lib/sensu/constants.rb
+- lib/sensu/logger.rb
 - lib/sensu/patches/amqp.rb
 - lib/sensu/patches/redis.rb
 - lib/sensu/patches/ruby.rb
+- lib/sensu/process.rb
 - lib/sensu/server.rb
-- lib/sensu/version.rb
+- lib/sensu/settings.rb
+- lib/sensu/socket.rb
 - lib/sensu.rb
 - sensu.gemspec
 - README.org
 - MIT-LICENSE.txt
-has_rdoc: true
-homepage: https://github.com/sonian/sensu
-licenses: 
+homepage: https://github.com/sensu/sensu
+licenses:
 - MIT
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
+required_ruby_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
+  requirements:
+  - - ! '>'
+    - !ruby/object:Gem::Version
+      version: 1.3.1
 requirements: []
-
 rubyforge_project: 
-rubygems_version: 1.3.7
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: A monitoring framework
 test_files: []
-
+has_rdoc: false