nagiosharder 0.3.0 → 0.4.0.rc1

data/lib/nagiosharder/cli.rb

@@ -0,0 +1,281 @@
+require 'terminal-table'
+require 'optparse'
+require 'nagiosharder'
+
+class NagiosHarder
+  class Cli
+
+    attr_reader :options, :command, :param, :the_rest, :host, :service
+
+    def initialize(argv)
+      @options          = parse_connection_options(argv)
+      @command, @param  = argv
+      @host, @service   = param.split("/") if param
+      @the_rest         = argv[2..-1]
+    end
+
+    def run
+      return_value = case command
+      when 'status'
+        service_table do
+          client.host_status(host).select do |name, s|
+            service.nil? || service == name
+          end.map { |name, s| s }
+        end
+        true
+      when /^(ack|acknowledged)$/
+        if service.nil?
+          client.acknowledge_host(host, the_rest.join(' '))
+        else
+          client.acknowledge_service(host, service, the_rest.join(' '))
+        end
+      when /^unack/
+        client.unacknowledge_service(host, service)
+      when /^(mute|disable_notifications)$/
+        client.disable_service_notifications(host, service)
+      when /^(unmute|enable_notifications)$/
+        client.enable_service_notifications(host, service)
+      when 'check'
+        if service
+          client.schedule_service_check(host, service)
+        else
+          client.schedule_host_check(host)
+        end
+      when 'downtime'
+        if service
+          client.schedule_service_downtime(host, service, :type => :fixed, :start_time => Time.now, :end_time => Time.now + the_rest.first.to_i.minutes)
+        else
+          client.schedule_host_downtime(host, :type => :fixed, :start_time => Time.now, :end_time => Time.now + the_rest.first.to_i.minutes)
+        end
+      when 'problems'
+        service_table do
+          params = {
+            :service_status_types => [:critical, :warning, :unknown],
+          }
+          params[:group] = param
+          client.service_status(params)
+        end
+        true
+      when /^(critical|warning|unknown)/
+        service_table do
+          params = {
+            :service_status_types => [
+              $1,
+            ],
+            :service_props => [
+              :hard_state,
+              :no_scheduled_downtime,
+              :state_unacknowledged,
+              :notifications_enabled
+            ]
+          }
+          params[:group] = param
+          client.service_status(params)
+        end
+        true
+      when /^(triage|unhandled)/
+        service_table do
+          params = {
+            :service_status_types => [
+              :critical,
+              :warning,
+              :unknown
+            ],
+            :host_props => [
+              :no_scheduled_downtime,
+              :state_unacknowledged,
+              :notifications_enabled
+            ],
+            :service_props => [
+              :no_scheduled_downtime,
+              :state_unacknowledged,
+              :notifications_enabled
+            ]
+          }
+          params[:group] = param
+          client.service_status(params)
+        end
+        true
+      when /^muted/
+        service_table do
+          params = {
+            :service_props => [
+              :notifications_disabled,
+            ]
+          }
+          params[:group] = param
+          client.service_status(params)
+        end
+        true
+      when /^(acked|acknowledged)/
+        service_table do
+          params = {
+            :service_status_types => [
+              :critical,
+              :warning,
+              :unknown
+            ],
+            :service_props => [
+              :state_acknowledged,
+            ]
+          }
+          params[:group] = param
+          client.service_status(params)
+        end
+        true
+      else
+        debug "'#{command}'"
+        raise ArgumentError, help
+      end
+      if return_value
+        0
+      else
+        puts "Sorry, bro, nagios didn't like that."
+        1
+      end
+    end
+
+    protected
+
+    def client
+      debug "loading client with options #{options.inspect}"
+      @client ||= NagiosHarder::Site.new(options['nagios_url'], options['user'], options['password'], options['version'], options['time'])
+    end
+
+    def parse_connection_options(argv)
+      options = {
+        'version' => 3
+      }
+
+      optparse = OptionParser.new do |opts|
+        opts.banner = "Usage: nagiosharder [options] [command]"
+
+        opts.on( '-h', '--help') do
+          raise ArgumentError, opts.help
+        end
+
+        opts.on( '-c', '--config [/path/to/file]', 'YAML config file [optional, but recommended]') do |file|
+          options.merge!(YAML.load_file(file))
+        end
+
+        opts.on( '-u', '--user USER', 'Nagios user') do |user|
+          options['user'] = user
+        end
+
+        opts.on( '-p', '--password PASSWORD', 'Nagios password') do |password|
+          options['password'] = password
+        end
+
+        opts.on( '-n', '--nagios_url URL', 'Nagios cgi url') do |nagios_url|
+          options['nagios_url'] = nagios_url
+        end
+
+        opts.on( '-v', '--version [3]', 'Nagios version (2 or 3, defaults to 3)') do |version|
+          options['version'] = version
+        end
+
+        opts.on( '-t', '--time [us|euro]', 'Nagios time format') do |time|
+          options['time'] = time
+        end
+
+      end
+
+      begin
+        optparse.parse!(argv)
+      rescue OptionParser::InvalidOption, OptionParser::MissingArgument => e
+        raise ArgumentError, e.message + "\n\n" + optparse.help
+      end
+
+      unless options['nagios_url'] && options['user'] && options['password']
+        raise ArgumentError, optparse.help
+      end
+
+      options
+    end
+
+    def service_table
+      table = Terminal::Table.new(:headings => ['Service', 'Status', 'Details']) do |t|
+        services = yield
+        services.each do |service|
+          t << service_row(service)
+        end
+        t
+      end
+      table.align_column(1, :right)
+      puts table
+    end
+
+    def service_row(service)
+      service['status'] << "/ACK" if service['acknowledged']
+      service['status'] << "/MUTE" if service['notifications_disabled']
+      service['status'] << "/COMMENT" if service['comments_url']
+      [
+        service['host']+"/"+service["service"],
+        service['status'],
+        wrap_text(service['extended_info'], 40)
+      ]
+    end
+
+    # wraps text at the specified column, 80 by default
+    def wrap_text(txt, col = 80)
+      txt.gsub(/(.{1,#{col}})( +|$\n?)|(.{1,#{col}})/,
+        "\\1\\3\n")
+    end
+
+    def debug(*args)
+      $stderr.puts *args if ENV['DEBUG']
+    end
+
+    def help
+      help = <<-HELP
+NagiosHarder
+
+USAGE:
+
+  nagiosharder --help
+  nagiosharder [options] [command]
+
+COMMANDS:
+
+    nagiosharder commands
+
+    nagiosharder status aux1
+    nagiosharder status aux1/http
+
+    nagiosharder acknowledge aux1 [message]
+
+    nagiosharder acknowledge aux1/http [message]
+    nagiosharder ack aux1/http [message]
+
+    nagiosharder unacknowledge aux1/http
+    nagiosharder unack aux1/http
+
+    nagiosharder mute aux1
+    nagiosharder mute aux1/http
+    nagiosharder disable_notifications aux1/http
+
+    nagiosharder unmute aux1
+    nagiosharder unmute aux1/http
+    nagiosharder enable_notifications aux1/http
+
+    nagiosharder check aux1/http
+
+    nagiosharder downtime aux1 15
+    nagiosharder downtime aux1/http 15
+
+    nagiosharder problems
+    nagiosharder problems http-services
+
+    nagiosharder acknowledged
+    nagiosharder acked http-services
+
+    nagiosharder muted
+    nagiosharder muted http-services
+
+    nagiosharder triage
+    nagiosharder unhandled
+    nagiosharder unhandled http-services
+      HELP
+    end
+  end
+end

data/lib/nagiosharder/filters.rb

@@ -0,0 +1,140 @@
+module Nagiosharder
+  module Filters
+
+    # irb> Nagiosharder::Filters.value(:host)
+    # => 0
+    # irb> Nagiosharder::Filters.value(:host, :state_acknowledged)
+    # => 4
+    # irb> Nagiosharder::Filters.value(:host, :state_acknowledged, :checks_disabled)
+    # => 20
+    def self.value(type, *args)
+      args.each.inject(0) do |i, arg|
+        begin
+          i += const_get("#{type}_#{arg}".upcase)
+        rescue NameError
+          i
+        end
+      end
+    end
+
+    #/********************* EXTENDED INFO CGI DISPLAY TYPES  *********************/
+
+    DISPLAY_PROCESS_INFO=0
+    DISPLAY_HOST_INFO=1
+    DISPLAY_SERVICE_INFO=2
+    DISPLAY_COMMENTS=3
+    DISPLAY_PERFORMANCE=4
+    DISPLAY_HOSTGROUP_INFO=5
+    DISPLAY_DOWNTIME=6
+    DISPLAY_SCHEDULING_QUEUE=7
+    DISPLAY_SERVICEGROUP_INFO=8
+
+
+    #/************************ COMMAND CGI COMMAND MODES *************************/
+
+    CMDMODE_NONE=0
+    CMDMODE_REQUEST=1
+    CMDMODE_COMMIT=2
+
+
+
+    #/******************** HOST AND SERVICE NOTIFICATION TYPES ******************/
+
+    NOTIFICATION_ALL=0 #/* all service and host notifications */
+    NOTIFICATION_SERVICE_ALL=1 #/* all types of service notifications */
+    NOTIFICATION_HOST_ALL=2 #/* all types of host notifications */
+    NOTIFICATION_SERVICE_WARNING=4
+    NOTIFICATION_SERVICE_UNKNOWN=8
+    NOTIFICATION_SERVICE_CRITICAL=16
+    NOTIFICATION_SERVICE_RECOVERY=32
+    NOTIFICATION_HOST_DOWN=64
+    NOTIFICATION_HOST_UNREACHABLE=128
+    NOTIFICATION_HOST_RECOVERY=256
+    NOTIFICATION_SERVICE_ACK=512
+    NOTIFICATION_HOST_ACK=1024
+    NOTIFICATION_SERVICE_FLAP=2048
+    NOTIFICATION_HOST_FLAP=4096
+    NOTIFICATION_SERVICE_CUSTOM=8192
+    NOTIFICATION_HOST_CUSTOM=16384
+
+
+    #/********************** HOST AND SERVICE ALERT TYPES **********************/
+
+    HISTORY_ALL=0 #/* all service and host alert */
+    HISTORY_SERVICE_ALL=1 #/* all types of service alerts */
+    HISTORY_HOST_ALL=2 #/* all types of host alerts */
+    HISTORY_SERVICE_WARNING=4
+    HISTORY_SERVICE_UNKNOWN=8
+    HISTORY_SERVICE_CRITICAL=16
+    HISTORY_SERVICE_RECOVERY=32
+    HISTORY_HOST_DOWN=64
+    HISTORY_HOST_UNREACHABLE=128
+    HISTORY_HOST_RECOVERY=256
+
+
+    #/****************************** SORT TYPES  *******************************/
+
+    SORT_NONE=0
+    SORT_ASCENDING=1
+    SORT_DESCENDING=2
+
+
+    #/***************************** SORT OPTIONS  ******************************/
+
+    SORT_NOTHING=0
+    SORT_HOSTNAME=1
+    SORT_SERVICENAME=2
+    SORT_SERVICESTATUS=3
+    SORT_LASTCHECKTIME=4
+    SORT_CURRENTATTEMPT=5
+    SORT_STATEDURATION=6
+    SORT_NEXTCHECKTIME=7
+    SORT_HOSTSTATUS=8
+    SORT_HOSTURGENCY=9
+
+
+    #/****************** HOST AND SERVICE FILTER PROPERTIES  *******************/
+
+    HOST_SCHEDULED_DOWNTIME=1
+    HOST_NO_SCHEDULED_DOWNTIME=2
+    HOST_STATE_ACKNOWLEDGED=4
+    HOST_STATE_UNACKNOWLEDGED=8
+    HOST_CHECKS_DISABLED=16
+    HOST_CHECKS_ENABLED=32
+    HOST_EVENT_HANDLER_DISABLED=64
+    HOST_EVENT_HANDLER_ENABLED=128
+    HOST_FLAP_DETECTION_DISABLED=256
+    HOST_FLAP_DETECTION_ENABLED=512
+    HOST_IS_FLAPPING=1024
+    HOST_IS_NOT_FLAPPING=2048
+    HOST_NOTIFICATIONS_DISABLED=4096
+    HOST_NOTIFICATIONS_ENABLED=8192
+    HOST_PASSIVE_CHECKS_DISABLED=16384
+    HOST_PASSIVE_CHECKS_ENABLED=32768
+    HOST_PASSIVE_CHECK=65536
+    HOST_ACTIVE_CHECK=131072
+    HOST_HARD_STATE=262144
+    HOST_SOFT_STATE=524288
+
+    SERVICE_SCHEDULED_DOWNTIME=1
+    SERVICE_NO_SCHEDULED_DOWNTIME=2
+    SERVICE_STATE_ACKNOWLEDGED=4
+    SERVICE_STATE_UNACKNOWLEDGED=8
+    SERVICE_CHECKS_DISABLED=16
+    SERVICE_CHECKS_ENABLED=32
+    SERVICE_EVENT_HANDLER_DISABLED=64
+    SERVICE_EVENT_HANDLER_ENABLED=128
+    SERVICE_FLAP_DETECTION_ENABLED=256
+    SERVICE_FLAP_DETECTION_DISABLED=512
+    SERVICE_IS_FLAPPING=1024
+    SERVICE_IS_NOT_FLAPPING=2048
+    SERVICE_NOTIFICATIONS_DISABLED=4096
+    SERVICE_NOTIFICATIONS_ENABLED=8192
+    SERVICE_PASSIVE_CHECKS_DISABLED=16384
+    SERVICE_PASSIVE_CHECKS_ENABLED=32768
+    SERVICE_PASSIVE_CHECK=65536
+    SERVICE_ACTIVE_CHECK=131072
+    SERVICE_HARD_STATE=262144
+    SERVICE_SOFT_STATE=524288
+  end
+end

data/nagiosharder.gemspec

@@ -1,66 +1,29 @@
-# Generated by jeweler
-# DO NOT EDIT THIS FILE DIRECTLY
-# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
 # -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
 
 Gem::Specification.new do |s|
-  s.name = %q{nagiosharder}
-  s.version = "0.3.0"
-
-  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
-  s.authors = ["Joshua Nichols"]
-  s.date = %q{2011-05-05}
+  s.name        = "nagiosharder"
+  s.version     = "0.4.0.rc1"
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Josh Nichols", "Jesse Newland"]
+  s.email       = ["josh@technicalpickles.com", "jesse@github.com"]
+  s.homepage    = "http://github.com/railsmachine/nagiosharder"
+  s.summary     = %q{Nagios access at your ruby fingertips}
   s.description = %q{Nagios access at your ruby fingertips}
-  s.email = %q{josh@technicalpickles.com}
-  s.extra_rdoc_files = [
-    "LICENSE",
-    "README.rdoc"
-  ]
-  s.files = [
-    ".document",
-    "LICENSE",
-    "README.rdoc",
-    "Rakefile",
-    "lib/nagiosharder.rb",
-    "nagiosharder.gemspec",
-    "spec/nagiosharder_spec.rb",
-    "spec/spec.opts",
-    "spec/spec_helper.rb"
-  ]
-  s.homepage = %q{http://github.com/railsmachine/nagiosharder}
-  s.require_paths = ["lib"]
-  s.rubygems_version = %q{1.4.2}
-  s.summary = %q{Nagios access at your ruby fingertips}
-  s.test_files = [
-    "spec/nagiosharder_spec.rb",
-    "spec/spec_helper.rb"
-  ]
 
-  if s.respond_to? :specification_version then
-    s.specification_version = 3
+  s.rubyforge_project = "nagiosharder"
 
-    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
-      s.add_runtime_dependency(%q<rest-client>, ["~> 1.6.1"])
-      s.add_runtime_dependency(%q<nokogiri>, ["~> 1.4.3"])
-      s.add_runtime_dependency(%q<activesupport>, [">= 0"])
-      s.add_runtime_dependency(%q<httparty>, ["~> 0.6.1"])
-      s.add_runtime_dependency(%q<hashie>, ["~> 1.0.0"])
-      s.add_development_dependency(%q<rspec>, [">= 1.2.9"])
-    else
-      s.add_dependency(%q<rest-client>, ["~> 1.6.1"])
-      s.add_dependency(%q<nokogiri>, ["~> 1.4.3"])
-      s.add_dependency(%q<activesupport>, [">= 0"])
-      s.add_dependency(%q<httparty>, ["~> 0.6.1"])
-      s.add_dependency(%q<hashie>, ["~> 1.0.0"])
-      s.add_dependency(%q<rspec>, [">= 1.2.9"])
-    end
-  else
-    s.add_dependency(%q<rest-client>, ["~> 1.6.1"])
-    s.add_dependency(%q<nokogiri>, ["~> 1.4.3"])
-    s.add_dependency(%q<activesupport>, [">= 0"])
-    s.add_dependency(%q<httparty>, ["~> 0.6.1"])
-    s.add_dependency(%q<hashie>, ["~> 1.0.0"])
-    s.add_dependency(%q<rspec>, [">= 1.2.9"])
-  end
-end
+  s.add_dependency 'rest-client', '~> 1.6.1'
+  s.add_dependency 'nokogiri'
+  s.add_dependency 'activesupport'
+  s.add_dependency 'i18n'
+  s.add_dependency 'terminal-table'
+  s.add_dependency 'httparty', '~> 0.8.3'
+  s.add_dependency 'hashie', '~> 1.2.0'
+  s.add_development_dependency "rspec", ">= 1.2.9"
 
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = ['nagiosharder']
+  s.require_paths = ["lib"]
+end