Skip to content

Instantly share code, notes, and snippets.

@sacreman
Created Nov 12, 2015
Embed
What would you like to do?
#! /usr/bin/env ruby
require 'main'
require 'yaml'
require 'httparty'
Main {
option('execute'){
argument :optional
description 'execute the check'
}
option('stack'){
argument :optional
description 'Specify The Stack to check'
}
option('username'){
argument :optional
description 'Specify the Username'
}
option('password'){
argument :optional
description 'Specify the Password'
}
option('authtoken'){
argument :optional
description 'Specify the authtoken'
}
option('account'){
argument :optional
description 'Specify the account'
}
option('url'){
argument :optional
defaults 'https://api.pingdom.com/api/2.0/checks'
description 'Optionally override the pingdom endpoint'
}
option('config'){
argument :optional
description 'Path of Config file'
}
def run
if params['execute'].value == true
# Capture a HUP signal and act accordingly
Signal.trap("HUP") { Process.kill("HUP"); Process.wait }
# Set variables from arguements
stack = "#{params['stack'].value}"
hostname = "#{params['stack'].value}.metapack.com"
username = "#{params['username'].value}"
password = "#{params['password'].value}"
authtoken = "#{params['authtoken'].value}"
account = "#{params['account'].value}"
url = "#{params['url'].value}"
# update params from config file if specified
if params['config'].value != "false"
config = YAML.load_file("#{params['config'].value}")
stack = config['stack']
hostname = "#{config['stack']}.metapack.com"
end
#Build response headers
auth =
{
:username => "#{username}",
:password => "#{password}"
}
headers =
{
"App-Key" => authtoken,
"Account-Email" => account
}
# Execute the api request
response =
HTTParty.get(
"#{url}?tags=dataloop",
:basic_auth => auth,
:headers => headers
)
# Select the HTTPS request for the stack in question
stack = response['checks'].select { |hash| hash['hostname'] == "#{stack}.metapack.com" }
https = stack
# Set the response vars
status = https.first['status']
lrtime = https.first['lastresponsetime']
# Build the nagios response
case status
when "up"
puts "OK | responsetime=#{lrtime};;;;"
exit_status 0
exit_success!
when "down"
puts "DOWN | responsetime=#{lrtime};;;;"
exit_status 2
exit_failure!
when "unconfirmed_down"
puts "UNCONFIRMED | responsetime=#{lrtime};;;;"
exit_status 1
exit_failure!
when "paused"
puts "PAUSED | responsetime=#{lrtime};;;;"
exit_status 1
exit_failure!
else
puts "UNKNOWN | responsetime=#{lrtime};;;;"
exit_status 3
exit_failure!
end
else
puts "--execute hasn't been specified check --help"
end
end
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment