From 996e5074ca43fbb049f496292a921012c3a49d63 Mon Sep 17 00:00:00 2001
From: Marcel Klehr <mklehr@gmx.net>
Date: Mon, 6 May 2024 09:44:48 +0200
Subject: [PATCH] feat: Remove stale files in AppData as well

Signed-off-by: Marcel Klehr <mklehr@gmx.net>
---
 .../RemoveOldTasksBackgroundJob.php           | 36 +++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/lib/private/TaskProcessing/RemoveOldTasksBackgroundJob.php b/lib/private/TaskProcessing/RemoveOldTasksBackgroundJob.php
index 54b63ac42fb..c68ead4e675 100644
--- a/lib/private/TaskProcessing/RemoveOldTasksBackgroundJob.php
+++ b/lib/private/TaskProcessing/RemoveOldTasksBackgroundJob.php
@@ -5,31 +5,63 @@ namespace OC\TaskProcessing;
 use OC\TaskProcessing\Db\TaskMapper;
 use OCP\AppFramework\Utility\ITimeFactory;
 use OCP\BackgroundJob\TimedJob;
+use OCP\Files\AppData\IAppDataFactory;
+use OCP\Files\NotFoundException;
+use OCP\Files\NotPermittedException;
+use OCP\Files\SimpleFS\ISimpleFolder;
 use Psr\Log\LoggerInterface;
 
 class RemoveOldTasksBackgroundJob extends TimedJob {
 	public const MAX_TASK_AGE_SECONDS = 60 * 50 * 24 * 7 * 4; // 4 weeks
+	private \OCP\Files\IAppData $appData;
 
 	public function __construct(
 		ITimeFactory $timeFactory,
 		private TaskMapper $taskMapper,
 		private LoggerInterface $logger,
+		IAppDataFactory $appDataFactory,
 	) {
 		parent::__construct($timeFactory);
 		$this->setInterval(60 * 60 * 24);
 		// can be deferred to maintenance window
 		$this->setTimeSensitivity(TimedJob::TIME_INSENSITIVE);
+		$this->appData = $appDataFactory->get('core');
 	}
 
 
 	/**
 	 * @inheritDoc
 	 */
-	protected function run($argument) {
+	protected function run($argument): void {
 		try {
 			$this->taskMapper->deleteOlderThan(self::MAX_TASK_AGE_SECONDS);
 		} catch (\OCP\DB\Exception $e) {
-			$this->logger->warning('Failed to delete stale language model tasks', ['exception' => $e]);
+			$this->logger->warning('Failed to delete stale task processing tasks', ['exception' => $e]);
+		}
+		try {
+			$this->clearFilesOlderThan($this->appData->getFolder('text2image'), self::MAX_TASK_AGE_SECONDS);
+			$this->clearFilesOlderThan($this->appData->getFolder('audio2text'), self::MAX_TASK_AGE_SECONDS);
+			$this->clearFilesOlderThan($this->appData->getFolder('TaskProcessing'), self::MAX_TASK_AGE_SECONDS);
+		} catch (NotFoundException $e) {
+			// noop
+		}
+	}
+
+	/**
+	 * @param ISimpleFolder $folder
+	 * @param int $ageInSeconds
+	 * @return void
+	 */
+	private function clearFilesOlderThan(ISimpleFolder $folder, int $ageInSeconds): void {
+		foreach($folder->getDirectoryListing() as $file) {
+			if ($file->getMTime() < time() - $ageInSeconds) {
+				try {
+					$file->delete();
+				} catch (NotPermittedException $e) {
+					$this->logger->warning('Failed to delete a stale task processing file', ['exception' => $e]);
+				}
+			}
 		}
 	}
+
 }
-- 
2.39.5