Untitled diff

Created Diff never expires
46 removals
204 lines
37 additions
194 lines
#!/usr/bin/env python
#!/usr/bin/env python


import httplib
import httplib
try:
try:
import simplejson as json
import simplejson as json
except ImportError:
except ImportError:
import json
import json
import os
import os
import sys
import sys
from urlparse import urljoin
from urlparse import urljoin


try:
try:
import requests
import requests
except ImportError:
except ImportError:
raise ImportError('Missing dependency "requests". Do ``pip install requests``.')
raise ImportError('Missing dependency requests. Do ``pip install requests``.')


try:
try:
import yaml
import yaml
except ImportError:
except ImportError:
raise ImportError('Missing dependency "pyyaml". Do ``pip install pyyaml``.')
raise ImportError('Missing dependency pyyaml. Do ``pip install pyyaml``.')


# ST2 configuration
# ST2 configuration
ST2_CONFIG_FILE = './config.yaml'


ST2_API_BASE_URL = 'http://localhost:9101/v1'
ST2_API_BASE_URL = 'http://localhost:9101/v1'
ST2_AUTH_BASE_URL = 'http://localhost:9100'
ST2_AUTH_BASE_URL = 'http://localhost:9100'
ST2_USERNAME = None
ST2_USERNAME = None
ST2_PASSWORD = None
ST2_PASSWORD = None
ST2_AUTH_TOKEN = None
ST2_AUTH_TOKEN = None


ST2_AUTH_PATH = 'tokens'
ST2_AUTH_PATH = 'tokens'
ST2_WEBHOOKS_PATH = 'webhooks/st2/'
ST2_WEBHOOKS_PATH = 'webhooks/st2/'
ST2_TRIGGERS_PATH = 'triggertypes/'
ST2_TRIGGERS_PATH = 'triggertypes/'
ST2_TRIGGERTYPE_PACK = 'e_nagios'
ST2_TRIGGERTYPE_PACK = 'nagios'
ST2_TRIGGERTYPE_NAME = 'service_state_change'
ST2_TRIGGERTYPE_NAME = 'service-state-change'
ST2_TRIGGERTYPE_REF = '.'.join([ST2_TRIGGERTYPE_PACK, ST2_TRIGGERTYPE_NAME])
ST2_TRIGGERTYPE_REF = '.'.join([ST2_TRIGGERTYPE_PACK, ST2_TRIGGERTYPE_NAME])


STATE_MESSAGE = {
'OK': 'All is well on the Western front.',
'WARNING': 'We gots a warning yo!',
'UNKNOWN': 'It be unknown...',
'CRITICAL': 'Critical!'
}

REGISTERED_WITH_ST2 = False
REGISTERED_WITH_ST2 = False


OK_CODES = [httplib.OK, httplib.CREATED, httplib.ACCEPTED, httplib.CONFLICT]
OK_CODES = [httplib.OK, httplib.CREATED, httplib.ACCEPTED, httplib.CONFLICT]


STATE_MESSAGE = {
'OK': '[OK] Service/Host is back to normal.',
'WARNING': '[WARNING] Service/Host warning alert!',
'UNKNOWN': '[UNKNOWN] Service/Host is in the unknown state.',
'CRITICAL': '[CRITICAL] Pay attention immediately!'
}


def _create_trigger_type():
def _create_trigger_type():
try:
try:
url = _get_st2_triggers_url()
url = _get_st2_triggers_url()
payload = {
payload = {
'name': ST2_TRIGGERTYPE_NAME,
'name': ST2_TRIGGERTYPE_NAME,
'pack': ST2_TRIGGERTYPE_PACK,
'pack': ST2_TRIGGERTYPE_PACK,
'description': 'Trigger type for nagios event handler.'
'description': 'Trigger type for nagios event handler.'
}
}
# sys.stdout.write('POST: %s: Body: %s\n' % (url, payload))
# sys.stdout.write('POST: %s: Body: %s\n' % (url, payload))
headers = {}
headers = {}
headers['Content-Type'] = 'application/json; charset=utf-8'
headers['Content-Type'] = 'application/json; charset=utf-8'


if ST2_AUTH_TOKEN:
if ST2_AUTH_TOKEN:
headers['X-Auth-Token'] = ST2_AUTH_TOKEN
headers['X-Auth-Token'] = ST2_AUTH_TOKEN


