You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
242 lines
7.3 KiB
242 lines
7.3 KiB
<?php declare(strict_types = 0);
|
|
/*
|
|
** Zabbix
|
|
** Copyright (C) 2001-2023 Zabbix SIA
|
|
**
|
|
** This program is free software; you can redistribute it and/or modify
|
|
** it under the terms of the GNU General Public License as published by
|
|
** the Free Software Foundation; either version 2 of the License, or
|
|
** (at your option) any later version.
|
|
**
|
|
** This program is distributed in the hope that it will be useful,
|
|
** but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
** GNU General Public License for more details.
|
|
**
|
|
** You should have received a copy of the GNU General Public License
|
|
** along with this program; if not, write to the Free Software
|
|
** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
**/
|
|
|
|
|
|
class CTriggerHelper extends CTriggerGeneralHelper {
|
|
|
|
/**
|
|
* @param string $src_templateid
|
|
* @param string $dst_templateid
|
|
*
|
|
* @return bool
|
|
*/
|
|
public static function cloneTemplateTriggers(string $src_templateid, string $dst_templateid): bool {
|
|
$src_options = [
|
|
'hostids' => $src_templateid,
|
|
'inherited' => false
|
|
];
|
|
|
|
$dst_options = ['templateids' => [$dst_templateid]];
|
|
|
|
return self::copy($src_options, $dst_options);
|
|
}
|
|
|
|
/**
|
|
* @param string $src_hostid
|
|
* @param string $dst_hostid
|
|
*
|
|
* @return bool
|
|
*/
|
|
public static function cloneHostTriggers(string $src_hostid, string $dst_hostid): bool {
|
|
$src_options = [
|
|
'hostids' => $src_hostid,
|
|
'inherited' => false,
|
|
'filter' => ['flags' => ZBX_FLAG_DISCOVERY_NORMAL]
|
|
];
|
|
|
|
$dst_options = ['hostids' => [$dst_hostid]];
|
|
|
|
return self::copy($src_options, $dst_options);
|
|
}
|
|
|
|
/**
|
|
* @param array $src_options
|
|
* @param array $dst_options
|
|
*
|
|
* @return bool
|
|
*/
|
|
public static function copy(array $src_options, array $dst_options): bool {
|
|
$src_triggers = self::getSourceTriggers($src_options);
|
|
|
|
if (!$src_triggers) {
|
|
return true;
|
|
}
|
|
elseif (!array_key_exists('hostids', $src_options)) {
|
|
foreach ($src_triggers as $src_trigger) {
|
|
if (count($src_trigger['hosts']) > 1) {
|
|
$error = reset($src_trigger['hosts'])['status'] == HOST_STATUS_TEMPLATE
|
|
? _('Cannot copy trigger "%1$s", because it has multiple templates in the expression.')
|
|
: _('Cannot copy trigger "%1$s", because it has multiple hosts in the expression.');
|
|
|
|
error(sprintf($error, $src_trigger['description']));
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
$src_triggerids = array_fill_keys(array_keys($src_triggers), true);
|
|
$src_dep_triggers = [];
|
|
$src_master_dep_triggers = [];
|
|
$src_hosts = [];
|
|
|
|
foreach ($src_triggers as $src_trigger) {
|
|
foreach ($src_trigger['dependencies'] as $master_trigger) {
|
|
if (array_key_exists($master_trigger['triggerid'], $src_triggerids)) {
|
|
$src_dep_triggers[$src_trigger['triggerid']] = $src_trigger;
|
|
$src_master_dep_triggers[$master_trigger['triggerid']][$src_trigger['triggerid']] = true;
|
|
|
|
unset($src_triggers[$src_trigger['triggerid']]);
|
|
}
|
|
elseif (array_key_exists('triggerids', $src_options)) {
|
|
$src_host = array_key_exists('hostids', $src_options)
|
|
? $src_trigger['hosts'][$src_options['hostids']]
|
|
: reset($src_trigger['hosts']);
|
|
|
|
$src_hosts[$master_trigger['triggerid']][$src_trigger['triggerid']] = $src_host;
|
|
}
|
|
}
|
|
}
|
|
|
|
$dst_hosts = array_key_exists('templateids', $dst_options)
|
|
? API::Template()->get([
|
|
'output' => ['host'],
|
|
'preservekeys' => true
|
|
] + $dst_options)
|
|
: API::Host()->get([
|
|
'output' => ['host', 'status'],
|
|
'preservekeys' => true
|
|
] + $dst_options);
|
|
|
|
try {
|
|
$dst_master_triggerids = self::getDestinationMasterTriggers($src_hosts, $dst_hosts);
|
|
}
|
|
catch (Exception $e) {
|
|
return false;
|
|
}
|
|
|
|
do {
|
|
$dst_triggers = [];
|
|
|
|
foreach ($dst_hosts as $dst_hostid => $dst_host) {
|
|
foreach ($src_triggers as $src_trigger) {
|
|
$dst_trigger = array_diff_key($src_trigger, array_flip(['triggerid', 'hosts']));
|
|
|
|
$src_host = array_key_exists('hostids', $src_options)
|
|
? $src_trigger['hosts'][$src_options['hostids']]
|
|
: reset($src_trigger['hosts']);
|
|
|
|
$dst_trigger['expression'] = self::getExpressionWithReplacedHost(
|
|
$src_trigger['expression'], $src_host['host'], $dst_host['host']
|
|
);
|
|
|
|
if ($src_trigger['recovery_mode'] == ZBX_RECOVERY_MODE_RECOVERY_EXPRESSION) {
|
|
$dst_trigger['recovery_expression'] = self::getExpressionWithReplacedHost(
|
|
$src_trigger['recovery_expression'], $src_host['host'], $dst_host['host']
|
|
);
|
|
}
|
|
|
|
foreach ($dst_trigger['dependencies'] as &$master_trigger) {
|
|
if (array_key_exists($master_trigger['triggerid'], $dst_master_triggerids)) {
|
|
$dst_triggerid = $dst_master_triggerids[$master_trigger['triggerid']][$dst_hostid];
|
|
|
|
$master_trigger['triggerid'] = is_array($dst_triggerid)
|
|
? $dst_triggerid[$src_host['hostid']]
|
|
: $dst_triggerid;
|
|
}
|
|
}
|
|
unset($master_trigger);
|
|
|
|
$dst_triggers[] = $dst_trigger;
|
|
}
|
|
}
|
|
|
|
$response = API::Trigger()->create($dst_triggers);
|
|
|
|
if ($response === false) {
|
|
return false;
|
|
}
|
|
|
|
$_src_triggers = [];
|
|
|
|
if ($src_dep_triggers) {
|
|
foreach ($dst_hosts as $dst_hostid => $foo) {
|
|
foreach ($src_triggers as $src_trigger) {
|
|
$dst_triggerid = array_shift($response['triggerids']);
|
|
|
|
if (array_key_exists($src_trigger['triggerid'], $src_master_dep_triggers)) {
|
|
$dst_master_triggerids[$src_trigger['triggerid']][$dst_hostid] = $dst_triggerid;
|
|
|
|
foreach ($src_master_dep_triggers[$src_trigger['triggerid']] as $src_dep_triggerid => $f) {
|
|
unset($src_master_dep_triggers[$src_trigger['triggerid']][$src_dep_triggerid]);
|
|
|
|
if (!$src_master_dep_triggers[$src_trigger['triggerid']]) {
|
|
unset($src_master_dep_triggers[$src_trigger['triggerid']]);
|
|
}
|
|
|
|
foreach ($src_dep_triggers[$src_dep_triggerid]['dependencies'] as $master_trigger) {
|
|
if (bccomp($master_trigger['triggerid'], $src_trigger['triggerid']) == 0) {
|
|
continue;
|
|
}
|
|
|
|
if (array_key_exists($master_trigger['triggerid'], $src_master_dep_triggers)
|
|
&& array_key_exists($src_dep_triggerid, $src_master_dep_triggers[$master_trigger['triggerid']])) {
|
|
continue 3;
|
|
}
|
|
}
|
|
|
|
$_src_triggers[] = $src_dep_triggers[$src_dep_triggerid];
|
|
unset($src_dep_triggers[$src_dep_triggerid]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
$src_triggers = $_src_triggers;
|
|
} while ($src_triggers);
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* @param array $src_options
|
|
*
|
|
* @return array
|
|
*/
|
|
private static function getSourceTriggers(array $src_options): array {
|
|
$src_triggers = API::Trigger()->get([
|
|
'output' => ['triggerid', 'expression', 'description', 'url_name', 'url', 'status', 'priority', 'comments',
|
|
'type', 'recovery_mode', 'recovery_expression', 'correlation_mode', 'correlation_tag', 'manual_close',
|
|
'opdata', 'event_name'
|
|
],
|
|
'selectDependencies' => ['triggerid'],
|
|
'selectTags' => ['tag', 'value'],
|
|
'selectHosts' => ['hostid', 'host', 'status'],
|
|
'preservekeys' => true
|
|
] + $src_options);
|
|
|
|
if (!$src_triggers) {
|
|
return [];
|
|
}
|
|
|
|
$src_triggers = CMacrosResolverHelper::resolveTriggerExpressions($src_triggers,
|
|
['sources' => ['expression', 'recovery_expression']]
|
|
);
|
|
|
|
foreach ($src_triggers as &$src_trigger) {
|
|
$src_trigger['hosts'] = array_column($src_trigger['hosts'], null, 'hostid');
|
|
}
|
|
unset($src_trigger);
|
|
|
|
return $src_triggers;
|
|
}
|
|
}
|