1 files changed, 97 insertions, 0 deletions
diff --git a/library/Cube/Ido/IdoServiceStatusCube.php b/library/Cube/Ido/IdoServiceStatusCube.php
new file mode 100644
index 0000000..a403645
--- /dev/null
+++ b/library/Cube/Ido/IdoServiceStatusCube.php
@@ -0,0 +1,97 @@
+<?php
+
+// Icinga Web 2 Cube Module | (c) 2019 Icinga GmbH | GPLv2
+
+namespace Icinga\Module\Cube\Ido;
+
+use Icinga\Module\Cube\CubeRenderer\ServiceStatusCubeRenderer;
+
+class IdoServiceStatusCube extends IdoCube
+{
+    public function getRenderer()
+    {
+        return new ServiceStatusCubeRenderer($this);
+    }
+
+    public function getAvailableFactColumns()
+    {
+        return [
+            'services_cnt'                => 'SUM(CASE WHEN ss.has_been_checked = 1 THEN 1 ELSE 0 END)',
+            'services_critical'           => 'SUM(CASE WHEN ss.has_been_checked = 1 AND ss.current_state = 2'
+                . ' THEN 1 ELSE 0 END)',
+            'services_unhandled_critical' => 'SUM(CASE WHEN ss.has_been_checked = 1 AND ss.current_state = 2'
+                . ' AND ss.problem_has_been_acknowledged = 0 AND ss.scheduled_downtime_depth = 0'
+                . ' THEN 1 ELSE 0 END)',
+            'services_warning'           => 'SUM(CASE WHEN ss.current_state = 1 THEN 1 ELSE 0 END)',
+            'services_unhandled_warning' => 'SUM(CASE WHEN ss.current_state = 1'
+                . ' AND ss.problem_has_been_acknowledged = 0 AND ss.scheduled_downtime_depth = 0'
+                . ' THEN 1 ELSE 0 END)',
+            'services_unknown'           => 'SUM(CASE WHEN ss.current_state = 3 THEN 1 ELSE 0 END)',
+            'services_unhandled_unknown' => 'SUM(CASE WHEN ss.current_state = 3'
+                . ' AND ss.problem_has_been_acknowledged = 0 AND ss.scheduled_downtime_depth = 0'
+                . ' THEN 1 ELSE 0 END)',
+        ];
+    }
+
+    /**
+     * This returns a list of all available Dimensions
+     *
+     * @return array
+     */
+    public function listAvailableDimensions()
+    {
+        $this->requireBackend();
+
+        $view = $this->backend->select()->from('servicestatus');
+
+        $view->applyFilter($this->getMonitoringRestriction());
+
+        $select = $view->getQuery()->clearOrder()->getSelectQuery();
+
+        $select
+            ->columns('cv.varname')
+            ->join(
+                ['cv' => $this->tableName('icinga_customvariablestatus')],
+                'cv.object_id = so.object_id',
+                []
+            )
+            ->group('cv.varname');
+
+        if (version_compare($this->getIdoVersion(), '1.12.0', '>=')) {
+            $select->where('cv.is_json = 0');
+        }
+
+        $select->order('cv.varname');
+
+        $dimensions = $this->filterProtectedCustomvars($this->db()->fetchCol($select));
+        $keys = array_map('strtolower', $dimensions);
+
+        return array_combine($keys, $dimensions);
+    }
+
+    public function prepareInnerQuery()
+    {
+        $this->requireBackend();
+
+        $view = $this->backend->select()->from('servicestatus');
+
+        $view->getQuery()->requireColumn('service_state');
+
+        $view->applyFilter($this->getMonitoringRestriction());
+
+        $select = $view->getQuery()->clearOrder()->getSelectQuery();
+
+        return $select;
+    }
+
+    public function createDimension($name)
+    {
+        $this->registerAvailableDimensions();
+
+        if (isset($this->availableDimensions[$name])) {
+            return clone $this->availableDimensions[$name];
+        }
+
+        return new CustomVarDimension($name, CustomVarDimension::TYPE_SERVICE);
+    }
+}