post_resp = requests.post(url, data=json.dumps(payload), headers=headers, verify=False)
post_resp = requests.post(url, data=json.dumps(payload), headers=headers)
except:
except:
sys.stderr.write('Unable to register trigger type with st2.')
sys.stderr.write('Unable to register trigger type with st2.')
raise
raise
else:
else:
status = post_resp.status_code
status = post_resp.status_code
if status not in OK_CODES:
if status not in OK_CODES:
sys.stderr.write('Failed to register trigger type with st2. HTTP_CODE: %d\n' %
sys.stderr.write('Failed to register trigger type with st2. HTTP_CODE: %d\n' %
status)
status)
raise
raise
else:
else:
sys.stdout.write('Registered trigger type with st2.\n')
sys.stdout.write('Registered trigger type with st2.\n')




def _get_auth_url():
def _get_auth_url():
return urljoin(ST2_AUTH_BASE_URL, ST2_AUTH_PATH)
return urljoin(ST2_AUTH_BASE_URL, ST2_AUTH_PATH)




def _get_auth_token():
def _get_auth_token():
global ST2_AUTH_TOKEN
global ST2_AUTH_TOKEN
auth_url = _get_auth_url()
auth_url = _get_auth_url()
try:
try:
resp = requests.post(auth_url, json.dumps({'ttl': 5 * 60}),
resp = requests.post(auth_url, json.dumps({'ttl': 5 * 60}),
auth=(ST2_USERNAME, ST2_PASSWORD), verify=False)
auth=(ST2_USERNAME, ST2_PASSWORD))
except:
except:
sys.stderr.write('Cannot get auth token from st2. Trying unauthed.')
raise Exception('Cannot get auth token from st2. Will try unauthed.')
else:
else:
if resp.status_code not in OK_CODES:
raise Exception("Cannot authenticate: %s\n" % resp.text)

ST2_AUTH_TOKEN = resp.json()['token']
ST2_AUTH_TOKEN = resp.json()['token']




def _register_with_st2():
def _register_with_st2():
global REGISTERED_WITH_ST2
global REGISTERED_WITH_ST2
try:
try:
url = urljoin(_get_st2_triggers_url(), ST2_TRIGGERTYPE_REF)
url = urljoin(_get_st2_triggers_url(), ST2_TRIGGERTYPE_REF)
# sys.stdout.write('GET: %s\n' % url)
# sys.stdout.write('GET: %s\n' % url)
if not ST2_AUTH_TOKEN:
if not ST2_AUTH_TOKEN:
_get_auth_token()
_get_auth_token()


if ST2_AUTH_TOKEN:
if ST2_AUTH_TOKEN:
get_resp = requests.get(url, headers={'X-Auth-Token': ST2_AUTH_TOKEN}, verify=False)
get_resp = requests.get(url, headers={'X-Auth-Token': ST2_AUTH_TOKEN})
else:
else:
get_resp = requests.get(url, verify=False)
get_resp = requests.get(url)


if get_resp.status_code != httplib.OK:
if get_resp.status_code != httplib.OK:
_create_trigger_type()
_create_trigger_type()
else:
else:
body = json.loads(get_resp.text)
body = json.loads(get_resp.text)
if len(body) == 0:
if len(body) == 0:
_create_trigger_type()
_create_trigger_type()
except:
except:
raise
raise
else:
else:
REGISTERED_WITH_ST2 = True
REGISTERED_WITH_ST2 = True




def _get_st2_triggers_url():
def _get_st2_triggers_url():
url = urljoin(ST2_API_BASE_URL, ST2_TRIGGERS_PATH)
url = urljoin(ST2_API_BASE_URL, ST2_TRIGGERS_PATH)
return url
return url




def _get_st2_webhooks_url():
def _get_st2_webhooks_url():
url = urljoin(ST2_API_BASE_URL, ST2_WEBHOOKS_PATH)
url = urljoin(ST2_API_BASE_URL, ST2_WEBHOOKS_PATH)
return url
return url




