Merge pull request #5510 from christianbeeznest/GH-5412-3

Migration: Add TrackEDownloads entries with null resourceLink - refs #5412

src/CoreBundle/Migrations/Schema/V200/Version20240515124400.php

@@ -0,0 +1,100 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Chamilo\CoreBundle\Migrations\Schema\V200;
+
+use Chamilo\CoreBundle\Entity\User;
+use Chamilo\CoreBundle\Migrations\AbstractMigrationChamilo;
+use Doctrine\DBAL\Schema\Schema;
+use Chamilo\CoreBundle\Entity\TrackEDownloads;
+use Chamilo\CoreBundle\Entity\ResourceFile;
+use Chamilo\CoreBundle\Entity\ResourceNode;
+
+final class Version20240515124400 extends AbstractMigrationChamilo
+{
+    public function getDescription(): string
+    {
+        return 'Find and process TrackEDownloads entries with null resourceLink, linking them to ResourceNodes and updating downloads.';
+    }
+
+    public function up(Schema $schema): void
+    {
+        $trackEDownloadsRepository = $this->entityManager->getRepository(TrackEDownloads::class);
+        $resourceFileRepository = $this->entityManager->getRepository(ResourceFile::class);
+        $resourceNodeRepository = $this->entityManager->getRepository(ResourceNode::class);
+        $userRepository = $this->entityManager->getRepository(User::class);
+
+        $batchSize = 100;
+        $offset = 0;
+        $success = true;
+
+        while (true) {
+            $downloads = $trackEDownloadsRepository->createQueryBuilder('t')
+                ->where('t.resourceLink IS NULL')
+                ->setFirstResult($offset)
+                ->setMaxResults($batchSize)
+                ->getQuery()
+                ->getResult();
+
+            if (empty($downloads)) {
+                break;
+            }
+
+            $this->entityManager->beginTransaction();
+
+            try {
+                foreach ($downloads as $download) {
+                    $downDocPath = $download->getDownDocPath();
+                    $fileName = basename($downDocPath);
+
+                    $resourceFile = $resourceFileRepository->findOneBy(['originalName' => $fileName]);
+
+                    if ($resourceFile) {
+                        $resourceFileId = $resourceFile->getId();
+                        $resourceNode = $resourceNodeRepository->findOneBy(['resourceFile' => $resourceFileId]);
+
+                        if ($resourceNode) {
+                            $downUserId = $download->getDownUserId();
+                            $user = $userRepository->find($downUserId);
+                            if (null === $user) {
+                                $user = $this->getAdmin();
+                            }
+
+                            $firstResourceLink = $resourceNode->getResourceLinks()->first();
+                            if ($firstResourceLink && $user) {
+                                $resourceLinkId = $firstResourceLink->getId();
+                                $url = $resourceNode->getResourceFile()->getOriginalName();
+                                echo "Resource link $resourceLinkId Down id {$download->getDownId()} for $url: user " . $user->getFullname() . "\n";
+
+                                $this->connection->executeUpdate("UPDATE track_e_downloads SET resource_link_id = ? WHERE down_id = ?", [$resourceLinkId, $download->getDownId()]);
+                            }
+                        }
+                    }
+                }
+
+                $this->entityManager->commit();
+            } catch (\Exception $e) {
+                $this->entityManager->rollback();
+                $success = false;
+                echo "Failed for download ID {$download->getDownId()}: " . $e->getMessage() . "\n";
+                break;
+            } finally {
+                $this->entityManager->clear();
+            }
+
+            $offset += $batchSize;
+        }
+
+        // Only delete if all updates were successful
+        if ($success) {
+            $this->connection->executeUpdate("DELETE FROM track_e_downloads WHERE resource_link_id IS NULL");
+        } else {
+            echo "Process failed. No records were deleted.\n";
+        }
+    }
+
+    public function down(Schema $schema): void
+    {
+    }
+}