icingadb-web/library/Icingadb/Compat/CompatObject.php

374 lines
9.9 KiB
PHP

<?php
/* Icinga DB Web | (c) 2020 Icinga GmbH | GPLv2 */
namespace Icinga\Module\Icingadb\Compat;
use Icinga\Exception\NotImplementedError;
use Icinga\Module\Icingadb\Common\Auth;
use Icinga\Module\Icingadb\Common\Database;
use Icinga\Module\Icingadb\Model\CustomvarFlat;
use Icinga\Module\Icingadb\Model\Host;
use Icinga\Module\Icingadb\Model\Service;
use Icinga\Module\Icingadb\Model\Servicegroup;
use Icinga\Module\Icingadb\Model\ServicestateSummary;
use InvalidArgumentException;
use ipl\Orm\Model;
use ipl\Orm\Query;
use ipl\Stdlib\Filter;
use LogicException;
use function ipl\Stdlib\get_php_type;
trait CompatObject
{
use Auth;
use Database;
/** @var array Non-obscured custom variables */
protected $rawCustomvars;
/** @var array Non-obscured host custom variables */
protected $rawHostCustomvars;
/** @var Model $object */
private $object;
public function __construct(Model $object)
{
$this->object = $object;
}
public static function fromModel(Model $object)
{
switch (true) {
case $object instanceof Host:
return new CompatHost($object);
case $object instanceof Service:
return new CompatService($object);
default:
throw new InvalidArgumentException(sprintf(
'Host or Service Model instance expected, got "%s" instead.',
get_php_type($object)
));
}
}
/**
* Get this object's name
*
* @return string
*/
public function getName(): string
{
return $this->object->name ?? '';
}
public function fetch(): bool
{
return true;
}
protected function fetchRawCustomvars(): self
{
if ($this->rawCustomvars !== null) {
return $this;
}
$vars = $this->object->customvar->execute();
$customVars = [];
foreach ($vars as $row) {
$customVars[$row->name] = $row->value;
}
$this->rawCustomvars = $customVars;
return $this;
}
protected function fetchRawHostCustomvars(): self
{
if ($this->rawHostCustomvars !== null) {
return $this;
}
$vars = $this->object->host->customvar->execute();
$customVars = [];
foreach ($vars as $row) {
$customVars[$row->name] = $row->value;
}
$this->rawHostCustomvars = $customVars;
return $this;
}
public function fetchComments()
{
$this->comments = [];
return $this;
}
public function fetchContactgroups()
{
$this->contactgroups = [];
return $this;
}
public function fetchContacts()
{
$this->contacts = [];
return $this;
}
public function fetchCustomvars(): self
{
if ($this->customvars !== null) {
return $this;
}
$this->customvars = (new CustomvarFlat())->unFlattenVars($this->object->customvar_flat);
return $this;
}
public function fetchHostVariables()
{
if (isset($this->hostVariables)) {
return $this;
}
$this->hostVariables = [];
foreach ($this->object->customvar as $customvar) {
$this->hostVariables[strtolower($customvar->name)] = json_decode($customvar->value);
}
return $this;
}
public function fetchServiceVariables()
{
if (isset($this->serviceVariables)) {
return $this;
}
$this->serviceVariables = [];
foreach ($this->object->customvar as $customvar) {
$this->serviceVariables[strtolower($customvar->name)] = json_decode($customvar->value);
}
return $this;
}
public function fetchDowntimes()
{
$this->downtimes = [];
return $this;
}
public function fetchEventhistory()
{
$this->eventhistory = [];
return $this;
}
public function fetchHostgroups()
{
if ($this->type === self::TYPE_HOST) {
$hostname = $this->object->name;
$hostgroupQuery = clone $this->object->hostgroup;
} else {
$hostname = $this->object->host->name;
$hostgroupQuery = clone $this->object->host->hostgroup;
}
$hostgroupQuery
->columns(['name', 'display_name'])
->filter(Filter::equal('host.name', $hostname));
/** @var Query $hostgroupQuery */
$this->hostgroups = [];
foreach ($hostgroupQuery as $hostgroup) {
$this->hostgroups[$hostgroup->name] = $hostgroup->display_name;
}
return $this;
}
public function fetchServicegroups()
{
if ($this->type === self::TYPE_HOST) {
$hostname = $this->object->name;
$query = Servicegroup::on($this->getDb());
} else {
$hostname = $this->object->host->name;
$query = (clone $this->object->servicegroup);
}
$query
->columns(['name', 'display_name'])
->filter(Filter::equal('host.name', $hostname));
if ($this->type === self::TYPE_SERVICE) {
$query->filter(Filter::equal('service.name', $this->object->name));
}
$this->servicegroups = [];
foreach ($query as $serviceGroup) {
$this->servicegroups[$serviceGroup->name] = $serviceGroup->display_name;
}
return $this;
}
public function fetchStats()
{
$query = ServicestateSummary::on($this->getDb());
if ($this->type === self::TYPE_HOST) {
$query->filter(Filter::equal('host.name', $this->object->name));
} else {
$query->filter(Filter::all(
Filter::equal('host.name', $this->object->host->name),
Filter::equal('service.name', $this->object->name)
));
}
$result = $query->first();
$this->stats = (object) [
'services_total' => $result->services_total,
'services_ok' => $result->services_ok,
'services_critical' => $result->services_critical_handled + $result->services_critical_unhandled,
'services_critical_unhandled' => $result->services_critical_unhandled,
'services_critical_handled' => $result->services_critical_handled,
'services_warning' => $result->services_warning_handled + $result->services_warning_unhandled,
'services_warning_unhandled' => $result->services_warning_unhandled,
'services_warning_handled' => $result->services_warning_handled,
'services_unknown' => $result->services_unknown_handled + $result->services_unknown_unhandled,
'services_unknown_unhandled' => $result->services_unknown_unhandled,
'services_unknown_handled' => $result->services_unknown_handled,
'services_pending' => $result->services_pending
];
return $this;
}
public function __get($name)
{
if (property_exists($this, $name)) {
if ($this->$name === null) {
$fetchMethod = 'fetch' . ucfirst($name);
$this->$fetchMethod();
}
return $this->$name;
}
if (preg_match('/^_(host|service)_(.+)/i', $name, $matches)) {
switch (strtolower($matches[1])) {
case $this->type:
$customvars = $this->fetchRawCustomvars()->rawCustomvars;
break;
case self::TYPE_HOST:
$customvars = $this->fetchRawHostCustomvars()->rawHostCustomvars;
break;
case self::TYPE_SERVICE:
throw new LogicException('Cannot fetch service custom variables for non-service objects');
}
$variableName = strtolower($matches[2]);
if (isset($customvars[$variableName])) {
return $customvars[$variableName];
}
return null; // Unknown custom variables MUST NOT throw an error
}
if (! array_key_exists($name, $this->legacyColumns) && ! $this->object->hasProperty($name)) {
if (isset($this->customvars[$name])) {
return $this->customvars[$name];
}
if (substr($name, 0, strlen($this->prefix)) !== $this->prefix) {
$name = $this->prefix . $name;
}
}
if (array_key_exists($name, $this->legacyColumns)) {
$opts = $this->legacyColumns[$name];
if ($opts === null) {
return null;
}
$path = $opts['path'];
$value = null;
if (! empty($path)) {
$value = $this->object;
do {
$col = array_shift($path);
$value = $value->$col;
} while (! empty($path) && $value !== null);
}
if (isset($opts['type'])) {
$method = 'get' . ucfirst($opts['type']) . 'Type';
$value = $this->$method($value);
}
return $value;
}
return $this->object->$name;
}
public function __isset($name)
{
if (property_exists($this, $name)) {
return isset($this->$name);
}
if (isset($this->legacyColumns[$name]) || isset($this->object->$name)) {
return true;
}
return false;
}
/**
* @throws NotImplementedError Don't use!
*/
protected function getDataView()
{
throw new NotImplementedError('getDataView() is not supported');
}
/**
* Get the bool type of the given value as an int
*
* @param bool|string $value
*
* @return ?int
*/
private function getBoolType($value)
{
if ($value === 'sticky') {
return 2;
}
if (is_string($value)) {
return null;
}
return (int) $value;
}
}