final class VectorizeCommand extends Command
{
    public function __construct(
        private readonly DocumentManager $dm,
        #[Autowire(service: 'ai.vectorizer.voyage')]
        private readonly VectorizerInterface $vectorizer,
    ) {
        parent::__construct();
    }

    protected function execute(InputInterface $input, OutputInterface $output): int
    {
        $verbosityLevelMap = [
            LogLevel::NOTICE => OutputInterface::VERBOSITY_NORMAL,
            LogLevel::INFO   => OutputInterface::VERBOSITY_NORMAL,
        ];

        $qb = $this->dm->createQueryBuilder(TcgCard::class);

        $logger = new ConsoleLogger($output, $verbosityLevelMap);

        $tcgCards = $qb->field('embeddingVector')->equals(null)->getQuery()->execute();

        $chunkSize = 500;
        $counter = 0;
        $chunk = [];

        $logger->info('Starting vectorizer: ' . count($tcgCards), ['total_documents' => count($tcgCards)]);

        foreach ($tcgCards as $document) {
            $chunk[] = $document;
            ++$counter;

            if ($chunkSize === \count($chunk)) {
                $logger->info("Processing chunk, at {$counter}", ['processed_documents' => $counter]);
                $this->vectorizeChunk($chunk, $qb);

                $chunk = [];
            }
        }

        if ([] !== $chunk) {
            $this->vectorizeChunk($chunk, $qb);
        }

        $logger->info('Document processing completed', ['total_documents' => $counter]);
        $this->dm->flush();

        return 0;
    }

    private function vectorizeChunk(array $chunk, Builder $qb): void
    {
        $vectorDocuments = $this->vectorizer->vectorize(
            $chunk,
            ['dimensions' => 512]
        );

        foreach ($vectorDocuments as $document) {
            $qb->findAndUpdate()
                ->field('id')->equals($document->id)
                ->field('embeddingVector')->set($document->vector->getData())
                ->getQuery()
                ->execute();
        }
    }