From 297d2e48ad0df78f00e66be23fd19b1360c37272 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thomas=20M=C3=BCller?=
 <DeepDiver1975@users.noreply.github.com>
Date: Tue, 23 Aug 2016 14:03:05 +0200
Subject: [PATCH] Add occ command to expire the trashbin (#25878)

* Add occ command to expire the trashbin

* Fix versions folder in setup check
---
 .../appinfo/register_command.php              |   6 +
 .../lib/Command/ExpireTrash.php               | 127 ++++++++++++++++++
 .../appinfo/register_command.php              |   5 +
 .../lib/Command/ExpireVersions.php            | 125 +++++++++++++++++
 4 files changed, 263 insertions(+)
 create mode 100644 apps/files_trashbin/lib/Command/ExpireTrash.php
 create mode 100644 apps/files_versions/lib/Command/ExpireVersions.php

diff --git a/apps/files_trashbin/appinfo/register_command.php b/apps/files_trashbin/appinfo/register_command.php
index 676e3f5891c..e0dafc60cd9 100644
--- a/apps/files_trashbin/appinfo/register_command.php
+++ b/apps/files_trashbin/appinfo/register_command.php
@@ -21,10 +21,16 @@
  */
 
 
+use OCA\Files_Trashbin\AppInfo\Application;
 use OCA\Files_Trashbin\Command\CleanUp;
+use OCA\Files_Trashbin\Command\ExpireTrash;
 
+$app = new Application();
+$expiration = $app->getContainer()->query('Expiration');
 $userManager = OC::$server->getUserManager();
 $rootFolder = \OC::$server->getRootFolder();
 $dbConnection = \OC::$server->getDatabaseConnection();
+
 /** @var Symfony\Component\Console\Application $application */
 $application->add(new CleanUp($rootFolder, $userManager, $dbConnection));
+$application->add(new ExpireTrash($userManager, $expiration));
diff --git a/apps/files_trashbin/lib/Command/ExpireTrash.php b/apps/files_trashbin/lib/Command/ExpireTrash.php
new file mode 100644
index 00000000000..ff827718885
--- /dev/null
+++ b/apps/files_trashbin/lib/Command/ExpireTrash.php
@@ -0,0 +1,127 @@
+<?php
+/**
+ * @author Thomas Müller <thomas.mueller@tmit.eu>
+ *
+ * @copyright Copyright (c) 2016, ownCloud GmbH.
+ * @license AGPL-3.0
+ *
+ * This code is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3,
+ * as published by the Free Software Foundation.
+ *
+ * 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, version 3,
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>
+ *
+ */
+
+namespace OCA\Files_Trashbin\Command;
+
+use OCP\IUser;
+use OCP\IUserManager;
+use OCA\Files_Trashbin\Expiration;
+use OCA\Files_Trashbin\Helper;
+use OCA\Files_Trashbin\Trashbin;
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Helper\ProgressBar;
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+class ExpireTrash extends Command {
+
+	/**
+	 * @var Expiration
+	 */
+	private $expiration;
+	
+	/**
+	 * @var IUserManager
+	 */
+	private $userManager;
+
+	/**
+	 * @param IUserManager|null $userManager
+	 * @param Expiration|null $expiration
+	 */
+	public function __construct(IUserManager $userManager = null,
+								Expiration $expiration = null) {
+		parent::__construct();
+
+		$this->userManager = $userManager;
+		$this->expiration = $expiration;
+	}
+
+	protected function configure() {
+		$this
+			->setName('trashbin:expire')
+			->setDescription('Expires the users trashbin')
+			->addArgument(
+				'user_id',
+				InputArgument::OPTIONAL | InputArgument::IS_ARRAY,
+				'expires the trashbin of the given user(s), if no user is given the trash for all users will be expired'
+			);
+	}
+
+	protected function execute(InputInterface $input, OutputInterface $output) {
+
+		$maxAge = $this->expiration->getMaxAgeAsTimestamp();
+		if (!$maxAge) {
+			$output->writeln("No expiry configured.");
+			return;
+		}
+
+		$users = $input->getArgument('user_id');
+		if (!empty($users)) {
+			foreach ($users as $user) {
+				if ($this->userManager->userExists($user)) {
+					$output->writeln("Remove deleted files of   <info>$user</info>");
+					$userObject = $this->userManager->get($user);
+					$this->expireTrashForUser($userObject);
+				} else {
+					$output->writeln("<error>Unknown user $user</error>");
+				}
+			}
+		} else {
+			$p = new ProgressBar($output);
+			$p->start();
+			$this->userManager->callForAllUsers(function(IUser $user) use ($p) {
+				$p->advance();
+				$this->expireTrashForUser($user);
+			});
+			$p->finish();
+			$output->writeln('');
+		}
+	}
+
+	function expireTrashForUser(IUser $user) {
+		$uid = $user->getUID();
+		if ($user->getLastLogin() === 0 || !$this->setupFS($uid)) {
+			return;
+		}
+		$dirContent = Helper::getTrashFiles('/', $uid, 'mtime');
+		Trashbin::deleteExpiredFiles($dirContent, $uid);
+	}
+
+	/**
+	 * Act on behalf on trash item owner
+	 * @param string $user
+	 * @return boolean
+	 */
+	protected function setupFS($user) {
+		\OC_Util::tearDownFS();
+		\OC_Util::setupFS($user);
+
+		// Check if this user has a trashbin directory
+		$view = new \OC\Files\View('/' . $user);
+		if (!$view->is_dir('/files_trashbin/files')) {
+			return false;
+		}
+
+		return true;
+	}
+}
diff --git a/apps/files_versions/appinfo/register_command.php b/apps/files_versions/appinfo/register_command.php
index b991c72b944..bca869075aa 100644
--- a/apps/files_versions/appinfo/register_command.php
+++ b/apps/files_versions/appinfo/register_command.php
@@ -21,9 +21,14 @@
  */
 
 
+use OCA\Files_Versions\AppInfo\Application;
 use OCA\Files_Versions\Command\CleanUp;
+use OCA\Files_Versions\Command\ExpireVersions;
 
+$app = new Application();
+$expiration = $app->getContainer()->query('Expiration');
 $userManager = OC::$server->getUserManager();
 $rootFolder = \OC::$server->getRootFolder();
 /** @var Symfony\Component\Console\Application $application */
 $application->add(new CleanUp($rootFolder, $userManager));
+$application->add(new ExpireVersions($userManager, $expiration));
diff --git a/apps/files_versions/lib/Command/ExpireVersions.php b/apps/files_versions/lib/Command/ExpireVersions.php
new file mode 100644
index 00000000000..f384420f22f
--- /dev/null
+++ b/apps/files_versions/lib/Command/ExpireVersions.php
@@ -0,0 +1,125 @@
+<?php
+/**
+ * @author Thomas Müller <thomas.mueller@tmit.eu>
+ *
+ * @copyright Copyright (c) 2016, ownCloud GmbH.
+ * @license AGPL-3.0
+ *
+ * This code is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3,
+ * as published by the Free Software Foundation.
+ *
+ * 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, version 3,
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>
+ *
+ */
+
+namespace OCA\Files_Versions\Command;
+
+use OCA\Files_Versions\Expiration;
+use OCA\Files_Versions\Storage;
+use OCP\IUser;
+use OCP\IUserManager;
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Helper\ProgressBar;
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+class ExpireVersions extends Command {
+
+	/**
+	 * @var Expiration
+	 */
+	private $expiration;
+	
+	/**
+	 * @var IUserManager
+	 */
+	private $userManager;
+
+	/**
+	 * @param IUserManager|null $userManager
+	 * @param Expiration|null $expiration
+	 */
+	public function __construct(IUserManager $userManager = null,
+								Expiration $expiration = null) {
+		parent::__construct();
+
+		$this->userManager = $userManager;
+		$this->expiration = $expiration;
+	}
+
+	protected function configure() {
+		$this
+			->setName('versions:expire')
+			->setDescription('Expires the users file versions')
+			->addArgument(
+				'user_id',
+				InputArgument::OPTIONAL | InputArgument::IS_ARRAY,
+				'expire file versions of the given user(s), if no user is given file versions for all users will be expired.'
+			);
+	}
+
+	protected function execute(InputInterface $input, OutputInterface $output) {
+
+		$maxAge = $this->expiration->getMaxAgeAsTimestamp();
+		if (!$maxAge) {
+			$output->writeln("No expiry configured.");
+			return;
+		}
+
+		$users = $input->getArgument('user_id');
+		if (!empty($users)) {
+			foreach ($users as $user) {
+				if ($this->userManager->userExists($user)) {
+					$output->writeln("Remove deleted files of   <info>$user</info>");
+					$userObject = $this->userManager->get($user);
+					$this->expireVersionsForUser($userObject);
+				} else {
+					$output->writeln("<error>Unknown user $user</error>");
+				}
+			}
+		} else {
+			$p = new ProgressBar($output);
+			$p->start();
+			$this->userManager->callForAllUsers(function(IUser $user) use ($p) {
+				$p->advance();
+				$this->expireVersionsForUser($user);
+			});
+			$p->finish();
+			$output->writeln('');
+		}
+	}
+
+	function expireVersionsForUser(IUser $user) {
+		$uid = $user->getUID();
+		if ($user->getLastLogin() === 0 || !$this->setupFS($uid)) {
+			return;
+		}
+		Storage::expireOlderThanMaxForUser($uid);
+	}
+
+	/**
+	 * Act on behalf on versions item owner
+	 * @param string $user
+	 * @return boolean
+	 */
+	protected function setupFS($user) {
+		\OC_Util::tearDownFS();
+		\OC_Util::setupFS($user);
+
+		// Check if this user has a version directory
+		$view = new \OC\Files\View('/' . $user);
+		if (!$view->is_dir('/files_versions')) {
+			return false;
+		}
+
+		return true;
+	}
+}
-- 
GitLab