first commit

libraries/vendor/joomla/di/src/ContainerAwareTrait.php

@@ -0,0 +1,61 @@
+<?php
+
+/**
+ * Part of the Joomla Framework DI Package
+ *
+ * @copyright  Copyright (C) 2013 - 2018 Open Source Matters, Inc. All rights reserved.
+ * @license    GNU General Public License version 2 or later; see LICENSE
+ */
+
+namespace Joomla\DI;
+
+use Joomla\DI\Exception\ContainerNotFoundException;
+
+/**
+ * Defines the trait for a Container Aware Class.
+ *
+ * @since  1.2
+ */
+trait ContainerAwareTrait
+{
+    /**
+     * DI Container
+     *
+     * @var    Container
+     * @since  1.2
+     */
+    private $container;
+
+    /**
+     * Get the DI container.
+     *
+     * @return  Container
+     *
+     * @since   1.2
+     * @throws  ContainerNotFoundException May be thrown if the container has not been set.
+     */
+    protected function getContainer()
+    {
+        if ($this->container) {
+            return $this->container;
+        }
+
+        throw new ContainerNotFoundException('Container not set in ' . \get_class($this));
+    }
+
+    /**
+     * Set the DI container.
+     *
+     * @param   Container  $container  The DI container.
+     *
+     * @return  $this
+     *
+     * @since   1.2
+     */
+    public function setContainer(Container $container)
+    {
+        $this->container = $container;
+
+        return $this;
+    }
+}