feat: Delete chat messages and assignments when user is deleted and stop running assignments

Signed-off-by: Jana Peper <jana.peper@nextcloud.com>

Author: janepie

lib/AppInfo/Application.php

@@ -23,6 +23,7 @@
 use OCA\Assistant\Listener\TaskSuccessfulListener;
 use OCA\Assistant\Listener\Text2Image\Text2ImageReferenceListener;
 use OCA\Assistant\Listener\Text2Image\Text2StickerListener;
+use OCA\Assistant\Listener\UserDeletedListener;
 use OCA\Assistant\Notification\Notifier;
 use OCA\Assistant\Reference\FreePromptReferenceProvider;
 use OCA\Assistant\Reference\SpeechToTextReferenceProvider;
@@ -48,6 +49,7 @@
 use OCP\TaskProcessing\Events\TaskFailedEvent;
 use OCP\TaskProcessing\Events\TaskSuccessfulEvent;
 use OCP\TaskProcessing\IManager;
+use OCP\User\Events\UserDeletedEvent;
 
 class Application extends App implements IBootstrap {
 
@@ -102,6 +104,8 @@ public function register(IRegistrationContext $context): void {
 
 		$context->registerEventListener(AddContentSecurityPolicyEvent::class, CSPListener::class);
 
+		$context->registerEventListener(UserDeletedEvent::class, UserDeletedListener::class);
+
 		if (class_exists('OCP\\TaskProcessing\\TaskTypes\\AudioToAudioChat')) {
 			$context->registerTaskProcessingProvider(AudioToAudioChatProvider::class);
 		}

lib/Db/AssignmentMapper.php

@@ -86,4 +86,15 @@ public function findDueAssignmentsForUser(string $userId): \Generator {
 			yield $assignment;
 		}
 	}
+
+	/**
+	 * @throws \OCP\DB\Exception
+	 */
+	public function deleteAllForUser(string $userId): void {
+		$qb = $this->db->getQueryBuilder();
+		$qb->delete($this->getTableName())
+			->where($qb->expr()->eq('user_id', $qb->createPositionalParameter($userId, IQueryBuilder::PARAM_STR)));
+
+		$qb->executeStatement();
+	}
 }

lib/Db/ChattyLLM/SessionMapper.php

@@ -82,6 +82,19 @@ public function getUserSessionForAssignment(string $userId, int $assignmentId):
 		return $this->findEntity($qb);
 	}
 
+	/**
+	 * @return \Generator<array-key, Session>
+	 * @throws \OCP\DB\Exception
+	 */
+	public function getAllUserSessions(string $userId): \Generator {
+		$qb = $this->db->getQueryBuilder();
+		$qb->select(Session::$columns)
+			->from($this->getTableName())
+			->where($qb->expr()->eq('user_id', $qb->createPositionalParameter($userId, IQueryBuilder::PARAM_STR)));
+
+		yield from $this->yieldEntities($qb);
+	}
+
 	/**
 	 * @param string $userId
 	 * @param bool $isAssignment
@@ -205,6 +218,17 @@ public function deleteSession(string $userId, int $sessionId) {
 		$qb->executeStatement();
 	}
 
+	/**
+	 * @throws \OCP\DB\Exception
+	 */
+	public function deleteAllSessionsForUser(string $userId): void {
+		$qb = $this->db->getQueryBuilder();
+		$qb->delete($this->getTableName())
+			->where($qb->expr()->eq('user_id', $qb->createPositionalParameter($userId, IQueryBuilder::PARAM_STR)));
+
+		$qb->executeStatement();
+	}
+
 	public function updateSessionIsRemembered(?string $userId, int $sessionId, bool $is_remembered) {
 		$session = $this->getUserSession($userId, $sessionId);
 		$session->setIsRemembered($is_remembered);

lib/Listener/UserDeletedListener.php

@@ -0,0 +1,51 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * SPDX-FileCopyrightText: 2026 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+namespace OCA\Assistant\Listener;
+
+use OCA\Assistant\Service\AssignmentsService;
+use OCA\Assistant\Service\ChatService;
+use OCA\Assistant\Service\InternalException;
+use OCP\EventDispatcher\Event;
+use OCP\EventDispatcher\IEventListener;
+use OCP\User\Events\UserDeletedEvent;
+use Psr\Log\LoggerInterface;
+
+/**
+ * @template-implements IEventListener<UserDeletedEvent>
+ */
+class UserDeletedListener implements IEventListener {
+
+	public function __construct(
+		private ChatService $chatService,
+		private AssignmentsService $assignmentsService,
+		private LoggerInterface $logger,
+	) {
+	}
+
+	public function handle(Event $event): void {
+		if (!($event instanceof UserDeletedEvent)) {
+			return;
+		}
+
+		$userId = $event->getUid();
+
+		try {
+			$this->assignmentsService->deleteAllForUser($userId);
+		} catch (InternalException $e) {
+			$this->logger->error('Error while deleting assignments for user ' . $userId, ['exception' => $e]);
+		}
+
+		try {
+			$this->chatService->deleteAllUserChatData($userId);
+		} catch (InternalException $e) {
+			$this->logger->error('Error while deleting chat data for user ' . $userId, ['exception' => $e]);
+		}
+	}
+}

lib/Service/AssignmentsService.php

@@ -21,6 +21,7 @@
 use OCP\DB\Exception;
 use OCP\IDateTimeZone;
 use OCP\IL10N;
+use OCP\IUserManager;
 use Psr\Log\LoggerInterface;
 
 class AssignmentsService {
@@ -33,6 +34,7 @@ public function __construct(
 		private IJobList $jobList,
 		private IL10N $l10n,
 		private IDateTimeZone $dateTimeZone,
+		private IUserManager $userManager,
 	) {
 	}
 
@@ -84,13 +86,31 @@ public function createAssignment(?string $userId, string $title, string $prompt,
 		return $assignment;
 	}
 
+	/**
+	 * @throws InternalException
+	 */
+	public function deleteAllForUser(string $userId): void {
+		try {
+			$this->assignmentMapper->deleteAllForUser($userId);
+		} catch (Exception $e) {
+			throw new InternalException(previous: $e);
+		}
+		if ($this->jobList->has(RunAssignmentsJob::class, ['userId' => $userId])) {
+			$this->jobList->remove(RunAssignmentsJob::class, ['userId' => $userId]);
+		}
+	}
+
 	/**
 	 * @throws InternalException|UnauthorizedException
 	 */
 	public function runDueAssignmentsForUser(?string $userId): void {
 		if ($userId === null) {
 			throw new UnauthorizedException();
 		}
+		if ($this->userManager->get($userId) === null) {
+			$this->deleteAllForUser($userId);
+			return;
+		}
 		try {
 			foreach ($this->assignmentMapper->findDueAssignmentsForUser($userId) as $assignment) {
 				if ($assignment === null) {

lib/Service/ChatService.php

@@ -152,6 +152,21 @@ public function deleteSession(?string $userId, int $sessionId): void {
 		}
 	}
 
+	/**
+	 * @throws InternalException
+	 */
+	public function deleteAllUserChatData(string $userId): void {
+		try {
+			$sessions = $this->sessionMapper->getAllUserSessions($userId);
+			foreach ($sessions as $session) {
+				$this->messageMapper->deleteMessagesBySession($session->getId());
+			}
+			$this->sessionMapper->deleteAllSessionsForUser($userId);
+		} catch (Exception|\RuntimeException $e) {
+			throw new InternalException(previous: $e);
+		}
+	}
+
 	/**
 	 * @return list<Session>
 	 * @throws InternalException