1 files changed, 139 insertions, 0 deletions
diff --git a/modules/monitoring/library/Monitoring/Backend/Ido/Query/ContactQuery.php b/modules/monitoring/library/Monitoring/Backend/Ido/Query/ContactQuery.php
new file mode 100644
index 0000000..ca10323
--- /dev/null
+++ b/modules/monitoring/library/Monitoring/Backend/Ido/Query/ContactQuery.php
@@ -0,0 +1,139 @@
+<?php
+/* Icinga Web 2 | (c) 2013 Icinga Development Team | GPLv2+ */
+
+namespace Icinga\Module\Monitoring\Backend\Ido\Query;
+
+use Icinga\Data\Filter\FilterExpression;
+use Zend_Db_Select;
+use Icinga\Data\Filter\Filter;
+
+/**
+ * Query for contacts
+ */
+class ContactQuery extends IdoQuery
+{
+    protected $columnMap = [
+        'contacts' => [
+            'contact_id'                        => 'c.contact_id',
+            'contact'                           => 'c.contact',
+            'contact_name'                      => 'c.contact_name',
+            'contact_alias'                     => 'c.contact_alias',
+            'contact_email'                     => 'c.contact_email',
+            'contact_pager'                     => 'c.contact_pager',
+            'contact_object_id'                 => 'c.contact_object_id',
+            'contact_has_host_notfications'     => 'c.contact_has_host_notfications',
+            'contact_has_service_notfications'  => 'c.contact_has_service_notfications',
+            'contact_can_submit_commands'       => 'c.contact_can_submit_commands',
+            'contact_notify_service_recovery'   => 'c.contact_notify_service_recovery',
+            'contact_notify_service_warning'    => 'c.contact_notify_service_warning',
+            'contact_notify_service_critical'   => 'c.contact_notify_service_critical',
+            'contact_notify_service_unknown'    => 'c.contact_notify_service_unknown',
+            'contact_notify_service_flapping'   => 'c.contact_notify_service_flapping',
+            'contact_notify_service_downtime'   => 'c.contact_notify_service_downtime',
+            'contact_notify_host_recovery'      => 'c.contact_notify_host_recovery',
+            'contact_notify_host_down'          => 'c.contact_notify_host_down',
+            'contact_notify_host_unreachable'   => 'c.contact_notify_host_unreachable',
+            'contact_notify_host_flapping'      => 'c.contact_notify_host_flapping',
+            'contact_notify_host_downtime'      => 'c.contact_notify_host_downtime',
+            'contact_notify_host_timeperiod'    => 'c.contact_notify_host_timeperiod',
+            'contact_notify_service_timeperiod' => 'c.contact_notify_service_timeperiod'
+        ]
+    ];
+
+    /** @var Zend_Db_Select The union */
+    protected $contactQuery;
+
+    /** @var IdoQuery[] Subqueries used for the contact query */
+    protected $subQueries = [];
+
+    public function allowsCustomVars()
+    {
+        foreach ($this->subQueries as $query) {
+            if (! $query->allowsCustomVars()) {
+                return false;
+            }
+        }
+
+        return true;
+    }
+
+    public function addFilter(Filter $filter)
+    {
+        $strangers = array_diff(
+            $filter->listFilteredColumns(),
+            array_keys($this->columnMap['contacts'])
+        );
+        if (! empty($strangers)) {
+            $this->transformToUnion();
+        }
+
+        foreach ($this->subQueries as $sub) {
+            $sub->applyFilter(clone $filter);
+        }
+
+        return $this;
+    }
+
+    protected function joinBaseTables()
+    {
+        $this->contactQuery = $this->createSubQuery('Hostcontact', array_keys($this->columnMap['contacts']));
+        $this->contactQuery->setIsSubQuery();
+        $this->subQueries[] = $this->contactQuery;
+
+        $this->select->from(
+            ['c' => $this->contactQuery],
+            []
+        );
+
+        $this->joinedVirtualTables['contacts'] = true;
+    }
+
+    public function order($columnOrAlias, $dir = null)
+    {
+        foreach ($this->subQueries as $sub) {
+            $sub->requireColumn($columnOrAlias);
+        }
+
+        return parent::order($columnOrAlias, $dir);
+    }
+
+    public function where($condition, $value = null)
+    {
+        $this->requireColumn($condition);
+        foreach ($this->subQueries as $sub) {
+            $sub->where($condition, $value);
+        }
+
+        return $this;
+    }
+
+    public function whereEx(FilterExpression $ex)
+    {
+        $this->requireColumn($ex->getColumn());
+        foreach ($this->subQueries as $sub) {
+            $sub->whereEx($ex);
+        }
+
+        return $this;
+    }
+
+    public function transformToUnion()
+    {
+        $this->contactQuery = $this->db->select();
+        $this->select->reset();
+        $this->subQueries = [];
+
+        $this->select->distinct()->from(
+            ['c' => $this->contactQuery],
+            []
+        );
+
+        $hosts = $this->createSubQuery('Hostcontact', array_keys($this->columnMap['contacts']));
+        $this->subQueries[] = $hosts;
+        $this->contactQuery->union([$hosts], Zend_Db_Select::SQL_UNION_ALL);
+
+        $services = $this->createSubQuery('Servicecontact', array_keys($this->columnMap['contacts']));
+        $this->subQueries[] = $services;
+        $this->contactQuery->union([$services], Zend_Db_Select::SQL_UNION_ALL);
+    }
+}