123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565 |
- <?php
- declare(strict_types=1);
- /**
- * SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
- * SPDX-License-Identifier: AGPL-3.0-or-later
- */
- namespace OC\Core\Controller;
- use OC\Core\ResponseDefinitions;
- use OC\Files\SimpleFS\SimpleFile;
- use OCP\AppFramework\Http;
- use OCP\AppFramework\Http\Attribute\AnonRateLimit;
- use OCP\AppFramework\Http\Attribute\ApiRoute;
- use OCP\AppFramework\Http\Attribute\ExAppRequired;
- use OCP\AppFramework\Http\Attribute\NoAdminRequired;
- use OCP\AppFramework\Http\Attribute\PublicPage;
- use OCP\AppFramework\Http\Attribute\UserRateLimit;
- use OCP\AppFramework\Http\DataDownloadResponse;
- use OCP\AppFramework\Http\DataResponse;
- use OCP\Files\File;
- use OCP\Files\GenericFileException;
- use OCP\Files\IAppData;
- use OCP\Files\IRootFolder;
- use OCP\Files\NotPermittedException;
- use OCP\IL10N;
- use OCP\IRequest;
- use OCP\Lock\LockedException;
- use OCP\TaskProcessing\EShapeType;
- use OCP\TaskProcessing\Exception\Exception;
- use OCP\TaskProcessing\Exception\NotFoundException;
- use OCP\TaskProcessing\Exception\PreConditionNotMetException;
- use OCP\TaskProcessing\Exception\UnauthorizedException;
- use OCP\TaskProcessing\Exception\ValidationException;
- use OCP\TaskProcessing\IManager;
- use OCP\TaskProcessing\ShapeEnumValue;
- use OCP\TaskProcessing\Task;
- use RuntimeException;
- /**
- * @psalm-import-type CoreTaskProcessingTask from ResponseDefinitions
- * @psalm-import-type CoreTaskProcessingTaskType from ResponseDefinitions
- */
- class TaskProcessingApiController extends \OCP\AppFramework\OCSController {
- public function __construct(
- string $appName,
- IRequest $request,
- private IManager $taskProcessingManager,
- private IL10N $l,
- private ?string $userId,
- private IRootFolder $rootFolder,
- private IAppData $appData,
- ) {
- parent::__construct($appName, $request);
- }
- /**
- * Returns all available TaskProcessing task types
- *
- * @return DataResponse<Http::STATUS_OK, array{types: array<string, CoreTaskProcessingTaskType>}, array{}>
- *
- * 200: Task types returned
- */
- #[PublicPage]
- #[ApiRoute(verb: 'GET', url: '/tasktypes', root: '/taskprocessing')]
- public function taskTypes(): DataResponse {
- $taskTypes = array_map(function (array $tt) {
- $tt['inputShape'] = array_values(array_map(function ($descriptor) {
- return $descriptor->jsonSerialize();
- }, $tt['inputShape']));
- $tt['outputShape'] = array_values(array_map(function ($descriptor) {
- return $descriptor->jsonSerialize();
- }, $tt['outputShape']));
- $tt['optionalInputShape'] = array_values(array_map(function ($descriptor) {
- return $descriptor->jsonSerialize();
- }, $tt['optionalInputShape']));
- $tt['optionalOutputShape'] = array_values(array_map(function ($descriptor) {
- return $descriptor->jsonSerialize();
- }, $tt['optionalOutputShape']));
- $tt['inputShapeEnumValues'] = array_values(array_map(function (array $enumValues) {
- return array_values(array_map(fn (ShapeEnumValue $enumValue) => $enumValue->jsonSerialize(), $enumValues));
- }, $tt['inputShapeEnumValues']));
- $tt['optionalInputShapeEnumValues'] = array_values(array_map(function (array $enumValues) {
- return array_values(array_map(fn (ShapeEnumValue $enumValue) => $enumValue->jsonSerialize(), $enumValues));
- }, $tt['optionalInputShapeEnumValues']));
- $tt['outputShapeEnumValues'] = array_values(array_map(function (array $enumValues) {
- return array_values(array_map(fn (ShapeEnumValue $enumValue) => $enumValue->jsonSerialize(), $enumValues));
- }, $tt['outputShapeEnumValues']));
- $tt['optionalOutputShapeEnumValues'] = array_values(array_map(function (array $enumValues) {
- return array_values(array_map(fn (ShapeEnumValue $enumValue) => $enumValue->jsonSerialize(), $enumValues));
- }, $tt['optionalOutputShapeEnumValues']));
- return $tt;
- }, $this->taskProcessingManager->getAvailableTaskTypes());
- return new DataResponse([
- 'types' => $taskTypes,
- ]);
- }
- /**
- * Schedules a task
- *
- * @param array<string, mixed> $input Task's input parameters
- * @param string $type Type of the task
- * @param string $appId ID of the app that will execute the task
- * @param string $customId An arbitrary identifier for the task
- * @param string|null $webhookUri URI to be requested when the task finishes
- * @param string|null $webhookMethod Method used for the webhook request (HTTP:GET, HTTP:POST, HTTP:PUT, HTTP:DELETE or AppAPI:APP_ID:GET, AppAPI:APP_ID:POST...)
- * @return DataResponse<Http::STATUS_OK, array{task: CoreTaskProcessingTask}, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_BAD_REQUEST|Http::STATUS_PRECONDITION_FAILED|Http::STATUS_UNAUTHORIZED, array{message: string}, array{}>
- *
- * 200: Task scheduled successfully
- * 400: Scheduling task is not possible
- * 412: Scheduling task is not possible
- * 401: Cannot schedule task because it references files in its input that the user doesn't have access to
- */
- #[PublicPage]
- #[UserRateLimit(limit: 20, period: 120)]
- #[AnonRateLimit(limit: 5, period: 120)]
- #[ApiRoute(verb: 'POST', url: '/schedule', root: '/taskprocessing')]
- public function schedule(
- array $input, string $type, string $appId, string $customId = '',
- ?string $webhookUri = null, ?string $webhookMethod = null,
- ): DataResponse {
- $task = new Task($type, $input, $appId, $this->userId, $customId);
- $task->setWebhookUri($webhookUri);
- $task->setWebhookMethod($webhookMethod);
- try {
- $this->taskProcessingManager->scheduleTask($task);
- /** @var CoreTaskProcessingTask $json */
- $json = $task->jsonSerialize();
- return new DataResponse([
- 'task' => $json,
- ]);
- } catch (PreConditionNotMetException) {
- return new DataResponse(['message' => $this->l->t('The given provider is not available')], Http::STATUS_PRECONDITION_FAILED);
- } catch (ValidationException $e) {
- return new DataResponse(['message' => $e->getMessage()], Http::STATUS_BAD_REQUEST);
- } catch (UnauthorizedException) {
- return new DataResponse(['message' => 'User does not have access to the files mentioned in the task input'], Http::STATUS_UNAUTHORIZED);
- } catch (Exception) {
- return new DataResponse(['message' => 'Internal server error'], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Gets a task including status and result
- *
- * Tasks are removed 1 week after receiving their last update
- *
- * @param int $id The id of the task
- *
- * @return DataResponse<Http::STATUS_OK, array{task: CoreTaskProcessingTask}, array{}>|DataResponse<Http::STATUS_NOT_FOUND|Http::STATUS_INTERNAL_SERVER_ERROR, array{message: string}, array{}>
- *
- * 200: Task returned
- * 404: Task not found
- */
- #[PublicPage]
- #[ApiRoute(verb: 'GET', url: '/task/{id}', root: '/taskprocessing')]
- public function getTask(int $id): DataResponse {
- try {
- $task = $this->taskProcessingManager->getUserTask($id, $this->userId);
- /** @var CoreTaskProcessingTask $json */
- $json = $task->jsonSerialize();
- return new DataResponse([
- 'task' => $json,
- ]);
- } catch (NotFoundException) {
- return new DataResponse(['message' => $this->l->t('Task not found')], Http::STATUS_NOT_FOUND);
- } catch (RuntimeException) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Deletes a task
- *
- * @param int $id The id of the task
- *
- * @return DataResponse<Http::STATUS_OK, null, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR, array{message: string}, array{}>
- *
- * 200: Task deleted
- */
- #[NoAdminRequired]
- #[ApiRoute(verb: 'DELETE', url: '/task/{id}', root: '/taskprocessing')]
- public function deleteTask(int $id): DataResponse {
- try {
- $task = $this->taskProcessingManager->getUserTask($id, $this->userId);
- $this->taskProcessingManager->deleteTask($task);
- return new DataResponse(null);
- } catch (NotFoundException) {
- return new DataResponse(null);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Returns tasks for the current user filtered by the appId and optional customId
- *
- * @param string $appId ID of the app
- * @param string|null $customId An arbitrary identifier for the task
- * @return DataResponse<Http::STATUS_OK, array{tasks: list<CoreTaskProcessingTask>}, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR, array{message: string}, array{}>
- *
- * 200: Tasks returned
- */
- #[NoAdminRequired]
- #[ApiRoute(verb: 'GET', url: '/tasks/app/{appId}', root: '/taskprocessing')]
- public function listTasksByApp(string $appId, ?string $customId = null): DataResponse {
- try {
- $tasks = $this->taskProcessingManager->getUserTasksByApp($this->userId, $appId, $customId);
- $json = array_map(static function (Task $task) {
- return $task->jsonSerialize();
- }, $tasks);
- return new DataResponse([
- 'tasks' => $json,
- ]);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Returns tasks for the current user filtered by the optional taskType and optional customId
- *
- * @param string|null $taskType The task type to filter by
- * @param string|null $customId An arbitrary identifier for the task
- * @return DataResponse<Http::STATUS_OK, array{tasks: list<CoreTaskProcessingTask>}, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR, array{message: string}, array{}>
- *
- * 200: Tasks returned
- */
- #[NoAdminRequired]
- #[ApiRoute(verb: 'GET', url: '/tasks', root: '/taskprocessing')]
- public function listTasks(?string $taskType, ?string $customId = null): DataResponse {
- try {
- $tasks = $this->taskProcessingManager->getUserTasks($this->userId, $taskType, $customId);
- $json = array_map(static function (Task $task) {
- return $task->jsonSerialize();
- }, $tasks);
- return new DataResponse([
- 'tasks' => $json,
- ]);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Returns the contents of a file referenced in a task
- *
- * @param int $taskId The id of the task
- * @param int $fileId The file id of the file to retrieve
- * @return DataDownloadResponse<Http::STATUS_OK, string, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_NOT_FOUND, array{message: string}, array{}>
- *
- * 200: File content returned
- * 404: Task or file not found
- */
- #[NoAdminRequired]
- #[Http\Attribute\NoCSRFRequired]
- #[ApiRoute(verb: 'GET', url: '/tasks/{taskId}/file/{fileId}', root: '/taskprocessing')]
- public function getFileContents(int $taskId, int $fileId): Http\DataDownloadResponse|DataResponse {
- try {
- $task = $this->taskProcessingManager->getUserTask($taskId, $this->userId);
- return $this->getFileContentsInternal($task, $fileId);
- } catch (NotFoundException) {
- return new DataResponse(['message' => $this->l->t('Not found')], Http::STATUS_NOT_FOUND);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Returns the contents of a file referenced in a task(ExApp route version)
- *
- * @param int $taskId The id of the task
- * @param int $fileId The file id of the file to retrieve
- * @return DataDownloadResponse<Http::STATUS_OK, string, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_NOT_FOUND, array{message: string}, array{}>
- *
- * 200: File content returned
- * 404: Task or file not found
- */
- #[ExAppRequired]
- #[ApiRoute(verb: 'GET', url: '/tasks_provider/{taskId}/file/{fileId}', root: '/taskprocessing')]
- public function getFileContentsExApp(int $taskId, int $fileId): Http\DataDownloadResponse|DataResponse {
- try {
- $task = $this->taskProcessingManager->getTask($taskId);
- return $this->getFileContentsInternal($task, $fileId);
- } catch (NotFoundException) {
- return new DataResponse(['message' => $this->l->t('Not found')], Http::STATUS_NOT_FOUND);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Upload a file so it can be referenced in a task result (ExApp route version)
- *
- * Use field 'file' for the file upload
- *
- * @param int $taskId The id of the task
- * @return DataResponse<Http::STATUS_CREATED, array{fileId: int}, array{}>|DataResponse<Http::STATUS_BAD_REQUEST|Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_NOT_FOUND, array{message: string}, array{}>
- *
- * 201: File created
- * 400: File upload failed or no file was uploaded
- * 404: Task not found
- */
- #[ExAppRequired]
- #[ApiRoute(verb: 'POST', url: '/tasks_provider/{taskId}/file', root: '/taskprocessing')]
- public function setFileContentsExApp(int $taskId): DataResponse {
- try {
- $task = $this->taskProcessingManager->getTask($taskId);
- $file = $this->request->getUploadedFile('file');
- if (!isset($file['tmp_name'])) {
- return new DataResponse(['message' => $this->l->t('Bad request')], Http::STATUS_BAD_REQUEST);
- }
- $handle = fopen($file['tmp_name'], 'r');
- if (!$handle) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- $fileId = $this->setFileContentsInternal($handle);
- return new DataResponse(['fileId' => $fileId], Http::STATUS_CREATED);
- } catch (NotFoundException) {
- return new DataResponse(['message' => $this->l->t('Not found')], Http::STATUS_NOT_FOUND);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * @throws NotPermittedException
- * @throws NotFoundException
- * @throws GenericFileException
- * @throws LockedException
- *
- * @return DataDownloadResponse<Http::STATUS_OK, string, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_NOT_FOUND, array{message: string}, array{}>
- */
- private function getFileContentsInternal(Task $task, int $fileId): Http\DataDownloadResponse|DataResponse {
- $ids = $this->extractFileIdsFromTask($task);
- if (!in_array($fileId, $ids)) {
- return new DataResponse(['message' => $this->l->t('Not found')], Http::STATUS_NOT_FOUND);
- }
- if ($task->getUserId() !== null) {
- \OC_Util::setupFS($task->getUserId());
- }
- $node = $this->rootFolder->getFirstNodeById($fileId);
- if ($node === null) {
- $node = $this->rootFolder->getFirstNodeByIdInPath($fileId, '/' . $this->rootFolder->getAppDataDirectoryName() . '/');
- if (!$node instanceof File) {
- throw new NotFoundException('Node is not a file');
- }
- } elseif (!$node instanceof File) {
- throw new NotFoundException('Node is not a file');
- }
- return new Http\DataDownloadResponse($node->getContent(), $node->getName(), $node->getMimeType());
- }
- /**
- * @param Task $task
- * @return list<int>
- * @throws NotFoundException
- */
- private function extractFileIdsFromTask(Task $task): array {
- $ids = [];
- $taskTypes = $this->taskProcessingManager->getAvailableTaskTypes();
- if (!isset($taskTypes[$task->getTaskTypeId()])) {
- throw new NotFoundException('Could not find task type');
- }
- $taskType = $taskTypes[$task->getTaskTypeId()];
- foreach ($taskType['inputShape'] + $taskType['optionalInputShape'] as $key => $descriptor) {
- if (in_array(EShapeType::getScalarType($descriptor->getShapeType()), [EShapeType::File, EShapeType::Image, EShapeType::Audio, EShapeType::Video], true)) {
- /** @var int|list<int> $inputSlot */
- $inputSlot = $task->getInput()[$key];
- if (is_array($inputSlot)) {
- $ids = array_merge($inputSlot, $ids);
- } else {
- $ids[] = $inputSlot;
- }
- }
- }
- if ($task->getOutput() !== null) {
- foreach ($taskType['outputShape'] + $taskType['optionalOutputShape'] as $key => $descriptor) {
- if (in_array(EShapeType::getScalarType($descriptor->getShapeType()), [EShapeType::File, EShapeType::Image, EShapeType::Audio, EShapeType::Video], true)) {
- /** @var int|list<int> $outputSlot */
- $outputSlot = $task->getOutput()[$key];
- if (is_array($outputSlot)) {
- $ids = array_merge($outputSlot, $ids);
- } else {
- $ids[] = $outputSlot;
- }
- }
- }
- }
- return $ids;
- }
- /**
- * Sets the task progress
- *
- * @param int $taskId The id of the task
- * @param float $progress The progress
- * @return DataResponse<Http::STATUS_OK, array{task: CoreTaskProcessingTask}, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_NOT_FOUND, array{message: string}, array{}>
- *
- * 200: Progress updated successfully
- * 404: Task not found
- */
- #[ExAppRequired]
- #[ApiRoute(verb: 'POST', url: '/tasks_provider/{taskId}/progress', root: '/taskprocessing')]
- public function setProgress(int $taskId, float $progress): DataResponse {
- try {
- $this->taskProcessingManager->setTaskProgress($taskId, $progress);
- $task = $this->taskProcessingManager->getTask($taskId);
- /** @var CoreTaskProcessingTask $json */
- $json = $task->jsonSerialize();
- return new DataResponse([
- 'task' => $json,
- ]);
- } catch (NotFoundException) {
- return new DataResponse(['message' => $this->l->t('Not found')], Http::STATUS_NOT_FOUND);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Sets the task result
- *
- * @param int $taskId The id of the task
- * @param array<string,mixed>|null $output The resulting task output, files are represented by their IDs
- * @param string|null $errorMessage An error message if the task failed
- * @return DataResponse<Http::STATUS_OK, array{task: CoreTaskProcessingTask}, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_NOT_FOUND, array{message: string}, array{}>
- *
- * 200: Result updated successfully
- * 404: Task not found
- */
- #[ExAppRequired]
- #[ApiRoute(verb: 'POST', url: '/tasks_provider/{taskId}/result', root: '/taskprocessing')]
- public function setResult(int $taskId, ?array $output = null, ?string $errorMessage = null): DataResponse {
- try {
- // set result
- $this->taskProcessingManager->setTaskResult($taskId, $errorMessage, $output, true);
- $task = $this->taskProcessingManager->getTask($taskId);
- /** @var CoreTaskProcessingTask $json */
- $json = $task->jsonSerialize();
- return new DataResponse([
- 'task' => $json,
- ]);
- } catch (NotFoundException) {
- return new DataResponse(['message' => $this->l->t('Not found')], Http::STATUS_NOT_FOUND);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Cancels a task
- *
- * @param int $taskId The id of the task
- * @return DataResponse<Http::STATUS_OK, array{task: CoreTaskProcessingTask}, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR|Http::STATUS_NOT_FOUND, array{message: string}, array{}>
- *
- * 200: Task canceled successfully
- * 404: Task not found
- */
- #[NoAdminRequired]
- #[ApiRoute(verb: 'POST', url: '/tasks/{taskId}/cancel', root: '/taskprocessing')]
- public function cancelTask(int $taskId): DataResponse {
- try {
- // Check if the current user can access the task
- $this->taskProcessingManager->getUserTask($taskId, $this->userId);
- // set result
- $this->taskProcessingManager->cancelTask($taskId);
- $task = $this->taskProcessingManager->getUserTask($taskId, $this->userId);
- /** @var CoreTaskProcessingTask $json */
- $json = $task->jsonSerialize();
- return new DataResponse([
- 'task' => $json,
- ]);
- } catch (NotFoundException) {
- return new DataResponse(['message' => $this->l->t('Not found')], Http::STATUS_NOT_FOUND);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * Returns the next scheduled task for the taskTypeId
- *
- * @param list<string> $providerIds The ids of the providers
- * @param list<string> $taskTypeIds The ids of the task types
- * @return DataResponse<Http::STATUS_OK, array{task: CoreTaskProcessingTask, provider: array{name: string}}, array{}>|DataResponse<Http::STATUS_NO_CONTENT, null, array{}>|DataResponse<Http::STATUS_INTERNAL_SERVER_ERROR, array{message: string}, array{}>
- *
- * 200: Task returned
- * 204: No task found
- */
- #[ExAppRequired]
- #[ApiRoute(verb: 'GET', url: '/tasks_provider/next', root: '/taskprocessing')]
- public function getNextScheduledTask(array $providerIds, array $taskTypeIds): DataResponse {
- try {
- // restrict $providerIds to providers that are configured as preferred for the passed task types
- $providerIds = array_values(array_intersect(array_unique(array_map(fn ($taskTypeId) => $this->taskProcessingManager->getPreferredProvider($taskTypeId)->getId(), $taskTypeIds)), $providerIds));
- // restrict $taskTypeIds to task types that can actually be run by one of the now restricted providers
- $taskTypeIds = array_values(array_filter($taskTypeIds, fn ($taskTypeId) => in_array($this->taskProcessingManager->getPreferredProvider($taskTypeId)->getId(), $providerIds, true)));
- if (count($providerIds) === 0 || count($taskTypeIds) === 0) {
- throw new NotFoundException();
- }
- $taskIdsToIgnore = [];
- while (true) {
- $task = $this->taskProcessingManager->getNextScheduledTask($taskTypeIds, $taskIdsToIgnore);
- $provider = $this->taskProcessingManager->getPreferredProvider($task->getTaskTypeId());
- if (in_array($provider->getId(), $providerIds, true)) {
- if ($this->taskProcessingManager->lockTask($task)) {
- break;
- }
- }
- $taskIdsToIgnore[] = (int)$task->getId();
- }
- /** @var CoreTaskProcessingTask $json */
- $json = $task->jsonSerialize();
- return new DataResponse([
- 'task' => $json,
- 'provider' => [
- 'name' => $provider->getId(),
- ],
- ]);
- } catch (NotFoundException) {
- return new DataResponse(null, Http::STATUS_NO_CONTENT);
- } catch (Exception) {
- return new DataResponse(['message' => $this->l->t('Internal error')], Http::STATUS_INTERNAL_SERVER_ERROR);
- }
- }
- /**
- * @param resource $data
- * @return int
- * @throws NotPermittedException
- */
- private function setFileContentsInternal($data): int {
- try {
- $folder = $this->appData->getFolder('TaskProcessing');
- } catch (\OCP\Files\NotFoundException) {
- $folder = $this->appData->newFolder('TaskProcessing');
- }
- /** @var SimpleFile $file */
- $file = $folder->newFile(time() . '-' . rand(1, 100000), $data);
- return $file->getId();
- }
- }
|