. * * @package LibreNMS * @link http://librenms.org * @copyright 2018 Neil Lathwood * @author Neil Lathwood */ use LibreNMS\Alerting\QueryBuilderParser; header('Content-type: application/json'); if (!Auth::user()->hasGlobalAdmin()) { die(json_encode([ 'status' => 'error', 'message' => 'ERROR: You need to be admin', ])); } $status = 'ok'; $message = ''; $builder_json = $vars['builder_json']; $override_query = $vars['override_query']; $options = [ 'override_query' => $override_query, ]; if ($override_query === 'on') { $query = $vars['adv_query']; } else { $query = QueryBuilderParser::fromJson($builder_json)->toSql(); } $rule_id = $_POST['rule_id']; $count = mres($_POST['count']); $delay = mres($_POST['delay']); $interval = mres($_POST['interval']); $mute = mres(isset($_POST['mute']) ? $_POST['mute'] : null); $invert = mres(isset($_POST['invert']) ? $_POST['invert'] : null); $name = mres($_POST['name']); $proc = mres($_POST['proc']); $recovery = ($vars['recovery']); $severity = mres($_POST['severity']); if (!is_numeric($count)) { $count = '-1'; } $delay_sec = convert_delay($delay); $interval_sec = convert_delay($interval); if ($mute == 'on') { $mute = true; } else { $mute = false; } if ($invert == 'on') { $invert = true; } else { $invert = false; } $recovery = empty($recovery) ? $recovery = false : true; $extra = array( 'mute' => $mute, 'count' => $count, 'delay' => $delay_sec, 'invert' => $invert, 'interval' => $interval_sec, 'recovery' => $recovery, 'options' => $options, ); $extra_json = json_encode($extra); if (is_numeric($rule_id) && $rule_id > 0) { if (dbUpdate( array( 'severity' => $severity, 'extra' => $extra_json, 'name' => $name, 'proc' => $proc, 'query' => $query, 'builder' => $builder_json ), 'alert_rules', 'id=?', array($rule_id) ) >= 0) { $message = "Edited Rule: $name"; } else { $message = "Failed to edit Rule $name"; $status = 'error'; } } else { if (empty($name)) { $status = 'error'; $message = 'No rule name provided'; } elseif (empty($builder_json) || empty($query)) { $status = 'error'; $message = 'No rules provided'; } else { $rule_id = dbInsert(array( 'rule' => '', 'severity' => $severity, 'extra' => $extra_json, 'disabled' => 0, 'name' => $name, 'proc' => $proc, 'query' => $query, 'builder' => $builder_json ), 'alert_rules'); if ($rule_id) { $message = "Added Rule: $name"; } else { if (dbFetchCell('SELECT 1 FROM alert_rules WHERE name=?', [$name])) { $message = "Rule named $name already exists"; } else { $message = "Failed to add Rule: $name"; } $status = 'error'; } } }//end if // update maps if (is_numeric($rule_id) && $rule_id > 0) { $devices = []; $groups = []; foreach ((array)$vars['maps'] as $item) { if (starts_with($item, 'g')) { $groups[] = (int)substr($item, 1); } else { $devices[] = (int)$item; } } dbSyncRelationship('alert_device_map', 'rule_id', $rule_id, 'device_id', $devices); dbSyncRelationship('alert_group_map', 'rule_id', $rule_id, 'group_id', $groups); //Update transport groups and transports - can't use dbSyncRelationship $transports = []; $groups = []; foreach ((array)$vars['transports'] as $item) { if (starts_with($item, 'g')) { $groups[] = (int)substr($item, 1); } else { $transports[] = (int)$item; } } // Fetch transport/group mappings already in db $sql = "SELECT `transport_or_group_id` FROM `alert_transport_map` WHERE `target_type`='single' AND `rule_id`=?"; $db_transports = dbFetchColumn($sql, [$rule_id]); $sql = "SELECT `transport_or_group_id` FROM `alert_transport_map` WHERE `target_type`='group' AND `rule_id`=?"; $db_groups = dbFetchColumn($sql, [$rule_id]); // Compare arrays to get add and removed transports/groups $t_add = array_diff($transports, $db_transports); $t_del = array_diff($db_transports, $transports); $g_add = array_diff($groups, $db_groups); $g_del = array_diff($db_groups, $groups); // Insert any new mappings $insert = []; foreach ($t_add as $transport_id) { $insert[] = array ( 'transport_or_group_id' => $transport_id, 'target_type' => 'single', 'rule_id' => $rule_id ); } foreach ($g_add as $group_id) { $insert[] = array( 'transport_or_group_id' => $group_id, 'target_type' => 'group', 'rule_id' => $rule_id ); } if (!empty($insert)) { $res = dbBulkInsert($insert, 'alert_transport_map'); } // Remove old mappings if (!empty($t_del)) { $db_t_values = array_merge([$rule_id], $t_del); dbDelete('alert_transport_map', 'target_type="single" AND rule_id=? AND transport_or_group_id IN ' . dbGenPlaceholders(count($t_del)), $db_t_values); } if (!empty($g_del)) { $db_g_values = array_merge([$rule_id], $g_del); dbDelete('alert_transport_map', 'target_type="group" AND rule_id=? AND transport_or_group_id IN ' . dbGenPlaceholders(count($g_del)), $db_g_values); } } die(json_encode([ 'status' => $status, 'message' => $message ]));