github
/
phpservermon
mirror of https://github.com/phpservermon/phpservermon
1
0
Fork 0
Code Issues Releases Wiki Activity

add time for statuscake

pull/352/head
Laurent RICHARD 2016-09-11 16:38:25 +02:00
parent 807f07eb46
commit 4c07c7e59d
1 changed files with 148 additions and 138 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

286
src/psm/Util/Server/Updater/StatusNotifier.php
View File

@ -131,6 +131,7 @@ class StatusNotifier {
), array( ), array(
'server_id', 'ip', 'port', 'label', 'type', 'pattern', 'status', 'rtime', 'header_name', 'header_value', 'error', 'active', 'email', 'sms', 'pushover', 'statuscake','statuscake_test_id', 'server_id', 'ip', 'port', 'label', 'type', 'pattern', 'status', 'rtime', 'header_name', 'header_value', 'error', 'active', 'email', 'sms', 'pushover', 'statuscake','statuscake_test_id',
)); ));
if(empty($this->server)) { if(empty($this->server)) {
return false; return false;
} }
@ -160,206 +161,215 @@ class StatusNotifier {
} }
// check if statuscake is enabled for this server // check if statuscake is enabled for this server
if( !is_null($this->send_statuscake_account_token) && !empty($this->send_statuscake_account_token) if( $this->server['status'] == 'on' && !is_null($this->send_statuscake_account_token) && !empty($this->send_statuscake_account_token)
&& $this->server['statuscake'] == 'yes' && $this->server['statuscake'] == 'yes'
&& !is_null($this->server['statuscake_test_id']) && !empty($this->server['statuscake_test_id']) && !is_null($this->server['statuscake_test_id']) && !empty($this->server['statuscake_test_id'])
) { ) {
// yay lets wake those nerds up! // yay lets wake those nerds up!
$this->notifyByStatusCake($users, $this->server['statuscake_test_id']); <<<<<<< HEAD
$this->notifyByStatusCake($users, $this->server['statuscake_test_id']);
} }
if(!$notify) { if(!$notify) {
return false; return false;
} }
=======
$this->notifyByStatusCake($users, $this->server['statuscake_test_id'], $this->server['rtime']);
}
if(!$notify) {
return false;
}
>>>>>>> add time for statuscake
// first add to log (we use the same text as the SMS message because its short..) // first add to log (we use the same text as the SMS message because its short..)
if($this->save_logs) { if($this->save_logs) {
psm_add_log( psm_add_log(
$this->server_id, $this->server_id,
'status', 'status',
psm_parse_msg($status_new, 'sms', $this->server) psm_parse_msg($status_new, 'sms', $this->server)
); );
} }
$users = $this->getUsers($this->server_id); $users = $this->getUsers($this->server_id);
if(empty($users)) {
return $notify;
}
// check if email is enabled for this server
if($this->send_emails && $this->server['email'] == 'yes') {
// send email
$this->notifyByEmail($users);
}
// check if sms is enabled for this server
if($this->send_sms && $this->server['sms'] == 'yes') {
// yay lets wake those nerds up!
$this->notifyByTxtMsg($users);
}
// check if pushover is enabled for this server
if($this->send_pushover && $this->server['pushover'] == 'yes') {
// yay lets wake those nerds up!
$this->notifyByPushover($users);
}
if(empty($users)) {
return $notify; return $notify;
} }
/** // check if email is enabled for this server
if($this->send_emails && $this->server['email'] == 'yes') {
// send email
$this->notifyByEmail($users);
}
// check if sms is enabled for this server
if($this->send_sms && $this->server['sms'] == 'yes') {
// yay lets wake those nerds up!
$this->notifyByTxtMsg($users);
}
// check if pushover is enabled for this server
if($this->send_pushover && $this->server['pushover'] == 'yes') {
// yay lets wake those nerds up!
$this->notifyByPushover($users);
}
return $notify;
}
/**
* This functions performs the email notifications * This functions performs the email notifications
* *
* @param array $users * @param array $users
* @return boolean * @return boolean
*/ */
protected function notifyByEmail($users) { protected function notifyByEmail($users) {
// build mail object with some default values // build mail object with some default values
$mail = psm_build_mail(); $mail = psm_build_mail();
$mail->Subject = utf8_decode(psm_parse_msg($this->status_new, 'email_subject', $this->server)); $mail->Subject = utf8_decode(psm_parse_msg($this->status_new, 'email_subject', $this->server));
$mail->Priority = 1; $mail->Priority = 1;
$body = psm_parse_msg($this->status_new, 'email_body', $this->server); $body = psm_parse_msg($this->status_new, 'email_body', $this->server);
$mail->Body = utf8_decode($body); $mail->Body = utf8_decode($body);
$mail->AltBody = str_replace('<br/>', "\n", $body); $mail->AltBody = str_replace('<br/>', "\n", $body);
if(psm_get_conf('log_email')) { if(psm_get_conf('log_email')) {
$log_id = psm_add_log($this->server_id, 'email', $body); $log_id = psm_add_log($this->server_id, 'email', $body);
}
// go through empl
foreach ($users as $user) {
if(!empty($log_id)) {
psm_add_log_user($log_id, $user['user_id']);
}
// we sent a seperate email to every single user.
$mail->AddAddress($user['email'], $user['name']);
$mail->Send();
$mail->ClearAddresses();
}
} }
protected function notifyByStatusCake($users, $statuscake_test_id) { // go through empl
foreach ($users as $user) {
if(!empty($log_id)) {
psm_add_log_user($log_id, $user['user_id']);
}
$statuscake = psm_build_statuscake(); // we sent a seperate email to every single user.
$statuscake->setTestId($statuscake_test_id); $mail->AddAddress($user['email'], $user['name']);
$statuscake->setTime(11); $mail->Send();
$statuscake->setStatusCode(200); $mail->ClearAddresses();
}
}
protected function notifyByStatusCake($users, $statuscake_test_id, $rtime) {
$statuscake = psm_build_statuscake();
$statuscake->setTestId($statuscake_test_id);
$statuscake->setTime(($rtime*1000));
$statuscake->setStatusCode(200);
// Log // Log
//if(psm_get_conf('log_statuscake')) { //if(psm_get_conf('log_statuscake')) {
// $log_id = psm_add_log($this->server_id, 'statuscake', $message); // $log_id = psm_add_log($this->server_id, 'statuscake', $message);
//} //}
$statuscake->send(); $statuscake->send();
} }
/** /**
* This functions performs the pushover notifications * This functions performs the pushover notifications
* *
* @param array $users * @param array $users
* @return boolean * @return boolean
*/ */
protected function notifyByPushover($users) { protected function notifyByPushover($users) {
// Remove users that have no pushover_key // Remove users that have no pushover_key
foreach($users as $k => $user) { foreach($users as $k => $user) {
if (trim($user['pushover_key']) == '') { if (trim($user['pushover_key']) == '') {
unset($users[$k]); unset($users[$k]);
}
} }
}
// Validation // Validation
if (empty($users)) { if (empty($users)) {
return; return;
} }
// Pushover // Pushover
$message = psm_parse_msg($this->status_new, 'pushover_message', $this->server); $message = psm_parse_msg($this->status_new, 'pushover_message', $this->server);
$pushover = psm_build_pushover(); $pushover = psm_build_pushover();
if($this->status_new === true) { if($this->status_new === true) {
$pushover->setPriority(0); $pushover->setPriority(0);
} else { } else {
$pushover->setPriority(2); $pushover->setPriority(2);
$pushover->setRetry(300); //Used with Priority = 2; Pushover will resend the notification every 60 seconds until the user accepts. $pushover->setRetry(300); //Used with Priority = 2; Pushover will resend the notification every 60 seconds until the user accepts.
$pushover->setExpire(3600); //Used with Priority = 2; Pushover will resend the notification every 60 seconds for 3600 seconds. After that point, it stops sending notifications. $pushover->setExpire(3600); //Used with Priority = 2; Pushover will resend the notification every 60 seconds for 3600 seconds. After that point, it stops sending notifications.
} }
$pushover->setTitle(psm_parse_msg($this->status_new, 'pushover_title', $this->server)); $pushover->setTitle(psm_parse_msg($this->status_new, 'pushover_title', $this->server));
$pushover->setMessage(str_replace('<br/>', "\n", $message)); $pushover->setMessage(str_replace('<br/>', "\n", $message));
$pushover->setUrl(psm_build_url()); $pushover->setUrl(psm_build_url());
$pushover->setUrlTitle(psm_get_lang('system', 'title')); $pushover->setUrlTitle(psm_get_lang('system', 'title'));
// Log // Log
if(psm_get_conf('log_pushover')) { if(psm_get_conf('log_pushover')) {
$log_id = psm_add_log($this->server_id, 'pushover', $message); $log_id = psm_add_log($this->server_id, 'pushover', $message);
} }
foreach($users as $user) { foreach($users as $user) {
// Log // Log
if(!empty($log_id)) { if(!empty($log_id)) {
psm_add_log_user($log_id, $user['user_id']); psm_add_log_user($log_id, $user['user_id']);
} }
// Set recipient + send // Set recipient + send
$pushover->setUser($user['pushover_key']); $pushover->setUser($user['pushover_key']);
if($user['pushover_device'] != '') { if($user['pushover_device'] != '') {
$pushover->setDevice($user['pushover_device']); $pushover->setDevice($user['pushover_device']);
} }
$pushover->send(); $pushover->send();
} }
} }
/** /**
* This functions performs the text message notifications * This functions performs the text message notifications
* *
* @param array $users * @param array $users
* @return boolean * @return boolean
*/ */
protected function notifyByTxtMsg($users) { protected function notifyByTxtMsg($users) {
$sms = psm_build_sms(); $sms = psm_build_sms();
if(!$sms) { if(!$sms) {
return false; return false;
}
$message = psm_parse_msg($this->status_new, 'sms', $this->server);
// Log
if(psm_get_conf('log_sms')) {
$log_id = psm_add_log($this->server_id, 'sms', $message);
}
// add all users to the recipients list
foreach ($users as $user) {
// Log
if(!empty($log_id)) {
psm_add_log_user($log_id, $user['user_id']);
}
$sms->addRecipients($user['mobile']);
}
// Send sms
$result = $sms->sendSMS($message);
return $result;
} }
/** $message = psm_parse_msg($this->status_new, 'sms', $this->server);
// Log
if(psm_get_conf('log_sms')) {
$log_id = psm_add_log($this->server_id, 'sms', $message);
}
// add all users to the recipients list
foreach ($users as $user) {
// Log
if(!empty($log_id)) {
psm_add_log_user($log_id, $user['user_id']);
}
$sms->addRecipients($user['mobile']);
}
// Send sms
$result = $sms->sendSMS($message);
return $result;
}
/**
* Get all users for the provided server id * Get all users for the provided server id
* @param int $server_id * @param int $server_id
* @return array * @return array
*/ */
public function getUsers($server_id) { public function getUsers($server_id) {
// find all the users with this server listed // find all the users with this server listed
$users = $this->db->query(" $users = $this->db->query("
SELECT `u`.`user_id`, `u`.`name`,`u`.`email`, `u`.`mobile`, `u`.`pushover_key`, `u`.`pushover_device` SELECT `u`.`user_id`, `u`.`name`,`u`.`email`, `u`.`mobile`, `u`.`pushover_key`, `u`.`pushover_device`
FROM `".PSM_DB_PREFIX."users` AS `u` FROM `".PSM_DB_PREFIX."users` AS `u`
JOIN `".PSM_DB_PREFIX."users_servers` AS `us` ON ( JOIN `".PSM_DB_PREFIX."users_servers` AS `us` ON (
`us`.`user_id`=`u`.`user_id` `us`.`user_id`=`u`.`user_id`
AND `us`.`server_id` = {$server_id} AND `us`.`server_id` = {$server_id}
) )
"); ");
return $users; return $users;
} }
} }