UploadCleanup.php 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. <?php
  2. declare(strict_types=1);
  3. /**
  4. * @copyright Copyright (c) 2018, Roeland Jago Douma <roeland@famdouma.nl>
  5. *
  6. * @author Roeland Jago Douma <roeland@famdouma.nl>
  7. *
  8. * @license GNU AGPL version 3 or any later version
  9. *
  10. * This program is free software: you can redistribute it and/or modify
  11. * it under the terms of the GNU Affero General Public License as
  12. * published by the Free Software Foundation, either version 3 of the
  13. * License, or (at your option) any later version.
  14. *
  15. * This program is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. * GNU Affero General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU Affero General Public License
  21. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  22. *
  23. */
  24. namespace OCA\DAV\BackgroundJob;
  25. use OCP\AppFramework\Utility\ITimeFactory;
  26. use OCP\BackgroundJob\IJobList;
  27. use OCP\BackgroundJob\TimedJob;
  28. use OCP\Files\File;
  29. use OCP\Files\Folder;
  30. use OCP\Files\IRootFolder;
  31. use OCP\Files\NotFoundException;
  32. class UploadCleanup extends TimedJob {
  33. /** @var IRootFolder */
  34. private $rootFolder;
  35. /** @var IJobList */
  36. private $jobList;
  37. public function __construct(ITimeFactory $time, IRootFolder $rootFolder, IJobList $jobList) {
  38. parent::__construct($time);
  39. $this->rootFolder = $rootFolder;
  40. $this->jobList = $jobList;
  41. // Run once a day
  42. $this->setInterval(60*60*24);
  43. }
  44. protected function run($argument) {
  45. $uid = $argument['uid'];
  46. $folder = $argument['folder'];
  47. $userFolder = $this->rootFolder->getUserFolder($uid);
  48. $userRoot = $userFolder->getParent();
  49. try {
  50. /** @var Folder $uploads */
  51. $uploads = $userRoot->get('uploads');
  52. /** @var Folder $uploadFolder */
  53. $uploadFolder = $uploads->get($folder);
  54. } catch (NotFoundException $e) {
  55. $this->jobList->remove(self::class, $argument);
  56. return;
  57. }
  58. $files = $uploadFolder->getDirectoryListing();
  59. // Remove if all files have an mtime of more than a day
  60. $time = $this->time->getTime() - 60 * 60 * 24;
  61. // The folder has to be more than a day old
  62. $initial = $uploadFolder->getMTime() < $time;
  63. $expire = array_reduce($files, function(bool $carry, File $file) use ($time) {
  64. return $carry && $file->getMTime() < $time;
  65. }, $initial);
  66. if ($expire) {
  67. $uploadFolder->delete();
  68. $this->jobList->remove(self::class, $argument);
  69. }
  70. }
  71. }