1 files changed, 99 insertions, 0 deletions
diff --git a/library/Director/Repository/RepositoryByObjectHelper.php b/library/Director/Repository/RepositoryByObjectHelper.php
new file mode 100644
index 0000000..0d1dda3
--- /dev/null
+++ b/library/Director/Repository/RepositoryByObjectHelper.php
@@ -0,0 +1,99 @@
+<?php
+
+namespace Icinga\Module\Director\Repository;
+
+use Icinga\Authentication\Auth;
+use Icinga\Module\Director\Db;
+use Icinga\Module\Director\Objects\IcingaObject;
+use RuntimeException;
+
+trait RepositoryByObjectHelper
+{
+    protected $type;
+
+    /** @var Db */
+    protected $connection;
+
+    /** @var Auth */
+    protected static $auth;
+
+    /** @var static[] */
+    protected static $instances = [];
+
+    protected function __construct($type, Db $connection)
+    {
+        $this->type = $type;
+        $this->connection = $connection;
+    }
+
+    /**
+     * @param string $type
+     * @return bool
+     */
+    public static function hasInstanceForType($type)
+    {
+        return array_key_exists($type, self::$instances);
+    }
+
+    /**
+     * @param string $type
+     * @param Db $connection
+     * @return static
+     */
+    public static function instanceByType($type, Db $connection)
+    {
+        if (! static::hasInstanceForType($type)) {
+            self::$instances[$type] = new static($type, $connection);
+        }
+
+        return self::$instances[$type];
+    }
+
+    /**
+     * @param IcingaObject $object
+     * @return bool
+     */
+    public static function hasInstanceForObject(IcingaObject $object)
+    {
+        return static::hasInstanceForType($object->getShortTableName());
+    }
+
+    /**
+     * @param IcingaObject $object
+     * @param Db|null $connection
+     * @return static
+     */
+    public static function instanceByObject(IcingaObject $object, Db $connection = null)
+    {
+        if (null === $connection) {
+            $connection = $object->getConnection();
+        }
+
+        if (! $connection) {
+            throw new RuntimeException(sprintf(
+                'Cannot use repository for %s "%s" as it has no DB connection',
+                $object->getShortTableName(),
+                $object->getObjectName()
+            ));
+        }
+
+        return static::instanceByType(
+            $object->getShortTableName(),
+            $connection
+        );
+    }
+
+    protected static function auth()
+    {
+        if (self::$auth === null) {
+            self::$auth = Auth::getInstance();
+        }
+
+        return self::$auth;
+    }
+
+    protected static function clearInstances()
+    {
+        self::$instances = [];
+    }
+}