CleanupUploadedFilesJob.php

<?php

/**
 * SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
 * SPDX-License-Identifier: AGPL-3.0-or-later
 */

namespace OCA\Forms\BackgroundJob;

use OCA\Forms\Constants;
use OCA\Forms\Db\FormMapper;
use OCA\Forms\Db\UploadedFileMapper;
use OCP\AppFramework\Db\DoesNotExistException;
use OCP\AppFramework\Utility\ITimeFactory;
use OCP\BackgroundJob\TimedJob;
use OCP\Files\Folder;
use OCP\Files\IRootFolder;
use Psr\Log\LoggerInterface;

class CleanupUploadedFilesJob extends TimedJob {
	private const FILE_LIFETIME = '-1 hour';

	public function __construct(
		private readonly IRootFolder $rootFolder,
		private readonly FormMapper $formMapper,
		private readonly UploadedFileMapper $uploadedFileMapper,
		private readonly LoggerInterface $logger,
		ITimeFactory $time,
	) {
		parent::__construct($time);

		$this->setInterval(60 * 60);
	}

	/**
	 * @param array $argument
	 */
	public function run($argument): void {
		$dateTime = new \DateTimeImmutable(self::FILE_LIFETIME);

		$this->logger->info('Deleting files that were uploaded before {before} and still not submitted.', [
			'before' => $dateTime->format(\DateTimeImmutable::ATOM),
		]);

		$uploadedFiles = $this->uploadedFileMapper->findUploadedEarlierThan($dateTime);

		$deleted = 0;
		$usersToCleanup = [];
		foreach ($uploadedFiles as $uploadedFile) {
			$this->logger->info('Deleting uploaded file "{originalFileName}" for form {formId}.', [
				'originalFileName' => $uploadedFile->getOriginalFileName(),
				'formId' => $uploadedFile->getFormId(),
			]);

			try {
				$form = $this->formMapper->findById($uploadedFile->getFormId());
			} catch (DoesNotExistException) {
				$this->logger->warning('Could not find form {formId} for uploaded file deletion.', [
					'formId' => $uploadedFile->getFormId(),
				]);
				$this->uploadedFileMapper->delete($uploadedFile);
				continue;
			}

			$usersToCleanup[$form->getOwnerId()] = true;
			$userFolder = $this->rootFolder->getUserFolder($form->getOwnerId());

			$nodes = $userFolder->getById($uploadedFile->getFileId());

			if (!empty($nodes)) {
				$node = $nodes[0];
				$node->delete();
			} else {
				$this->logger->warning('Could not find uploaded file "{fileId}" for deletion.', [
					'fileId' => $uploadedFile->getFileId(),
				]);
			}

			$this->uploadedFileMapper->delete($uploadedFile);

			$deleted++;
		}

		$this->logger->info('Deleted {deleted} uploaded files.', ['deleted' => $deleted]);

		// now delete empty folders in user folders
		$deleted = 0;
		foreach (array_keys($usersToCleanup) as $userId) {
			$this->logger->info('Cleaning up empty folders for user {userId}.', ['userId' => $userId]);
			$userFolder = $this->rootFolder->getUserFolder($userId);

			$unsubmittedFilesFolder = $userFolder->get(Constants::UNSUBMITTED_FILES_FOLDER);
			if (!$unsubmittedFilesFolder instanceof Folder) {
				continue;
			}

			foreach ($unsubmittedFilesFolder->getDirectoryListing() as $node) {
				if ($node->getName() < $dateTime->getTimestamp()) {
					$node->delete();
					$deleted++;
				}
			}
		}

		$this->logger->info('Deleted {deleted} folders.', ['deleted' => $deleted]);
	}
}