From 3fd076811c646b6208417d28a44805241cda9d02 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Cygankiewicz?= <m.cygankiewicz@amtsolution.pl>
Date: Fri, 19 Apr 2024 13:09:16 +0200
Subject: [PATCH] Permissions: button for inline generation

---
 .../CustomFileControlsEventListener.php       | 10 ++-
 .../ModifyFilelistButtonBarEventListener.php  | 10 ++-
 Classes/Backend/Hooks/ButtonBarHook.php       | 25 ++++--
 Classes/Controller/AiImageController.php      | 84 -------------------
 Classes/Controller/AjaxController.php         | 59 ++++++++++++-
 Classes/Controller/BaseController.php         | 32 ++++++-
 .../InlineControl/ImageGenerationButton.php   |  7 ++
 Classes/Utility/PermissionsUtility.php        | 20 ++++-
 Configuration/Backend/AjaxRoutes.php          |  2 +-
 Configuration/Services.yaml                   | 12 +++
 Resources/Private/Language/de.locallang.xlf   |  4 +
 .../Language/de.locallang_contentai.xlf       |  4 +
 Resources/Private/Language/locallang.xlf      |  3 +
 .../Private/Language/locallang_contentai.xlf  |  3 +
 .../Private/Templates/AiImage/Prompt.html     |  4 +
 ext_tables.php                                |  3 +
 16 files changed, 183 insertions(+), 99 deletions(-)

diff --git a/Classes/Backend/EventListener/CustomFileControlsEventListener.php b/Classes/Backend/EventListener/CustomFileControlsEventListener.php
index 6687eea..528494c 100644
--- a/Classes/Backend/EventListener/CustomFileControlsEventListener.php
+++ b/Classes/Backend/EventListener/CustomFileControlsEventListener.php
@@ -17,6 +17,7 @@
 
 namespace DMK\MkContentAi\Backend\EventListener;
 
+use DMK\MkContentAi\Utility\PermissionsUtility;
 use TYPO3\CMS\Backend\Form\Event\CustomFileControlsEvent;
 use TYPO3\CMS\Backend\Form\NodeFactory;
 use TYPO3\CMS\Core\Imaging\Icon;
@@ -36,14 +37,21 @@ final class CustomFileControlsEventListener
      */
     public $iconFactory;
 
