1 files changed, 124 insertions, 0 deletions
diff --git a/apps/files_trashbin/lib/Command/Size.php b/apps/files_trashbin/lib/Command/Size.php
new file mode 100644
index 00000000000..9c19d4d92b3
--- /dev/null
+++ b/apps/files_trashbin/lib/Command/Size.php
@@ -0,0 +1,124 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * SPDX-FileCopyrightText: 2020 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+namespace OCA\Files_Trashbin\Command;
+
+use OC\Core\Command\Base;
+use OCP\Command\IBus;
+use OCP\IConfig;
+use OCP\IUser;
+use OCP\IUserManager;
+use OCP\Util;
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
+use Symfony\Component\Console\Output\OutputInterface;
+
+class Size extends Base {
+	public function __construct(
+		private IConfig $config,
+		private IUserManager $userManager,
+		private IBus $commandBus,
+	) {
+		parent::__construct();
+	}
+
+	protected function configure() {
+		parent::configure();
+		$this
+			->setName('trashbin:size')
+			->setDescription('Configure the target trashbin size')
+			->addOption('user', 'u', InputOption::VALUE_REQUIRED, 'configure the target size for the provided user, if no user is given the default size is configured')
+			->addArgument(
+				'size',
+				InputArgument::OPTIONAL,
+				'the target size for the trashbin, if not provided the current trashbin size will be returned'
+			);
+	}
+
+	protected function execute(InputInterface $input, OutputInterface $output): int {
+		$user = $input->getOption('user');
+		$size = $input->getArgument('size');
+
+		if ($size) {
+			$parsedSize = Util::computerFileSize($size);
+			if ($parsedSize === false) {
+				$output->writeln('<error>Failed to parse input size</error>');
+				return -1;
+			}
+			if ($user) {
+				$this->config->setUserValue($user, 'files_trashbin', 'trashbin_size', (string)$parsedSize);
+				$this->commandBus->push(new Expire($user));
+			} else {
+				$this->config->setAppValue('files_trashbin', 'trashbin_size', (string)$parsedSize);
+				$output->writeln('<info>Warning: changing the default trashbin size will automatically trigger cleanup of existing trashbins,</info>');
+				$output->writeln('<info>a users trashbin can exceed the configured size until they move a new file to the trashbin.</info>');
+			}
+		} else {
+			$this->printTrashbinSize($input, $output, $user);
+		}
+
+		return 0;
+	}
+
+	private function printTrashbinSize(InputInterface $input, OutputInterface $output, ?string $user) {
+		$globalSize = (int)$this->config->getAppValue('files_trashbin', 'trashbin_size', '-1');
+		if ($globalSize < 0) {
+			$globalHumanSize = 'default (50% of available space)';
+		} else {
+			$globalHumanSize = Util::humanFileSize($globalSize);
+		}
+
+		if ($user) {
+			$userSize = (int)$this->config->getUserValue($user, 'files_trashbin', 'trashbin_size', '-1');
+
+			if ($userSize < 0) {
+				$userHumanSize = ($globalSize < 0) ? $globalHumanSize : "default($globalHumanSize)";
+			} else {
+				$userHumanSize = Util::humanFileSize($userSize);
+			}
+
+			if ($input->getOption('output') == self::OUTPUT_FORMAT_PLAIN) {
+				$output->writeln($userHumanSize);
+			} else {
+				$userValue = ($userSize < 0) ? 'default' : $userSize;
+				$globalValue = ($globalSize < 0) ? 'default' : $globalSize;
+				$this->writeArrayInOutputFormat($input, $output, [
+					'user_size' => $userValue,
+					'global_size' => $globalValue,
+					'effective_size' => ($userSize < 0) ? $globalValue : $userValue,
+				]);
+			}
+		} else {
+			$users = [];
+			$this->userManager->callForSeenUsers(function (IUser $user) use (&$users): void {
+				$users[] = $user->getUID();
+			});
+			$userValues = $this->config->getUserValueForUsers('files_trashbin', 'trashbin_size', $users);
+
+			if ($input->getOption('output') == self::OUTPUT_FORMAT_PLAIN) {
+				$output->writeln("Default size: $globalHumanSize");
+				$output->writeln('');
+				if (count($userValues)) {
+					$output->writeln('Per-user sizes:');
+					$this->writeArrayInOutputFormat($input, $output, array_map(function ($size) {
+						return Util::humanFileSize($size);
+					}, $userValues));
+				} else {
+					$output->writeln('No per-user sizes configured');
+				}
+			} else {
+				$globalValue = ($globalSize < 0) ? 'default' : $globalSize;
+				$this->writeArrayInOutputFormat($input, $output, [
+					'global_size' => $globalValue,
+					'user_sizes' => $userValues,
+				]);
+			}
+		}
+	}
+}