* * @author Roeland Jago Douma * * @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 . * */ 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(); } }