[packages/nagios-alert-jabber.git] / nagios-jabber.alert

#!/usr/bin/python -u
# arekm@pld-linux.org, 2006-01
# glen@pld-linux.org, 2006-03-14
# glen@pld-linux.org,arekm@pld-linux.org, 2006-10-30 - added ssl support (for gmail.com)
# glen@pld-linux.org 2006-11-03 - made it work with jabber.pld-linux.org again
# glen@pld-linux.org,arekm@pld-linux.org, 2006-11-13 - added config file support
# glen@pld-linux.org, 2006-12-07 - added html messages support (-x), thx goes to to jajcus
# luzik@pld-linux.org, 2007-03 - added digest auth method(jabber.gda.pl)
# arekm@pld-linux.org, 2009-07 - added fallback accounts support
# usage:
#   jabber.alert [-x] [-a account_id][,otheraccount_id] [-J from_jid -P password] to_jid1 to_jid2 to_jid3

import os
import re
import sys
import getopt
import logging
import string
import time
import libxml2

from pyxmpp.jid import JID
from pyxmpp.message import Message
from pyxmpp.jabber.client import JabberClient
from pyxmpp.streamtls import TLSSettings

try:
	opts, args = getopt.getopt(sys.argv[1:], "J:P:a:dx")
except getopt.GetoptError, e:
	print >> sys.stderr, "%s: %s " % (sys.argv[0], e)
	sys.exit(1)

jids = []
html = False
debug = False

tjid = None
for o, a in opts:
	if o == '-d':
		debug = True
	if o == '-x':
		html = True
	if o == '-J':
		tjid = a
	if o == '-P':
		jids.append({ 'jid': tjid, 'password': a })
	if o == '-a':
		import ConfigParser

		config = ConfigParser.ConfigParser()
		config.read('/etc/nagios/jabber-notify.ini')

		for section in a.split(','):
			jids.append({ 'jid': config.get(section, 'jid'), 'password': config.get(section, 'password')})

recpt = args

for section in jids:
	if not section['jid'] or not section['password']:
		print >> sys.stderr, "%s: jid (-J) and password (-P) are required for `%s'" % (sys.argv[0], section)
		sys.exit(1)

if not jids:
	print >> sys.stderr, "%s: no configured jid accounts found" % sys.argv[0]
	sys.exit(1)

if not recpt:
	print >> sys.stderr, "%s: recipient jids are required" % sys.argv[0]
	sys.exit(1)

if debug:
	logger=logging.getLogger()
	logger.addHandler(logging.StreamHandler())
	logger.setLevel(logging.DEBUG)

subject = "Nagios alert"

body = ""
stdin_body = ""
do_print = True
for line in sys.stdin.readlines():
	stdin_body += line

body += stdin_body

if len(body.strip()) == 0:
	body = "(nagios-jabber.alert warning: missing message body)";

message_type = 'chat'

class Client(JabberClient):
	def session_started(self):
		if (html == True):
			import re
			message = re.sub('<.*?>', '', body)
			doc = libxml2.parseDoc('<body>' + body + '</body>')
			doc_element = doc.getRootElement().children
		else:
			message = body

		for r in recpt:

			jid_r = JID(r)
			msg = Message(to_jid = jid_r, body = message, subject = subject, stanza_type = message_type)

			if (html == True):
				node = msg.add_new_content('http://jabber.org/protocol/xhtml-im', 'html')
				xbody = node.newChild(None, "body", None)
				html_ns = xbody.newNs('http://www.w3.org/1999/xhtml', None)
				xbody.setNs(html_ns)
				xbody.addChildList(doc_element.docCopyNodeList(xbody.doc))

			self.stream.send(msg)
		self.disconnect()

	def stream_state_changed(self,state,arg):
		if debug:
			print "*** State changed: %s %r ***" % (state,arg)

err = []
for section in jids:
	jid = JID(section['jid'])
	if not jid.resource:
		jid = JID(jid.node, jid.domain, "Nagios")

	c = Client(jid, section['password'], auth_methods = ['sasl:DIGEST-MD5', 'sasl:PLAIN', 'digest'],
			tls_settings = TLSSettings(require = False, verify_peer = False))
	try:
		c.connect()
		try:
			c.loop(1)
		except Exception, e:
			err.append("ERROR1: %s: %s" % (section['jid'], e))
			c.disconnect()
			continue
		c.disconnect()
		# stop after first succeeded attempt
		sys.exit(0)
	except Exception, e:
		err.append("ERROR2: %s: %s" % (section['jid'], e))

print >> sys.stderr, "\n".join(err)
sys.exit(1)
Commit	Line	Data
02673eb0 ER	1	#!/usr/bin/python -u
	2	# arekm@pld-linux.org, 2006-01
	3	# glen@pld-linux.org, 2006-03-14
ad7ac473	4	# glen@pld-linux.org,arekm@pld-linux.org, 2006-10-30 - added ssl support (for gmail.com)
8a0d8b38	5	# glen@pld-linux.org 2006-11-03 - made it work with jabber.pld-linux.org again
20bb0aca	6	# glen@pld-linux.org,arekm@pld-linux.org, 2006-11-13 - added config file support
ebbef88a	7	# glen@pld-linux.org, 2006-12-07 - added html messages support (-x), thx goes to to jajcus
071905aa	8	# luzik@pld-linux.org, 2007-03 - added digest auth method(jabber.gda.pl)
b6621e54	9	# arekm@pld-linux.org, 2009-07 - added fallback accounts support
02673eb0	10	# usage:
b6621e54	11	# jabber.alert [-x] [-a account_id][,otheraccount_id] [-J from_jid -P password] to_jid1 to_jid2 to_jid3
02673eb0 ER	12
	13	import os
	14	import re
	15	import sys
	16	import getopt
