From 3ef601c0402d99672757addcab8e99c513bb32cf Mon Sep 17 00:00:00 2001
From: Jana Peper <jana.peper@nextcloud.com>
Date: Sun, 8 Dec 2024 19:16:19 +0100
Subject: [PATCH] feat: filter disabled apps in task types requests

Signed-off-by: Jana Peper <jana.peper@nextcloud.com>
---
 .../lib/Settings/Admin/ArtificialIntelligence.php  |  2 +-
 apps/settings/src/components/AdminAI.vue           |  4 +---
 lib/private/TaskProcessing/Manager.php             | 14 +++++++++++++-
 lib/public/TaskProcessing/IManager.php             |  3 ++-
 4 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/apps/settings/lib/Settings/Admin/ArtificialIntelligence.php b/apps/settings/lib/Settings/Admin/ArtificialIntelligence.php
index 197ea0de283d8..c606963536715 100644
--- a/apps/settings/lib/Settings/Admin/ArtificialIntelligence.php
+++ b/apps/settings/lib/Settings/Admin/ArtificialIntelligence.php
@@ -114,7 +114,7 @@ public function getForm() {
 		}
 		$taskProcessingTaskTypes = [];
 		$taskProcessingTypeSettings = [];
-		foreach ($this->taskProcessingManager->getAvailableTaskTypes() as $taskTypeId => $taskTypeDefinition) {
+		foreach ($this->taskProcessingManager->getAvailableTaskTypes(true) as $taskTypeId => $taskTypeDefinition) {
 			$taskProcessingTaskTypes[] = [
 				'id' => $taskTypeId,
 				'name' => $taskTypeDefinition['name'],
diff --git a/apps/settings/src/components/AdminAI.vue b/apps/settings/src/components/AdminAI.vue
index 0d6f32a9aa9e0..fe8044d93b527 100644
--- a/apps/settings/src/components/AdminAI.vue
+++ b/apps/settings/src/components/AdminAI.vue
@@ -18,7 +18,7 @@
 					<NcSelect v-model="settings['ai.taskprocessing_provider_preferences'][type.id]"
 						class="provider-select"
 						:clearable="false"
-						:disabled="!type.enabled"
+						:disabled="!settings['ai.taskprocessing_type_preferences'][type.id]"
 						:options="taskProcessingProviders.filter(p => p.taskType === type.id).map(p => p.id)"
 						@input="saveChanges">
 						<template #option="{label}">
@@ -189,7 +189,6 @@ export default {
 			this.saveChanges()
 		},
 		async saveChanges() {
-			console.warn(this.settings)
 			this.loading = true
 			await nextTick()
 			const data = { settings: this.settings }
@@ -199,7 +198,6 @@ export default {
 				console.error('could not save changes', err)
 			}
 			this.loading = false
-			console.warn('done')
 		},
 		getTextProcessingTaskType(type) {
 		  if (!Array.isArray(this.textProcessingTaskTypes)) {
diff --git a/lib/private/TaskProcessing/Manager.php b/lib/private/TaskProcessing/Manager.php
index 4e9668021ad3f..bf72154061916 100644
--- a/lib/private/TaskProcessing/Manager.php
+++ b/lib/private/TaskProcessing/Manager.php
@@ -561,6 +561,14 @@ private function _getTaskTypes(): array {
 		return $taskTypes;
 	}
 
+	/**
+	 * @return array
+	 */
+	private function _getTaskTypeSettings(): array {
+		$json = $this->config->getAppValue('core', 'ai.taskprocessing_type_preferences', '');
+		return json_decode($json, true);
+	}
+
 	/**
 	 * @param ShapeDescriptor[] $spec
 	 * @param array<array-key, string|numeric> $defaults
@@ -718,12 +726,16 @@ public function getPreferredProvider(string $taskTypeId) {
 		throw new \OCP\TaskProcessing\Exception\Exception('No matching provider found');
 	}
 
-	public function getAvailableTaskTypes(): array {
+	public function getAvailableTaskTypes(bool $showDisabled = false): array {
 		if ($this->availableTaskTypes === null) {
 			$taskTypes = $this->_getTaskTypes();
+			$taskTypeSettings = $this->_getTaskTypeSettings();
 
 			$availableTaskTypes = [];
 			foreach ($taskTypes as $taskType) {
+				if ((!$showDisabled) && isset($taskTypeSettings[$taskType->getId()]) && !$taskTypeSettings[$taskType->getId()]) {
+					continue;
+				}
 				try {
 					$provider = $this->getPreferredProvider($taskType->getId());
 				} catch (\OCP\TaskProcessing\Exception\Exception $e) {
diff --git a/lib/public/TaskProcessing/IManager.php b/lib/public/TaskProcessing/IManager.php
index 1b59bb12ad4ae..a4ce9e8bbbe15 100644
--- a/lib/public/TaskProcessing/IManager.php
+++ b/lib/public/TaskProcessing/IManager.php
@@ -46,10 +46,11 @@ public function getProviders(): array;
 	public function getPreferredProvider(string $taskTypeId);
 
 	/**
+	 * @param bool $showDisabled if false, disabled task types will be filtered
 	 * @return array<string, array{name: string, description: string, inputShape: ShapeDescriptor[], inputShapeEnumValues: ShapeEnumValue[][], inputShapeDefaults: array<array-key, numeric|string>, optionalInputShape: ShapeDescriptor[], optionalInputShapeEnumValues: ShapeEnumValue[][], optionalInputShapeDefaults: array<array-key, numeric|string>, outputShape: ShapeDescriptor[], outputShapeEnumValues: ShapeEnumValue[][], optionalOutputShape: ShapeDescriptor[], optionalOutputShapeEnumValues: ShapeEnumValue[][]}>
 	 * @since 30.0.0
 	 */
-	public function getAvailableTaskTypes(): array;
+	public function getAvailableTaskTypes(bool $showDisabled = false): array;
 
 	/**
 	 * @param Task $task The task to run