nagios-gearman-downtime 1.0.0 → 1.0.1

data/.gitignore

@@ -0,0 +1,20 @@
+.DS_store
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+bundle
+*.swp

data/README.md

@@ -0,0 +1,57 @@
+# Nagios Gearman Downtime
+
+Gearman Client & Worker for sending downtime and enable / disable notifications for service groups
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'nagios-gearman-downtime', 
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install nagios-gearman-downtime
+
+## Usage:
+  downtime_client set --duration=duration in minutes --gearman-server=Gearman server address
+
+## Available options are:
+
+  --object_type [host, 
+                 hostgroup_host, 
+                 hostgroup_svc, 
+                 servicegroup_host, 
+                 servicegroup_svc,
+                 service,
+                 enable_servicegroup,
+                 disable_servicegroup], default => host
+
+  --object_name  default => 'hostname -f', the name of the object type
+
+
+  --encryption true/false - default => false
+  --key Key to encrypt request - must be identical to the gearman_server key
+  --queue gearman server queue default => downtime
+
+
+  --start_time  default => now
+  --end_time    default => now + duration
+  --duration    time in minutes, default => 1 minute
+
+  --author    Author of downtime (default hostname) 
+  --comment   default => 'Downtime recieved by gearman server'
+
+Send Downtime request to Gearman server - downtime queue
+
+ 
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,8 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+end
+
+desc "Run tests"
+task :default => :test

data/bin/downtime_client

@@ -3,7 +3,7 @@ require 'gearman'
 require 'base64'
 require "openssl"
 require "thor"
-require "../lib/nagios-gearman-downtime"
+require "nagios-gearman-downtime"
 
 class Downtime < Thor
 

data/lib/nagios-gearman-downtime.rb

@@ -0,0 +1,76 @@
+module Nagios
+  module Gearman
+    module Downtime
+
+      def self.create_object_type(obj, obj_name)
+        case obj
+        when 'host' 
+          object_info = "SCHEDULE_HOST_SVC_DOWNTIME;#{obj_name}"
+        when 'hostgroup_host' 
+          object_info = "SCHEDULE_HOSTGROUP_HOST_DOWNTIME;#{obj_name}"
+        when 'hostgroup_svc' 
+          object_info = "SCHEDULE_HOSTGROUP_SVC_DOWNTIME;#{obj_name}"
+        when 'servicegroup_host' 
+          object_info = "SCHEDULE_SERVICEGROUP_HOST_DOWNTIME;#{obj_name}"
+        when 'servicegroup_svc' 
+          object_info = "SCHEDULE_SERVICEGROUP_SVC_DOWNTIME;#{obj_name}"
+        when 'service' 
+          object_info = "SCHEDULE_HOSTGROUP_HOST_DOWNTIME;#{obj_name}"
+        when 'enable_servicegroup'
+          object_info = "ENABLE_SERVICEGROUP_SVC_NOTIFICATIONS;#{obj_name}"
+        when 'disable_servicegroup'
+          object_info = "DISABLE_SERVICEGROUP_SVC_NOTIFICATIONS;#{obj_name}"
+        end
+      end
+
+
+      def self.build_payload(args)
+        object_info = create_object_type(args.delete(:object_type), args.delete(:object_name))
+        
+        if object_info.match(/enable|disable/i)
+          common_args = ''
+        else
+          common_args = "#{args.delete(:start_time)};#{args.delete(:end_time)};#{args.delete(:fixed)};#{args.delete(:trigger_id)};#{args.delete(:duration)};#{args.delete(:author)};#{args.delete(:comment)}"
+        end
+        
+        payload = "[#{Time.now.to_i}] #{object_info};#{common_args}"
+
+        if args.delete(:encryption)
+          begin
+            payload = aes256_encrypt(args.delete(:key), payload)
+          rescue Exception => e
+            puts "unable to encrypt: #{e}"
+          end
+        end
+        payload
+      end
+
+      def self.send_external_cmd(options)
+        puts "send_external_cmd options: #{options}"
+        client  = ::Gearman::Client.new(options[:gearman_job_server])
+        taskset = ::Gearman::TaskSet.new(client)
+        encoded_job = Base64.encode64(options[:job])
+        task = ::Gearman::Task.new(options[:queue], encoded_job) 
+        begin
+          puts "[client] Sending task: #{task.inspect}"
+          result = taskset.add_task(task)
+        rescue Exception => e
+          "Send external command failed: #{e}"
+        end
+      end
+
+      def self.aes256_encrypt(key, data)
+        key = null_padding(key)
+        aes = OpenSSL::Cipher.new('AES-256-ECB')
+        aes.encrypt
+        aes.key = key
+        aes.update(data) + aes.final
+      end
+
+      def self.null_padding(key)
+        padding = (32 - key.bytesize) if(key.kind_of?(String) && 32 != key.bytesize)
+        key += "\0" * padding
+      end
+    end
+  end
+end

data/lib/nagios-gearman-downtime/version.rb

@@ -0,0 +1,7 @@
+module Nagios
+  module Gearman
+    module Downtime
+      VERSION = "1.0.1"
+    end
+  end
+end

data/nagios-gearman-downtime.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+$LOAD_PATH.push File.expand_path("../lib", __FILE__)
+require 'nagios-gearman-downtime/version'
+
+Gem::Specification.new do |s|
+  s.name        = "nagios-gearman-downtime"
+  s.authors     = ["Ami Mahloof"]
+  s.email       = "ami.mahloof@gmail.com"
+  s.homepage    = "https://github.com/gtforge/nagios-gearman-downtime"
+  s.summary     = "Send external commands to mod-gearman"
+  s.description = "Gearman Client & Worker for sending downtime and enable / disable notifications for service groups"
+  s.add_dependency 'gearman-ruby'
+  s.add_dependency 'thor'
+  s.add_dependency 'bluepill'
+  s.files            = `git ls-files`.split("\n")
+  s.require_paths    = ["lib"]
+  s.executables = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.license = 'MIT'
+  s.version     = ::Nagios::Gearman::Downtime::VERSION
+end

data/worker_daemon/downtime_worker.pill

@@ -0,0 +1,10 @@
+downtime_worker = `which downtime_worker`
+Bluepill.application("downtime_worker", :log_file => "/var/log/bluepill-downtime_worker.log") do |app|
+  app.process("downtime_worker") do |process|
+    process.start_command      = "#{downtime_worker}"
+    process.stop_command       = "kill -QUIT {{PID}}"
+    process.restart_command    = "kill -HUP {{PID}}"
+    process.stdout             = process.stderr = "/var/log/bluepill-downtime_worker.log"
+    process.daemonize          = true
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: nagios-gearman-downtime
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-03-03 00:00:00.000000000 Z
+date: 2014-03-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gearman-ruby
@@ -43,6 +43,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: bluepill
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Gearman Client & Worker for sending downtime and enable / disable notifications
   for service groups
 email: ami.mahloof@gmail.com
@@ -52,8 +68,15 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- .gitignore
+- README.md
+- Rakefile
 - bin/downtime_client
 - bin/downtime_worker
+- lib/nagios-gearman-downtime.rb
+- lib/nagios-gearman-downtime/version.rb
+- nagios-gearman-downtime.gemspec
+- worker_daemon/downtime_worker.pill
 homepage: https://github.com/gtforge/nagios-gearman-downtime
 licenses:
 - MIT