diff options
Diffstat (limited to 'lib/private/Collaboration/Resources/Collection.php')
-rw-r--r-- | lib/private/Collaboration/Resources/Collection.php | 155 |
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; + } +} |