feat(files): Add support for storage-id and path in RepairTree command

This allows to run the command on specific part of the filecache. When the filecache is big, it prevents overloading the DB.

Signed-off-by: Louis Chmn <[email protected]>

Author: artonge

apps/files/lib/Command/RepairTree.php

@@ -11,6 +11,7 @@
 use OCP\IDBConnection;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
 
 class RepairTree extends Command {
@@ -26,11 +27,16 @@ protected function configure(): void {
 		$this
 			->setName('files:repair-tree')
 			->setDescription('Try and repair malformed filesystem tree structures (may be necessary to run multiple times for nested malformations)')
-			->addOption('dry-run');
+			->addOption('dry-run')
+			->addOption('storage-id', 's', InputOption::VALUE_OPTIONAL, 'If set, only repair files within the given storage ID', null)
+			->addOption('path', 'p', InputOption::VALUE_OPTIONAL, 'If set, only repair files within the given path', null);
 	}
 
 	public function execute(InputInterface $input, OutputInterface $output): int {
-		$rows = $this->findBrokenTreeBits();
+		$rows = $this->findBrokenTreeBits(
+			$input->getOption('storage-id'),
+			$input->getOption('path'),
+		);
 		$fix = !$input->getOption('dry-run');
 
 		$output->writeln('Found ' . count($rows) . ' file entries with an invalid path');
@@ -88,7 +94,7 @@ private function deleteById(int $fileId): void {
 		$query->executeStatement();
 	}
 
-	private function findBrokenTreeBits(): array {
+	private function findBrokenTreeBits(string|null $storageId, string|null $path): array {
 		$query = $this->connection->getQueryBuilder();
 
 		$query->select('f.fileid', 'f.path', 'f.parent', 'f.name')
@@ -108,6 +114,14 @@ private function findBrokenTreeBits(): array {
 				$query->expr()->neq('f.storage', 'p.storage')
 			));
 
+		if ($storageId !== null) {
+			$query->andWhere($query->expr()->eq('f.storage', $query->createNamedParameter($storageId)));
+		}
+
+		if ($path !== null) {
+			$query->andWhere($query->expr()->like('f.path', $query->createNamedParameter($path . '%')));
+		}
+
 		return $query->executeQuery()->fetchAllAssociative();
 	}
 }