nagiosharder 0.5.0.rc1 → 0.5.0

data/lib/nagiosharder/cli.rb

@@ -17,10 +17,24 @@ class NagiosHarder
     def run
       return_value = case command
       when 'status'
+        status_checks = []
+        hostnames_and_check_names = [param,*the_rest].compact
+
+        # for each hostname or checkname we got
+        hostnames_and_check_names.each do |check|
+          host,service = check.split("/")
+          check_info = client.host_status(host)
+
+          # If we were asked about a specific service, get rid of the ones
+          # that they didn't ask for
+          check_info.reject! { |name,s| service and service != name }
+
+          # accumulate our results
+          status_checks += check_info.map { |name,service| service }
+        end
+
         service_table do
-          client.host_status(host).select do |name, s|
-            service.nil? || service == name
-          end.map { |name, s| s }
+          status_checks
         end
         true
       when /^(ack|acknowledged)$/
@@ -42,10 +56,15 @@ class NagiosHarder
           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)
+        if the_rest == []
+          puts "Downtime requires duration in minutes."
+          false
         else
-          client.schedule_host_downtime(host, :type => :fixed, :start_time => Time.now, :end_time => Time.now + the_rest.first.to_i.minutes)
+          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
         end
       when 'problems'
         service_table do
@@ -96,6 +115,16 @@ class NagiosHarder
           client.service_status(params)
         end
         true
