nextcloud-recognize/lib/Service/IgnoreService.php

<?php

/*
 * Copyright (c) 2022 The Recognize contributors.
 * This file is licensed under the Affero General Public License version 3 or later. See the COPYING file.
 */
declare(strict_types=1);
namespace OCA\Recognize\Service;

use OC\Files\Cache\CacheQueryBuilder;
use OCP\DB\QueryBuilder\IQueryBuilder;
use OCP\FilesMetadata\IFilesMetadataManager;
use OCP\ICache;
use OCP\ICacheFactory;
use OCP\IDBConnection;

final class IgnoreService {
	private array $inMemoryCache = [];
	private ICache $localCache;

	public function __construct(
		private IDBConnection $db,
		ICacheFactory $cacheFactory,
		private IFilesMetadataManager $metadataManager) {
		$this->localCache = $cacheFactory->createLocal('recognize-ignored-directories');
	}

	/**
	 * @param int $storageId
	 * @param string[] $ignoreMarkers
	 * @return list<string>
	 * @throws \OCP\DB\Exception
	 */
	public function getIgnoredDirectories(int $storageId, array $ignoreMarkers): array {
		$cacheKey = $storageId . '-' . implode(',', $ignoreMarkers);
		if (isset($this->inMemoryCache[$cacheKey])) {
			return $this->inMemoryCache[$cacheKey];
		}
		/** @var list<string> $directories */
		$directories = $this->localCache->get($cacheKey);
		if ($directories !== null) {
			return $directories;
		}

		$qb = new CacheQueryBuilder($this->db->getQueryBuilder(), $this->metadataManager);
		$result = $qb->selectFileCache()
			->andWhere($qb->expr()->in('name', $qb->createNamedParameter($ignoreMarkers, IQueryBuilder::PARAM_STR_ARRAY)))
			->andWhere($qb->expr()->eq('storage', $qb->createNamedParameter($storageId, IQueryBuilder::PARAM_INT)))
			->executeQuery();
		/**
		 * @var list<array{path: string}> $ignoreFiles
		 */
		$ignoreFiles = $result->fetchAll();
		$directories = array_map(static fn ($file): string => dirname($file['path']) === '.' ? '' : dirname($file['path']), $ignoreFiles);
		$this->inMemoryCache[$cacheKey] = $directories;
		$this->localCache->set($cacheKey, $directories);
		return $directories;
	}

	public function clearCacheForStorage(int $storageId) : void {
		$this->inMemoryCache = [];
		$this->localCache->clear($storageId . '-');
	}
}