This repository has been archived by the owner on Jan 3, 2023. It is now read-only.
/
add_oncall_weekly.php
156 lines (141 loc) · 8.44 KB
/
add_oncall_weekly.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
<?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);
$hide_event = isset($n['hide_event']) ? '1' : '0';
$id = isset($n['id']) ? $n['id'] : false; // Will only be present for events already in the database. Used for UPDATEs.
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','$hide_event')";
// 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,
'hide_event' => $hide_event,
'id' => $id);
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.
} else {
logline("'event_versioning' appears to be defined but with an unknown value. Check phplib/config.php and be certain it's set to either 'on' or 'off'.");
}
}
// 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, hide_event) 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);
// We need to match id because sometimes our providers return duplicate entries where all values match.
$update_query = "UPDATE oncall_weekly SET {$update_values_string} WHERE alert_id='{$update_hash['alert_id']}' AND id='{$update_hash['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');
}
}