sensu-plugin 0.0.1

data/lib/sensu-handler.rb

@@ -0,0 +1,48 @@
+require 'json'
+
+module Sensu
+  class Handler
+
+    # Implementing classes should override this.
+
+    def handle(event)
+      puts 'ignoring event -- no handler defined'
+    end
+
+    # Overriding filtering logic is optional. Returns truthy if the
+    # event should be handled and falsy if it should not.
+
+    def filter(event)
+      if event['check']['alert'] == false
+        puts 'alert disabled -- filtered event ' + short_name(event)
+        exit 0
+      end
+      refresh = (60.fdiv(event['check']['interval']) * 30).to_i
+      event['occurrences'] == 1 || event['occurrences'] % refresh == 0
+    end
+
+    def short_name(event)
+      event['client']['name'] + '/' + event['check']['name']
+    end
+
+    # This works just like Plugin::CLI's autorun.
+
+    @@autorun = self
+    class << self
+      def method_added(name)
+        if name == :handle
+          @@autorun = self
+        end
+      end
+    end
+
+    at_exit do
+      handler = @@autorun.new
+      event = ::JSON.parse(STDIN.read)
+      if handler.filter(event)
+        handler.handle(event)
+      end
+    end
+
+  end
+end

data/lib/sensu-plugin.rb

@@ -0,0 +1,11 @@
+module Sensu
+  module Plugin
+    VERSION = "0.0.1"
+    EXIT_CODES = {
+      'OK' => 0,
+      'WARNING' => 1,
+      'CRITICAL' => 2,
+      'UNKNOWN' => 3,
+    }
+  end
+end

data/lib/sensu-plugin/check/cli.rb

@@ -0,0 +1,25 @@
+require 'sensu-plugin/cli'
+
+module Sensu
+  module Plugin
+    class Check
+      class CLI < Sensu::Plugin::CLI
+
+        class << self
+          def check_name(name=nil)
+            if name
+              @check_name = name
+            else
+              @check_name || self.to_s
+            end
+          end
+        end
+
+        def format_output(status, output)
+          "#{self.class.check_name}: #{status} - #{output}"
+        end
+
+      end
+    end
+  end
+end

data/lib/sensu-plugin/cli.rb

@@ -0,0 +1,60 @@
+require 'sensu-plugin'
+require 'mixlib/cli'
+
+module Sensu
+  module Plugin
+    class CLI
+      include Mixlib::CLI
+
+      # Implementing classes should override this to produce appropriate
+      # output for their handler.
+
+      def format_output(status, output)
+        output
+      end
+
+      # This will define 'ok', 'warning', 'critical', and 'unknown'
+      # methods, which the plugin should call to exit.
+
+      Sensu::Plugin::EXIT_CODES.each do |status, code|
+        define_method(status.downcase) do |*args|
+          puts format_output(status, *args)
+          exit(code)
+        end
+      end
+
+      # Implementing classes must override this.
+
+      def run
+        unknown "Not implemented! You should override Sensu::Plugin::CLI#run."
+      end
+
+      # Behind-the-scenes stuff below. If you do something crazy like
+      # define two plugins in one script, the last one will 'win' in
+      # terms of what gets auto-run.
+
+      @@autorun = self
+      class << self
+        def method_added(name)
+          if name == :run
+            @@autorun = self
+          end
+        end
+      end
+
+      at_exit do
+        begin
+          check = @@autorun.new
+          check.parse_options
+          check.run
+        rescue SystemExit => e
+          exit e.status
+        rescue Exception => e
+          self.new.critical "Check failed to run: #{e}"
+        end
+        self.new.warning "Check did not exit! You should call an exit code method."
+      end
+
+    end
+  end
+end

data/lib/sensu-plugin/metric/cli.rb

@@ -0,0 +1,17 @@
+require 'sensu-plugin/cli'
+require 'json'
+
+module Sensu
+  module Plugin
+    class Metric
+      class CLI < Sensu::Plugin::CLI
+
+        def format_output(status, output)
+          output[:timestamp] ||= Time.now.to_i
+          ::JSON.generate(output)
+        end
+
+      end
+    end
+  end
+end

data/lib/sensu-plugin/util/procs.rb

@@ -0,0 +1,25 @@
+require 'sensu-plugin/check/cli'
+
+module Sensu
+  module Plugin
+    module Util
+      module Procs
+
+        class << self
+          def get_procs
+            `which tasklist`; $? == 0 ? `tasklist` : `ps aux`
+          end
+
+          def find_proc(name)
+            get_procs.split("\n").find {|ln| ln.include?(name) }
+          end
+
+          def find_proc_regex(pat)
+            get_procs.split("\n").find {|ln| ln =~ pat }
+          end
+        end
+
+      end
+    end
+  end
+end

data/test/check.rb

@@ -0,0 +1,37 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'minitest/autorun'
+
+class TestCheck < MiniTest::Unit::TestCase
+
+  def setup
+    @script = File.join(File.dirname(__FILE__), 'checks/check-options')
+  end
+
+  def test_ok
+    system @script, '-o'
+    assert $?.exitstatus == 0
+  end
+
+  def test_warning
+    system @script, '-w'
+    assert $?.exitstatus == 1
+  end
+
+  def test_critical
+    system @script, '-c'
+    assert $?.exitstatus == 2
+  end
+
+  def test_unknown
+    system @script, '-u'
+    assert $?.exitstatus == 3
+  end
+
+  def test_fallthrough
+    system @script
+    assert $?.exitstatus == 1
+  end
+
+end

metadata

@@ -0,0 +1,76 @@
+--- !ruby/object:Gem::Specification
+name: sensu-plugin
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Decklin Foster
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-12-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: &70296568612360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70296568612360
+- !ruby/object:Gem::Dependency
+  name: mixlib-cli
+  requirement: &70296568611860 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70296568611860
+description: Plugins and helper libraries for Sensu, a monitoring framework
+email:
+- decklin@red-bean.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/sensu-handler.rb
+- lib/sensu-plugin/check/cli.rb
+- lib/sensu-plugin/cli.rb
+- lib/sensu-plugin/metric/cli.rb
+- lib/sensu-plugin/util/procs.rb
+- lib/sensu-plugin.rb
+- test/check.rb
+homepage: https://github.com/sonian/sensu-plugin
+licenses:
+- MIT
+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.11
+signing_key: 
+specification_version: 3
+summary: Sensu Plugins
+test_files:
+- test/check.rb