1 files changed, 112 insertions, 0 deletions
diff --git a/library/Cube/Ido/IdoHostStatusCube.php b/library/Cube/Ido/IdoHostStatusCube.php
new file mode 100644
index 0000000..4881a9b
--- /dev/null
+++ b/library/Cube/Ido/IdoHostStatusCube.php
@@ -0,0 +1,112 @@
+<?php
+
+// Icinga Web 2 Cube Module | (c) 2016 Icinga GmbH | GPLv2
+
+namespace Icinga\Module\Cube\Ido;
+
+use Icinga\Module\Cube\CubeRenderer\HostStatusCubeRenderer;
+use Icinga\Module\Cube\Ido\DataView\Hoststatus;
+
+class IdoHostStatusCube extends IdoCube
+{
+    public function getRenderer()
+    {
+        return new HostStatusCubeRenderer($this);
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function getAvailableFactColumns()
+    {
+        return array(
+            'hosts_cnt'            => 'SUM(CASE WHEN hs.has_been_checked = 1 THEN 1 ELSE 0 END)',
+            'hosts_down'           => 'SUM(CASE WHEN hs.has_been_checked = 1 AND hs.current_state = 1'
+                . ' THEN 1 ELSE 0 END)',
+            'hosts_unhandled_down' => 'SUM(CASE WHEN hs.has_been_checked = 1 AND hs.current_state = 1'
+                . ' AND hs.problem_has_been_acknowledged = 0 AND hs.scheduled_downtime_depth = 0'
+                . ' THEN 1 ELSE 0 END)',
+            'hosts_unreachable'           => 'SUM(CASE WHEN hs.current_state = 2 THEN 1 ELSE 0 END)',
+            'hosts_unhandled_unreachable' => 'SUM(CASE WHEN hs.current_state = 2'
+                . ' AND hs.problem_has_been_acknowledged = 0 AND hs.scheduled_downtime_depth = 0'
+                . ' THEN 1 ELSE 0 END)',
+        );
+    }
+
+    public function createDimension($name)
+    {
+        $this->registerAvailableDimensions();
+
+        if (isset($this->availableDimensions[$name])) {
+            return clone $this->availableDimensions[$name];
+        }
+
+        return new CustomVarDimension($name, CustomVarDimension::TYPE_HOST);
+    }
+
+    /**
+     * Add a specific named dimension
+     *
+     * Right now this are just custom vars, we might support group memberships
+     * or other properties in future
+     *
+     * @param string $name
+     * @return $this
+     */
+    public function addDimensionByName($name)
+    {
+        if (count($this->filterProtectedCustomvars(array($name))) === 1) {
+            $this->addDimension($this->createDimension($name));
+        }
+
+        return $this;
+    }
+
+    /**
+     * This returns a list of all available Dimensions
+     *
+     * @return array
+     */
+    public function listAvailableDimensions()
+    {
+        $this->requireBackend();
+
+        $view = $this->backend->select()->from('hoststatus');
+
+        $view->applyFilter($this->getMonitoringRestriction());
+
+        $select = $view->getQuery()->clearOrder()->getSelectQuery();
+
+        $select
+            ->columns('cv.varname')
+            ->join(
+                ['cv' => $this->tableName('icinga_customvariablestatus')],
+                'cv.object_id = ho.object_id',
+                []
+            )
+            ->group('cv.varname');
+
+        if (version_compare($this->getIdoVersion(), '1.12.0', '>=')) {
+            $select->where('cv.is_json = 0');
+        }
+
+        $select->order('cv.varname');
+
+        return $this->filterProtectedCustomvars($this->db()->fetchCol($select));
+    }
+
+    public function prepareInnerQuery()
+    {
+        $this->requireBackend();
+
+        $view = new Hoststatus($this->backend);
+
+        $view->getQuery()->requireColumn('host_state');
+
+        $view->applyFilter($this->getMonitoringRestriction());
+
+        $select = $view->getQuery()->clearOrder()->getSelectQuery();
+
+        return $select;
+    }
+}