scripts/cachet_notify

191 lines
5.6 KiB
Plaintext
Raw Normal View History

2021-01-21 16:58:32 -05:00
#!/usr/bin/php5
<?php
if ($argc != 6) {
echo 'Usage: ' . basename(__FILE__) . ' cachet_component service_name service_state service_state_type service_output' . "\n";
exit(1);
}
$api_path = dirname(__FILE__) . '/.api_key';
if ( is_file($api_path) ) {
$api = parse_ini_file($api_path);
} else {
exit(1);
}
$cachet_url = $api['url'];
$api_key = $api['key'];
$incident_prefix = '[Icinga2]';
$cachet_notify_subscribers = true; // Enable subscribers notifcation for incidents creation and updates
$cachet_component = $argv[1];
$service_name = $argv[2];
$service_status = $argv[3];
$service_status_type = $argv[4];
$service_output = $argv[5];
define('CATCHET_STATUS_INVESTIGATING', 1);
define('CATCHET_STATUS_IDENTIFIED', 2);
define('CATCHET_STATUS_WATCHING', 3);
define('CATCHET_STATUS_FIXED', 4);
define('CATCHET_COMP_OPERATIONAL', 1);
define('CATCHET_COMP_PERFISSUES', 2);
define('CATCHET_COMP_PARTIALOUTAGE', 3);
define('CATCHET_COMP_MAJOROUTAGE', 4);
function cachet_query($api_part, $action = 'GET', $data = null) {
global $api_key, $cachet_url;
$ch = curl_init();
curl_setopt($ch, CURLOPT_URL, $cachet_url . $api_part);
curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, false);
curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
if (in_array($action, array('GET', 'POST', 'PUT'))) {
curl_setopt($ch, CURLOPT_CUSTOMREQUEST, $action);
}
if ($data !== null && is_array($data)) {
$ch_data = http_build_query($data);
curl_setopt($ch, CURLOPT_POST, 1);
curl_setopt($ch, CURLOPT_POSTFIELDS, $ch_data);
}
$ch_headers = array(
'X-Cachet-Token: ' . $api_key
);
curl_setopt($ch, CURLOPT_HTTPHEADER, $ch_headers);
curl_setopt($ch, CURLOPT_HEADER, false); // Don't return headers
curl_setopt($ch, CURLOPT_RETURNTRANSFER, true); // Return body
$http_body = curl_exec($ch);
$http_code = curl_getinfo($ch, CURLINFO_HTTP_CODE);
curl_close($ch);
return array('code' => $http_code, 'body' => json_decode($http_body));
}
/* Find Cachet component ID */
$result = cachet_query('components');
if ($result['code'] != 200) {
echo 'Can\'t query components' . "\n";
exit(1);
}
$cachet_component_id = false;
foreach ($result['body']->data as $component) {
if ($cachet_component == $component->name) { // We nailed it
$cachet_component_id = $component->id;
break; // Yes, bad.
}
}
if ($cachet_component_id === false) {
echo 'Can\'t find component "' . $cachet_component . '"' . "\n";
exit(1);
}
/*
Determine what to to:
- if PROBLEM and SOFT then don't cry just yet
- if PROBLEM and HARD then create incident
- if RECOVERY and SOFT then update incident
- if RECOVERY and HARD then update incident
PROBLEM = !OK = (WARNING | CRITICAL | UNKONWN)
RECOVERY = OK
*/
if ($service_status != 'OK' && $service_status_type == 'SOFT') { // Hope it will be back soon
echo 'KO SOFT: not doing anything' . "\n";
exit(0);
} elseif ($service_status != 'OK' && $service_status_type == 'HARD') { // Something went wrong, let's notify
echo 'KO HARD: creating incident' . "\n";
$query = array(
'name' => $incident_prefix . ' ' . $service_name,
'message' => $service_output,
'status' => CATCHET_STATUS_INVESTIGATING,
'component_id' => $cachet_component_id,
'component_status' => CATCHET_COMP_MAJOROUTAGE,
'notify' => $cachet_notify_subscribers,
);
$result = cachet_query('incidents', 'POST', $query);
if ($result['code'] != 200) {
echo 'Can\'t create incident' . "\n";
//exit(1);
}
} elseif ($service_status == 'OK' && $service_status_type == 'SOFT') { // Recovery underway
echo 'OK SOFT: updating incident' . "\n";
/* Get the incident ID */
$results = cachet_query('incidents');
if ($result['code'] != 200) {
echo 'Can\'t get incidents' . "\n";
exit(1);
}
$cachet_incident_id = false;
foreach ($results['body']->data as $incident) {
if ($incident->name == $incident_prefix . ' ' . $service_name) {
$cachet_incident_id = $incident->id;
break; // Yes, bad.
}
}
if ($cachet_incident_id === false) {
echo 'Can\'t find incident "' . $incident_prefix . ' ' . $service_name . '"' . "\n";
exit(1);
}
/* Update the incident */
$query = array(
'name' => $incident_prefix . ' ' . $service_name,
'message' => $service_output,
'status' => CATCHET_STATUS_WATCHING,
'component_id' => $cachet_component_id,
'component_status' => CATCHET_COMP_PARTIALOUTAGE,
'notify' => $cachet_notify_subscribers,
);
$result = cachet_query('incidents/' . $cachet_incident_id, 'PUT', $query);
if ($result['code'] != 200) {
echo 'Can\'t update incident' . "\n";
exit(1);
}
} elseif ($service_status == 'OK' && $service_status_type == 'HARD') { // Recovery completed
echo 'OK HARD: updating incident' . "\n";
/* Get the incident ID */
$results = cachet_query('incidents');
if ($result['code'] != 200) {
echo 'Can\'t get incidents' . "\n";
exit(1);
}
$cachet_incident_id = false;
foreach ($results['body']->data as $incident) {
if ($incident->name == $incident_prefix . ' ' . $service_name) {
$cachet_incident_id = $incident->id;
break; // Yes, bad.
}
}
if ($cachet_incident_id === false) {
echo 'Can\'t find incident "' . $incident_prefix . ' ' . $service_name . '"' . "\n";
exit(1);
}
/* Update the incident */
$query = array(
'name' => $incident_prefix . ' ' . $service_name,
'message' => $service_output,
'status' => CATCHET_STATUS_FIXED,
'component_id' => $cachet_component_id,
'component_status' => CATCHET_COMP_OPERATIONAL,
'notify' => $cachet_notify_subscribers,
);
$result = cachet_query('incidents/' . $cachet_incident_id, 'PUT', $query);
if ($result['code'] != 200) {
echo 'Can\'t update incident' . "\n";
exit(1);
}
} else {
echo 'Bad arguments' . "\n";
exit(1);
}
exit(0);