1 files changed, 133 insertions, 0 deletions
diff --git a/modules/monitoring/library/Monitoring/Object/HostList.php b/modules/monitoring/library/Monitoring/Object/HostList.php
new file mode 100644
index 0000000..8b1947d
--- /dev/null
+++ b/modules/monitoring/library/Monitoring/Object/HostList.php
@@ -0,0 +1,133 @@
+<?php
+/* Icinga Web 2 | (c) 2014 Icinga Development Team | GPLv2+ */
+
+namespace Icinga\Module\Monitoring\Object;
+
+use Icinga\Data\DataArray\ArrayDatasource;
+use Icinga\Data\Filter\Filter;
+use Icinga\Data\Filter\FilterOr;
+use Icinga\Data\SimpleQuery;
+use Icinga\Util\StringHelper;
+
+/**
+ * A host list
+ */
+class HostList extends ObjectList
+{
+    protected $dataViewName = 'hoststatus';
+
+    protected $columns = array('host_name');
+
+    protected function fetchObjects()
+    {
+        $hosts = array();
+        $query = $this->backend->select()->from($this->dataViewName, $this->columns)->applyFilter($this->filter)
+            ->getQuery()->getSelectQuery()->query();
+        foreach ($query as $row) {
+            /** @var object $row */
+            $host = new Host($this->backend, $row->host_name);
+            $host->setProperties($row);
+            $hosts[] = $host;
+        }
+        return $hosts;
+    }
+
+    /**
+     * Create a state summary of all hosts that can be consumed by hostssummary.phtml
+     *
+     * @return  SimpleQuery
+     */
+    public function getStateSummary()
+    {
+        $hostStates = array_fill_keys(self::getHostStatesSummaryEmpty(), 0);
+        foreach ($this as $host) {
+            $unhandled = (bool) $host->problem === true && (bool) $host->handled === false;
+
+            $stateName = 'hosts_' . $host::getStateText($host->state);
+            ++$hostStates[$stateName];
+            ++$hostStates[$stateName. ($unhandled ? '_unhandled' : '_handled')];
+        }
+
+        $hostStates['hosts_total'] = count($this);
+
+        $ds = new ArrayDatasource(array((object) $hostStates));
+        return $ds->select();
+    }
+
+    /**
+     * Return an empty array with all possible host state names
+     *
+     * @return array    An array containing all possible host states as keys and 0 as values.
+     */
+    public static function getHostStatesSummaryEmpty()
+    {
+        return StringHelper::cartesianProduct(
+            array(
+                array('hosts'),
+                array(
+                    Host::getStateText(Host::STATE_UP),
+                    Host::getStateText(Host::STATE_DOWN),
+                    Host::getStateText(Host::STATE_UNREACHABLE),
+                    Host::getStateText(Host::STATE_PENDING)
+                ),
+                array(null, 'handled', 'unhandled')
+            ),
+            '_'
+        );
+    }
+
+    /**
+     * Returns a Filter that matches all hosts in this list
+     *
+     * @return Filter
+     */
+    public function objectsFilter($columns = array('host' => 'host'))
+    {
+        $filterExpression = array();
+        foreach ($this as $host) {
+            $filterExpression[] = Filter::where($columns['host'], $host->getName());
+        }
+        return FilterOr::matchAny($filterExpression);
+    }
+
+    /**
+     * Get the comments
+     *
+     * @return \Icinga\Module\Monitoring\DataView\Hostcomment
+     */
+    public function getComments()
+    {
+        return $this->backend
+            ->select()
+            ->from('hostcomment', array('host_name'))
+            ->applyFilter(clone $this->filter);
+    }
+
+    /**
+     * Get the scheduled downtimes
+     *
+     * @return \Icinga\Module\Monitoring\DataView\Hostdowntime
+     */
+    public function getScheduledDowntimes()
+    {
+        return $this->backend
+            ->select()
+            ->from('hostdowntime', array('host_name'))
+            ->applyFilter(clone $this->filter);
+    }
+
+    /**
+     * @return ObjectList
+     */
+    public function getUnacknowledgedObjects()
+    {
+        $unhandledObjects = array();
+        foreach ($this as $object) {
+            if (! in_array((int) $object->state, array(0, 99)) &&
+                (bool) $object->host_acknowledged === false) {
+                $unhandledObjects[] = $object;
+            }
+        }
+        return $this->newFromArray($unhandledObjects);
+    }
+}