Nagios Support Forum

Code: Select all

#!/usr/bin/env ruby

require 'optparse'
require 'fileutils'

def self.parse(args)
options = {}
options[:port] = 5693

opt_parser = OptionParser.new do |opts|
opts.banner = 'Usage: createImportcfg.rb [options] -m -r --help'

# Mandatory arguments
opts.separator ' '
opts.separator 'Mandatory arguments:'

opts.on('-m', '--map-path', OptionParser::REQUIRED_ARGUMENT, "Path to NagVis maps '/usr/local/nagivs/etc/maps'") do |m|
options[:map_path] = m.to_s
end

opts.on('-r', '--reconfigure-nagios', OptionParser::REQUIRED_ARGUMENT, 'Path to reconfigure_nagios.sh script') do |r|
options[:reconfigure_script] = r.to_s
end

opts.on('--help') do
puts opts
exit
end

if ARGV.empty?
puts opts
exit
end
end

opt_parser.parse!(args)
options
end

def self.updateFiles(a, b)
Dir.open(Dir.pwd).each do |file|
next if File.directory? file
f_current = File.read(file)
f_updated = f_current.gsub(a, b)
File.open(file, 'w') do |f|
f.write(f_updated)
end
end
end

def self.maintenanceMode
options = parse(ARGV)

# Check if map location exists and put it in maintenanceMode
if File.directory?(options[:map_path]) && File.directory?(options[:reconfigure_script])
Dir.chdir(options[:map_path])
updateFiles("in_maintenance=0", "in_maintenance=1")
# Sleep for 15 seconds for maintenance popup to load
sleep(15)
begin
Dir.chdir(options[:reconfigure_script])
`#{options[:reconfigure_script]}/reconfigure_nagios.sh`
rescue
Dir.chdir(options[:map_path])
updateFiles("in_maintenance=1", "in_maintenance=0")
puts "Warning: Unable to execute #{options[:reconfigure_script]}/reconfigure_nagios.sh, rolling back maintenance."
return 1
end

# Sleep for 60 seconds for nagios to finish reconfiguration
sleep(60)

# Remove map from maintenanceMode
Dir.chdir(options[:map_path])
updateFiles("in_maintenance=1", "in_maintenance=0")

# Report success status
puts "SUCCESS: Reconfigured nagios and entered/exited maintenance mode successfully."
return 0
else
puts 'UNKNOWN: The paths passed as parameters do not exist, cannot reload nagios in maintenance mode.'
return 1
end
end

maintenanceMode