Merge pull request #47177 from nextcloud/fix/noid/preferred-taskprocessing-providers

[Task processing] Fix preferred providers

2 files changed, 16 insertions, 11 deletions

diff --git a/lib/private/TaskProcessing/Manager.php b/lib/private/TaskProcessing/Manager.php
index a804115a631..37533df088f 100644
--- a/lib/private/TaskProcessing/Manager.php
+++ b/lib/private/TaskProcessing/Manager.php
@@ -649,24 +649,24 @@ class Manager implements IManager {
 		return $this->providers;
 	}
 
-	public function getPreferredProvider(string $taskType) {
+	public function getPreferredProvider(string $taskTypeId) {
 		try {
 			$preferences = json_decode($this->config->getAppValue('core', 'ai.taskprocessing_provider_preferences', 'null'), associative: true, flags: JSON_THROW_ON_ERROR);
 			$providers = $this->getProviders();
-			if (isset($preferences[$taskType])) {
-				$provider = current(array_values(array_filter($providers, fn ($provider) => $provider->getId() === $preferences[$taskType])));
+			if (isset($preferences[$taskTypeId])) {
+				$provider = current(array_values(array_filter($providers, fn ($provider) => $provider->getId() === $preferences[$taskTypeId])));
 				if ($provider !== false) {
 					return $provider;
 				}
 			}
 			// By default, use the first available provider
 			foreach ($providers as $provider) {
-				if ($provider->getTaskTypeId() === $taskType) {
+				if ($provider->getTaskTypeId() === $taskTypeId) {
 					return $provider;
 				}
 			}
 		} catch (\JsonException $e) {
-			$this->logger->warning('Failed to parse provider preferences while getting preferred provider for task type ' . $taskType, ['exception' => $e]);
+			$this->logger->warning('Failed to parse provider preferences while getting preferred provider for task type ' . $taskTypeId, ['exception' => $e]);
 		}
 		throw new \OCP\TaskProcessing\Exception\Exception('No matching provider found');
 	}
@@ -674,14 +674,14 @@ class Manager implements IManager {
 	public function getAvailableTaskTypes(): array {
 		if ($this->availableTaskTypes === null) {
 			$taskTypes = $this->_getTaskTypes();
-			$providers = $this->getProviders();
 
 			$availableTaskTypes = [];
-			foreach ($providers as $provider) {
-				if (!isset($taskTypes[$provider->getTaskTypeId()])) {
+			foreach ($taskTypes as $taskType) {
+				try {
+					$provider = $this->getPreferredProvider($taskType->getId());
+				} catch (\OCP\TaskProcessing\Exception\Exception $e) {
 					continue;
 				}
-				$taskType = $taskTypes[$provider->getTaskTypeId()];
 				try {
 					$availableTaskTypes[$provider->getTaskTypeId()] = [
 						'name' => $taskType->getName(),
diff --git a/lib/private/TaskProcessing/SynchronousBackgroundJob.php b/lib/private/TaskProcessing/SynchronousBackgroundJob.php
index 093882d4c1e..85a8fbc21f6 100644
--- a/lib/private/TaskProcessing/SynchronousBackgroundJob.php
+++ b/lib/private/TaskProcessing/SynchronousBackgroundJob.php
@@ -43,9 +43,14 @@ class SynchronousBackgroundJob extends QueuedJob {
 			if (!$provider instanceof ISynchronousProvider) {
 				continue;
 			}
-			$taskType = $provider->getTaskTypeId();
+			$taskTypeId = $provider->getTaskTypeId();
+			// only use this provider if it is the preferred one
+			$preferredProvider = $this->taskProcessingManager->getPreferredProvider($taskTypeId);
+			if ($provider->getId() !== $preferredProvider->getId()) {
+				continue;
+			}
 			try {
-				$task = $this->taskProcessingManager->getNextScheduledTask([$taskType]);
+				$task = $this->taskProcessingManager->getNextScheduledTask([$taskTypeId]);
 			} catch (NotFoundException $e) {
 				continue;
 			} catch (Exception $e) {