naplug 1.6.1

data/lib/naplug.rb

@@ -0,0 +1,163 @@
+require 'rubygems'
+require 'naplug/plugin'
+
+module Naplug
+
+  class Error < StandardError; end
+
+  module ClassMethods
+
+    attr_reader :plugins
+
+    def plugin(tag = :main, &block)
+      @plugins = Hash.new unless @plugins
+      @plugins[tag] = create_metaplugin tag, block
+    end
+
+    def tags
+      self.plugins.keys
+    end
+
+    private
+
+    def create_metaplugin(tag,block)
+      module_eval do
+        define_method "#{tag}".to_sym  do; plugin;         end    # <tag> methods for quick access to plugins
+        define_method "#{tag}!".to_sym do; self.exec! tag; end    # <tag>! methods to involke exec! on a given plugin
+      end
+      Plugin.new tag, :meta, block
+    end
+
+  end
+
+  module InstanceMethods
+
+    attr_reader :plugins
+
+    def initialize(args = {})
+      @plugins = Hash.new
+      plugins!
+
+      @_args = Hash.new
+      args! args
+    end
+
+    def args
+      @_args
+    end
+
+    def args!(args)
+      @_args.merge! args
+      @plugins.each do |tag,plugin|
+        plugin_args = args.key?(tag) ? args[tag] : {}
+        shared_args = args.select { |t,a| not @plugins.keys.include? t }
+        plugin.args! shared_args.merge! plugin_args
+      end
+    end
+
+    def to_str(tag = default_plugin.tag)
+      s_format = perfdata(tag) ? '%s: %s | %s' : '%s: %s'
+      s_array = perfdata(tag) ? [@plugins[tag].status,@plugins[tag].output,perfdata(tag)] : [@plugins[tag].status,@plugins[tag].output]
+      s_format % s_array
+    end
+
+    def exec!(tag = default_plugin.tag)
+      exec tag
+      eval tag
+      exit tag
+    end
+
+    def exec(tag = default_plugin.tag)
+      rexec @plugins[tag]
+    end
+
+    def eval(tag = default_plugin.tag)
+      @plugins[tag].eval
+    end
+
+    def eject!(payload = nil)
+      o = case payload
+            when String then payload
+            when Exception then "#{payload.backtrace[1][/.+:\d+/]}: #{payload.message}"
+            else nil
+              caller[0][/.+:\d+/]
+          end
+      print "UNKNOWN: plugin eject! in %s\n" % [o]
+      Kernel::exit 3
+    end
+
+    private
+
+    def rexec(plug)
+      if plug.has_plugins?
+        plug.plugins.each_value { |p| rexec p }
+      else
+        plexec plug
+      end
+    end
+
+    def plexec(p)
+      begin
+        @_running = p.tag
+        instance_exec p, &p.block
+        @_running = nil
+      rescue Naplug::Error => e
+          p.status.unknown!
+          p.output! "#{e.backtrace[1][/[^\/]+:\d+/]}: #{e.message}"
+      rescue => e
+        p.status.unknown!
+        p.output!  "#{e.backtrace[0][/[^\/]+:\d+/]}: #{e.message}"
+        p.payload! e
+      ensure
+        @_runinng = nil
+      end
+    end
+
+    def plugins!
+      self.class.plugins.each do |tag,plugin|
+        @plugins[tag] = Plugin.new tag, plugin.block
+      end
+    end
+
+    def default_plugin
+      return @plugins[:main] if @plugins.key? :main
+      return @plugins[@plugins.keys[0]] if @plugins.size == 1
+      raise Naplug::Error, 'unable to determine default plugin'
+    end
+
+    def perfdata(tag = default_plugin.tag)
+      plugin = @plugins[tag]
+      if plugin.has_plugins?
+        plugin.plugins.values.map { |plug| plug.perfdata }.join(' ').gsub(/^\s+$/,'').strip!
+      else
+        plugin.perfdata
+      end
+    end
+
+    def exit(tag = default_plugin.tag)
+      print "%s\n" % [to_str(tag)]
+      Kernel::exit @plugins[tag].status.to_i
+    end
+
+    def method_missing(method, *args, &block)
+      message = "undefined instance variable or method #{method}"
+      case @_runinng
+        when nil?
+          begin; raise Naplug::Error, message; rescue => e; eject! e ; end
+        else
+          raise Naplug::Error, message
+      end
+    end
+
+    def respond_to_missing?(method, *)
+      @plugins.keys? method || super
+    end
+
+  end
+
+  def self.included(klass)
+    klass.send :include, InstanceMethods
+    klass.extend ClassMethods
+  end
+
+end

metadata

@@ -0,0 +1,67 @@
+--- !ruby/object:Gem::Specification
+name: naplug
+version: !ruby/object:Gem::Version
+  version: 1.6.1
+  prerelease: 
+platform: ruby
+authors:
+- Gerardo López-Fernádez
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-02-24 00:00:00.000000000 Z
+dependencies: []
+description: ! 'A Ruby library for Nagios plugins '
+email: gerir@evernote.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/naplug/about.rb
+- lib/naplug/helpers.rb
+- lib/naplug/output.rb
+- lib/naplug/performancedata.rb
+- lib/naplug/plugin.rb
+- lib/naplug/status.rb
+- lib/naplug/version.rb
+- lib/naplug.rb
+- examples/almostalwaysok
+- examples/alwaysok
+- examples/alwaysunknown
+- examples/dupplugin
+- examples/exception
+- examples/exception+
+- examples/markerfile
+- examples/markerfile++
+- examples/multiplug
+- examples/multiplugin
+- examples/multiplugins
+- examples/status
+- LICENSE
+- README.md
+homepage: https://github.com/gerirgaudi/naplug
+licenses:
+- Apache License, Version 2.0
+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: 1.3.5
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: A Ruby library for Nagios plugins
+test_files: []