1 files changed, 109 insertions, 0 deletions
diff --git a/library/Cube/IcingaDb/IcingaDbServiceStatusCube.php b/library/Cube/IcingaDb/IcingaDbServiceStatusCube.php
new file mode 100644
index 0000000..9336cdf
--- /dev/null
+++ b/library/Cube/IcingaDb/IcingaDbServiceStatusCube.php
@@ -0,0 +1,109 @@
+<?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\CustomvarFlat;
+use Icinga\Module\Icingadb\Model\Service;
+use Icinga\Module\Icingadb\Model\ServicestateSummary;
+use ipl\Stdlib\Filter;
+
+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()
+    {
+        $db = $this->getDb();
+
+        $query = CustomvarFlat::on($db);
+        $this->applyRestrictions($query);
+
+        $query
+            ->columns('flatname')
+            ->orderBy('flatname')
+            ->filter(Filter::like('service.id', '*'));
+        $query->getSelectBase()->groupBy('flatname');
+
+        $dimensions = [];
+        foreach ($query as $row) {
+            // Replaces array index notations with [*] to get results for arbitrary indexes
+            $name = preg_replace('/\\[\d+](?=\\.|$)/', '[*]', $row->flatname);
+            $dimensions[$name] = $name;
+        }
+
+        return $dimensions;
+    }
+
+    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;
+    }
+}