Lesstime/src/Mcp/Tool/TimeEntry/UpdateTimeEntryTool.php

<?php

declare(strict_types=1);

namespace App\Mcp\Tool\TimeEntry;

use App\Mcp\Tool\Serializer;
use App\Repository\ClientTicketRepository;
use App\Repository\ProjectRepository;
use App\Repository\TaskRepository;
use App\Repository\TaskTagRepository;
use App\Repository\TimeEntryRepository;
use DateTimeImmutable;
use Doctrine\ORM\EntityManagerInterface;
use InvalidArgumentException;
use Mcp\Capability\Attribute\McpTool;

use function sprintf;

#[McpTool(name: 'update-time-entry', description: 'Update a time entry. Use to stop an active timer by providing stoppedAt, or to correct start time. userId is not updatable.')]
class UpdateTimeEntryTool
{
    public function __construct(
        private readonly TimeEntryRepository $timeEntryRepository,
        private readonly ProjectRepository $projectRepository,
        private readonly TaskRepository $taskRepository,
        private readonly TaskTagRepository $taskTagRepository,
        private readonly ClientTicketRepository $clientTicketRepository,
        private readonly EntityManagerInterface $entityManager,
    ) {}

    public function __invoke(
        int $id,
        ?string $title = null,
        ?string $startedAt = null,
        ?string $stoppedAt = null,
        ?int $projectId = null,
        ?int $taskId = null,
        ?array $tagIds = null,
        ?string $description = null,
        ?int $clientTicketId = null,
    ): string {
        $entry = $this->timeEntryRepository->find($id);

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

        if (null !== $title) {
            $entry->setTitle($title);
        }
        if (null !== $startedAt) {
            $entry->setStartedAt(new DateTimeImmutable($startedAt));
        }
        if (null !== $stoppedAt) {
            $entry->setStoppedAt(new DateTimeImmutable($stoppedAt));
        }
        if (null !== $description) {
            $entry->setDescription($description);
        }
        if (null !== $projectId) {
            $project = $this->projectRepository->find($projectId);
            if (null === $project) {
                throw new InvalidArgumentException(sprintf('Project with ID %d not found.', $projectId));
            }
            $entry->setProject($project);
        }
        if (null !== $taskId) {
            $task = $this->taskRepository->find($taskId);
            if (null === $task) {
                throw new InvalidArgumentException(sprintf('Task with ID %d not found.', $taskId));
            }
            $entry->setTask($task);
        }
        if (null !== $clientTicketId) {
            $clientTicket = $this->clientTicketRepository->find($clientTicketId);
            if (null === $clientTicket) {
                throw new InvalidArgumentException(sprintf('ClientTicket with ID %d not found.', $clientTicketId));
            }
            $entry->setClientTicket($clientTicket);
        }
        if (null !== $tagIds) {
            foreach ($entry->getTags()->toArray() as $existingTag) {
                $entry->removeTag($existingTag);
            }
            foreach ($tagIds as $tagId) {
                $tag = $this->taskTagRepository->find($tagId);
                if (null === $tag) {
                    throw new InvalidArgumentException(sprintf('TaskTag with ID %d not found.', $tagId));
                }
                $entry->addTag($tag);
            }
        }

        $this->entityManager->flush();

        return json_encode(Serializer::timeEntry($entry));
    }
}