1 files changed, 94 insertions, 0 deletions
diff --git a/library/Cube/IcingaDb/IcingaDbServiceStatusCube.php b/library/Cube/IcingaDb/IcingaDbServiceStatusCube.php
new file mode 100644
index 0000000..ac59de2
--- /dev/null
+++ b/library/Cube/IcingaDb/IcingaDbServiceStatusCube.php
@@ -0,0 +1,94 @@
+<?php
+
+// Icinga Web 2 Cube Module | (c) 2022 Icinga GmbH | GPLv2
+
+namespace Icinga\Module\Cube\IcingaDb;
+
+use Icinga\Module\Cube\CubeRenderer\ServiceStatusCubeRenderer;
+use Icinga\Module\Icingadb\Model\Service;
+use Icinga\Module\Icingadb\Model\ServicestateSummary;
+use ipl\Stdlib\Filter;
+use ipl\Stdlib\Str;
+
+class IcingaDbServiceStatusCube extends IcingaDbCube
+{
+    public function getRenderer()
+    {
+        return new ServiceStatusCubeRenderer($this);
+    }
+
+    public function createDimension($name)
+    {
+        $this->registerAvailableDimensions();
+
+        if (isset($this->availableDimensions[$name])) {
+            return clone $this->availableDimensions[$name];
+        }
+
+        return new CustomVariableDimension($name);
+    }
+
+    public function getAvailableFactColumns()
+    {
+        return [
+            'services_cnt' => 'services_total',
+            'services_critical' => 'services_critical_handled + f.services_critical_unhandled',
+            'services_unhandled_critical' => 'services_critical_unhandled',
+            'services_warning' => 'services_warning_handled + f.services_warning_unhandled',
+            'services_unhandled_warning' => 'services_warning_unhandled',
+            'services_unknown' => 'services_unknown_handled + f.services_unknown_unhandled',
+            'services_unhandled_unknown' => 'services_unknown_unhandled',
+        ];
+    }
+
+    public function listAvailableDimensions()
+    {
+        return array_merge(
+            $this->fetchServiceVariableDimensions(),
+            $this->fetchHostVariableDimensions()
+        );
+    }
+
+    public function prepareInnerQuery()
+    {
+        $query = ServicestateSummary::on($this->getDb());
+        $query->columns(array_diff_key($query->getModel()->getColumns(), (new Service())->getColumns()));
+        $query->disableDefaultSort();
+        $this->applyRestrictions($query);
+
+        return $query;
+    }
+
+    /**
+     * Return Filter for Services cube.
+     *
+     * @return Filter\Any|Filter\Chain
+     */
+    public function getObjectsFilter()
+    {
+        if ($this->objectsFilter === null) {
+            $this->finalizeInnerQuery();
+
+            $services = $this->innerQuery()->columns([
+                'host_name' => 'host.name',
+                'service_name' => 'service.name'
+            ]);
+
+            $services->getSelectBase()->resetGroupBy();
+            $filter = Filter::any();
+
+            foreach ($services as $service) {
+                $filter->add(
+                    Filter::all(
+                        Filter::equal('service.name', $service->service_name),
+                        Filter::equal('host.name', $service->host_name)
+                    )
+                );
+            }
+
+            $this->objectsFilter = $filter;
+        }
+
+        return $this->objectsFilter;
+    }
+}