def _post_event_to_st2(url, body):
def _post_event_to_st2(url, body):
headers = {}
headers = {}
headers['X-ST2-Integration'] = 'nagios.'
headers['X-ST2-Integration'] = 'nagios.'
headers['Content-Type'] = 'application/json; charset=utf-8'
headers['Content-Type'] = 'application/json; charset=utf-8'
if ST2_AUTH_TOKEN:
if ST2_AUTH_TOKEN:
headers['X-Auth-Token'] = ST2_AUTH_TOKEN
headers['X-Auth-Token'] = ST2_AUTH_TOKEN
try:
try:
sys.stdout.write('POST: url: %s, body: %s\n' % (url, body))
# sys.stdout.write('POST: url: %s, body: %s\n' % (url, body))
r = requests.post(url, data=json.dumps(body), headers=headers, verify=False)
r = requests.post(url, data=json.dumps(body), headers=headers)
except:
except:
sys.stderr.write('Cannot connect to st2 endpoint.')
sys.stderr.write('Cannot connect to st2 endpoint.')
else:
else:
status = r.status_code
status = r.status_code
if status not in OK_CODES:
if status not in OK_CODES:
sys.stderr.write('Failed posting nagios event to st2. HTTP_CODE: %d\n' % status)
sys.stderr.write('Failed posting nagios event to st2. HTTP_CODE: %d\n' % status)
else:
else:
sys.stdout.write('Sent nagios event to st2. HTTP_CODE: %d\n' % status)
sys.stdout.write('Sent nagios event to st2. HTTP_CODE: %d\n' % status)


def _get_payload(host, service, event_id, state, state_id, state_type, attempt):

def _get_payload(host, service, event_id, state, state_type, attempt):
payload = {}
payload = {}
payload['host'] = host
payload['host'] = host
payload['service'] = service
payload['service'] = service
payload['event_id'] = event_id
payload['event_id'] = event_id
payload['state'] = state
payload['state'] = state
payload['state_id'] = state_id
payload['state_type'] = state_type
payload['state_type'] = state_type
payload['attempt'] = attempt
payload['attempt'] = attempt
payload['msg'] = STATE_MESSAGE.get(state, 'Undefined state.')
payload['msg'] = STATE_MESSAGE.get(state, 'Undefined state.')
return payload
return payload



def main(args):
def main(args):
event_id = args[1]
event_id = args[1]
service = args[2]
service = args[2]
state = args[3]
state = args[3]
state_id = args[4]
state_type = args[4]
state_type = args[5]
attempt = args[5]
attempt = args[6]
host = args[6]
host = args[7]

payload = _get_payload(host, service, event_id, state, state_type, attempt)
body = {}
body = {}
body['trigger'] = ST2_TRIGGERTYPE_REF
body['trigger'] = ST2_TRIGGERTYPE_REF
body['payload'] = _get_payload(host, service, event_id, state, state_id, state_type, attempt)
body['payload'] = payload
_post_event_to_st2(_get_st2_webhooks_url(), body)
_post_event_to_st2(_get_st2_webhooks_url(), body)



if __name__ == '__main__':
if __name__ == '__main__':
if len(sys.argv) > 1:
st2_config_file = sys.argv[1]
sys.argv.pop(1)
else:
sys.stderr.write('Error: config file missing.\n')
sys.stderr.write('Usage: %s ST2_CONFIG_FILE\n' % sys.argv[0])
sys.exit(2)

try:
try:
if not os.path.exists(st2_config_file):
if not os.path.exists(ST2_CONFIG_FILE):
sys.stderr.write('Configuration file not found. Exiting.\n')
sys.stderr.write('Configuration file not found. Exiting.\n')
sys.exit(1)
sys.exit(1)


with open(st2_config_file) as f:
with open(ST2_CONFIG_FILE) as f:
config = yaml.safe_load(f)
config = yaml.safe_load(f)
ST2_USERNAME = config['st2_username']
ST2_USERNAME = config['st2_username']
ST2_PASSWORD = config['st2_password']
ST2_PASSWORD = config['st2_password']
ST2_API_BASE_URL = config['st2_api_base_url']
ST2_API_BASE_URL = config['st2_api_base_url']
ST2_AUTH_BASE_URL = config['st2_auth_base_url']
ST2_AUTH_BASE_URL = config['st2_auth_base_url']


if not REGISTERED_WITH_ST2:
if not REGISTERED_WITH_ST2:
_register_with_st2()
_register_with_st2()
except Exception as e:
except:
sys.stderr.write(
sys.stderr.write('Failed registering with st2. Won\'t post event.\n')
'Failed registering with st2. Won\'t post event.\n%s' % e)
# import traceback
# traceback.print_exc()
else:
else:
main(sys.argv)
main(sys.argv)