diff options
author | Robin Appelman <robin@icewind.nl> | 2023-03-27 17:55:24 +0200 |
---|---|---|
committer | Robin Appelman <robin@icewind.nl> | 2023-04-03 18:59:02 +0200 |
commit | 9a731ad617816110d98fceb5c46360a372e2c750 (patch) | |
tree | 6d36ae5f6812b7609423c9193b220a8f0ff1178b /apps | |
parent | 5024f295dcfdf850d2d95f419bcf122bf08be5c8 (diff) | |
download | nextcloud-server-9a731ad617816110d98fceb5c46360a372e2c750.tar.gz nextcloud-server-9a731ad617816110d98fceb5c46360a372e2c750.zip |
add a bit more verbose option for trashbin cleanup
Signed-off-by: Robin Appelman <robin@icewind.nl>
Diffstat (limited to 'apps')
-rw-r--r-- | apps/files_trashbin/lib/Command/CleanUp.php | 27 | ||||
-rw-r--r-- | apps/files_trashbin/tests/Command/CleanUpTest.php | 52 |
2 files changed, 52 insertions, 27 deletions
diff --git a/apps/files_trashbin/lib/Command/CleanUp.php b/apps/files_trashbin/lib/Command/CleanUp.php index e3ed527e535..a00a96d5ee2 100644 --- a/apps/files_trashbin/lib/Command/CleanUp.php +++ b/apps/files_trashbin/lib/Command/CleanUp.php @@ -78,13 +78,14 @@ class CleanUp extends Command { protected function execute(InputInterface $input, OutputInterface $output): int { $users = $input->getArgument('user_id'); + $verbose = $input->getOption('verbose'); if ((!empty($users)) and ($input->getOption('all-users'))) { throw new InvalidOptionException('Either specify a user_id or --all-users'); } elseif (!empty($users)) { foreach ($users as $user) { if ($this->userManager->userExists($user)) { $output->writeln("Remove deleted files of <info>$user</info>"); - $this->removeDeletedFiles($user); + $this->removeDeletedFiles($user, $output, $verbose); } else { $output->writeln("<error>Unknown user $user</error>"); return 1; @@ -104,7 +105,7 @@ class CleanUp extends Command { $users = $backend->getUsers('', $limit, $offset); foreach ($users as $user) { $output->writeln(" <info>$user</info>"); - $this->removeDeletedFiles($user); + $this->removeDeletedFiles($user, $output, $verbose); } $offset += $limit; } while (count($users) >= $limit); @@ -117,19 +118,31 @@ class CleanUp extends Command { /** * remove deleted files for the given user - * - * @param string $uid */ - protected function removeDeletedFiles($uid) { + protected function removeDeletedFiles(string $uid, OutputInterface $output, bool $verbose): void { \OC_Util::tearDownFS(); \OC_Util::setupFS($uid); - if ($this->rootFolder->nodeExists('/' . $uid . '/files_trashbin')) { - $this->rootFolder->get('/' . $uid . '/files_trashbin')->delete(); + $path = '/' . $uid . '/files_trashbin'; + if ($this->rootFolder->nodeExists($path)) { + $node = $this->rootFolder->get($path); + + if ($verbose) { + $output->writeln("Deleting <info>" . \OC_Helper::humanFileSize($node->getSize()) . "</info> in trash for <info>$uid</info>."); + } + $node->delete(); + if ($this->rootFolder->nodeExists($path)) { + $output->writeln("<error>Trash folder sill exists after attempting to delete it</error>"); + return; + } $query = $this->dbConnection->getQueryBuilder(); $query->delete('files_trash') ->where($query->expr()->eq('user', $query->createParameter('uid'))) ->setParameter('uid', $uid); $query->execute(); + } else { + if ($verbose) { + $output->writeln("No trash found for <info>$uid</info>"); + } } } } diff --git a/apps/files_trashbin/tests/Command/CleanUpTest.php b/apps/files_trashbin/tests/Command/CleanUpTest.php index 949d0529fd6..2f582ed428b 100644 --- a/apps/files_trashbin/tests/Command/CleanUpTest.php +++ b/apps/files_trashbin/tests/Command/CleanUpTest.php @@ -32,6 +32,7 @@ use OCP\Files\IRootFolder; use OCP\IDBConnection; use Symfony\Component\Console\Exception\InvalidOptionException; use Symfony\Component\Console\Input\InputInterface; +use Symfony\Component\Console\Output\NullOutput; use Symfony\Component\Console\Output\OutputInterface; use Test\TestCase; @@ -101,27 +102,27 @@ class CleanUpTest extends TestCase { * @dataProvider dataTestRemoveDeletedFiles * @param boolean $nodeExists */ - public function testRemoveDeletedFiles($nodeExists) { + public function testRemoveDeletedFiles(bool $nodeExists) { $this->initTable(); - $this->rootFolder->expects($this->once()) + $this->rootFolder ->method('nodeExists') ->with('/' . $this->user0 . '/files_trashbin') - ->willReturn($nodeExists); + ->willReturnOnConsecutiveCalls($nodeExists, false); if ($nodeExists) { - $this->rootFolder->expects($this->once()) + $this->rootFolder ->method('get') ->with('/' . $this->user0 . '/files_trashbin') ->willReturn($this->rootFolder); - $this->rootFolder->expects($this->once()) + $this->rootFolder ->method('delete'); } else { $this->rootFolder->expects($this->never())->method('get'); $this->rootFolder->expects($this->never())->method('delete'); } - $this->invokePrivate($this->cleanup, 'removeDeletedFiles', [$this->user0]); + $this->invokePrivate($this->cleanup, 'removeDeletedFiles', [$this->user0, new NullOutput(), false]); if ($nodeExists) { - // if the delete operation was execute only files from user1 + // if the delete operation was executed only files from user1 // should be left. $query = $this->dbConnection->getQueryBuilder(); $query->select('user') @@ -136,7 +137,7 @@ class CleanUpTest extends TestCase { $this->assertSame('user1', $r['user']); } } else { - // if no delete operation was execute we should still have all 10 + // if no delete operation was executed we should still have all 10 // database entries $getAllQuery = $this->dbConnection->getQueryBuilder(); $result = $getAllQuery->select('id') @@ -171,9 +172,14 @@ class CleanUpTest extends TestCase { ->method('userExists')->willReturn(true); $inputInterface = $this->getMockBuilder('\Symfony\Component\Console\Input\InputInterface') ->disableOriginalConstructor()->getMock(); - $inputInterface->expects($this->once())->method('getArgument') + $inputInterface->method('getArgument') ->with('user_id') ->willReturn($userIds); + $inputInterface->method('getOption') + ->willReturnMap([ + ['all-users', false], + ['verbose', false], + ]); $outputInterface = $this->getMockBuilder('\Symfony\Component\Console\Output\OutputInterface') ->disableOriginalConstructor()->getMock(); $this->invokePrivate($instance, 'execute', [$inputInterface, $outputInterface]); @@ -190,7 +196,7 @@ class CleanUpTest extends TestCase { ->setConstructorArgs([$this->rootFolder, $this->userManager, $this->dbConnection]) ->getMock(); $backend = $this->createMock(\OCP\UserInterface::class); - $backend->expects($this->once())->method('getUsers') + $backend->method('getUsers') ->with('', 500, 0) ->willReturn($backendUsers); $instance->expects($this->exactly(count($backendUsers))) @@ -199,14 +205,16 @@ class CleanUpTest extends TestCase { $this->assertTrue(in_array($user, $backendUsers)); }); $inputInterface = $this->createMock(InputInterface::class); - $inputInterface->expects($this->once())->method('getArgument') + $inputInterface->method('getArgument') ->with('user_id') ->willReturn($userIds); $inputInterface->method('getOption') - ->with('all-users') - ->willReturn(true); + ->willReturnMap([ + ['all-users', true], + ['verbose', false], + ]); $outputInterface = $this->createMock(OutputInterface::class); - $this->userManager->expects($this->once()) + $this->userManager ->method('getBackends') ->willReturn([$backend]); $this->invokePrivate($instance, 'execute', [$inputInterface, $outputInterface]); @@ -214,12 +222,14 @@ class CleanUpTest extends TestCase { public function testExecuteNoUsersAndNoAllUsers() { $inputInterface = $this->createMock(InputInterface::class); - $inputInterface->expects($this->once())->method('getArgument') + $inputInterface->method('getArgument') ->with('user_id') ->willReturn([]); $inputInterface->method('getOption') - ->with('all-users') - ->willReturn(false); + ->willReturnMap([ + ['all-users', false], + ['verbose', false], + ]); $outputInterface = $this->createMock(OutputInterface::class); $this->expectException(InvalidOptionException::class); @@ -230,12 +240,14 @@ class CleanUpTest extends TestCase { public function testExecuteUsersAndAllUsers() { $inputInterface = $this->createMock(InputInterface::class); - $inputInterface->expects($this->once())->method('getArgument') + $inputInterface->method('getArgument') ->with('user_id') ->willReturn(['user1', 'user2']); $inputInterface->method('getOption') - ->with('all-users') - ->willReturn(true); + ->willReturnMap([ + ['all-users', true], + ['verbose', false], + ]); $outputInterface = $this->createMock(OutputInterface::class); $this->expectException(InvalidOptionException::class); |