Adding upstream version 2.5.0.upstream/2.5.0upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

2 files changed, 112 insertions, 0 deletions

diff --git a/library/Businessprocess/Web/Navigation/Renderer/ProcessProblemsBadge.php b/library/Businessprocess/Web/Navigation/Renderer/ProcessProblemsBadge.php
new file mode 100644
index 0000000..575dc5e
--- /dev/null
+++ b/library/Businessprocess/Web/Navigation/Renderer/ProcessProblemsBadge.php
@@ -0,0 +1,59 @@
+<?php
+
+namespace Icinga\Module\Businessprocess\Web\Navigation\Renderer;
+
+use Icinga\Module\Businessprocess\Storage\LegacyStorage;
+use Icinga\Web\Navigation\Renderer\BadgeNavigationItemRenderer;
+
+class ProcessProblemsBadge extends BadgeNavigationItemRenderer
+{
+    /**
+     * Cached count
+     *
+     * @var int
+     */
+    protected $count;
+
+    /** @var string */
+    private $bpConfigName;
+
+    public function getCount()
+    {
+        $count = 0;
+        if ($this->count === null) {
+            $storage = LegacyStorage::getInstance();
+            $bp = $storage->loadProcess($this->getBpConfigName());
+            foreach ($bp->getRootNodes() as $rootNode) {
+                if (! $rootNode->isEmpty() &&
+                    $rootNode->getState() !== $rootNode::ICINGA_PENDING
+                    && $rootNode->hasProblems()) {
+                    $count++;
+                }
+            }
+
+            $this->count = $count;
+            $this->setState(self::STATE_CRITICAL);
+        }
+
+        if ($count) {
+            $this->setTitle(sprintf(
+                tp('One unhandled root node critical', '%d unhandled root nodes critical', $count),
+                $count
+            ));
+        }
+
+        return $this->count;
+    }
+
+    public function setBpConfigName($bpConfigName)
+    {
+        $this->bpConfigName = $bpConfigName;
+
+        return $this;
+    }
+
+    public function getBpConfigName()
+    {
+        return $this->bpConfigName;
+    }
+}
diff --git a/library/Businessprocess/Web/Navigation/Renderer/ProcessesProblemsBadge.php b/library/Businessprocess/Web/Navigation/Renderer/ProcessesProblemsBadge.php
new file mode 100644
index 0000000..dd419a2
--- /dev/null
+++ b/library/Businessprocess/Web/Navigation/Renderer/ProcessesProblemsBadge.php
@@ -0,0 +1,53 @@
+<?php
+
+namespace Icinga\Module\Businessprocess\Web\Navigation\Renderer;
+
+use Icinga\Application\Modules\Module;
+use Icinga\Module\Businessprocess\ProvidedHook\Icingadb\IcingadbSupport;
+use Icinga\Module\Businessprocess\State\IcingaDbState;
+use Icinga\Module\Businessprocess\State\MonitoringState;
+use Icinga\Module\Businessprocess\Storage\LegacyStorage;
+use Icinga\Web\Navigation\Renderer\BadgeNavigationItemRenderer;
+
+class ProcessesProblemsBadge extends BadgeNavigationItemRenderer
+{
+    /**
+     * Cached count
+     *
+     * @var int
+     */
+    protected $count;
+
+    public function getCount()
+    {
+        if ($this->count === null) {
+            $storage = LegacyStorage::getInstance();
+            $count = 0;
+
+            foreach ($storage->listProcessNames() as $processName) {
+                $bp = $storage->loadProcess($processName);
+                if (Module::exists('icingadb') &&
+                    (! $bp->hasBackendName() && IcingadbSupport::useIcingaDbAsBackend())
+                ) {
+                    IcingaDbState::apply($bp);
+                } else {
+                    MonitoringState::apply($bp);
+                }
+
+                foreach ($bp->getRootNodes() as $rootNode) {
+                    if (! $rootNode->isEmpty() &&
+                        $rootNode->getState() !== $rootNode::ICINGA_PENDING
+                        && $rootNode->hasProblems()) {
+                        $count++;
+                        break;
+                    }
+                }
+            }
+
+            $this->count = $count;
+            $this->setState(self::STATE_CRITICAL);
+        }
+
+        return $this->count;
+    }
+}