diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/private/TaskProcessing/Manager.php | 14 | ||||
| -rw-r--r-- | lib/public/TaskProcessing/IManager.php | 3 |
2 files changed, 15 insertions, 2 deletions
diff --git a/lib/private/TaskProcessing/Manager.php b/lib/private/TaskProcessing/Manager.php index ffa39912a01..f24b1b73da4 100644 --- a/lib/private/TaskProcessing/Manager.php +++ b/lib/private/TaskProcessing/Manager.php @@ -565,6 +565,14 @@ class Manager implements IManager { } /** + * @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 * @param array<array-key, ShapeEnumValue[]> $enumValues @@ -721,12 +729,16 @@ class Manager implements IManager { 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 1b59bb12ad4..a4ce9e8bbbe 100644 --- a/lib/public/TaskProcessing/IManager.php +++ b/lib/public/TaskProcessing/IManager.php @@ -46,10 +46,11 @@ interface IManager { 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 |