Adding upstream version 2.4.0.upstream/2.4.0upstream

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

1 files changed, 94 insertions, 0 deletions

diff --git a/library/Businessprocess/IcingaDbObject.php b/library/Businessprocess/IcingaDbObject.php
new file mode 100644
index 0000000..cad459f
--- /dev/null
+++ b/library/Businessprocess/IcingaDbObject.php
@@ -0,0 +1,94 @@
+<?php
+
+namespace Icinga\Module\Businessprocess;
+
+use Icinga\Module\Icingadb\Common\Auth;
+use Icinga\Module\Icingadb\Common\Database as IcingadbDatabase;
+use Icinga\Module\Icingadb\Model\Host;
+use Icinga\Module\Icingadb\Model\Service;
+use ipl\Sql\Connection as IcingaDbConnection;
+use ipl\Web\Filter\QueryString;
+
+class IcingaDbObject
+{
+    use IcingadbDatabase;
+
+    use Auth;
+
+    /** @var BpConfig */
+    protected $config;
+
+    /** @var IcingaDbConnection */
+    protected $conn;
+
+    public function __construct()
+    {
+        $this->conn = $this->getDb();
+    }
+
+    public function fetchHosts($filter = null)
+    {
+
+        $hosts = Host::on($this->conn);
+
+        if ($filter !== null) {
+            $filterQuery = QueryString::parse($filter);
+
+            $hosts->filter($filterQuery);
+        }
+
+        $hosts->orderBy('host.name');
+
+        $this->applyIcingaDbRestrictions($hosts);
+
+        return $hosts;
+    }
+
+    public function fetchServices($filter)
+    {
+        $services = Service::on($this->conn)
+            ->with('host');
+
+        if ($filter !== null) {
+            $filterQuery = QueryString::parse($filter);
+
+            $services->filter($filterQuery);
+        }
+
+        $services->orderBy('service.name');
+
+        $this->applyIcingaDbRestrictions($services);
+
+        return $services;
+    }
+
+    public function yieldHostnames($filter = null)
+    {
+        foreach ($this->fetchHosts($filter) as $host) {
+            yield $host->name;
+        }
+    }
+
+    public function yieldServicenames($host)
+    {
+        $filter = "host.name=$host";
+
+        foreach ($this->fetchServices($filter) as $service) {
+            yield $service->name;
+        }
+    }
+
+    public static function applyIcingaDbRestrictions($query)
+    {
+        $object = new self;
+        $object->applyRestrictions($query);
+
+        return $object;
+    }
+
+    public static function fetchDb()
+    {
+        $object = new self;
+        return $object->getDb();
+    }
+}