From 4f78d5d43d6cce9aeb99f1ea4b5d16949e48af26 Mon Sep 17 00:00:00 2001
From: louis <louis@systemli.org>
Date: Wed, 20 Dec 2023 21:52:08 +0100
Subject: [PATCH] Add Migration for Enabled Field

---
 migrations/Version20231220194844.php | 37 +++++++++++++++++++
 migrations/Version20231220203032.php | 53 ++++++++++++++++++++++++++++
 src/DataFixtures/LoadUserData.php    |  1 +
 src/Enum/Roles.php                   |  1 +
 4 files changed, 92 insertions(+)
 create mode 100644 migrations/Version20231220194844.php
 create mode 100644 migrations/Version20231220203032.php

diff --git a/migrations/Version20231220194844.php b/migrations/Version20231220194844.php
new file mode 100644
index 00000000..20a178e0
--- /dev/null
+++ b/migrations/Version20231220194844.php
@@ -0,0 +1,37 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DoctrineMigrations;
+
+use Doctrine\DBAL\Schema\Schema;
+use Doctrine\Migrations\AbstractMigration;
+
+/**
+ * Add enabled field to User entity.
+ */
+final class Version20231220194844 extends AbstractMigration
+{
+    public function getDescription(): string
+    {
+        return 'Add enabled field to User entity.';
+    }
+
+    public function up(Schema $schema): void
+    {
+        if ($schema->getTable('virtual_users')->hasColumn('enabled')) {
+            return;
+        }
+
+        $this->addSql('ALTER TABLE virtual_users ADD enabled TINYINT(1) NOT NULL');
+    }
+
+    public function down(Schema $schema): void
+    {
+        if (!$schema->getTable('virtual_users')->hasColumn('enabled')) {
+            return;
+        }
+
+        $this->addSql('ALTER TABLE virtual_users DROP enabled');
+    }
+}
diff --git a/migrations/Version20231220203032.php b/migrations/Version20231220203032.php
new file mode 100644
index 00000000..90c9e9be
--- /dev/null
+++ b/migrations/Version20231220203032.php
@@ -0,0 +1,53 @@
+<?php
+
+declare(strict_types=1);
+
+namespace DoctrineMigrations;
+
+use App\Entity\User;
+use App\Enum\Roles;
+use Doctrine\DBAL\Schema\Schema;
+use Doctrine\Migrations\AbstractMigration;
+use Symfony\Component\DependencyInjection\ContainerAwareInterface;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+
+/**
+ * Remove Roles::SPAM from users and disable them.
+ */
+final class Version20231220203032 extends AbstractMigration implements ContainerAwareInterface
+{
+    private $container;
+
+    public function setContainer(ContainerInterface $container = null): void
+    {
+        $this->container = $container;
+    }
+
+    public function getDescription(): string
+    {
+        return 'Remove Roles::SPAM from users and disable them.';
+    }
+
+    public function up(Schema $schema): void
+    {
+        $entityManager = $this->container->get('doctrine.orm.entity_manager');
+        /** @var \App\Entity\User[] $users */
+        $users = $entityManager->getRepository(User::class)->findAll();
+        foreach ($users as $user) {
+            if (!$user->hasRole(Roles::SPAM) || $user->isDeleted()) {
+                continue;
+            }
+
+            $user->setRoles(array_filter($user->getRoles(), function ($role) {
+                return $role !== Roles::SPAM;
+            }));
+            $user->setEnabled(false);
+            $entityManager->persist($user);
+        }
+        $entityManager->flush();
+    }
+
+    public function down(Schema $schema): void
+    {
+    }
+}
diff --git a/src/DataFixtures/LoadUserData.php b/src/DataFixtures/LoadUserData.php
index e8d850f9..6cea9914 100644
--- a/src/DataFixtures/LoadUserData.php
+++ b/src/DataFixtures/LoadUserData.php
@@ -21,6 +21,7 @@ class LoadUserData extends Fixture implements OrderedFixtureInterface, Container
         ['email' => 'user@example.org', 'roles' => [Roles::USER]],
         ['email' => 'support@example.org', 'roles' => [Roles::MULTIPLIER]],
         ['email' => 'suspicious@example.org', 'roles' => [Roles::SUSPICIOUS]],
+        ['email' => 'spammer@example.org', 'roles' => [Roles::SPAM]],
         ['email' => 'domain@example.com', 'roles' => [Roles::DOMAIN_ADMIN]],
     ];
 
diff --git a/src/Enum/Roles.php b/src/Enum/Roles.php
index 92e9e829..f80865f9 100644
--- a/src/Enum/Roles.php
+++ b/src/Enum/Roles.php
@@ -6,6 +6,7 @@ final class Roles
 {
     public const PERMANENT = 'ROLE_PERMANENT';
     public const MULTIPLIER = 'ROLE_MULTIPLIER';
+    /** @deprecated */
     public const SPAM = 'ROLE_SPAM';
     public const SUSPICIOUS = 'ROLE_SUSPICIOUS';
     public const USER = 'ROLE_USER';