From e3c08c242b838edc3e67c19c099ea01cbe35be2b Mon Sep 17 00:00:00 2001
From: Thomas Gelf <thomas@gelf.net>
Date: Sun, 12 Jun 2016 11:19:06 +0200
Subject: [PATCH] IcingaTemplateResolver: initial import

Should help to accomplish various template resolving tasks
---
 .../Objects/IcingaTemplateResolver.php        | 236 ++++++++++++++++++
 1 file changed, 236 insertions(+)
 create mode 100644 library/Director/Objects/IcingaTemplateResolver.php

diff --git a/library/Director/Objects/IcingaTemplateResolver.php b/library/Director/Objects/IcingaTemplateResolver.php
new file mode 100644
index 00000000..0963e3bf
--- /dev/null
+++ b/library/Director/Objects/IcingaTemplateResolver.php
@@ -0,0 +1,236 @@
+<?php
+
+namespace Icinga\Module\Director\Objects;
+
+use Icinga\Module\Director\Db;
+
+class IcingaTemplateResolver
+{
+    protected $object;
+
+    protected $connection;
+
+    protected $db;
+
+    protected $type;
+
+    protected static $templates = array();
+
+    protected static $idIdx = array();
+
+    protected static $nameIdx = array();
+
+    public function __construct(IcingaObject $object)
+    {
+        $this->setObject($object);
+    }
+
+    public function setObject(IcingaObject $object)
+    {
+        $this->object     = $object;
+        $this->type       = $object->getShortTableName();
+        $this->table      = $object->getTableName();
+        $this->connection = $object->getConnection();
+        $this->db         = $this->connection->getDbAdapter();
+
+        return $this;
+    }
+
+    /**
+     * Forget all template relation of the given object type
+     *
+     * @return self
+     */
+    public function clearCache()
+    {
+        $type = $object->getShortTableName();
+        unset(self::$templates[$type]);
+    }
+
+    public function fetchParents()
+    {
+        // TODO: involve lookup cache
+        $res = array();
+        $class = $this->object;
+        foreach ($this->listParentIds() as $id) {
+            $res[] = $class::loadWithAutoIncId($id);
+        }
+
+        return $res;
+    }
+
+    public function listParentIds($id = null)
+    {
+        $this->requireTemplates();
+
+        if ($id === null) {
+            $id = $this->object->id;
+        }
+
+        $type = $this->type;
+
+        if (array_key_exists($id, self::$idIdx[$type])) {
+            return array_keys(self::$idIdx[$type][$id]);
+        }
+
+        return array();
+    }
+
+    public function listParentNames($name = null)
+    {
+        $this->requireTemplates();
+
+        if ($name === null) {
+            $name = $this->object->object_name;
+        }
+
+        $type = $this->type;
+
+        if (array_key_exists($name, self::$nameIdx[$type])) {
+            return array_keys(self::$nameIdx[$type][$name]);
+        }
+
+        return array();
+    }
+
+    public function fetchResolvedParents()
+    {
+        // TODO: involve lookup cache
+        $res = array();
+        $class = $this->object;
+        $connection = $this->connection;
+
+        foreach ($this->listResolvedParentIds() as $id) {
+            $res[] = $class::loadWithAutoIncId($id, $connection);
+        }
+
+        return $res;
+    }
+
+    public function listResolvedParentIds()
+    {
+        $this->requireTemplates();
+        return $this->resolveParentIds($this->object->id);
+    }
+
+    public function listResolvedParentNames()
+    {
+        $this->requireTemplates();
+        return $this->resolveParentNames($this->object->object_name);
+    }
+
+    public function resolveParentIds($id)
+    {
+        $res = array();
+
+        foreach ($this->listParentIds($id) as $parentId) {
+            foreach ($this->resolveParentIds($parentId) as $gpId) {
+                $res[] = $gpId;
+            }
+            $res[] = $parentId;
+        }
+
+        return $res;
+    }
+
+    public function resolveParentNames($name)
+    {
+        $res = array();
+        foreach ($this->listParentNames($name) as $parentName) {
+            foreach ($this->resolveParentNames($parentName) as $gpName) {
+                $res[] = $gpName;
+            }
+            $res[] = $parentName;
+        }
+
+        return $res;
+    }
+
+    /*
+    public function listChildren()
+    {
+    }
+
+    public function listChildrenIds
+    */
+
+    protected function requireTemplates()
+    {
+        if (! array_key_exists($this->type, self::$templates)) {
+            $this->prepareLookupTables();
+        }
+
+        return $this;
+    }
+
+    protected function prepareLookupTables()
+    {
+        $type = $this->type;
+
+        $templates = static::fetchTemplates(
+            $this->db,
+            $type
+        );
+
+        $ids = array();
+        $names = array();
+
+        foreach ($templates as $row) {
+            if ($row->parent_id === null) {
+                continue;
+            }
+            if (array_key_exists($row->id, $ids)) {
+                $ids[$row->id][$row->parent_id] = $row->parent_name;
+                $names[$row->name][$row->parent_name] = $row->parent_id;
+            } else {
+                $ids[$row->id] = array(
+                    $row->parent_id => $row->parent_name
+                );
+
+                $names[$row->name] = array(
+                    $row->parent_name => $row->parent_id
+                );
+            }
+        }
+
+        self::$idIdx[$type]     = $ids;
+        self::$nameIdx[$type]   = $names;
+        self::$templates[$type] = $templates;
+    }
+
+    protected function fetchTemplates()
+    {
+        $db    = $this->db;
+        $type  = $this->type;
+        $table = $this->object->getTableName();
+
+        $query = $db->select()->from(
+            array('o' => $table),
+            array(
+                'id'          => 'o.id',
+                'name'        => 'o.object_name',
+                'parent_id'   => 'p.id',
+                'parent_name' => 'p.object_name',
+            )
+        )->joinLeft(
+            array('i' => $table . '_inheritance'),
+            'o.id = i.' . $type . '_id',
+            array()
+        )->joinLeft(
+            array('p' => $table),
+            'p.id = i.parent_' . $type . '_id',
+            array()
+        )//->where("o.object_type = 'template'")
+         ->order('o.id')
+         ->order('i.weight');
+
+        return $db->fetchAll($query);
+    }
+
+    public function __destruct()
+    {
+        unset($this->connection);
+        unset($this->db);
+        unset($this->object);
+    }
+}