Signed-off-by: Julien Veyssier <julien-nc@posteo.net>pull/57235/head
parent
72ad2edcd7
commit
e1d19b2c24
@ -0,0 +1,93 @@ |
||||
<?php |
||||
|
||||
/** |
||||
* SPDX-FileCopyrightText: 2025 Nextcloud GmbH and Nextcloud contributors |
||||
* SPDX-License-Identifier: AGPL-3.0-or-later |
||||
*/ |
||||
|
||||
declare(strict_types=1); |
||||
|
||||
namespace OCA\Testing\TaskProcessing; |
||||
|
||||
use OCA\Testing\AppInfo\Application; |
||||
use OCP\AppFramework\Services\IAppConfig; |
||||
use OCP\Files\File; |
||||
use OCP\TaskProcessing\Exception\ProcessingException; |
||||
use OCP\TaskProcessing\ISynchronousProvider; |
||||
use OCP\TaskProcessing\TaskTypes\ImageToTextOpticalCharacterRecognition; |
||||
use RuntimeException; |
||||
|
||||
class FakeOcrProvider implements ISynchronousProvider { |
||||
|
||||
public function __construct( |
||||
protected IAppConfig $appConfig, |
||||
) { |
||||
} |
||||
|
||||
public function getId(): string { |
||||
return Application::APP_ID . '-image2text-ocr'; |
||||
} |
||||
|
||||
public function getName(): string { |
||||
return 'Fake OCR task processing provider'; |
||||
} |
||||
|
||||
public function getTaskTypeId(): string { |
||||
return ImageToTextOpticalCharacterRecognition::ID; |
||||
} |
||||
|
||||
public function getExpectedRuntime(): int { |
||||
return 1; |
||||
} |
||||
|
||||
public function getInputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getInputShapeDefaults(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalInputShape(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalInputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalInputShapeDefaults(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOutputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalOutputShape(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function getOptionalOutputShapeEnumValues(): array { |
||||
return []; |
||||
} |
||||
|
||||
public function process(?string $userId, array $input, callable $reportProgress): array { |
||||
if ($this->appConfig->getAppValueBool('fail-' . $this->getId())) { |
||||
throw new ProcessingException('Failing as set by AppConfig'); |
||||
} |
||||
|
||||
if (!isset($input['input']) || !is_array($input['input'])) { |
||||
throw new RuntimeException('Invalid input'); |
||||
} |
||||
$outputs = []; |
||||
foreach ($input['input'] as $i => $inputImage) { |
||||
if (!($inputImage instanceof File) || !$inputImage->isReadable()) { |
||||
throw new RuntimeException('Invalid input images'); |
||||
} |
||||
$outputs[] = '[' . $i . '] This is a fake OCR result.'; |
||||
} |
||||
|
||||
return ['output' => $outputs]; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue