1 files changed, 155 insertions, 0 deletions
diff --git a/lib/private/Collaboration/Resources/Collection.php b/lib/private/Collaboration/Resources/Collection.php
new file mode 100644
index 00000000000..8509b4442de
--- /dev/null
+++ b/lib/private/Collaboration/Resources/Collection.php
@@ -0,0 +1,155 @@
+<?php
+declare(strict_types=1);
+/**
+ * @copyright Copyright (c) 2018 Joas Schilling <coding@schilljs.com>
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+namespace OC\Collaboration\Resources;
+
+
+use Doctrine\DBAL\Exception\ConstraintViolationException;
+use OCP\Collaboration\Resources\IManager;
+use OCP\Collaboration\Resources\ResourceException;
+use OCP\Collaboration\Resources\ICollection;
+use OCP\Collaboration\Resources\IResource;
+use OCP\DB\QueryBuilder\IQueryBuilder;
+use OCP\IDBConnection;
+
+class Collection implements ICollection {
+
+	/** @var IManager */
+	protected $manager;
+
+	/** @var IDBConnection */
+	protected $connection;
+
+	/** @var int */
+	protected $id;
+
+	/** @var IResource[] */
+	protected $resources;
+
+	public function __construct(IManager $manager, IDBConnection $connection, int $id) {
+		$this->manager = $manager;
+		$this->connection = $connection;
+		$this->id = $id;
+		$this->resources = [];
+	}
+
+	/**
+	 * @return IResource[]
+	 * @since 15.0.0
+	 */
+	public function getResources(): array {
+		if (empty($this->resources)) {
+			$query = $this->connection->getQueryBuilder();
+			$query->select('resource_type', 'resource_id')
+				->from('collres_resources')
+				->where($query->expr()->eq('collection_id', $query->createNamedParameter($this->id, IQueryBuilder::PARAM_INT)));
+
+			$result = $query->execute();
+			while ($row = $result->fetch()) {
+				$this->resources[] = $this->manager->getResource($row['resource_type'], $row['resource_id']);
+			}
+			$result->closeCursor();
+		}
+
+		return $this->resources;
+	}
+
+	/**
+	 * Adds a resource to a collection
+	 *
+	 * @param IResource $resource
+	 * @throws ResourceException when the resource is already part of the collection
+	 * @since 15.0.0
+	 */
+	public function addResource(IResource $resource) {
+		array_map(function(IResource $r) use ($resource) {
+			if ($this->isSameResource($r, $resource)) {
+				throw new ResourceException('Already part of the collection');
+			}
+		}, $this->resources);
+
+		$this->resources[] = $resource;
+
+		if ($this->id === 0) {
+			$this->makeCollectionPersistent();
+		}
+
+		$query = $this->connection->getQueryBuilder();
+		$query->insert('collres_resources')
+			->values([
+				'collection_id' => $query->createNamedParameter($this->id, IQueryBuilder::PARAM_INT),
+				'resource_type' => $query->createNamedParameter($resource->getType()),
+				'resource_id' => $query->createNamedParameter($resource->getId()),
+			]);
+
+		try {
+			$query->execute();
+		} catch (ConstraintViolationException $e) {
+			throw new ResourceException('Already part of the collection');
+		}
+	}
+
+	/**
+	 * Removes a resource from a collection
+	 *
+	 * @param IResource $resource
+	 * @since 15.0.0
+	 */
+	public function removeResource(IResource $resource) {
+		$this->resources = array_filter($this->resources, function(IResource $r) use ($resource) {
+			return !$this->isSameResource($r, $resource);
+		});
+
+		$query = $this->connection->getQueryBuilder();
+		$query->delete('collres_resources')
+			->where($query->expr()->eq('collection_id', $query->createNamedParameter($this->id, IQueryBuilder::PARAM_INT)))
+			->andWhere($query->expr()->eq('resource_type', $query->createNamedParameter($resource->getType())))
+			->andWhere($query->expr()->eq('resource_id', $query->createNamedParameter($resource->getId())));
+		$query->execute();
+
+		if (empty($this->resources)) {
+			$this->makeCollectionUnsteady();
+		}
+	}
+
+	protected function isSameResource(IResource $resource1, IResource $resource2): bool {
+		return $resource1->getType() === $resource2->getType() &&
+			$resource1->getId() === $resource2->getId();
+	}
+
+	protected function makeCollectionPersistent() {
+		$query = $this->connection->getQueryBuilder();
+		$query->insert('collres_collections');
+		$query->execute();
+
+		$this->id = $query->getLastInsertId();
+	}
+
+	protected function makeCollectionUnsteady() {
+		$query = $this->connection->getQueryBuilder();
+		$query->delete('collres_collections')
+			->where($query->expr()->eq('id', $query->createNamedParameter($this->id, IQueryBuilder::PARAM_INT)));
+		$query->execute();
+
+		$this->id = 0;
+	}
+}