feat(new-file-menu): adjustments

Signed-off-by: Julien Veyssier <julien-nc@posteo.net>

Author: julien-nc

lib/AppInfo/Application.php

@@ -15,7 +15,6 @@
 use OCA\Assistant\Listener\FileActionTaskSuccessfulListener;
 use OCA\Assistant\Listener\FreePrompt\FreePromptReferenceListener;
 use OCA\Assistant\Listener\LoadAdditionalScriptsListener;
-use OCA\Assistant\Listener\LoadAdditionalScriptsListener2;
 use OCA\Assistant\Listener\SpeechToText\SpeechToTextReferenceListener;
 use OCA\Assistant\Listener\TaskFailedListener;
 use OCA\Assistant\Listener\TaskOutputFileReferenceListener;
@@ -70,7 +69,6 @@ public function register(IRegistrationContext $context): void {
 
 		$context->registerEventListener(BeforeTemplateRenderedEvent::class, BeforeTemplateRenderedListener::class);
 		$context->registerEventListener(LoadAdditionalScriptsEvent::class, LoadAdditionalScriptsListener::class);
-		$context->registerEventListener(LoadAdditionalScriptsEvent::class, LoadAdditionalScriptsListener2::class);
 
 		$context->registerEventListener(TaskSuccessfulEvent::class, TaskSuccessfulListener::class);
 		$context->registerEventListener(TaskFailedEvent::class, TaskFailedListener::class);

lib/Listener/LoadAdditionalScriptsListener.php

@@ -12,8 +12,10 @@
 use OCP\AppFramework\Services\IInitialState;
 use OCP\EventDispatcher\Event;
 use OCP\EventDispatcher\IEventListener;
+use OCP\IAppConfig;
 use OCP\TaskProcessing\IManager as ITaskProcessingManager;
 use OCP\TaskProcessing\TaskTypes\AudioToText;
+use OCP\TaskProcessing\TaskTypes\TextToImage;
 use OCP\TaskProcessing\TaskTypes\TextToTextSummary;
 use OCP\Util;
 
@@ -23,6 +25,7 @@
 class LoadAdditionalScriptsListener implements IEventListener {
 
 	public function __construct(
+		private IAppConfig $appConfig,
 		private IInitialState $initialStateService,
 		private ITaskProcessingManager $taskProcessingManager,
 	) {
@@ -34,15 +37,25 @@ public function handle(Event $event): void {
 		}
 
 		$availableTaskTypes = $this->taskProcessingManager->getAvailableTaskTypes();
+
+		// file actions
 		$summarizeAvailable = array_key_exists(TextToTextSummary::ID, $availableTaskTypes);
 		$sttAvailable = array_key_exists(AudioToText::ID, $availableTaskTypes);
 		$ttsAvailable = class_exists('OCP\\TaskProcessing\\TaskTypes\\TextToSpeech')
 			&& array_key_exists(\OCP\TaskProcessing\TaskTypes\TextToSpeech::ID, $availableTaskTypes);
-
 		$this->initialStateService->provideInitialState('stt-available', $sttAvailable);
 		$this->initialStateService->provideInitialState('tts-available', $ttsAvailable);
 		$this->initialStateService->provideInitialState('summarize-available', $summarizeAvailable);
-
 		Util::addInitScript(Application::APP_ID, Application::APP_ID . '-fileActions');
+
+		// New file menu to generate images
+		$isNewFileMenuEnabled = $this->appConfig->getValueInt(Application::APP_ID, 'newFileMenuPlugin', 1) === 1;
+		if ($isNewFileMenuEnabled) {
+			$hasText2Image = array_key_exists(TextToImage::ID, $availableTaskTypes);
+			$this->initialStateService->provideInitialState('new-file-generate-image', [
+				'hasText2Image' => $hasText2Image,
+			]);
+			Util::addScript(Application::APP_ID, Application::APP_ID . '-filesNewMenu');
+		}
 	}
 }

lib/Listener/LoadAdditionalScriptsListener2.php

@@ -69,7 +69,7 @@ public function handle(Event $event): void {
 
 		if ($hasTargetDirectory) {
 			$directoryId = (int)$matches[1];
-			$fileId = (int) $task->getOutput()['images'][0];
+			$fileId = (int)$task->getOutput()['images'][0];
 			try {
 				$file = $this->assistantService->saveNewFileMenuActionFile($task->getUserId(), $task->getId(), $fileId, $directoryId);
 				$notificationTarget = $this->url->linkToRouteAbsolute(