feat: files "new file" menu generate image action

Signed-off-by: Andrey Borysenko <andrey18106x@gmail.com>

Author: andrey18106

lib/AppInfo/Application.php

@@ -12,6 +12,7 @@
 use OCA\Assistant\Listener\ChattyLLMTaskListener;
 use OCA\Assistant\Listener\CSPListener;
 use OCA\Assistant\Listener\FreePrompt\FreePromptReferenceListener;
+use OCA\Assistant\Listener\LoadAdditionalScriptsListener;
 use OCA\Assistant\Listener\SpeechToText\SpeechToTextReferenceListener;
 use OCA\Assistant\Listener\TaskFailedListener;
 use OCA\Assistant\Listener\TaskOutputFileReferenceListener;
@@ -24,6 +25,7 @@
 use OCA\Assistant\Reference\Text2ImageReferenceProvider;
 use OCA\Assistant\TaskProcessing\AudioToAudioChatProvider;
 use OCA\Assistant\TaskProcessing\ContextAgentAudioInteractionProvider;
+use OCA\Files\Event\LoadAdditionalScriptsEvent;
 use OCP\AppFramework\App;
 use OCP\AppFramework\Bootstrap\IBootContext;
 
@@ -64,6 +66,7 @@ public function register(IRegistrationContext $context): void {
 		$context->registerEventListener(RenderReferenceEvent::class, TaskOutputFileReferenceListener::class);
 
 		$context->registerEventListener(BeforeTemplateRenderedEvent::class, BeforeTemplateRenderedListener::class);
+		$context->registerEventListener(LoadAdditionalScriptsEvent::class, LoadAdditionalScriptsListener::class);
 
 		$context->registerEventListener(TaskSuccessfulEvent::class, TaskSuccessfulListener::class);
 		$context->registerEventListener(TaskFailedEvent::class, TaskFailedListener::class);

lib/Listener/LoadAdditionalScriptsListener.php

@@ -0,0 +1,58 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * SPDX-FileCopyrightText: 2025 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+namespace OCA\Assistant\Listener;
+
+use OCA\Assistant\AppInfo\Application;
+use OCA\Assistant\Service\AssistantService;
+use OCA\Files\Event\LoadAdditionalScriptsEvent;
+use OCP\AppFramework\Services\IInitialState;
+use OCP\EventDispatcher\Event;
+use OCP\EventDispatcher\IEventListener;
+use OCP\IConfig;
+use OCP\Util;
+
+/**
+ * @template-extends IEventListener<LoadAdditionalScriptsEvent>
+ */
+class LoadAdditionalScriptsListener implements IEventListener {
+
+	public function __construct(
+		private readonly IConfig $config,
+		private readonly IInitialState $initialState,
+		private readonly AssistantService $assistantService,
+	) {
+	}
+
+	public function handle(Event $event): void {
+		if (!$event instanceof LoadAdditionalScriptsEvent) {
+			return;
+		}
+
+		if ($this->config->getAppValue(Application::APP_ID, 'disableFilesNewMenuPlugin', '0') === '1') {
+			return;
+		}
+
+		$taskTypes = $this->assistantService->getAvailableTaskTypes();
+		$hasText2Image = false;
+		foreach ($taskTypes as $taskType) {
+			if ($taskType['id'] === 'core:text2image') {
+				$hasText2Image = true;
+				break;
+			}
+		}
+
+		$this->initialState->provideInitialState('new-file-generate-image', [
+			'hasText2Image' => $hasText2Image,
+			'taskTypes' => $taskTypes,
+		]);
+
+		Util::addScript(Application::APP_ID, Application::APP_ID . '-filesNewMenu');
+	}
+}

lib/Listener/TaskSuccessfulListener.php

@@ -10,11 +10,14 @@
 use OCA\Assistant\AppInfo\Application;
 use OCA\Assistant\Db\TaskNotificationMapper;
 use OCA\Assistant\Event\BeforeAssistantNotificationEvent;
+use OCA\Assistant\Service\AssistantService;
 use OCA\Assistant\Service\NotificationService;
 use OCP\EventDispatcher\Event;
 use OCP\EventDispatcher\IEventDispatcher;
 use OCP\EventDispatcher\IEventListener;
+use OCP\IURLGenerator;
 use OCP\TaskProcessing\Events\TaskSuccessfulEvent;
+use Psr\Log\LoggerInterface;
 
 /**
  * @template-implements IEventListener<Event>
@@ -25,6 +28,9 @@ public function __construct(
 		private TaskNotificationMapper $taskNotificationMapper,
 		private NotificationService $notificationService,
 		private IEventDispatcher $eventDispatcher,
+		private AssistantService $assistantService,
+		private LoggerInterface $logger,
+		private IUrlGenerator $url,
 	) {
 	}
 
@@ -38,7 +44,11 @@ public function handle(Event $event): void {
 			return;
 		}
 
-		if ($this->taskNotificationMapper->getByTaskId($task->getId()) === null) {
+		$customIdPattern = '/^new-image-file:(\d+)$/';
+		$hasTargetDirectory = preg_match($customIdPattern, $task->getCustomId(), $matches) === 1;
+
+		// For tasks with customId "new-image-file:<directoryIdNumber>" we always send a notification
+		if ($this->taskNotificationMapper->getByTaskId($task->getId()) === null && !$hasTargetDirectory) {
 			return;
 		}
 
@@ -57,7 +67,31 @@ public function handle(Event $event): void {
 			$notificationActionLabel = $beforeAssistantNotificationEvent->getNotificationActionLabel();
 		}
 
+		if ($hasTargetDirectory) {
+			$directoryId = (int)$matches[1];
+			$fileId = (int) $task->getOutput()['images'][0];
+			try {
+				$file = $this->assistantService->saveNewFileMenuActionFile($task->getUserId(), $task->getId(), $fileId, $directoryId);
+				$notificationTarget = $this->url->linkToRouteAbsolute(
+					'files.viewcontroller.showFile',
+					[
+						'fileid' => $file->getId(),
+						'opendetails' => 'true',
+						'openfile' => 'false',
+					],
+				);
+			} catch (\Exception $e) {
+				$this->logger->error('TaskSuccessfulListener: Failed to save new file menu action file.', [
+					'task' => $task->jsonSerialize(),
+					'exception' => $e,
+				]);
+			}
+		}
+
 		$this->notificationService->sendNotification($task, $notificationTarget, $notificationActionLabel);
-		$this->taskNotificationMapper->deleteByTaskId($task->getId());
+
+		if (!$hasTargetDirectory) {
+			$this->taskNotificationMapper->deleteByTaskId($task->getId());
+		}
 	}
 }

lib/Service/AssistantService.php

@@ -533,6 +533,25 @@ private function saveFile(string $userId, int $ocpTaskId, int $fileId): File {
 		}
 	}
 
+	/**
+	 * @throws TaskProcessingException
+	 * @throws NotPermittedException
+	 * @throws NotFoundException
+	 * @throws Exception
+	 * @throws LockedException
+	 * @throws NoUserException
+	 */
+	public function saveNewFileMenuActionFile(string $userId, int $ocpTaskId, int $fileId, int $targetDirectoryId): File {
+		$taskOutputFile = $this->getTaskOutputFile($userId, $ocpTaskId, $fileId);
+		$userFolder = $this->rootFolder->getUserFolder($userId);
+		$targetDirectory = $userFolder->getFirstNodeById($targetDirectoryId);
+		if (!$targetDirectory instanceof Folder) {
+			throw new NotFoundException('Target directory not found');
+		}
+		$targetFileName = $this->getTargetFileName($taskOutputFile);
+		return $targetDirectory->newFile($targetFileName, $taskOutputFile->fopen('rb'));
+	}
+
 	/**
 	 * @param string $userId
 	 * @param int $ocpTaskId

package-lock.json

@@ -45,6 +45,7 @@
 		"@nextcloud/axios": "^2.0.0",
 		"@nextcloud/dialogs": "^7.0.0-rc.1",
 		"@nextcloud/event-bus": "^3.1.0",
+		"@nextcloud/files": "^3.11.0",
 		"@nextcloud/initial-state": "^2.0.0",
 		"@nextcloud/l10n": "^3.1.0",
 		"@nextcloud/moment": "^1.3.1",