<?php

include_once 'phplib/base.php';

if (!db::connect()) {

echo "Database connection failed, cannot continue. ";

} else {

$username = getUsername();

$range_start = db::escape($_POST['oncall']['range_start']);

$range_end = db::escape($_POST['oncall']['range_end']);

logline("Started adding a new oncall update for {$username} with range_start: {$range_start} and range_end: {$range_end}...");

if (count($_POST['oncall']['notifications']) > 0) {

// See if this user is enrolled in sleep tracking

$profile_data = checkForUserProfile($username);

if ($profile_data && $profile_data['sleeptracking_provider'] != "none" && $profile_data['sleeptracking_provider'] != "") {

$sleep_provider = $profile_data['sleeptracking_provider'];

logline("Sleeptracking enabled: {$sleep_provider}");

$sleep = true;

// Get the user settings into an array by decoding the JSON we put in the database for them

$sleeptracking_settings = json_decode($profile_data['sleeptracking_settings'], 1);

// Get the provider settings from the config array

$sleepprovider_settings = $sleep_providers[$sleep_provider];

// Load the sleeptracking provider's PHP code

include_once($sleepprovider_settings['lib']);

} else {

logline("No need to do sleep tracking because {$username} has no provider chosen");

}

$event_versioning = getTeamConfig('event_versioning');

// We'll create an array of value strings for both INSERT and UPDATE

// statements that will be submitted after iterating over all the

// notifications we've found. For notifications found to already exist

// in the database, we'll perform UPDATEs. Otherwise we'll INSERT a new

// row.

$insert_values = array();

$update_values = array();

foreach($_POST['oncall']['notifications'] as $id => $n) {

$sleep_state = -1;

$mtts = -1;

$sleep_level = -1;

$confidence = -1;

$timestamp = db::escape($n['time']);

$hostname = db::escape($n['hostname']);

$output = db::escape($n['output']);

$service = db::escape($n['service']);

$state = db::escape($n['state']);

$tag = db::escape($n['tag']);

$notes = db::escape(htmlentities($n['notes'], ENT_QUOTES));

$alert_id = generateOnCallAlertID($timestamp, $hostname, $service);

if ($sleep) {

// Run the sleep tracking provider for this alert

$sleep_info = getSleepDetailAtTimestamp($timestamp, $sleeptracking_settings[$sleep_provider], $sleepprovider_settings);

if ($sleep_info) {

$sleep_state = $sleep_info['sleep_state'];

$mtts = $sleep_info['mtts'];

$sleep_level = $sleep_info['sleep_level'];

$confidence = $sleep_info['confidence'];

}

}

// By default, assume events will be INSERTs. This is important in case

// we want event versioning and can simply push this string into

// the $insert_values array to be used in a single INSERT statement.

$values = "('$alert_id', '$range_start', '$range_end', '$timestamp', '$hostname', '$service', '$state', '$username', '$output', '$tag', '$sleep_state', '$mtts', '$sleep_level', '$confidence','$notes')";

// If event versioning is not defined or is explicitly enabled...

if (!$event_versioning || (isset($event_versioning) && (strcmp($event_versioning, "on")) == 0)) {

array_push($insert_values, $values); # Push to the $insert_values array and move on.

// If event versioning is explicitly disabled...

} elseif (isset($event_versioning) && (strcmp($event_versioning, "off")) == 0) {

// See if this event exists in the database already.

$select_query = "SELECT 1 from oncall_weekly WHERE alert_id='{$alert_id}'";

$result = db::query($select_query);

if ($result) {

// If there's at least 1 row returned, the event has already

// been stored. We'll update it.

if ($result->num_rows > 0) {

// Create a hash representing the values. We'll use the keys to

// specifiy the column names we're updating.

$values = array('alert_id' => $alert_id,

'range_start' => $range_start,

'range_end' => $range_end,

'timestamp' => $timestamp,

'hostname' => $hostname,

'service' => $service,

'state' => $state,

'contact' => $username,

'output' => $output,

'tag' => $tag,

'sleep_state' => $sleep_state,

'mtts' => $mtts,

'sleep_level' => $sleep_level,

'sleep_confidence' => $confidence,

'notes' => $notes);

array_push($update_values, $values);

// This is a fresh event.

} else {

array_push($insert_values, $values);

}

// We failed query the database for the event.

} else {

echo "Database select failed, error: " . db::error();

logline("Database select failed, error: " . db::error());

}

$result->free(); // Free it up, y'all.

}

}

// Do we have INSERTs and/or UPDATEs to execute?

if (count($insert_values) > 0) {

$insert_values_string = implode(', ', $insert_values);

$insert_query = "INSERT INTO oncall_weekly (alert_id, range_start, range_end, timestamp, hostname, service, state, contact, output, tag, sleep_state, mtts, sleep_level, sleep_confidence, notes) VALUES {$insert_values_string}";

if (!db::query($insert_query)) {

echo "Database insert failed, error: " . db::error();

logline("Database insert failed, error: " . db::error());

}

}

// NOTE: This doesn't solve for entries that are already duplicated.

// But this should prevent future entries from being duplicated.

if (count($update_values) > 0) {

foreach ($update_values as $update_hash) {

$set_columns = array();

foreach ($update_hash as $column => $value) {

array_push($set_columns, "{$column}='{$value}'");

}

$update_values_string = implode(', ', $set_columns);

$update_query = "UPDATE oncall_weekly SET {$update_values_string} WHERE alert_id='{$update_hash['alert_id']}'";

if (!db::query($update_query)) {

echo "Database update failed, error: " . db::error();

logline("Database update failed, error: " . db::error());

}

}

}

logline("Everything worked great, redirecting the user with success");

Header('Location: add.php?oncall_succ=hellyeah');

} else {

logline("We didn't find any notifications to process, redirect user back to add page");

Header('Location: add.php');

}

}