From f66ab8dae2f3d0418759f81a3a64dc9517a62449 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 14 Apr 2024 15:17:31 +0200 Subject: Adding upstream version 1.10.2. Signed-off-by: Daniel Baumann --- .../Objects/IcingaObjectMultiRelations.php | 454 +++++++++++++++++++++ 1 file changed, 454 insertions(+) create mode 100644 library/Director/Objects/IcingaObjectMultiRelations.php (limited to 'library/Director/Objects/IcingaObjectMultiRelations.php') diff --git a/library/Director/Objects/IcingaObjectMultiRelations.php b/library/Director/Objects/IcingaObjectMultiRelations.php new file mode 100644 index 0000000..a1ec9a2 --- /dev/null +++ b/library/Director/Objects/IcingaObjectMultiRelations.php @@ -0,0 +1,454 @@ +object = $object; + $this->propertyName = $propertyName; + + if (is_object($config) || is_array($config)) { + foreach ($config as $k => $v) { + $this->$k = $v; + } + } else { + $this->relatedObjectClass = $config; + } + } + + public function getObjects() + { + return $this->relations; + } + + #[\ReturnTypeWillChange] + public function count() + { + return count($this->relations); + } + + #[\ReturnTypeWillChange] + public function rewind() + { + $this->position = 0; + } + + public function hasBeenModified() + { + return $this->modified; + } + + #[\ReturnTypeWillChange] + public function current() + { + if (! $this->valid()) { + return null; + } + + return $this->relations[$this->idx[$this->position]]; + } + + #[\ReturnTypeWillChange] + public function key() + { + return $this->idx[$this->position]; + } + + #[\ReturnTypeWillChange] + public function next() + { + ++$this->position; + } + + #[\ReturnTypeWillChange] + public function valid() + { + return array_key_exists($this->position, $this->idx); + } + + public function get($key) + { + if (array_key_exists($key, $this->relations)) { + return $this->relations[$key]; + } + + return null; + } + + public function set($relation) + { + if (! is_array($relation)) { + if ($relation === null) { + $relation = array(); + } else { + $relation = array($relation); + } + } + + $existing = array_keys($this->relations); + $new = array(); + $class = $this->getRelatedClassName(); + $unset = array(); + + foreach ($relation as $k => $ro) { + if ($ro instanceof $class) { + $new[] = $ro->object_name; + } else { + if (empty($ro)) { + $unset[] = $k; + continue; + } + + $new[] = $ro; + } + } + + foreach ($unset as $k) { + unset($relation[$k]); + } + + sort($existing); + sort($new); + if ($existing === $new) { + return $this; + } + + $this->relations = array(); + if (empty($relation)) { + $this->modified = true; + $this->refreshIndex(); + return $this; + } + + return $this->add($relation); + } + + /** + * Magic isset check + * + * @return boolean + */ + public function __isset($relation) + { + return array_key_exists($relation, $this->relations); + } + + public function remove($relation) + { + if (array_key_exists($relation, $this->relations)) { + unset($this->relations[$relation]); + } + + $this->modified = true; + $this->refreshIndex(); + } + + protected function refreshIndex() + { + ksort($this->relations); + $this->idx = array_keys($this->relations); + } + + public function add($relation, $onError = 'fail') + { + // TODO: only one query when adding array + if (is_array($relation)) { + foreach ($relation as $r) { + $this->add($r, $onError); + } + return $this; + } + + if (array_key_exists($relation, $this->relations)) { + return $this; + } + + $class = $this->getRelatedClassName(); + + if ($relation instanceof $class) { + $this->relations[$relation->object_name] = $relation; + } elseif (is_string($relation)) { + $connection = $this->object->getConnection(); + try { + // Related services can only be objects, used by ServiceSets + if ($class === 'Icinga\\Module\\Director\\Objects\\IcingaService') { + $relation = $class::load(array( + 'object_name' => $relation, + 'object_type' => 'template' + ), $connection); + } else { + $relation = $class::load($relation, $connection); + } + } catch (Exception $e) { + switch ($onError) { + case 'autocreate': + $relation = $class::create(array( + 'object_type' => 'object', + 'object_name' => $relation + )); + $relation->store($connection); + // TODO + case 'fail': + throw new ProgrammingError( + 'The related %s "%s" doesn\'t exists: %s', + $this->getRelatedTableName(), + $relation, + $e->getMessage() + ); + break; + case 'ignore': + return $this; + } + } + } else { + throw new ProgrammingError( + 'Invalid related object: %s', + var_export($relation, 1) + ); + } + + $this->relations[$relation->object_name] = $relation; + $this->modified = true; + $this->refreshIndex(); + + return $this; + } + + protected function getPropertyName() + { + return $this->propertyName; + } + + protected function getRelatedShortName() + { + if ($this->relatedShortName === null) { + /** @var IcingaObject $class */ + $class = $this->getRelatedClassName(); + $this->relatedShortName = $class::create()->getShortTableName(); + } + + return $this->relatedShortName; + } + + protected function getTableName() + { + return $this->object->getTableName() . '_' . $this->getRelatedShortName(); + } + + protected function getRelatedTableName() + { + if ($this->relatedTableName === null) { + /** @var IcingaObject $class */ + $class = $this->getRelatedClassName(); + $this->relatedTableName = $class::create()->getTableName(); + } + + return $this->relatedTableName; + } + + protected function getRelationIdColumn() + { + if ($this->relationIdColumn === null) { + $this->relationIdColumn = $this->getRelatedShortName(); + } + + return $this->relationIdColumn; + } + + public function listRelatedNames() + { + return array_keys($this->relations); + } + + public function listOriginalNames() + { + return array_keys($this->stored); + } + + public function getType() + { + return $this->object->getShortTableName(); + } + + protected function loadFromDb() + { + $db = $this->getDb(); + $connection = $this->object->getConnection(); + + $type = $this->getType(); + $objectIdCol = $type . '_id'; + $relationIdCol = $this->getRelationIdColumn() . '_id'; + + $query = $db->select()->from( + array('r' => $this->getTableName()), + array() + )->join( + array('ro' => $this->getRelatedTableName()), + sprintf('r.%s = ro.id', $relationIdCol), + '*' + )->where( + sprintf('r.%s = ?', $objectIdCol), + (int) $this->object->id + )->order('ro.object_name'); + + $class = $this->getRelatedClassName(); + $this->relations = $class::loadAll($connection, $query, 'object_name'); + $this->setBeingLoadedFromDb(); + + return $this; + } + + public function store() + { + $db = $this->getDb(); + $stored = array_keys($this->stored); + $relations = array_keys($this->relations); + + $objectId = $this->object->id; + $type = $this->getType(); + $objectCol = $type . '_id'; + $relationCol = $this->getRelationIdColumn() . '_id'; + + $toDelete = array_diff($stored, $relations); + foreach ($toDelete as $relation) { + // We work with cloned objects. (why?) + // As __clone drops the id, we need to access original properties + $orig = $this->stored[$relation]->getOriginalProperties(); + $where = sprintf( + $objectCol . ' = %d AND ' . $relationCol . ' = %d', + $objectId, + $orig['id'] + ); + + $db->delete( + $this->getTableName(), + $where + ); + } + + $toAdd = array_diff($relations, $stored); + foreach ($toAdd as $related) { + $db->insert( + $this->getTableName(), + array( + $objectCol => $objectId, + $relationCol => $this->relations[$related]->id + ) + ); + } + $this->setBeingLoadedFromDb(); + + return true; + } + + public function setBeingLoadedFromDb() + { + $this->stored = array(); + foreach ($this->relations as $k => $v) { + $this->stored[$k] = clone($v); + } + } + + protected function getRelatedClassName() + { + return __NAMESPACE__ . '\\' . $this->relatedObjectClass; + } + + protected function getDb() + { + if ($this->db === null) { + $this->db = $this->object->getDb(); + } + + return $this->db; + } + + public static function loadForStoredObject(IcingaObject $object, $propertyName, $relatedObjectClass) + { + $relations = new static($object, $propertyName, $relatedObjectClass); + return $relations->loadFromDb(); + } + + public function toConfigString() + { + $relations = array_keys($this->relations); + + if (empty($relations)) { + return ''; + } + + return c::renderKeyValue($this->propertyName, c::renderArray($relations)); + } + + public function __toString() + { + try { + return $this->toConfigString(); + } catch (Exception $e) { + trigger_error($e); + $previousHandler = set_exception_handler( + function () { + } + ); + restore_error_handler(); + if ($previousHandler !== null) { + call_user_func($previousHandler, $e); + die(); + } else { + die($e->getMessage()); + } + } + } + + public function toLegacyConfigString() + { + $relations = array_keys($this->relations); + + if (empty($relations)) { + return ''; + } + + if ($this->legacyPropertyName === null) { + return ' # not supported in legacy: ' . + c1::renderKeyValue($this->propertyName, c1::renderArray($relations), ''); + } + + return c1::renderKeyValue($this->legacyPropertyName, c1::renderArray($relations)); + } +} -- cgit v1.2.3