Lesstime/src/Module/ProjectManagement/Infrastructure/Mcp/Tool/Task/UpdateTaskTool.php

<?php

declare(strict_types=1);

namespace App\Module\ProjectManagement\Infrastructure\Mcp\Tool\Task;

use App\Module\Core\Infrastructure\Doctrine\DoctrineUserRepository;
use App\Module\ProjectManagement\Domain\Repository\TaskEffortRepositoryInterface;
use App\Module\ProjectManagement\Domain\Repository\TaskGroupRepositoryInterface;
use App\Module\ProjectManagement\Domain\Repository\TaskPriorityRepositoryInterface;
use App\Module\ProjectManagement\Domain\Repository\TaskRepositoryInterface;
use App\Module\ProjectManagement\Domain\Repository\TaskStatusRepositoryInterface;
use App\Module\ProjectManagement\Domain\Repository\TaskTagRepositoryInterface;
use App\Module\ProjectManagement\Infrastructure\Service\CalDavService;
use App\Shared\Infrastructure\Mcp\Serializer;
use DateTimeImmutable;
use Doctrine\ORM\EntityManagerInterface;
use InvalidArgumentException;
use Mcp\Capability\Attribute\McpTool;
use Symfony\Bundle\SecurityBundle\Security;
use Symfony\Component\Security\Core\Exception\AccessDeniedException;

use function sprintf;

#[McpTool(name: 'update-task', description: 'Update an existing task. Only provided fields are changed. Use list-statuses, list-priorities, etc. to discover valid IDs. The status parameter must reference a status that belongs to the task\'s project workflow — otherwise the call is rejected with a validation error.')]
class UpdateTaskTool
{
    public function __construct(
        private readonly EntityManagerInterface $entityManager,
        private readonly TaskRepositoryInterface $taskRepository,
        private readonly TaskStatusRepositoryInterface $taskStatusRepository,
        private readonly TaskPriorityRepositoryInterface $taskPriorityRepository,
        private readonly TaskEffortRepositoryInterface $taskEffortRepository,
        private readonly TaskGroupRepositoryInterface $taskGroupRepository,
        private readonly TaskTagRepositoryInterface $taskTagRepository,
        private readonly DoctrineUserRepository $userRepository,
        private readonly Security $security,
        private readonly CalDavService $calDavService,
    ) {}

    /**
     * @param int[] $tagIds          IDs of the tags to attach
     * @param int[] $collaboratorIds IDs of the collaborators to attach
     */
    public function __invoke(
        int $id,
        ?string $title = null,
        ?string $description = null,
        ?int $statusId = null,
        ?int $priorityId = null,
        ?int $effortId = null,
        ?int $assigneeId = null,
        ?int $groupId = null,
        ?array $tagIds = null,
        ?array $collaboratorIds = null,
        ?bool $archived = null,
        ?string $scheduledStart = null,
        ?string $scheduledEnd = null,
        ?string $deadline = null,
        ?bool $syncToCalendar = null,
    ): string {
        if (!$this->security->isGranted('ROLE_USER')) {
            throw new AccessDeniedException('Access denied: ROLE_USER required.');
        }

        $task = $this->taskRepository->findById($id);

        if (null === $task) {
            throw new InvalidArgumentException(sprintf('Task with ID %d not found.', $id));
        }

        if (null !== $title) {
            $task->setTitle($title);
        }
        if (null !== $description) {
            $task->setDescription($description);
        }
        if (null !== $statusId) {
            $status = $this->taskStatusRepository->findById($statusId);
            if (null === $status) {
                throw new InvalidArgumentException(sprintf('TaskStatus with ID %d not found.', $statusId));
            }
            $task->setStatus($status);
        }
        if (null !== $priorityId) {
            $priority = $this->taskPriorityRepository->findById($priorityId);
            if (null === $priority) {
                throw new InvalidArgumentException(sprintf('TaskPriority with ID %d not found.', $priorityId));
            }
            $task->setPriority($priority);
        }
        if (null !== $effortId) {
            $effort = $this->taskEffortRepository->findById($effortId);
            if (null === $effort) {
                throw new InvalidArgumentException(sprintf('TaskEffort with ID %d not found.', $effortId));
            }
            $task->setEffort($effort);
        }
        if (null !== $assigneeId) {
            $assignee = $this->userRepository->find($assigneeId);
            if (null === $assignee) {
                throw new InvalidArgumentException(sprintf('User with ID %d not found.', $assigneeId));
            }
            $task->setAssignee($assignee);
        }
        if (null !== $groupId) {
            $group = $this->taskGroupRepository->findById($groupId);
            if (null === $group) {
                throw new InvalidArgumentException(sprintf('TaskGroup with ID %d not found.', $groupId));
            }
            $task->setGroup($group);
        }
        if (null !== $tagIds) {
            // Clear existing tags and set new ones
            foreach ($task->getTags()->toArray() as $existingTag) {
                $task->removeTag($existingTag);
            }
            foreach ($tagIds as $tagId) {
                $tag = $this->taskTagRepository->findById($tagId);
                if (null === $tag) {
                    throw new InvalidArgumentException(sprintf('TaskTag with ID %d not found.', $tagId));
                }
                $task->addTag($tag);
            }
        }
        if (null !== $collaboratorIds) {
            foreach ($task->getCollaborators()->toArray() as $existing) {
                $task->removeCollaborator($existing);
            }
            $assignee = $task->getAssignee();
            foreach ($collaboratorIds as $collaboratorId) {
                $collaborator = $this->userRepository->find($collaboratorId);
                if (null === $collaborator) {
                    throw new InvalidArgumentException(sprintf('User with ID %d not found.', $collaboratorId));
                }
                if (null !== $assignee && $collaborator->getId() === $assignee->getId()) {
                    throw new InvalidArgumentException('A collaborator cannot be the assignee.');
                }
                $task->addCollaborator($collaborator);
            }
        }
        if (null !== $archived) {
            $task->setArchived($archived);
        }
        if (null !== $scheduledStart) {
            $task->setScheduledStart(new DateTimeImmutable($scheduledStart));
        }
        if (null !== $scheduledEnd) {
            $task->setScheduledEnd(new DateTimeImmutable($scheduledEnd));
        }
        if (null !== $deadline) {
            $task->setDeadline(new DateTimeImmutable($deadline));
        }
        if (null !== $syncToCalendar) {
            $task->setSyncToCalendar($syncToCalendar);
        }

        $this->entityManager->flush();
        $this->calDavService->syncTask($task);
        $this->entityManager->flush();

        return json_encode([
            'id'             => $task->getId(),
            'number'         => $task->getNumber(),
            'title'          => $task->getTitle(),
            'description'    => $task->getDescription(),
            'status'         => Serializer::status($task->getStatus()),
            'priority'       => Serializer::priority($task->getPriority()),
            'effort'         => Serializer::effort($task->getEffort()),
            'assignee'       => Serializer::user($task->getAssignee()),
            'collaborators'  => Serializer::users($task->getCollaborators()),
            'group'          => Serializer::groupRef($task->getGroup()),
            'project'        => Serializer::projectRef($task->getProject()),
            'tags'           => Serializer::tags($task->getTags()),
            'archived'       => $task->isArchived(),
            'scheduledStart' => $task->getScheduledStart()?->format('c'),
            'scheduledEnd'   => $task->getScheduledEnd()?->format('c'),
            'deadline'       => $task->getDeadline()?->format('c'),
            'syncToCalendar' => $task->isSyncToCalendar(),
        ]);
    }
}