From dc6ec2a41dfd53c67d09da0ddcc32efb44e84ceb Mon Sep 17 00:00:00 2001
From: Thomas Gelf <thomas@gelf.net>
Date: Wed, 14 Jun 2017 14:55:26 +0200
Subject: [PATCH] CoreApi: one more trait, for CoreApi functionality

---
 .../Web/Controller/Extension/CoreApi.php      | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 library/Director/Web/Controller/Extension/CoreApi.php

diff --git a/library/Director/Web/Controller/Extension/CoreApi.php b/library/Director/Web/Controller/Extension/CoreApi.php
new file mode 100644
index 00000000..75cba505
--- /dev/null
+++ b/library/Director/Web/Controller/Extension/CoreApi.php
@@ -0,0 +1,46 @@
+<?php
+
+namespace Icinga\Module\Director\Web\Controller\Extension;
+
+use Icinga\Module\Director\Objects\IcingaEndpoint;
+use Icinga\Module\Director\Core\CoreApi as Api;
+
+trait CoreApi
+{
+    /** @var Api */
+    private $api;
+
+    /**
+     * @return Api|null
+     */
+    public function getApiIfAvailable()
+    {
+        if ($this->api === null) {
+            if ($this->db()->hasDeploymentEndpoint()) {
+                $endpoint = $this->db()->getDeploymentEndpoint();
+                $this->api = $endpoint->api();
+            }
+        }
+
+        return $this->api;
+    }
+
+    /**
+     * @param string $endpointName
+     * @return Api
+     */
+    public function api($endpointName = null)
+    {
+        if ($this->api === null) {
+            if ($endpointName === null) {
+                $endpoint = $this->db()->getDeploymentEndpoint();
+            } else {
+                $endpoint = IcingaEndpoint::load($endpointName, $this->db());
+            }
+
+            $this->api = $endpoint->api();
+        }
+
+        return $this->api;
+    }
+}