diff options
Diffstat (limited to 'ncp-previewgenerator/lib/Watcher.php')
-rw-r--r-- | ncp-previewgenerator/lib/Watcher.php | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/ncp-previewgenerator/lib/Watcher.php b/ncp-previewgenerator/lib/Watcher.php new file mode 100644 index 00000000..5344ff7b --- /dev/null +++ b/ncp-previewgenerator/lib/Watcher.php @@ -0,0 +1,77 @@ +<?php +declare(strict_types=1); +/** + * @copyright Copyright (c) 2016, Roeland Jago Douma <roeland@famdouma.nl> + * + * @author Roeland Jago Douma <roeland@famdouma.nl> + * + * @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 OCA\PreviewGenerator; + +use OCP\Files\Folder; +use OCP\Files\Node; +use OCP\IDBConnection; +use OCP\IUserManager; + +class Watcher { + + /** @var IDBConnection */ + private $connection; + + /** @var IUserManager */ + private $userManager; + + public function __construct(IDBConnection $connection, + IUserManager $userManager) { + $this->connection = $connection; + $this->userManager = $userManager; + } + + public function postWrite(Node $node) { + $absPath = ltrim($node->getPath(), '/'); + $owner = explode('/', $absPath)[0]; + + if ($node instanceof Folder || !$this->userManager->userExists($owner)) { + return; + } + + $qb = $this->connection->getQueryBuilder(); + $qb->select('id') + ->from('preview_generation') + ->where( + $qb->expr()->andX( + $qb->expr()->eq('uid', $qb->createNamedParameter($owner)), + $qb->expr()->eq('file_id', $qb->createNamedParameter($node->getId())) + ) + )->setMaxResults(1); + $cursor = $qb->execute(); + $inTable = $cursor->fetch() !== false; + $cursor->closeCursor(); + + // Don't insert if there is already such an entry + if ($inTable) { + return; + } + + $qb = $this->connection->getQueryBuilder(); + $qb->insert('preview_generation') + ->setValue('uid', $qb->createNamedParameter($owner)) + ->setValue('file_id', $qb->createNamedParameter($node->getId())); + $qb->execute(); + } +} |