-    public function __construct()
+    private PermissionsUtility $permissionsUtility;
+
+    public function __construct(PermissionsUtility $permissionsUtility)
     {
         $this->nodeFactory = GeneralUtility::makeInstance(NodeFactory::class);
         $this->iconFactory = GeneralUtility::makeInstance(IconFactory::class);
+        $this->permissionsUtility = $permissionsUtility;
     }
 
     public function handleEvent(CustomFileControlsEvent $event): void
     {
+        if (!$this->permissionsUtility->userHasAccessToImageGenerationPromptButton()) {
+            return;
+        }
+
         $translatedMessage = LocalizationUtility::translate('labelAiGenerateText', 'mkcontentai') ?? '';
         $item = ' <div class="form-control-wrap"><button type="button" class="btn btn-default t3js-prompt" id="prompt">';
         $item .= $this->iconFactory->getIcon('actions-image', Icon::SIZE_SMALL)->render().' ';
diff --git a/Classes/Backend/EventListener/ModifyFilelistButtonBarEventListener.php b/Classes/Backend/EventListener/ModifyFilelistButtonBarEventListener.php
index 79726dd..b5186e1 100644
--- a/Classes/Backend/EventListener/ModifyFilelistButtonBarEventListener.php
+++ b/Classes/Backend/EventListener/ModifyFilelistButtonBarEventListener.php
@@ -15,6 +15,7 @@
 
 namespace DMK\MkContentAi\Backend\EventListener;
 
+use DMK\MkContentAi\Utility\PermissionsUtility;
 use TYPO3\CMS\Backend\Routing\UriBuilder;
 use TYPO3\CMS\Backend\Template\Components\ButtonBar;
 use TYPO3\CMS\Backend\Template\Components\ModifyButtonBarEvent;
@@ -31,13 +32,20 @@ class ModifyFilelistButtonBarEventListener
      */
     protected $iconFactory;
 
-    public function __construct()
+    private PermissionsUtility $permissionsUtility;
+
+    public function __construct(PermissionsUtility $permissionsUtility)
     {
         $this->iconFactory = GeneralUtility::makeInstance(IconFactory::class);
+        $this->permissionsUtility = $permissionsUtility;
     }
 
     public function handleEvent(ModifyButtonBarEvent $event): void
     {
+        if (!$this->permissionsUtility->userHasAccessToImageGenerationPromptButton()) {
+            return;
+        }
+
         $translatedMessage = LocalizationUtility::translate('labelAiGenerateText', 'mkcontentai') ?? '';
         $url = $this->buildUriToControllerAction();
         $buttons = $event->getButtons();
diff --git a/Classes/Backend/Hooks/ButtonBarHook.php b/Classes/Backend/Hooks/ButtonBarHook.php
index 0272561..7db2acb 100644
--- a/Classes/Backend/Hooks/ButtonBarHook.php
+++ b/Classes/Backend/Hooks/ButtonBarHook.php
@@ -15,6 +15,7 @@
 
 namespace DMK\MkContentAi\Backend\Hooks;
 
+use DMK\MkContentAi\Utility\PermissionsUtility;
 use TYPO3\CMS\Backend\Routing\UriBuilder;
 use TYPO3\CMS\Backend\Template\Components\ButtonBar;
 use TYPO3\CMS\Core\Http\ServerRequestFactory;
@@ -25,6 +26,13 @@
 
 class ButtonBarHook
 {
+    private PermissionsUtility $permissionsUtility;
+
+    public function __construct(PermissionsUtility $permissionsUtility)
+    {
+        $this->permissionsUtility = $permissionsUtility;
+    }
+
     /**
      * Retrieves and returns buttons for the button bar.
      *
@@ -40,18 +48,19 @@ public function getButtons(array $params, ButtonBar $buttonBar)
         $url = $this->buildUriToControllerAction();
         $request = ServerRequestFactory::fromGlobals();
         $currentUri = $request->getQueryParams()['route'];
-        if ('/module/file/FilelistList' === $currentUri) {
-            $iconFactory = GeneralUtility::makeInstance(IconFactory::class);
-            $button = $buttonBar->makeLinkButton();
-            $button->setShowLabelText(true);
-            $button->setIcon($iconFactory->getIcon('actions-image', Icon::SIZE_SMALL));
-            $button->setTitle($translatedMessage);
-            $button->setHref($url);
-            $buttons[ButtonBar::BUTTON_POSITION_LEFT][1][] = $button;
 
+        if ('/module/file/FilelistList' !== $currentUri || !$this->permissionsUtility->userHasAccessToImageGenerationPromptButton()) {
             return $buttons;
         }
 
+        $iconFactory = GeneralUtility::makeInstance(IconFactory::class);
+        $button = $buttonBar->makeLinkButton();
+        $button->setShowLabelText(true);
+        $button->setIcon($iconFactory->getIcon('actions-image', Icon::SIZE_SMALL));
+        $button->setTitle($translatedMessage);
+        $button->setHref($url);
+        $buttons[ButtonBar::BUTTON_POSITION_LEFT][1][] = $button;
+
         return $buttons;
     }
 
diff --git a/Classes/Controller/AiImageController.php b/Classes/Controller/AiImageController.php
index 1b6843d..42fa9b7 100644
--- a/Classes/Controller/AiImageController.php
+++ b/Classes/Controller/AiImageController.php
@@ -17,15 +17,12 @@
 
 namespace DMK\MkContentAi\Controller;
 
-use DMK\MkContentAi\Domain\Model\Image;
 use DMK\MkContentAi\Http\Client\ImageApiInterface;
 use DMK\MkContentAi\Http\Client\OpenAiClient;
 use DMK\MkContentAi\Http\Client\StabilityAiClient;
 use DMK\MkContentAi\Http\Client\StableDiffusionClient;
 use DMK\MkContentAi\Service\FileService;
 use Psr\Http\Message\ResponseInterface;
-use Psr\Http\Message\ServerRequestInterface;
-use TYPO3\CMS\Core\Http\JsonResponse;
 use TYPO3\CMS\Core\Messaging\AbstractMessage;
 use TYPO3\CMS\Core\Utility\GeneralUtility;
 use TYPO3\CMS\Extbase\Domain\Model\File;
@@ -93,32 +90,6 @@ public function initializeAction(): void
         parent::initializeAction();
     }
 
-    /**
-     * @return array{client?:ImageApiInterface, clientClass?:string, error?:string}
-     */
-    private function initializeClient(): array
-    {
-        try {
-            $imageEngineKey = SettingsController::getImageAiEngine();
-            $client = GeneralUtility::makeInstance($this::GENERATOR_ENGINE[$imageEngineKey]);
-            if (is_a($client, ImageApiInterface::class)) {
-                return [
-                    'client' => $client,
-                    'clientClass' => get_class($client),
-                ];
-            }
-            $errorTranslated = LocalizationUtility::translate('labelError', 'mkcontentai') ?? '';
-
-            return [
-                'error' => $errorTranslated,
-            ];
-        } catch (\Exception $e) {
-            return [
-                'error' => $e->getMessage(),
-            ];
-        }
-    }
-
     /**
      * @return ResponseInterface
      *
@@ -151,61 +122,6 @@ protected function handleResponse(): ResponseInterface
         return $this->htmlResponse($moduleTemplate->renderContent());
     }
 
-    /**
-     * @return ResponseInterface
-     *
-     * @throws \TYPO3\CMS\Core\Resource\Exception\ExistingTargetFileNameException
-     */
-    public function promptResultAjaxAction(ServerRequestInterface $request)
-    {
-        $clientResponse = $this->initializeClient();
-
-        if (isset($clientResponse['error'])) {
-            return new JsonResponse(
-                [
-                    'error' => $clientResponse['error'],
-                ],
-                500);
-        }
-        if (!isset($clientResponse['client'])) {
-            $translatedMessage = LocalizationUtility::translate('labelErrorClientIsNotDefined', 'mkcontentai') ?? '';
-
-            throw new \Exception($translatedMessage, 1623345720);
-        }
-        $client = $clientResponse['client'];
-
-        if (empty($request->getParsedBody()['promptText'])) {
-            $translatedMessage = LocalizationUtility::translate('labelErrorPromptText', 'mkcontentai') ?? '';
-
-            return new JsonResponse(
-                [
-                    'error' => $translatedMessage,
-                ],
-                500);
-        }
-        $text = $request->getParsedBody()['promptText'];
-
-        try {
-            $images = $client->image($text);
-            /** @var Image[] $images */
-            foreach ($images as $key => $image) {
-                $images[$key] = $image->toArray();
-            }
-            $data = [
-                'name' => get_class($client),
-                'images' => $images,
-            ];
-        } catch (\Exception $e) {
-            return new JsonResponse(
-                [
-                    'error' => $e->getMessage(),
-                ],
-                500);
-        }
-
-        return new JsonResponse($data, 200);
-    }
-
     /**
      * @return ResponseInterface
      */
diff --git a/Classes/Controller/AjaxController.php b/Classes/Controller/AjaxController.php
index 07a9831..4cb3346 100644
--- a/Classes/Controller/AjaxController.php
+++ b/Classes/Controller/AjaxController.php
@@ -17,16 +17,18 @@
 
 namespace DMK\MkContentAi\Controller;
 
+use DMK\MkContentAi\Domain\Model\Image;
 use DMK\MkContentAi\Service\AiAltTextService;
 use DMK\MkContentAi\Service\FileService;
 use DMK\MkContentAi\Service\SiteLanguageService;
 use Psr\Http\Message\ResponseInterface;
 use Psr\Http\Message\ServerRequestInterface;
+use TYPO3\CMS\Core\Http\JsonResponse;
 use TYPO3\CMS\Core\Http\Response;
 use TYPO3\CMS\Core\Utility\GeneralUtility;
 use TYPO3\CMS\Extbase\Utility\LocalizationUtility;
 
-class AjaxController
+class AjaxController extends BaseController
 {
     private FileService $fileService;
 
@@ -144,4 +146,59 @@ public function altTextSaveAction(ServerRequestInterface $request): ResponseInte
 
         return $response->withHeader('Content-Type', 'text/plain')->withStatus(200);
     }
+
+    /**
+     * @return ResponseInterface
+     *
+     * @throws \TYPO3\CMS\Core\Resource\Exception\ExistingTargetFileNameException
+     */
+    public function promptResultAjaxAction(ServerRequestInterface $request)
+    {
+        $clientResponse = $this->initializeClient();
+
+        if (isset($clientResponse['error'])) {
+            return new JsonResponse(
+                [
+                    'error' => $clientResponse['error'],
+                ],
+                500);
+        }
+        if (!isset($clientResponse['client'])) {
+            $translatedMessage = LocalizationUtility::translate('labelErrorClientIsNotDefined', 'mkcontentai') ?? '';
+
+            throw new \Exception($translatedMessage, 1623345720);
+        }
+        $client = $clientResponse['client'];
+
+        if (empty($request->getParsedBody()['promptText'])) {
+            $translatedMessage = LocalizationUtility::translate('labelErrorPromptText', 'mkcontentai') ?? '';
+
+            return new JsonResponse(
+                [
+                    'error' => $translatedMessage,
+                ],
+                500);
+        }
+        $text = $request->getParsedBody()['promptText'];
+
+        try {
+            $images = $client->image($text);
+            /** @var Image[] $images */
+            foreach ($images as $key => $image) {
+                $images[$key] = $image->toArray();
+            }
+            $data = [
+                'name' => get_class($client),
+                'images' => $images,
+            ];
+        } catch (\Exception $e) {
+            return new JsonResponse(
+                [
+                    'error' => $e->getMessage(),
+                ],
+                500);
+        }
+
+        return new JsonResponse($data, 200);
+    }
 }
diff --git a/Classes/Controller/BaseController.php b/Classes/Controller/BaseController.php
index 2c9976f..6a1caf5 100644
--- a/Classes/Controller/BaseController.php
+++ b/Classes/Controller/BaseController.php
@@ -15,9 +15,13 @@
 
 namespace DMK\MkContentAi\Controller;
 
+use DMK\MkContentAi\Http\Client\ImageApiInterface;
 use TYPO3\CMS\Backend\Template\ModuleTemplateFactory;
+use TYPO3\CMS\Core\Page\PageRenderer;
+use TYPO3\CMS\Core\Utility\GeneralUtility;
 use TYPO3\CMS\Core\Utility\PathUtility;
 use TYPO3\CMS\Extbase\Mvc\Controller\ActionController;
+use TYPO3\CMS\Extbase\Utility\LocalizationUtility;
 
 class BaseController extends ActionController
 {
@@ -30,7 +34,7 @@ public function injectModuleTemplateFactory(ModuleTemplateFactory $moduleTemplat
 
     public function initializeAction(): void
     {
-        $pageRenderer = \TYPO3\CMS\Core\Utility\GeneralUtility::makeInstance(\TYPO3\CMS\Core\Page\PageRenderer::class);
+        $pageRenderer = GeneralUtility::makeInstance(PageRenderer::class);
         $pageRenderer->loadRequireJsModule('TYPO3/CMS/Mkcontentai/MkContentAi');
         $pageRenderer->addRequireJsConfiguration(
             [
@@ -43,4 +47,30 @@ public function initializeAction(): void
             ]
         );
     }
+
+    /**
+     * @return array{client?:ImageApiInterface, clientClass?:string, error?:string}
+     */
+    protected function initializeClient(): array
+    {
+        try {
+            $imageEngineKey = SettingsController::getImageAiEngine();
+            $client = GeneralUtility::makeInstance(AiImageController::GENERATOR_ENGINE[$imageEngineKey]);
+            if (is_a($client, ImageApiInterface::class)) {
+                return [
+                    'client' => $client,
+                    'clientClass' => get_class($client),
+                ];
+            }
+            $errorTranslated = LocalizationUtility::translate('labelError', 'mkcontentai') ?? '';
+
+            return [
+                'error' => $errorTranslated,
+            ];
+        } catch (\Exception $e) {
+            return [
+                'error' => $e->getMessage(),
+            ];
+        }
+    }
 }
diff --git a/Classes/User/InlineControl/ImageGenerationButton.php b/Classes/User/InlineControl/ImageGenerationButton.php
index c9f9056..dc8ce33 100644
--- a/Classes/User/InlineControl/ImageGenerationButton.php
+++ b/Classes/User/InlineControl/ImageGenerationButton.php
@@ -15,6 +15,7 @@
 
 namespace DMK\MkContentAi\User\InlineControl;
 
+use DMK\MkContentAi\Utility\PermissionsUtility;
 use TYPO3\CMS\Backend\Form\NodeFactory;
 use TYPO3\CMS\Core\Imaging\Icon;
 use TYPO3\CMS\Core\Imaging\IconFactory;
@@ -49,6 +50,12 @@ public function __construct()
      */
     public function render(array $parameters): string
     {
+        $permissionsUtility = GeneralUtility::makeInstance(PermissionsUtility::class);
+
+        if (!$permissionsUtility->userHasAccessToImageGenerationPromptButton()) {
+            return '';
+        }
+
         $translatedMessage = LocalizationUtility::translate('labelAiGenerateText', 'mkcontentai') ?? '';
         $item = ' <div class="form-control-wrap"><button type="button" class="btn btn-default t3js-prompt" id="prompt">';
         $item .= $this->iconFactory->getIcon('actions-image', Icon::SIZE_SMALL)->render().' ';
diff --git a/Classes/Utility/PermissionsUtility.php b/Classes/Utility/PermissionsUtility.php
index 737f515..113a112 100644
--- a/Classes/Utility/PermissionsUtility.php
+++ b/Classes/Utility/PermissionsUtility.php
@@ -33,11 +33,27 @@ public function injectContext(Context $context): void
      */
     public function userHasAccessToSettings(): bool
     {
-        $user = $this->context->getAspect('backend.user');
-        if ($user->isAdmin()) {
+        if ($this->getUserAspect()) {
             return true;
         }
 
         return true === $GLOBALS['BE_USER']->check('custom_options', 'mkcontentaiSettingsPermissions:settingsPermissions');
     }
+
+    /**
+     * @SuppressWarnings(PHPMD.Superglobals)
+     */
+    public function userHasAccessToImageGenerationPromptButton(): bool
+    {
+        if ($this->getUserAspect()) {
+            return true;
+        }
+
+        return true === $GLOBALS['BE_USER']->check('custom_options', 'mkcontentaiSettingsPermissions:tt_contentImagePrompt');
+    }
+
+    public function getUserAspect(): bool
+    {
+        return $this->context->getAspect('backend.user')->isAdmin();
+    }
 }
diff --git a/Configuration/Backend/AjaxRoutes.php b/Configuration/Backend/AjaxRoutes.php
index 7777540..ea69431 100644
--- a/Configuration/Backend/AjaxRoutes.php
+++ b/Configuration/Backend/AjaxRoutes.php
@@ -20,7 +20,7 @@
     ],
     'image_prompt' => [
         'path' => '/image/prompt',
-        'target' => DMK\MkContentAi\Controller\AiImageController::class.'::promptResultAjaxAction',
+        'target' => DMK\MkContentAi\Controller\AjaxController::class.'::promptResultAjaxAction',
     ],
     'alt_text' => [
         'path' => '/image/alt-text',
diff --git a/Configuration/Services.yaml b/Configuration/Services.yaml
index 7e9e314..9b3c4e1 100644
--- a/Configuration/Services.yaml
+++ b/Configuration/Services.yaml
@@ -35,6 +35,18 @@ services:
                 description: 'Generate alt texts in specific directory - example: "1:/mkcontentai/openai"'
                 schedulable: true
 
+    DMK\MkContentAi\Utility\PermissionsUtility:
+        calls:
+            - method: 'injectContext'
+              arguments:
+                  $context: '@TYPO3\CMS\Core\Context\Context'
+        public: true
+
+    DMK\MkContentAi\Backend\Hooks\ButtonBarHook:
+        arguments:
+            $permissionsUtility: '@DMK\MkContentAi\Utility\PermissionsUtility'
+        public: true
+
     DMK\MkContentAi\Service\SiteLanguageService:
         public: true
 
diff --git a/Resources/Private/Language/de.locallang.xlf b/Resources/Private/Language/de.locallang.xlf
index fd08d90..143b8e0 100644
--- a/Resources/Private/Language/de.locallang.xlf
+++ b/Resources/Private/Language/de.locallang.xlf
@@ -161,6 +161,10 @@
 			<trans-unit id="labelErrorSettingsPermissions" resname="labelErrorSettingsPermissions">
             	<source>You don't have permissions to go Settings Page!</source>
             	<target>Sie haben keine Berechtigung, die Seite Einstellungen aufzurufen!</target>
+            </trans-unit>
+			<trans-unit id="labelErrorAiImagePromptPermissions" resname="labelErrorAiImagePromptPermissions">
+            	<source>You don't have permissions to go Image generation Page!</source>
+            	<target>Sie haben keine Berechtigung, die Seite Bilderstellungen aufzurufen!</target>
             </trans-unit>
 		</body>
 	</file>
diff --git a/Resources/Private/Language/de.locallang_contentai.xlf b/Resources/Private/Language/de.locallang_contentai.xlf
index 1e9c4a7..d0727b5 100644
--- a/Resources/Private/Language/de.locallang_contentai.xlf
+++ b/Resources/Private/Language/de.locallang_contentai.xlf
@@ -216,6 +216,10 @@
 			<trans-unit id="labelMkcontentaiPermissions" resname="labelMkcontentaiPermissions">
             	<source>MKContentAI permissions</source>
             	<target>MKContentAI Berechtigungen</target>
+            </trans-unit>
+			<trans-unit id="labelSettingsImageGenerationPrompt" resname="labelSettingsImageGenerationPrompt">
+            	<source>Enable image generation prompt in tt_content:media field</source>
+            	<target>Aufforderung zur Bilderzeugung im Feld tt_content:media aktivieren</target>
             </trans-unit>
 		</body>
 	</file>
diff --git a/Resources/Private/Language/locallang.xlf b/Resources/Private/Language/locallang.xlf
index 2197191..1a09b63 100644
--- a/Resources/Private/Language/locallang.xlf
+++ b/Resources/Private/Language/locallang.xlf
@@ -134,6 +134,9 @@
             </trans-unit>
 			<trans-unit id="labelErrorSettingsPermissions" resname="labelErrorSettingsPermissions">
             	<source>You don't have permissions to go Settings Page!</source>
+            </trans-unit>
+			<trans-unit id="labelErrorAiImagePromptPermissions" resname="labelErrorAiImagePromptPermissions">
+            	<source>You don't have permissions to go Image generation Page!</source>
             </trans-unit>
 		</body>
 	</file>
diff --git a/Resources/Private/Language/locallang_contentai.xlf b/Resources/Private/Language/locallang_contentai.xlf
index f7cce5a..2fb16f5 100644
--- a/Resources/Private/Language/locallang_contentai.xlf
+++ b/Resources/Private/Language/locallang_contentai.xlf
@@ -164,6 +164,9 @@
             </trans-unit>
 			<trans-unit id="labelMkcontentaiPermissions" resname="labelMkcontentaiPermissions">
             	<source>MKContentAI permissions</source>
+            </trans-unit>
+			<trans-unit id="labelSettingsImageGenerationPrompt" resname="labelSettingsImageGenerationPrompt">
+            	<source>Enable image generation prompt in tt_content:media field</source>
             </trans-unit>
 		</body>
 	</file>
diff --git a/Resources/Private/Templates/AiImage/Prompt.html b/Resources/Private/Templates/AiImage/Prompt.html
index d491a81..85a2c7d 100644
--- a/Resources/Private/Templates/AiImage/Prompt.html
+++ b/Resources/Private/Templates/AiImage/Prompt.html
@@ -4,6 +4,10 @@
     data-namespace-typo3-fluid="true">
 
 <f:layout name="Default" />
+<f:section name="InsufficientPermissions">
+    <f:render partial="Navigation"/>
+    <f:flashMessages/>
+</f:section>
 
 <f:section name="Content">
     <h1><f:translate key="LLL:EXT:mkcontentai/Resources/Private/Language/locallang_contentai.xlf:labelPromptAction"/></h1>
diff --git a/ext_tables.php b/ext_tables.php
index 6d5def6..73ef10b 100644
--- a/ext_tables.php
+++ b/ext_tables.php
@@ -43,5 +43,8 @@
         'settingsPermissions' => [
             'LLL:EXT:mkcontentai/Resources/Private/Language/locallang_contentai.xlf:labelSettingsPageCurrentGroup',
         ],
+        'tt_contentImagePrompt' => [
+            'LLL:EXT:mkcontentai/Resources/Private/Language/locallang_contentai.xlf:labelSettingsImageGenerationPrompt',
+        ],
     ],
 ];