+      when /^servicegroups/
+        groups_table do
+          client.servicegroups_summary()
+        end
+        true
+      when /^hostgroups/
+        groups_table do
+          client.hostgroups_summary()
+        end
+        true
       when /^muted/
         service_table do
           params = {
@@ -107,6 +136,17 @@ class NagiosHarder
           client.service_status(params)
         end
         true
+      when /^scheduled_downtime/
+        service_table do
+          params = {
+            :service_props => [
+              :scheduled_downtime,
+            ]
+          }
+          params[:group] = param
+          client.service_status(params)
+        end
+        true
       when /^(acked|acknowledged)/
         service_table do
           params = {
@@ -130,7 +170,7 @@ class NagiosHarder
       if return_value
         0
       else
-        puts "Sorry, bro, nagios didn't like that."
+        puts "Sorry yo, nagios didn't like that."
         1
       end
     end
@@ -193,6 +233,30 @@ class NagiosHarder
       options
     end
 
+    def groups_table
+      table = Terminal::Table.new(:headings => ['Group', 'Host Up', 'Host Down', 'Service Ok', 'Service Warning', 'Service Critical', 'Service Unknown']) do |t|
+        groups = yield
+        groups.each do |name, group|
+          t << group_row(group)
+        end
+        t
+      end
+      table.align_column(1, :right)
+      puts table
+    end
+
+    def group_row(group)
+      [
+        group['group'],
+        group['host_status_counts']['up'],
+        group['host_status_counts']['down'],
+        group['service_status_counts']['ok'],
+        group['service_status_counts']['warning'],
+        group['service_status_counts']['critical'],
+        group['service_status_counts']['unknown']
+      ]
+    end
+
     def service_table
       table = Terminal::Table.new(:headings => ['Service', 'Status', 'Details']) do |t|
         services = yield
@@ -208,6 +272,7 @@ class NagiosHarder
     def service_row(service)
       service['status'] << "/ACK" if service['acknowledged']
       service['status'] << "/MUTE" if service['notifications_disabled']
+      service['status'] << "/DOWNTIME" if service['downtime']
       service['status'] << "/COMMENT" if service['comments_url']
       [
         service['host']+"/"+service["service"],
@@ -275,6 +340,9 @@ COMMANDS:
     nagiosharder triage
     nagiosharder unhandled
     nagiosharder unhandled http-services
+
+    nagiosharder hostgroups
+    nagiosharder servicegroups
       HELP
     end
   end

data/lib/nagiosharder/commands.rb

@@ -0,0 +1,173 @@
+class NagiosHarder
+  COMMANDS = {
+    :none => 0,
+    :add_host_comment => 1,
+    :del_host_comment => 2,
+    :add_service_comment => 3,
+    :del_service_comment => 4,
+    :enable_service_check => 5,
+    :disable_service_check => 6,
+    :schedule_service_check => 7,
+    :delay_service_notification => 9,
+    :delay_host_notification => 10,
+    :disable_notifications => 11,
+    :enable_notifications => 12,
+    :restart_process => 13,
+    :shutdown_process => 14,
+    :enable_host_service_checks => 15,
+    :disable_host_service_checks => 16,
+    :schedule_host_service_checks => 17,
+    :delay_host_service_notifications => 19,
+    :del_all_host_comments => 20,
+    :del_all_service_comments => 21,
+    :enable_service_notifications => 22,
+    :disable_service_notifications => 23,
+    :enable_host_notifications => 24,
+    :disable_host_notifications => 25,
+    :enable_all_notifications_beyond_host => 26,
+    :disable_all_notifications_beyond_host => 27,
+    :enable_host_service_notifications => 28,
+    :disable_host_service_notifications => 29,
+    :process_service_check_result => 30,
+    :save_state_information => 31,
+    :read_state_information => 32,
+    :acknowledge_host_problem => 33,
+    :acknowledge_service_problem => 34,
+    :start_executing_service_checks => 35,
+    :stop_executing_service_checks => 36,
+    :start_accepting_passive_service_checks => 37,
+    :stop_accepting_passive_service_checks => 38,
+    :enable_passive_service_checks => 39,
+    :disable_passive_service_checks => 40,
+    :enable_event_handlers => 41,
+    :disable_event_handlers => 42,
+    :enable_host_event_handler => 43,
+    :disable_host_event_handler => 44,
+    :enable_service_event_handler => 45,
+    :disable_service_event_handler => 46,
+    :enable_host_check => 47,
+    :disable_host_check => 48,
+    :start_obsessing_over_service_checks => 49,
+    :stop_obsessing_over_service_checks => 50,
+    :remove_host_acknowledgement => 51,
+    :remove_service_acknowledgement => 52,
+    :schedule_forced_host_service_checks => 53,
+    :schedule_forced_service_check => 54,
+    :schedule_host_downtime => 55,
+    :schedule_service_downtime => 56,
+    :enable_host_flap_detection => 57,
+    :disable_host_flap_detection => 58,
+    :enable_service_flap_detection => 59,
+    :disable_service_flap_detection => 60,
+    :enable_flap_detection => 61,
+    :disable_flap_detection => 62,
+    :enable_hostgroup_service_notifications => 63,
+    :disable_hostgroup_service_notifications => 64,
+    :enable_hostgroup_host_notifications => 65,
+    :disable_hostgroup_host_notifications => 66,
+    :enable_hostgroup_service_checks => 67,
+    :disable_hostgroup_service_checks => 68,
+    :unimplemented_69 => 69,
+    :unimplemented_70 => 70,
+    :unimplemented_71 => 71,
+    :unimplemented_72 => 72,
+    :unimplemented_73 => 73,
+    :unimplemented_74 => 74,
+    :unimplemented_75 => 75,
+    :unimplemented_76 => 76,
+    :unimplemented_77 => 77,
+    :del_host_downtime => 78,
+    :del_service_downtime => 79,
+    :enable_performance_data => 82,
+    :disable_performance_data => 83,
+    :schedule_hostgroup_host_downtime => 84,
+    :schedule_hostgroup_service_downtime => 85,
+    :schedule_host_service_downtime => 86,
+    :process_host_check_result => 87,
+    :start_executing_host_checks => 88,
+    :stop_executing_host_checks => 89,
+    :start_accepting_passive_host_checks => 90,
+    :stop_accepting_passive_host_checks => 91,
+    :enable_passive_host_checks => 92,
+    :disable_passive_host_checks => 93,
+    :start_obsessing_over_host_checks => 94,
+    :stop_obsessing_over_host_checks => 95,
+    :schedule_host_check => 96,
+    :schedule_forced_host_check => 98,
+    :start_obsessing_over_svc => 99,
+    :stop_obsessing_over_svc => 100,
+    :start_obsessing_over_host => 101,
+    :stop_obsessing_over_host => 102,
+    :enable_hostgroup_host_checks => 103,
+    :disable_hostgroup_host_checks => 104,
+    :enable_hostgroup_passive_service_checks => 105,
+    :disable_hostgroup_passive_service_checks => 106,
+    :enable_hostgroup_passive_host_checks => 107,
+    :disable_hostgroup_passive_host_checks => 108,
+    :enable_servicegroup_service_notifications => 109,
+    :disable_servicegroup_service_notifications => 110,
+    :enable_servicegroup_host_notifications => 111,
+    :disable_servicegroup_host_notifications => 112,
+    :enable_servicegroup_service_checks => 113,
+    :disable_servicegroup_service_checks => 114,
+    :enable_servicegroup_host_checks => 115,
+    :disable_servicegroup_host_checks => 116,
+    :enable_servicegroup_passive_service_checks => 117,
+    :disable_servicegroup_passive_service_checks => 118,
+    :enable_servicegroup_passive_host_checks => 119,
+    :disable_servicegroup_passive_host_checks => 120,
+    :schedule_servicegroup_host_downtime => 121,
+    :schedule_servicegroup_service_downtime => 122,
+    :change_global_host_event_handler => 123,
+    :change_global_service_event_handler => 124,
+    :change_host_event_handler => 125,
+    :change_service_event_handler => 126,
+    :change_host_check_command => 127,
+    :change_service_check_command => 128,
+    :change_normal_host_check_interval => 129,
+    :change_normal_service_check_interval => 130,
+    :change_retry_service_check_interval => 131,
+    :change_max_host_check_attempts => 132,
+    :change_max_service_check_attempts => 133,
+    :schedule_and_propagate_triggered_host_downtime => 134,
+    :enable_host_and_child_notifications => 135,
+    :disable_host_and_child_notifications => 136,
+    :schedule_and_propagate_host_downtime => 137,
+    :enable_service_freshness_checks => 138,
+    :disable_service_freshness_checks => 139,
+    :enable_host_freshness_checks => 140,
+    :disable_host_freshness_checks => 141,
+    :set_host_notification_number => 142,
+    :set_service_notification_number => 143,
+    :change_host_check_timeperiod => 144,
+    :change_service_check_timeperiod => 145,
+    :process_file => 146,
+    :change_custom_host_var => 147,
+    :change_custom_service_var => 148,
+    :change_custom_contact_var => 149,
+    :enable_contact_host_notifications => 150,
+    :disable_contact_host_notifications => 151,
+    :enable_contact_service_notifications => 152,
+    :disable_contact_service_notifications => 153,
+    :enable_contactgroup_host_notifications => 154,
+    :disable_contactgroup_host_notifications => 155,
+    :enable_contactgroup_service_notifications => 156,
+    :disable_contactgroup_service_notifications => 157,
+    :change_retry_host_check_interval => 158,
+    :send_custom_host_notification => 159,
+    :send_custom_service_notification => 160,
+    :change_host_notification_timeperiod => 161,
+    :change_service_notification_timeperiod => 162,
+    :change_contact_host_notification_timeperiod => 163,
+    :change_contact_service_notification_timeperiod => 164,
+    :change_host_modattr => 165,
+    :change_service_modattr => 166,
+    :change_contact_modattr => 167,
+    :change_contact_modhattr => 168,
+    :change_contact_modsattr => 169,
+    :del_downtime_by_host_name => 170,
+    :del_downtime_by_hostgroup_name => 171,
+    :del_downtime_by_start_time_comment => 172,
+    :custom_command => 999
+  }.freeze unless defined? COMMANDS
+end

data/lib/nagiosharder/filters.rb

@@ -56,6 +56,13 @@ module Nagiosharder
     NOTIFICATION_HOST_FLAP=4096
     NOTIFICATION_SERVICE_CUSTOM=8192
     NOTIFICATION_HOST_CUSTOM=16384
+    
+    
+    #/********************** HOST AND SERVICE STATE TYPES **********************/
+
+    HISTORY_STATE_ALL=0
+    HISTORY_STATE_SOFT=1
+    HISTORY_STATE_HARD=2
 
 
     #/********************** HOST AND SERVICE ALERT TYPES **********************/

data/nagiosharder.gemspec

@@ -1,29 +1,31 @@
-# -*- encoding: utf-8 -*-
-$:.push File.expand_path("../lib", __FILE__)
+# encoding: utf-8
+$:.push File.expand_path('../lib', __FILE__)
 
 Gem::Specification.new do |s|
-  s.name        = "nagiosharder"
-  s.version     = "0.5.0.rc1"
+  s.name        = 'nagiosharder'
+  s.version     = '0.5.0'
+  s.date        = Time.now.strftime('%Y-%m-%d')
   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.authors     = ['Josh Nichols', 'Jesse Newland', 'Travis Graham']
+  s.email       = ['josh@technicalpickles.com', 'jesse@github.com', 'travis@railsmachine.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.rubyforge_project = "nagiosharder"
+  s.rubyforge_project = 'nagiosharder'
 
-  s.add_dependency 'rest-client', '~> 1.6.1'
+  s.add_dependency 'rest-client'
   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 'httparty'
   s.add_dependency 'hashie', '~> 1.2.0'
-  s.add_development_dependency "rspec", ">= 1.2.9"
+  s.add_development_dependency 'rspec', '>= 1.2.9'
+  s.add_development_dependency 'webmock'
 
   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"]
+  s.require_paths = ['lib']
 end

data/spec/site_spec.rb

@@ -0,0 +1,141 @@
+require 'spec_helper'
+
+describe 'NagiosHarder::Site' do
+  let(:client) do
+    client = NagiosHarder::Site.new(ENV['NAGIOS_URL']      || "",
+                                    ENV['NAGIOS_USER']     || "",
+                                    ENV['NAGIOS_PASSWORD'] || "")
+    client.stub(:post)
+    client
+  end
+
+  let(:successful_response) do
+    Class.new() do
+      def code
+        200
+      end
+
+      def body
+        "successful"
+      end
+    end.new
+  end
+
+  it 'should initialize' do
+    client
+  end
+
+  it 'should call post in post_command' do
+    client.should_receive(:post) do |url, params|
+      params[:body][:foo].should == :bar
+      # Return an instance that makes the rest of the method work
+      successful_response
+    end
+
+    client.post_command({:foo => :bar}).should == true
+  end
+
+  it 'should let you acknowledge a service' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should       == 'example.com'
+      params[:service].should    == 'http'
+      params[:com_data].should   == 'Looking'
+      params[:com_author].should == client.user
+      params[:cmd_typ].should    == 34
+    end
+
+    client.acknowledge_service('example.com', 'http', 'Looking')
+  end
+
+  it 'should let you acknowledge a host' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should       == 'example.com'
+      params[:com_data].should   == 'Looking'
+      params[:com_author].should == client.user
+      params[:cmd_typ].should    == 33
+    end
+
+    client.acknowledge_host('example.com', 'Looking')
+  end
+
+  it 'should let you unacknowledge a service' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should       == 'example.com'
+      params[:service].should    == 'http'
+      params[:cmd_typ].should    == 52
+    end
+
+    client.unacknowledge_service('example.com', 'http')
+  end
+
+  it 'should let you schedule service downtime' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should       == 'example.com'
+      params[:service].should    == 'http'
+      params[:cmd_typ].should    == 56
+      params[:com_author].should_not be_blank
+      params[:com_data].should_not   be_blank
+    end
+
+    client.schedule_service_downtime('example.com', 'http')
+  end
+
+  it 'should let you schedule host downtime' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should       == 'example.com'
+      params[:cmd_typ].should    == 55
+      params[:com_author].should_not be_blank
+      params[:com_data].should_not   be_blank
+    end
+
+    client.schedule_host_downtime('example.com')
+  end
+
+  it 'should let you cancel downtime' do
+    client.should_receive(:post_command) do |params|
+      params[:cmd_typ].should == 78
+      params[:down_id].should == 0
+    end
+
+    client.cancel_downtime(0)
+  end
+
+  it 'should let you schedule a host check' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should        == 'example.com'
+      params[:cmd_typ].should     == 96
+      params[:force_check].should == true
+    end
+
+    client.schedule_host_check('example.com')
+  end
+
+  it 'should let you schedule a service check' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should        == 'example.com'
+      params[:service].should     == 'http'
+      params[:cmd_typ].should     == 7
+      params[:force_check].should == true
+    end
+
+    client.schedule_service_check('example.com','http')
+  end
+
+  it 'should let you disable service notifications' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should        == 'example.com'
+      params[:service].should     == 'http'
+      params[:cmd_typ].should     == 23
+    end
+    client.disable_service_notifications('example.com', 'http')
+  end
+
+  it 'should let you enable service notifications' do
+    client.should_receive(:post_command) do |params|
+      params[:host].should        == 'example.com'
+      params[:service].should     == 'http'
+      params[:cmd_typ].should     == 22
+    end
+    client.enable_service_notifications('example.com', 'http')
+  end
+end