diff --git a/lib/private/TaskProcessing/Manager.php b/lib/private/TaskProcessing/Manager.php index 3609f309cc47b..165cde87bf052 100644 --- a/lib/private/TaskProcessing/Manager.php +++ b/lib/private/TaskProcessing/Manager.php @@ -307,13 +307,18 @@ private function _getTextToImageProviders(): array { $oldProviders = $this->textToImageManager->getProviders(); $newProviders = []; foreach ($oldProviders as $oldProvider) { - $newProvider = new class($oldProvider, $this->appData) implements IProvider, ISynchronousProvider { + $newProvider = new class($oldProvider, $this->appData, $this->l10nFactory, $this->userManager) implements IProvider, ISynchronousProvider { private \OCP\TextToImage\IProvider $provider; private IAppData $appData; + private IFactory $l10nFactory; - public function __construct(\OCP\TextToImage\IProvider $provider, IAppData $appData) { + private IUserManager $userManager; + + public function __construct(\OCP\TextToImage\IProvider $provider, IAppData $appData, IFactory $l10nFactory, IUserManager $userManager) { $this->provider = $provider; $this->appData = $appData; + $this->l10nFactory = $l10nFactory; + $this->userManager = $userManager; } public function getId(): string { @@ -346,6 +351,21 @@ public function process(?string $userId, array $input, callable $reportProgress) } catch (\OCP\Files\NotFoundException) { $folder = $this->appData->newFolder('text2image'); } + if ($input['numberOfImages'] > 12) { + throw new UserFacingProcessingException( + 'numberOfImages cannot be greater than 12', + userFacingMessage: + $this->l10nFactory->get('core', $this->l10nFactory->getUserLanguage($this->userManager->get($userId))) + ->t('Cannot generate more than 12 images') + ); + } + if ($input['numberOfImages'] < 1) { + throw new UserFacingProcessingException( + 'numberOfImages must be greater than 0', + userFacingMessage: + $this->l10nFactory->get('core', $this->l10nFactory->getUserLanguage($this->userManager->get($userId))) + ->t('Cannot generate less than 1 image')); + } $resources = []; $files = []; for ($i = 0; $i < $input['numberOfImages']; $i++) {