nextcloud/lib/private/Collaboration/Resources/Collection.php

156 lines
4.6 KiB
PHP
Raw Normal View History

<?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;
}
}