#!/usr/bin/env ruby
# 
# Plugin for Nagios to check pgpool-II backends
#
# Lukasz Jagiello (c) 2008
# lukasz.jagiello@gforces.pl
# 
# pcp_node_info
# Usage: pcp_node_info timeout hostname port# username password nodeID
#
# check_pgpool.rb
# Usage: check_pgpool.rb nodeID

time='30'
port='9898'
user='user'
pass='password'
host='hostname'
pcp='/usr/bin/pcp_node_info'

if (ARGV.length != 1)
  puts "Usage: check_pgpool.rb nodeID"
  exit
end

node_status=`#{pcp} #{time} #{host} #{port} #{user} #{pass} #{ARGV[0]}`.split(/ /)

case node_status[2].to_i
when 2
	puts 'WORKING'
	exit
when 3
	puts 'CRITICAL' + node_status[0]
	abort("CRITICAL")
else
	puts 'UNKNOWN - ' + node_status[0]
	abort("UNKNOWN")
end