eca05a2f	17	import logging
02673eb0 ER	18	import string
02673eb0 ER	19	import time
ebbef88a	20	import libxml2
02673eb0 ER	21
	22	from pyxmpp.jid import JID
	23	from pyxmpp.message import Message
	24	from pyxmpp.jabber.client import JabberClient
ad7ac473	25	from pyxmpp.streamtls import TLSSettings
02673eb0 ER	26
02673eb0 ER	27	try:
f985e683	28	opts, args = getopt.getopt(sys.argv[1:], "J:P:a:dx")
02673eb0	29	except getopt.GetoptError, e:
b6621e54	30	print >> sys.stderr, "%s: %s " % (sys.argv[0], e)
02673eb0 ER	31	sys.exit(1)
02673eb0 ER	32
b6621e54	33	jids = []
ebbef88a	34	html = False
f985e683	35	debug = False
02673eb0	36
b6621e54	37	tjid = None
02673eb0	38	for o, a in opts:
f985e683 AM	39	if o == '-d':
f985e683 AM	40	debug = True
ebbef88a ER	41	if o == '-x':
ebbef88a ER	42	html = True
02673eb0	43	if o == '-J':
b6621e54	44	tjid = a
02673eb0	45	if o == '-P':
b6621e54	46	jids.append({ 'jid': tjid, 'password': a })
20bb0aca ER	47	if o == '-a':
	48	import ConfigParser
	49
	50	config = ConfigParser.ConfigParser()
	51	config.read('/etc/nagios/jabber-notify.ini')
	52
b6621e54 AM	53	for section in a.split(','):
b6621e54 AM	54	jids.append({ 'jid': config.get(section, 'jid'), 'password': config.get(section, 'password')})
02673eb0 ER	55
	56	recpt = args
	57
b6621e54 AM	58	for section in jids:
	59	if not section['jid'] or not section['password']:
	60	print >> sys.stderr, "%s: jid (-J) and password (-P) are required for `%s'" % (sys.argv[0], section)
	61	sys.exit(1)
	62
	63	if not jids:
	64	print >> sys.stderr, "%s: no configured jid accounts found" % sys.argv[0]
02673eb0 ER	65	sys.exit(1)
02673eb0 ER	66
b6621e54 AM	67	if not recpt:
b6621e54 AM	68	print >> sys.stderr, "%s: recipient jids are required" % sys.argv[0]
02673eb0 ER	69	sys.exit(1)
02673eb0 ER	70
f985e683 AM	71	if debug:
	72	logger=logging.getLogger()
	73	logger.addHandler(logging.StreamHandler())
	74	logger.setLevel(logging.DEBUG)
02673eb0 ER	75
02673eb0 ER	76	subject = "Nagios alert"
02673eb0 ER	77
	78	body = ""
	79	stdin_body = ""
	80	do_print = True
	81	for line in sys.stdin.readlines():
	82	stdin_body += line
	83
	84	body += stdin_body
	85
8ccfe18b AM	86	if len(body.strip()) == 0:
	87	body = "(nagios-jabber.alert warning: missing message body)";
	88
02673eb0 ER	89	message_type = 'chat'
02673eb0 ER	90
02673eb0 ER	91	class Client(JabberClient):
02673eb0 ER	92	def session_started(self):
ebbef88a ER	93	if (html == True):
	94	import re
	95	message = re.sub('<.*?>', '', body)
b538c51f ER	96	doc = libxml2.parseDoc('<body>' + body + '</body>')
b538c51f ER	97	doc_element = doc.getRootElement().children
ebbef88a ER	98	else:
	99	message = body
	100
02673eb0	101	for r in recpt:
ebbef88a	102
02673eb0	103	jid_r = JID(r)
ebbef88a ER	104	msg = Message(to_jid = jid_r, body = message, subject = subject, stanza_type = message_type)
	105
	106	if (html == True):
	107	node = msg.add_new_content('http://jabber.org/protocol/xhtml-im', 'html')
	108	xbody = node.newChild(None, "body", None)
	109	html_ns = xbody.newNs('http://www.w3.org/1999/xhtml', None)
	110	xbody.setNs(html_ns)
b538c51f	111	xbody.addChildList(doc_element.docCopyNodeList(xbody.doc))
ebbef88a	112
02673eb0 ER	113	self.stream.send(msg)
02673eb0 ER	114	self.disconnect()
ad7ac473	115
eca05a2f AM	116	def stream_state_changed(self,state,arg):
	117	if debug:
	118	print "* State changed: %s %r *" % (state,arg)
	119
b6621e54 AM	120	err = []
	121	for section in jids:
	122	jid = JID(section['jid'])
	123	if not jid.resource:
	124	jid = JID(jid.node, jid.domain, "Nagios")
	125
	126	c = Client(jid, section['password'], auth_methods = ['sasl:DIGEST-MD5', 'sasl:PLAIN', 'digest'],
	127	tls_settings = TLSSettings(require = False, verify_peer = False))
	128	try:
	129	c.connect()
	130	try:
	131	c.loop(1)
	132	except Exception, e:
	133	err.append("ERROR1: %s: %s" % (section['jid'], e))
	134	c.disconnect()
	135	continue
	136	c.disconnect()
	137	# stop after first succeeded attempt
	138	sys.exit(0)
	139	except Exception, e:
	140	err.append("ERROR2: %s: %s" % (section['jid'], e))
	141
	142	print >> sys.stderr, "\n".join(err)
	143	sys.exit(1)