feat(TaskProcessing): Allow defining custom task types

lib/AppInfo/Application.php

@@ -96,6 +96,7 @@ public function register(IRegistrationContext $context): void {
 			/** @var TaskProcessingService $taskProcessingService */
 			$taskProcessingService = $container->get(TaskProcessingService::class);
 			$taskProcessingService->registerExAppTaskProcessingProviders($context, $container->getServer());
+			$taskProcessingService->registerExAppTaskProcessingCustomTaskTypes($context);
 		} catch (NotFoundExceptionInterface|ContainerExceptionInterface) {
 		}
 		$context->registerEventListener(NodeCreatedEvent::class, FileEventsListener::class);

lib/Controller/TaskProcessingController.php

@@ -35,14 +35,15 @@ public function __construct(
 	public function registerProvider(
 		string $name,
 		string $displayName,
-		string $taskType
+		string $taskType,
+		?array $customTaskType,
 	): DataResponse {
 		if (!$this->isSupported()) {
 			return new DataResponse([], Http::STATUS_NOT_IMPLEMENTED);
 		}
 
 		$provider = $this->taskProcessingService->registerTaskProcessingProvider(
-			$this->request->getHeader('EX-APP-ID'), $name, $displayName, $taskType,
+			$this->request->getHeader('EX-APP-ID'), $name, $displayName, $taskType, $customTaskType,
 		);
 
 		if ($provider === null) {

lib/Db/TaskProcessing/TaskProcessingProvider.php

@@ -20,19 +20,23 @@
  * @method string getDisplayName()
  * @method void setTaskType(string $taskType)
  * @method string getTaskType()
+ * @method void setCustomTaskType(string|null $customTaskType)
+ * @method string|null getCustomTaskType()
  */
 class TaskProcessingProvider extends Entity implements JsonSerializable {
 	protected ?string $appId = null;
 	protected ?string $name = null;
 	protected ?string $displayName = null;
 	protected ?string $actionHandler = null;
 	protected ?string $taskType = null;
+	protected ?string $customTaskType = null;
 
 	public function __construct(array $params = []) {
 		$this->addType('app_id', 'string');
 		$this->addType('name', 'string');
 		$this->addType('display_name', 'string');
 		$this->addType('task_type', 'string');
+		$this->addType('custom_task_type', 'string');
 
 		if (isset($params['app_id'])) {
 			$this->setAppId($params['app_id']);
@@ -46,6 +50,9 @@ public function __construct(array $params = []) {
 		if (isset($params['task_type'])) {
 			$this->setTaskType($params['task_type']);
 		}
+		if (isset($params['custom_task_type'])) {
+			$this->setCustomTaskType($params['custom_task_type']);
+		}
 	}
 
 	public function jsonSerialize(): array {
@@ -54,6 +61,7 @@ public function jsonSerialize(): array {
 			'name' => $this->name,
 			'display_name' => $this->displayName,
 			'task_type' => $this->taskType,
+			'custom_task_type' => $this->customTaskType,
 		];
 	}
 }

lib/Migration/Version2800Date20240711080316.php

@@ -0,0 +1,38 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+namespace OCA\AppAPI\Migration;
+
+use Closure;
+use OCP\DB\ISchemaWrapper;
+use OCP\DB\Types;
+use OCP\Migration\IOutput;
+use OCP\Migration\SimpleMigrationStep;
+
+class Version2800Date20240711080316 extends SimpleMigrationStep {
+	/**
+	 * @param IOutput $output
+	 * @param Closure(): ISchemaWrapper $schemaClosure
+	 * @param array $options
+	 * @return null|ISchemaWrapper
+	 */
+	public function changeSchema(IOutput $output, Closure $schemaClosure, array $options): ?ISchemaWrapper {
+		/** @var ISchemaWrapper $schema */
+		$schema = $schemaClosure();
+
+		$table = $schema->getTable('ex_task_processing');
+		if (!$table->hasColumn('custom_task_type')) {
+			$table->addColumn('custom_task_type', Types::TEXT, [
+				'notnull' => false,
+			]);
+		}
+
+		return $schema;
+	}
+}

lib/Service/ProvidersAI/TaskProcessingService.php

@@ -14,7 +14,10 @@
 use OCP\ICache;
 use OCP\ICacheFactory;
 use OCP\IServerContainer;
+use OCP\TaskProcessing\EShapeType;
 use OCP\TaskProcessing\IProvider;
+use OCP\TaskProcessing\ITaskType;
+use OCP\TaskProcessing\ShapeDescriptor;
 use Psr\Log\LoggerInterface;
 
 class TaskProcessingService {
@@ -71,7 +74,8 @@ public function registerTaskProcessingProvider(
 		string $appId,
 		string $name,
 		string $displayName,
-		string $taskType
+		string $taskType,
+		?array $customTaskType,
 	): ?TaskProcessingProvider {
 		try {
 			$taskProcessingProvider = $this->mapper->findByAppidName($appId, $name);
@@ -84,6 +88,7 @@ public function registerTaskProcessingProvider(
 				'name' => $name,
 				'display_name' => $displayName,
 				'task_type' => $taskType,
+				'custom_task_type' => json_encode($customTaskType, JSON_THROW_ON_ERROR),
 			]);
 
 			if ($taskProcessingProvider !== null) {
@@ -186,4 +191,64 @@ public function unregisterExAppTaskProcessingProviders(string $appId): int {
 		$this->resetCacheEnabled();
 		return $result;
 	}
+
+	/**
+	 * @param IRegistrationContext $context
+	 *
+	 * @return void
+	 */
+	public function registerExAppTaskProcessingCustomTaskTypes(IRegistrationContext $context): void {
+		$exAppsProviders = $this->getRegisteredTaskProcessingProviders();
+		foreach ($exAppsProviders as $exAppProvider) {
+			if ($exAppProvider->getCustomTaskType() === null) {
+				continue;
+			}
+
+			$className = '\\OCA\\AppAPI\\' . $exAppProvider->getAppId() . '\\' . $exAppProvider->getName() . '\\TaskType';
+			$taskType = $this->getAnonymousTaskType(json_decode($exAppProvider->getCustomTaskType(), true, 512, JSON_THROW_ON_ERROR));
+			$context->registerService($className, function () use ($taskType) {
+				return $taskType;
+			});
+			$context->registerTaskProcessingTaskType($className);
+		}
+	}
+
+	private function getAnonymousTaskType(
+		array $customTaskType,
+	): ITaskType {
+		return new class($customTaskType) implements ITaskType {
+			public function __construct(
+				private readonly array $customTaskType,
+			) {
+			}
+
+			public function getId(): string {
+				return $this->customTaskType['id'];
+			}
+
+			public function getName(): string {
+				return $this->customTaskType['name'];
+			}
+
+			public function getDescription(): string {
+				return $this->customTaskType['description'];
+			}
+
+			public function getInputShape(): array {
+				return array_map(static fn (array $shape) => new ShapeDescriptor(
+					$shape['name'],
+					$shape['description'],
+					EShapeType::from($shape['type']),
+				), $this->customTaskType['input_shape']);
+			}
+
+			public function getOutputShape(): array {
+				return array_map(static fn (array $shape) => new ShapeDescriptor(
+					$shape['name'],
+					$shape['description'],
+					EShapeType::from($shape['type']),
+				), $this->customTaskType['output_shape']);
+			}
+		};
+	}
 }