1 files changed, 150 insertions, 0 deletions
diff --git a/library/Director/CheckPlugin/CheckResults.php b/library/Director/CheckPlugin/CheckResults.php
new file mode 100644
index 0000000..7e20225
--- /dev/null
+++ b/library/Director/CheckPlugin/CheckResults.php
@@ -0,0 +1,150 @@
+<?php
+
+namespace Icinga\Module\Director\CheckPlugin;
+
+use Exception;
+
+class CheckResults
+{
+    /** @var string */
+    protected $name;
+
+    /** @var PluginState */
+    protected $state;
+
+    /** @var CheckResult[] */
+    protected $results = [];
+
+    protected $stateCounters = [
+        0 => 0,
+        1 => 0,
+        2 => 0,
+        3 => 0,
+    ];
+
+    public function __construct($name)
+    {
+        $this->name = $name;
+        $this->state = new PluginState(0);
+    }
+
+    public function getName()
+    {
+        return $this->name;
+    }
+
+    public function add(CheckResult $result)
+    {
+        $this->results[] = $result;
+        $this->state->raise($result->getState());
+        $this->stateCounters[$result->getState()->getNumeric()]++;
+
+        return $this;
+    }
+
+    public function getStateCounters()
+    {
+        return $this->stateCounters;
+    }
+
+    public function getProblemSummary()
+    {
+        $summary = [];
+        for ($i = 1; $i <= 3; $i++) {
+            $count = $this->stateCounters[$i];
+            if ($count === 0) {
+                continue;
+            }
+            $summary[PluginState::create($i)->getName()] = $count;
+        }
+
+        return $summary;
+    }
+
+    public function getStateSummaryString()
+    {
+        $summary = [sprintf(
+            '%d tests OK',
+            $this->stateCounters[0]
+        )];
+
+        for ($i = 1; $i <= 3; $i++) {
+            $count = $this->stateCounters[$i];
+            if ($count === 0) {
+                continue;
+            }
+            $summary[] = sprintf(
+                '%dx %s',
+                $count,
+                PluginState::create($i)->getName()
+            );
+        }
+
+        return implode(', ', $summary);
+    }
+
+    public function getOutput()
+    {
+        $output = sprintf(
+            "%s: %s\n",
+            $this->name,
+            $this->getStateSummaryString()
+        );
+
+        foreach ($this->results as $result) {
+            $output .= sprintf(
+                "[%s] %s\n",
+                $result->getState()->getName(),
+                $result->getOutput()
+            );
+        }
+
+        return $output;
+    }
+
+    public function getResults()
+    {
+        return $this->results;
+    }
+
+    public function getState()
+    {
+        return $this->state;
+    }
+
+    public function hasProblems()
+    {
+        return $this->getState()->getNumeric() !== 0;
+    }
+
+    public function hasErrors()
+    {
+        $state = $this->getState()->getNumeric();
+        return $state !== 0 && $state !== 1;
+    }
+
+    public function succeed($message)
+    {
+        $this->add(new CheckResult($message));
+
+        return $this;
+    }
+
+    public function warn($message)
+    {
+        $this->add(new CheckResult($message, 1));
+
+        return $this;
+    }
+
+    public function fail($message, $errorState = 'CRITICAL')
+    {
+        if ($message instanceof Exception) {
+            $message = $message->getMessage();
+        }
+
+        $this->add(new CheckResult($message, $errorState));
+
+        return $this;
+    }
+}