135 lines
3.7 KiB
PHP
135 lines
3.7 KiB
PHP
<?php
|
|
|
|
namespace Icinga\Module\Director\Objects;
|
|
|
|
use Icinga\Module\Director\IcingaConfig\IcingaConfig;
|
|
use Icinga\Module\Director\IcingaConfig\IcingaConfigHelper as c;
|
|
|
|
class IcingaCommand extends IcingaObject
|
|
{
|
|
protected $table = 'icinga_command';
|
|
|
|
protected $type = 'CheckCommand';
|
|
|
|
protected $defaultProperties = array(
|
|
'id' => null,
|
|
'object_name' => null,
|
|
'object_type' => null,
|
|
'disabled' => 'n',
|
|
'methods_execute' => null,
|
|
'command' => null,
|
|
'timeout' => null,
|
|
'zone_id' => null,
|
|
);
|
|
|
|
protected $supportsCustomVars = true;
|
|
|
|
protected $supportsFields = true;
|
|
|
|
protected $supportsImports = true;
|
|
|
|
protected $supportsArguments = true;
|
|
|
|
protected $intervalProperties = array(
|
|
'timeout' => 'timeout',
|
|
);
|
|
|
|
protected $relations = array(
|
|
'zone' => 'IcingaZone',
|
|
);
|
|
|
|
protected static $pluginDir;
|
|
|
|
/**
|
|
* Render the 'medhods_execute' property as 'execute'
|
|
*
|
|
* Execute is a reserved word in SQL, column name was prefixed
|
|
*
|
|
* Avoid complaints for method names with underscore:
|
|
* @codingStandardsIgnoreStart
|
|
*
|
|
* @return string
|
|
*/
|
|
protected function renderMethods_execute()
|
|
{
|
|
// @codingStandardsIgnoreEnd
|
|
return c::renderKeyValue('execute', $this->methods_execute);
|
|
}
|
|
|
|
protected function renderObjectHeader()
|
|
{
|
|
$execute = $this->getResolvedProperty('methods_execute');
|
|
|
|
if ($execute === 'PluginNotification') {
|
|
return $this->renderObjectHeaderWithType('NotificationCommand');
|
|
} elseif ($execute === 'PluginEvent') {
|
|
return $this->renderObjectHeaderWithType('EventCommand');
|
|
} else {
|
|
return parent::renderObjectHeader();
|
|
}
|
|
}
|
|
|
|
protected function renderObjectHeaderWithType($type)
|
|
{
|
|
return sprintf(
|
|
"%s %s %s {\n",
|
|
$this->getObjectTypeName(),
|
|
$type,
|
|
c::renderString($this->getObjectName())
|
|
);
|
|
}
|
|
|
|
public function mungeCommand($value)
|
|
{
|
|
if (is_array($value)) {
|
|
$value = implode(' ', $value);
|
|
} elseif (is_object($value)) {
|
|
// { type => Function } -> really??
|
|
return null;
|
|
return $value;
|
|
}
|
|
|
|
if (self::$pluginDir !== null) {
|
|
if (($pos = strpos($value, self::$pluginDir)) === 0) {
|
|
$value = substr($value, strlen(self::$pluginDir) + 1);
|
|
}
|
|
}
|
|
|
|
return $value;
|
|
}
|
|
|
|
public function getRenderingZone(IcingaConfig $config = null)
|
|
{
|
|
return $this->connection->getDefaultGlobalZoneName();
|
|
}
|
|
|
|
protected function renderCommand()
|
|
{
|
|
$command = $this->command;
|
|
$prefix = '';
|
|
if (preg_match('~^([A-Z][A-Za-z0-9_]+\s\+\s)(.+?)$~', $command, $m)) {
|
|
$prefix = $m[1];
|
|
$command = $m[2];
|
|
} elseif (! $this->isAbsolutePath($command)) {
|
|
$prefix = 'PluginDir + ';
|
|
$command = '/' . $command;
|
|
}
|
|
$parts = preg_split('/\s+/', $command, -1, PREG_SPLIT_NO_EMPTY);
|
|
array_unshift($parts, c::alreadyRendered($prefix . c::renderString(array_shift($parts))));
|
|
|
|
return c::renderKeyValue('command', c::renderArray($parts));
|
|
}
|
|
|
|
protected function isAbsolutePath($path)
|
|
{
|
|
return $path[0] === '/'
|
|
|| $path[0] === '\\'
|
|
|| preg_match('/^[A-Za-z]:\\\/', substr($path, 0, 3));
|
|
}
|
|
|
|
public static function setPluginDir($pluginDir)
|
|
{
|
|
self::$pluginDir = $pluginDir;
|
|
}
|
|
}
|