1 files changed, 110 insertions, 0 deletions
diff --git a/library/Director/Objects/IcingaZone.php b/library/Director/Objects/IcingaZone.php
new file mode 100644
index 0000000..8d77e47
--- /dev/null
+++ b/library/Director/Objects/IcingaZone.php
@@ -0,0 +1,110 @@
+<?php
+
+namespace Icinga\Module\Director\Objects;
+
+use Icinga\Module\Director\Db;
+use Icinga\Module\Director\IcingaConfig\IcingaConfig;
+use Icinga\Module\Director\IcingaConfig\IcingaConfigHelper as c;
+
+class IcingaZone extends IcingaObject
+{
+    protected $table = 'icinga_zone';
+
+    protected $uuidColumn = 'uuid';
+
+    protected $defaultProperties = [
+        'id'          => null,
+        'uuid'        => null,
+        'object_name' => null,
+        'object_type' => null,
+        'disabled'    => 'n',
+        'parent_id'   => null,
+        'is_global'   => 'n',
+    ];
+
+    protected $booleans = [
+        // Global is a reserved word in SQL, column name was prefixed
+        'is_global' => 'global'
+    ];
+
+    protected $relations = [
+        'parent' => 'IcingaZone',
+    ];
+
+    protected $supportsImports = true;
+
+    protected static $globalZoneNames;
+
+    private $endpointList;
+
+    protected function renderCustomExtensions()
+    {
+        $endpoints = $this->listEndpoints();
+        if (empty($endpoints)) {
+            return '';
+        }
+
+        return c::renderKeyValue('endpoints', c::renderArray($endpoints));
+    }
+
+    public function isGlobal()
+    {
+        return $this->get('is_global') === 'y';
+    }
+
+    public static function zoneNameIsGlobal($name, Db $connection)
+    {
+        if (self::$globalZoneNames === null) {
+            $db = $connection->getDbAdapter();
+            self::setCachedGlobalZoneNames($db->fetchCol(
+                $db->select()->from('icinga_zone', 'object_name')->where('is_global = ?', 'y')
+            ));
+        }
+
+        return \in_array($name, self::$globalZoneNames);
+    }
+
+    public static function setCachedGlobalZoneNames($names)
+    {
+        self::$globalZoneNames = $names;
+    }
+
+    public function getRenderingZone(IcingaConfig $config = null)
+    {
+        // If the zone has a parent zone...
+        if ($this->get('parent_id')) {
+            // ...we render the zone object to the parent zone
+            return $this->get('parent');
+        } elseif ($this->get('is_global') === 'y') {
+            // ...additional global zones are rendered to our global zone...
+            return $this->connection->getDefaultGlobalZoneName();
+        } else {
+            // ...and all the other zones are rendered to our master zone
+            return $this->connection->getMasterZoneName();
+        }
+    }
+
+    public function setEndpointList($list)
+    {
+        $this->endpointList = $list;
+
+        return $this;
+    }
+
+    // TODO: Move this away, should be prefetchable:
+    public function listEndpoints()
+    {
+        $id = $this->get('id');
+        if ($id && $this->endpointList === null) {
+            $db = $this->getDb();
+            $query = $db->select()
+                ->from('icinga_endpoint', 'object_name')
+                ->where('zone_id = ?', $id)
+                ->order('object_name');
+
+            $this->endpointList = $db->fetchCol($query);
+        }
+
+        return $this->endpointList;
+    }
+}